Package com.bytedesk.ai.springai.providers.ollama

Class SpringAIOllamaChatService

java.lang.Object

com.bytedesk.ai.springai.service.BaseSpringAIService

com.bytedesk.ai.springai.providers.ollama.SpringAIOllamaChatService

All Implemented Interfaces:

SpringAIService

@Service
@ConditionalOnProperty(prefix="spring.ai.ollama.chat",
                       name="enabled",
                       havingValue="true",
                       matchIfMissing=false)
public class SpringAIOllamaChatService
extends BaseSpringAIService

Nested Class Summary

Nested classes/interfaces inherited from class com.bytedesk.ai.springai.service.BaseSpringAIService

BaseSpringAIService.PromptResult

Field Summary

Fields

Modifier and Type	Field	Description
private org.springframework.ai.ollama.api.OllamaApi	bytedeskOllamaApi
private org.springframework.ai.ollama.OllamaChatModel	bytedeskOllamaChatModel

Fields inherited from class com.bytedesk.ai.springai.service.BaseSpringAIService

applicationEventPublisher, chunkElasticService, chunkVectorService, faqElasticService, faqVectorService, messagePersistCache, messageRestService, messageSendService, robotMessageCache, robotRestService, textElasticService, textVectorService, threadRestService, uidUtils

Constructor Summary

Constructors

Constructor	Description
SpringAIOllamaChatService()

Method Summary

Modifier and Type	Method	Description
org.springframework.ai.ollama.api.OllamaApi	createOllamaApi(String apiUrl)
private org.springframework.ai.ollama.api.OllamaOptions	createOllamaOptions(RobotLlm llm)	根据机器人配置创建动态的OllamaOptions
Boolean	isModelExists(OllamaRequest request)	检查模型是否存在
protected void	processPromptSse(org.springframework.ai.chat.prompt.Prompt prompt, RobotProtobuf robot, MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply, org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter, String fullPromptContent)
protected String	processPromptSync(String message, RobotProtobuf robot, String fullPromptContent)
protected void	processPromptWebsocket(org.springframework.ai.chat.prompt.Prompt prompt, RobotProtobuf robot, MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply, String fullPromptContent)	Token usage data class

Methods inherited from class com.bytedesk.ai.springai.service.BaseSpringAIService

createDynamicOptions, extractFullPromptContent, extractTextFromResponse, extractTokenUsage, getTokenUnitPrice, handleSseError, isEmitterCompleted, persistMessage, persistMessage, persistMessage, persistMessage, processDirectLlmRequest, processLlmResponseWebsocket, publishAiTokenUsageEvent, recordAiTokenUsage, searchKnowledgeBase, sendMessageWebsocket, sendSseMessage, sendStreamEndMessage, sendStreamMessage, sendStreamStartMessage, sendSyncMessage, sendWebsocketMessage

Methods inherited from class java.lang.Object

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

Field Details

bytedeskOllamaApi

@Autowired
@Qualifier("bytedeskOllamaApi")
private org.springframework.ai.ollama.api.OllamaApi bytedeskOllamaApi

bytedeskOllamaChatModel

@Autowired(required=false)
@Qualifier("bytedeskOllamaChatModel")
private org.springframework.ai.ollama.OllamaChatModel bytedeskOllamaChatModel

Constructor Details

SpringAIOllamaChatService

public SpringAIOllamaChatService()

Method Details

createOllamaOptions

private org.springframework.ai.ollama.api.OllamaOptions createOllamaOptions(RobotLlm llm)

根据机器人配置创建动态的OllamaOptions

Parameters:

llm - 机器人LLM配置

Returns:

根据机器人配置创建的选项

createOllamaApi

public org.springframework.ai.ollama.api.OllamaApi createOllamaApi(String apiUrl)

isModelExists

public Boolean isModelExists(OllamaRequest request)

检查模型是否存在

Parameters:

model - 模型名称

Returns:

如果模型存在返回true，否则返回false

processPromptWebsocket

protected void processPromptWebsocket(org.springframework.ai.chat.prompt.Prompt prompt,
 RobotProtobuf robot,
 MessageProtobuf messageProtobufQuery,
 MessageProtobuf messageProtobufReply,
 String fullPromptContent)

Description copied from class: BaseSpringAIService

Token usage data class

Specified by:

processPromptWebsocket in class BaseSpringAIService

processPromptSync

protected String processPromptSync(String message,
 RobotProtobuf robot,
 String fullPromptContent)

Specified by:

processPromptSync in class BaseSpringAIService

processPromptSse

protected void processPromptSse(org.springframework.ai.chat.prompt.Prompt prompt,
 RobotProtobuf robot,
 MessageProtobuf messageProtobufQuery,
 MessageProtobuf messageProtobufReply,
 org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter,
 String fullPromptContent)

Specified by:

processPromptSse in class BaseSpringAIService