Package com.developer.nefarious.zjoule.plugin.chat.ollama

Class OllamaClient

java.lang.Object

com.developer.nefarious.zjoule.plugin.chat.ollama.OllamaClient

All Implemented Interfaces:

IAIClient

public class OllamaClient
extends Object
implements IAIClient

Constructor Summary

Constructors

Constructor	Description
OllamaClient(IMemoryMessageHistory memoryMessageHistory, IMemoryObject<String> memoryOllamaEndpoint, IOllamaClientHelper ollamaClientHelper)

Method Summary

Modifier and Type	Method	Description
IChatMessage	chatCompletion(List<IChatMessage> messages)	Sends a list of chat messages to the AI model and retrieves the AI-generated response.
IChatMessage	createMessage(Role role, String userPrompt)	Creates a new chat message with the specified role and content.
List<IChatMessage>	getMessageHistory()	Retrieves the chat message history.
void	setMessageHistory(List<IChatMessage> chatMessages)	Updates the chat message history with the provided list of messages.

Methods inherited from class java.lang.Object

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

Constructor Details

OllamaClient

public OllamaClient(IMemoryMessageHistory memoryMessageHistory,
 IMemoryObject<String> memoryOllamaEndpoint,
 IOllamaClientHelper ollamaClientHelper)

Method Details

chatCompletion

public IChatMessage chatCompletion(List<IChatMessage> messages)
                            throws IOException,
InterruptedException

Description copied from interface: IAIClient

Sends a list of chat messages to the AI model and retrieves the AI-generated response.

Specified by:

chatCompletion in interface IAIClient

Parameters:

messages - the list of IChatMessage objects representing the chat history and the user's current prompt.

Returns:

the AI-generated response as an IChatMessage.

Throws:

IOException - if an I/O error occurs during the request.

InterruptedException - if the operation is interrupted.

createMessage

public IChatMessage createMessage(Role role,
 String userPrompt)

Description copied from interface: IAIClient

Creates a new chat message with the specified role and content. Roles, such as Role.USER or Role.SYSTEM, determine the context of the message within the chat.

Specified by:

createMessage in interface IAIClient

Parameters:

role - the role of the message (e.g., user, system, assistant).

userPrompt - the content of the message as a String.

Returns:

a newly created IChatMessage.

getMessageHistory

public List<IChatMessage> getMessageHistory()

Description copied from interface: IAIClient

Retrieves the chat message history. This method returns a list of previously exchanged messages that are stored in memory.

Specified by:

getMessageHistory in interface IAIClient

Returns:

a list of IChatMessage objects representing the chat history.

setMessageHistory

public void setMessageHistory(List<IChatMessage> chatMessages)

Description copied from interface: IAIClient

Updates the chat message history with the provided list of messages.

Specified by:

setMessageHistory in interface IAIClient

Parameters:

chatMessages - the list of IChatMessage objects to save as the updated chat history.