Package org.ametys.plugins.ai.provider.impl

Class OllamaProvider

java.lang.Object

org.ametys.runtime.plugin.component.AbstractLogEnabled

org.ametys.plugins.ai.provider.impl.AbstractAIProvider

org.ametys.plugins.ai.provider.impl.OllamaProvider

All Implemented Interfaces:

AIProvider, DeferredServiceable, LogEnabled, PluginAware, Disposable, Initializable, Component, Configurable

public class OllamaProvider
extends AbstractAIProvider

Implementation of the Ollama Provider

Nested Class Summary

Nested classes/interfaces inherited from class org.ametys.plugins.ai.provider.impl.AbstractAIProvider

AbstractAIProvider.EmbeddingModelInfo

Field Summary

Fields inherited from class org.ametys.plugins.ai.provider.impl.AbstractAIProvider

_aiHelper, _embeddingModel, _embeddingModelInfos, _httpClient, _id, _imageModel, _imageModelContextWindow, _imageModelNames, _label, _pluginName, _textModel, _textModelContextWindow, _textModelNames, CONFIG_API_KEY, CONFIG_EMBEDDING_MODEL, CONFIG_IMAGE_MODEL, CONFIG_IMAGE_TIMEOUT, CONFIG_SERVER_URL, CONFIG_TEXT_MODEL

Constructor Summary

Constructors

Constructor	Description
OllamaProvider()

Method Summary

Modifier and Type	Method	Description
protected dev.langchain4j.model.chat.ChatModel	createChatModel(String modelName)	Creates the text model implementation
protected dev.langchain4j.model.embedding.EmbeddingModel	createEmbeddingModel(String modelName)	Creates the embedding model implementation
protected boolean	isEmbeddingNormalizationRequired()	Is normalization required for the embeddings of this provider

Methods inherited from class org.ametys.plugins.ai.provider.impl.AbstractAIProvider

_download, _getTokenCountEstimator, askChatbot, computeEmbedding, configure, createImageModel, deferredService, dispose, getAPIKey, getEmbeddingDimension, getEmbeddingKey, getId, getKnownEmbeddingModels, getKnownImageModels, getKnownTextModels, getLabel, getServerURL, initialize, isEmbeddingSupported, isImageGenerationSupported, normalize, setPluginInfo, textToImage, textToSummary

Methods inherited from class org.ametys.runtime.plugin.component.AbstractLogEnabled

getLogger, setLogger

Methods inherited from class java.lang.Object

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

Constructor Details

OllamaProvider

public OllamaProvider()

Method Details

createChatModel

protected dev.langchain4j.model.chat.ChatModel createChatModel(String modelName)

Description copied from class: AbstractAIProvider

Creates the text model implementation

Specified by:

createChatModel in class AbstractAIProvider

Parameters:

modelName - The model name

Returns:

The chat model

createEmbeddingModel

protected dev.langchain4j.model.embedding.EmbeddingModel createEmbeddingModel(String modelName)

Description copied from class: AbstractAIProvider

Creates the embedding model implementation

Specified by:

createEmbeddingModel in class AbstractAIProvider

Parameters:

modelName - The model name

Returns:

The embedding model

isEmbeddingNormalizationRequired

protected boolean isEmbeddingNormalizationRequired()

Description copied from class: AbstractAIProvider

Is normalization required for the embeddings of this provider

Overrides:

isEmbeddingNormalizationRequired in class AbstractAIProvider

Returns:

true if normalization is required