Package com.bytedesk.ai.zhipuai

Class ZhipuaiMultiModelService

java.lang.Object

com.bytedesk.ai.service.BaseSpringAIService

com.bytedesk.ai.zhipuai.ZhipuaiMultiModelService

All Implemented Interfaces:

SpringAIService

@Service
public class ZhipuaiMultiModelService
extends BaseSpringAIService

Field Summary

Fields

Modifier and Type	Field	Description
private static final String	DEFAULT_MULTI_MODEL
private ai.z.openapi.ZhipuAiClient	defaultClient
private LlmProviderRestService	llmProviderRestService
private TokenUsageHelper	tokenUsageHelper
private static final String	ZAI_FILE_URL
private static final String	ZAI_IMAGE_URL
private static final String	ZAI_ROLE_SYSTEM
private static final String	ZAI_TEXT
private static final String	ZAI_THINKING_DISABLED
private static final String	ZAI_THINKING_ENABLED
private static final String	ZAI_VIDEO_URL

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

applicationEventPublisher, chunkElasticService, chunkVectorService, faqElasticService, faqVectorService, knowledgeBaseSearchHelper, messagePersistCache, messagePersistenceHelper, messageRestService, messageSendService, promptHelper, robotMessageCache, robotRestService, sseMessageHelper, textElasticService, textVectorService, threadRestService, uidUtils, webpageElasticService, webpageVectorService

Constructor Summary

Constructors

Constructor	Description
ZhipuaiMultiModelService()

Method Summary

Modifier and Type	Method	Description
private List<ai.z.openapi.service.model.MessageContent>	buildUserContents(String text)
private List<ai.z.openapi.service.model.MessageContent>	buildUserContentsFromMessage(MessageProtobuf messageProtobufQuery)
private List<ai.z.openapi.service.model.ChatMessage>	buildZaiMessagesFromPrompt(org.springframework.ai.chat.prompt.Prompt prompt)
private ai.z.openapi.ZhipuAiClient	createDynamicClient(RobotLlm llm)	根据机器人配置动态创建 ZhipuAiClient，优先使用 provider apiKey，失败则回退默认 Bean。
private long	estimateTokens(String text)
private String	extractDeltaText(ai.z.openapi.service.model.Delta delta)
private String	extractFinalTextFromResponse(ai.z.openapi.service.model.ChatCompletionResponse response)
private String	getModel(RobotProtobuf robot)
String	processMultiModalSyncRequest(MessageProtobuf messageProtobuf, RobotProtobuf robot, boolean searchKnowledgeBase)	多模态同步请求处理，支持图片等媒体类型
protected void	processPromptSse(org.springframework.ai.chat.prompt.Prompt prompt, RobotProtobuf robot, MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply, List<RobotContent.SourceReference> sourceReferences, org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter)
protected String	processPromptSync(String message, RobotProtobuf robot)
protected void	processPromptWebsocket(org.springframework.ai.chat.prompt.Prompt prompt, RobotProtobuf robot, MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply)
private String	stripThinkTags(String text)

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

processSyncRequest, sendSseMessage, 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

defaultClient

@Autowired(required=false)
@Qualifier("zhipuAiClient")
private ai.z.openapi.ZhipuAiClient defaultClient

tokenUsageHelper

@Autowired
private TokenUsageHelper tokenUsageHelper

DEFAULT_MULTI_MODEL

private static final String DEFAULT_MULTI_MODEL

See Also:

• Constant Field Values

ZAI_ROLE_SYSTEM

private static final String ZAI_ROLE_SYSTEM

See Also:

• Constant Field Values

ZAI_THINKING_ENABLED

private static final String ZAI_THINKING_ENABLED

See Also:

• Constant Field Values

ZAI_THINKING_DISABLED

private static final String ZAI_THINKING_DISABLED

See Also:

• Constant Field Values

ZAI_TEXT

private static final String ZAI_TEXT

See Also:

• Constant Field Values

ZAI_IMAGE_URL

private static final String ZAI_IMAGE_URL

See Also:

• Constant Field Values

ZAI_VIDEO_URL

private static final String ZAI_VIDEO_URL

See Also:

• Constant Field Values

ZAI_FILE_URL

private static final String ZAI_FILE_URL

See Also:

• Constant Field Values

Constructor Details

ZhipuaiMultiModelService

public ZhipuaiMultiModelService()

Method Details

getModel

private String getModel(RobotProtobuf robot)

createDynamicClient

private ai.z.openapi.ZhipuAiClient createDynamicClient(RobotLlm llm)

根据机器人配置动态创建 ZhipuAiClient，优先使用 provider apiKey，失败则回退默认 Bean。

buildZaiMessagesFromPrompt

private List<ai.z.openapi.service.model.ChatMessage> buildZaiMessagesFromPrompt(org.springframework.ai.chat.prompt.Prompt prompt)

buildUserContentsFromMessage

private List<ai.z.openapi.service.model.MessageContent> buildUserContentsFromMessage(MessageProtobuf messageProtobufQuery)

buildUserContents

private List<ai.z.openapi.service.model.MessageContent> buildUserContents(String text)

extractFinalTextFromResponse

private String extractFinalTextFromResponse(ai.z.openapi.service.model.ChatCompletionResponse response)

extractDeltaText

private String extractDeltaText(ai.z.openapi.service.model.Delta delta)

stripThinkTags

private String stripThinkTags(String text)

estimateTokens

private long estimateTokens(String text)

processPromptWebsocket

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

Specified by:

processPromptWebsocket in class BaseSpringAIService

processPromptSync

protected String processPromptSync(String message,
 RobotProtobuf robot)

Specified by:

processPromptSync in class BaseSpringAIService

processMultiModalSyncRequest

public String processMultiModalSyncRequest(MessageProtobuf messageProtobuf,
 RobotProtobuf robot,
 boolean searchKnowledgeBase)

多模态同步请求处理，支持图片等媒体类型

processPromptSse

protected void processPromptSse(org.springframework.ai.chat.prompt.Prompt prompt,
 RobotProtobuf robot,
 MessageProtobuf messageProtobufQuery,
 MessageProtobuf messageProtobufReply,
 List<RobotContent.SourceReference> sourceReferences,
 org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter)

Specified by:

processPromptSse in class BaseSpringAIService