ai-mocks-ollama/dev.mokksy.aimocks.ollama.generate/OllamaGenerateRequestSpecification

OllamaGenerateRequestSpecification

class OllamaGenerateRequestSpecification : AbstractInferenceRequestSpecification<GenerateRequest> (source)

Specification for matching generate completion requests.

This class is used to specify the criteria for matching requests to the generate endpoint. It allows specifying the model, prompt, system message, template, and stream flag.

Constructors

constructor()

Properties

maxTokens

var maxTokens: Long?

model

var model: String?

requestBody

val requestBody: MutableList<Matcher<GenerateRequest?>>

requestBodyString

val requestBodyString: MutableList<Matcher<String?>>

stream

var stream: Boolean?

Whether to match streaming requests

temperature

var temperature: Double?

template

var template: String?

The template to match in the request

topK

var topK: Long?

topP

var topP: Double?

Functions

maxTokens

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

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

model

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

requestBodyContains

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

requestBodyContainsIgnoringCase

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

requestBodyDoesNotContains

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

requestBodyDoesNotContainsIgnoringCase

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

requestBodyEquals

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

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

requestBodyEqualsJson

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

requestMatches

fun requestMatches(matcher: Matcher<GenerateRequest?>)

requestMatchesPredicate

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

requestSatisfies

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

stream

fun stream(stream: Boolean): OllamaGenerateRequestSpecification

Sets the stream flag to indicate whether to match streaming requests.

systemMessageContains

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

open override fun systemMessageContains(substring: String)

Specifies that the system message in the request must contain the given substring.

temperature

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

template

fun template(template: String): OllamaGenerateRequestSpecification

Sets the template to match in the generate request.

topK

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

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

topP

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

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

userMessageContains

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

open override fun userMessageContains(substring: String)

Specifies that the request's prompt must contain the given substring.