Package org.mockserver.llm.codec

Class OllamaCodec

java.lang.Object

org.mockserver.llm.codec.OllamaCodec

All Implemented Interfaces:

ProviderCodec

public class OllamaCodec
extends Object
implements ProviderCodec

Codec for Ollama /api/chat endpoint (version ollama-2025).

Ollama uses a distinct JSON shape with a top-level message object (not an array of choices). Token counts are exposed as prompt_eval_count and eval_count.

Streaming format: Ollama's native wire format is NDJSON (newline-delimited JSON, application/x-ndjson), not SSE. Each streaming chunk is a single JSON object followed by a newline character (<json>\n). This codec internally represents chunks as SseEvent objects (reusing the existing data/delay model), but declares StreamingFormat.NDJSON via streamingFormat() so that the downstream write handler emits raw NDJSON framing instead of SSE data: prefixes.

Tool calls use Ollama 0.3+ format where arguments is a JSON object (not a JSON-as-string like OpenAI).

Constructor Summary

Constructors

Constructor	Description
OllamaCodec()

Method Summary

Modifier and Type	Method	Description
String	apiVersion()
ParsedConversation	decode(HttpRequest request)
HttpResponse	encode(Completion completion, String model)
HttpResponse	encodeEmbedding(EmbeddingResponse embedding, String input)	Encodes an Ollama embedding response.
HttpResponse	encodeEmbedding(EmbeddingResponse embedding, String input, String model)	Model-aware embedding encode.
List<SseEvent>	encodeStreaming(Completion completion, String model, StreamingPhysics physics)
Provider	provider()
StreamingFormat	streamingFormat()	The wire format this provider uses for streaming responses.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.mockserver.llm.ProviderCodec

encodeRerank

Constructor Details

OllamaCodec

public OllamaCodec()

Method Details

provider

public Provider provider()

Specified by:

provider in interface ProviderCodec

apiVersion

public String apiVersion()

Specified by:

apiVersion in interface ProviderCodec

streamingFormat

public StreamingFormat streamingFormat()

Description copied from interface: ProviderCodec

The wire format this provider uses for streaming responses. Defaults to StreamingFormat.SSE; override for providers that use a different format (e.g. Ollama uses NDJSON).

Specified by:

streamingFormat in interface ProviderCodec

encode

public HttpResponse encode(Completion completion,
 String model)

Specified by:

encode in interface ProviderCodec

encodeStreaming

public List<SseEvent> encodeStreaming(Completion completion,
 String model,
 StreamingPhysics physics)

Specified by:

encodeStreaming in interface ProviderCodec

decode

public ParsedConversation decode(HttpRequest request)

Specified by:

decode in interface ProviderCodec

encodeEmbedding

public HttpResponse encodeEmbedding(EmbeddingResponse embedding,
 String input)

Encodes an Ollama embedding response. Ollama exposes two endpoints:

• the newer POST /api/embed returning a batch shape ({"embeddings":[[...]]}, plus model and prompt_eval_count), and
• the legacy POST /api/embeddings returning a single-vector shape ({"embedding":[...]}).

This codec emits the /api/embed batch shape with a single embedding (the modern, recommended endpoint), which also satisfies clients that read the embeddings array. The Ollama default dimensionality for nomic-embed-text is 768.

Specified by:

encodeEmbedding in interface ProviderCodec

encodeEmbedding

public HttpResponse encodeEmbedding(EmbeddingResponse embedding,
 String input,
 String model)

Description copied from interface: ProviderCodec

Model-aware embedding encode. Most providers have a single embedding wire shape and ignore the model, so the default delegates to ProviderCodec.encodeEmbedding(EmbeddingResponse, String). Providers whose embedding shape varies by model family (e.g. Bedrock Titan vs Cohere) override this to branch on model.

Specified by:

encodeEmbedding in interface ProviderCodec