ai-mocks-ollama/dev.mokksy.aimocks.ollama.embed/OllamaEmbedRequestSpecification

OllamaEmbedRequestSpecification

class OllamaEmbedRequestSpecification(source)

Specification for matching embedding requests.

This class is used to specify the criteria for matching requests to the embed endpoint. It allows specifying the model, input (string or list of strings), truncate flag, options, and keepAlive.

Constructors

constructor()

Properties

keepAlive

var keepAlive: String?

Controls how long the model will stay loaded into memory

model

var model: String?

The model to match in the request

options

var options: Map<String, String>?

Additional model parameters to match in the request

requestBody

var requestBody: EmbeddingsRequest?

The request body to match

requestBodyString

val requestBodyString: MutableList<String>

Additional string matchers for the request body

stringInput

var stringInput: String?

The string input to match in the request

stringListInput

var stringListInput: List<String>?

The list of string inputs to match in the request

truncate

var truncate: Boolean?

Whether to truncate the input to fit within context length

Functions

keepAlive

fun keepAlive(keepAlive: String): OllamaEmbedRequestSpecification

Sets the duration for which the model remains loaded in memory.

model

fun model(model: String): OllamaEmbedRequestSpecification

Sets the model name criterion for matching embedding requests.

options

fun options(options: Map<String, String>): OllamaEmbedRequestSpecification

Sets additional model parameters for the embedding request specification.

requestBody

fun requestBody(requestBody: EmbeddingsRequest): OllamaEmbedRequestSpecification

Sets the full request body for embedding request matching.

requestBodyString

fun requestBodyString(bodyString: String): OllamaEmbedRequestSpecification

Adds a string matcher to the list of request body matchers for this specification.

stringInput

fun stringInput(input: String): OllamaEmbedRequestSpecification

Sets a single string input for embedding requests, clearing any existing list input.

stringListInput

fun stringListInput(inputs: List<String>): OllamaEmbedRequestSpecification

Sets the list of string inputs for embedding, clearing any previously set single string input.

truncate

fun truncate(truncate: Boolean): OllamaEmbedRequestSpecification

Sets whether to truncate the input to fit within the model's context length.