ai-mocks-ollama/dev.mokksy.aimocks.ollama.chat/OllamaChatRequestSpecification

OllamaChatRequestSpecification

class OllamaChatRequestSpecification : AbstractInferenceRequestSpecification<ChatRequest> (source)

Specification for matching chat completion requests.

This class is used to specify the criteria for matching requests to the chat endpoint. It allows specifying the model, messages, tools, and stream flag.

Constructors

constructor()

Properties

maxTokens

var maxTokens: Long?

messages

var messages: List<Message>

The messages to match in the request

model

var model: String?

requestBody

val requestBody: MutableList<Matcher<ChatRequest?>>

requestBodyString

val requestBodyString: MutableList<Matcher<String?>>

seed

var seed: Int?

stream

var stream: Boolean?

Whether to match streaming requests

temperature

var temperature: Double?

topK

var topK: Long?

topP

var topP: Double?

Functions

maxTokens

fun maxTokens(value: Long): AbstractInferenceRequestSpecification<ChatRequest>

fun maxTokens(value: Number): AbstractInferenceRequestSpecification<ChatRequest>

messages

fun messages(messages: List<Message>): OllamaChatRequestSpecification

Sets the list of messages to match in the chat request.

model

fun model(model: String): ModelRequestSpecification<ChatRequest>

requestBodyContains

fun requestBodyContains(substring: String): ModelRequestSpecification<ChatRequest>

requestBodyContainsIgnoringCase

fun requestBodyContainsIgnoringCase(substring: String): ModelRequestSpecification<ChatRequest>

requestBodyDoesNotContains

fun requestBodyDoesNotContains(substring: String): ModelRequestSpecification<ChatRequest>

requestBodyDoesNotContainsIgnoringCase

fun requestBodyDoesNotContainsIgnoringCase(substring: String): ModelRequestSpecification<ChatRequest>

requestBodyEquals

fun requestBodyEquals(requestObject: ChatRequest): ModelRequestSpecification<ChatRequest>

fun requestBodyEquals(payload: String): ModelRequestSpecification<ChatRequest>

requestBodyEqualsJson

fun requestBodyEqualsJson(payload: String): ModelRequestSpecification<ChatRequest>

requestMatches

fun requestMatches(matcher: Matcher<ChatRequest?>)

requestMatchesPredicate

fun requestMatchesPredicate(predicate: (ChatRequest) -> Boolean)

requestSatisfies

fun requestSatisfies(description: String?, call: (ChatRequest?) -> Unit)

stream

fun stream(stream: Boolean): OllamaChatRequestSpecification

Sets the streaming flag for the request specification.

systemMessageContains

inline fun systemMessageContains(builderAction: StringBuilder.() -> Unit)

open override fun systemMessageContains(substring: String)

Adds a predicate to match requests where any system message contains the specified substring.

temperature

fun temperature(temperature: Double): AbstractInferenceRequestSpecification<ChatRequest>

topK

fun topK(value: Long): AbstractInferenceRequestSpecification<ChatRequest>

fun topK(value: Number): AbstractInferenceRequestSpecification<ChatRequest>

topP

fun topP(value: Double): AbstractInferenceRequestSpecification<ChatRequest>

fun topP(value: Number): AbstractInferenceRequestSpecification<ChatRequest>

userMessageContains

inline fun userMessageContains(builderAction: StringBuilder.() -> Unit)

open override fun userMessageContains(substring: String)

Adds a predicate to match requests where any user message contains the specified substring.