Package com.bytedesk.ai.springai.providers.zhipuai

Class SpringAIZhipuaiService

java.lang.Object

com.bytedesk.ai.springai.service.BaseSpringAIService

com.bytedesk.ai.springai.providers.zhipuai.SpringAIZhipuaiService

All Implemented Interfaces:

SpringAIService

@Service
public class SpringAIZhipuaiService
extends BaseSpringAIService

Spring AI ZhiPuAI服务

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 LlmProviderRestService	llmProviderRestService

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
SpringAIZhipuaiService()

Method Summary

Modifier and Type	Method	Description
org.springframework.ai.zhipuai.api.ZhiPuAiApi	createZhipuaiApi(String apiKey)
private org.springframework.ai.zhipuai.ZhiPuAiChatModel	createZhipuaiChatModel(RobotLlm llm)	根据机器人配置创建动态的ZhiPuAiChatModel
private org.springframework.ai.zhipuai.ZhiPuAiChatOptions	createZhipuaiOptions(RobotLlm llm)	根据机器人配置创建动态的ZhiPuAiChatOptions
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

llmProviderRestService

@Autowired
private LlmProviderRestService llmProviderRestService

Constructor Details

SpringAIZhipuaiService

public SpringAIZhipuaiService()

Method Details

createZhipuaiOptions

private org.springframework.ai.zhipuai.ZhiPuAiChatOptions createZhipuaiOptions(RobotLlm llm)

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

Parameters:

llm - 机器人LLM配置

Returns:

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

createZhipuaiApi

public org.springframework.ai.zhipuai.api.ZhiPuAiApi createZhipuaiApi(String apiKey)

createZhipuaiChatModel

private org.springframework.ai.zhipuai.ZhiPuAiChatModel createZhipuaiChatModel(RobotLlm llm)

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

Parameters:

llm - 机器人LLM配置

Returns:

配置了特定模型的ZhiPuAiChatModel

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