Class OllamaEmbeddingModel

java.lang.Object

org.springframework.ai.embedding.AbstractEmbeddingModel

org.springframework.ai.ollama.OllamaEmbeddingModel

All Implemented Interfaces:

EmbeddingModel, Model<EmbeddingRequest, EmbeddingResponse>

public class OllamaEmbeddingModel
extends AbstractEmbeddingModel

EmbeddingModel implementation for Ollama. Ollama allows developers to run large language models and generate embeddings locally. It supports open-source models available on [Ollama AI Library](...) and on Hugging Face. Please refer to the official Ollama website for the most up-to-date information on available models.

Since:

0.8.0

Author:

Christian Tzolov, Thomas Vitale, Ilayaperumal Gopinathan, Jonghoon Park

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static final class	OllamaEmbeddingModel.Builder

Field Summary

Fields inherited from class AbstractEmbeddingModel

embeddingDimensions

Constructor Summary

Constructors

Constructor	Description
OllamaEmbeddingModel(OllamaApi ollamaApi, OllamaEmbeddingOptions defaultOptions, io.micrometer.observation.ObservationRegistry observationRegistry, ModelManagementOptions modelManagementOptions)

Method Summary

Modifier and Type	Method	Description
static OllamaEmbeddingModel.Builder	builder()
EmbeddingResponse	call(EmbeddingRequest request)	Executes a method call to the AI model.
float[]	embed(Document document)	Embeds the given document's content into a vector.
void	setObservationConvention(EmbeddingModelObservationConvention observationConvention)	Use the provided convention for reporting observation data

Methods inherited from class AbstractEmbeddingModel

dimensions, dimensions

Methods inherited from class Object

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

Methods inherited from interface EmbeddingModel

embed, embed, embed, embedForResponse

Constructor Details

OllamaEmbeddingModel

public OllamaEmbeddingModel(OllamaApi ollamaApi,
 OllamaEmbeddingOptions defaultOptions,
 io.micrometer.observation.ObservationRegistry observationRegistry,
 ModelManagementOptions modelManagementOptions)

Method Details

builder

public static OllamaEmbeddingModel.Builder builder()

embed

public float[] embed(Document document)

Description copied from interface: EmbeddingModel

Embeds the given document's content into a vector.

Parameters:

document - the document to embed.

Returns:

the embedded vector.

call

public EmbeddingResponse call(EmbeddingRequest request)

Description copied from interface: Model

Executes a method call to the AI model.

Parameters:

request - the request object to be sent to the AI model

Returns:

the response from the AI model

setObservationConvention

public void setObservationConvention(EmbeddingModelObservationConvention observationConvention)

Use the provided convention for reporting observation data

Parameters:

observationConvention - The provided convention