ai-mocks-ollama/dev.mokksy.aimocks.ollama/MockOllama

MockOllama

open class MockOllama(port: Int = 0, verbose: Boolean = true) : AbstractMockLlm(source)

Mock implementation of an Ollama-compatible service for testing purposes.

This class provides an HTTP mock server to simulate Ollama APIs, specifically for generate completions, chat completions, and other Ollama endpoints. It is designed to mimic the behavior of the Ollama APIs locally and facilitate integration testing and development.

Extends AbstractMockLlm to provide Ollama-specific functionality.

Author

Konstantin Pavlov

Parameters

port

The port on which the mock server will run. Defaults to 0, which allows the server to select an available port.

verbose

Controls whether the mock server's operations are logged in detail. Defaults to true.

Constructors

constructor(port: Int = 0, verbose: Boolean = true)

Functions

open override fun baseUrl(): String

Returns the base URL of the mock Ollama server.

fun chat(name: String? = null, block: OllamaChatRequestSpecification.() -> Unit): OllamaChatBuildingStep

Sets up a mock handler for the Ollama chat completion (/api/chat) endpoint.

fun MockOllama.chat(name: String? = null, block: Consumer<OllamaChatRequestSpecification>): OllamaChatBuildingStep

Provides a Java-compatible overload for configuring a mock Ollama chat request using a Consumer.

fun embed(name: String? = null, block: OllamaEmbedRequestSpecification.() -> Unit): OllamaEmbedBuildingStep

Sets up a mock handler for the Ollama /api/embed endpoint, allowing configuration of request matching for embedding requests.

fun MockOllama.embed(name: String? = null, block: Consumer<OllamaEmbedRequestSpecification>): OllamaEmbedBuildingStep

Provides a Java-compatible overload for configuring a mock /api/embed endpoint using a Consumer.

fun generate(name: String? = null, block: OllamaGenerateRequestSpecification.() -> Unit): OllamaGenerateBuildingStep

Sets up a mock handler for the Ollama /api/generate completion endpoint.

fun MockOllama.generate(name: String? = null, block: Consumer<OllamaGenerateRequestSpecification>): OllamaGenerateBuildingStep

Configures a mock /api/generate endpoint using a Java Consumer to specify the request criteria.

fun port(): Int

resetMatchState

fun resetMatchState()

open fun shutdown(gracePeriodMillis: Long, timeoutMillis: Long)

verifyNoUnexpectedRequests

fun verifyNoUnexpectedRequests()

verifyNoUnmatchedRequests

fun ~~verifyNoUnmatchedRequests~~()