From db34fb5f1f0916d47bd176a97f6951ce6d6853f0 Mon Sep 17 00:00:00 2001 From: lhpqaq <657407891@qq.com> Date: Thu, 2 Jan 2025 22:22:01 +0800 Subject: [PATCH] remove convert --- .../manager/ai/assistant/GeneralAssistantFactory.java | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/bigtop-manager-ai/bigtop-manager-ai-assistant/src/main/java/org/apache/bigtop/manager/ai/assistant/GeneralAssistantFactory.java b/bigtop-manager-ai/bigtop-manager-ai-assistant/src/main/java/org/apache/bigtop/manager/ai/assistant/GeneralAssistantFactory.java index 6bce79bd..bf4b5290 100644 --- a/bigtop-manager-ai/bigtop-manager-ai-assistant/src/main/java/org/apache/bigtop/manager/ai/assistant/GeneralAssistantFactory.java +++ b/bigtop-manager-ai/bigtop-manager-ai-assistant/src/main/java/org/apache/bigtop/manager/ai/assistant/GeneralAssistantFactory.java @@ -20,7 +20,6 @@ import org.apache.bigtop.manager.ai.assistant.config.GeneralAssistantConfig; import org.apache.bigtop.manager.ai.assistant.provider.ChatMemoryStoreProvider; -import org.apache.bigtop.manager.ai.assistant.provider.LocSystemPromptProvider; import org.apache.bigtop.manager.ai.core.AbstractAIAssistantFactory; import org.apache.bigtop.manager.ai.core.config.AIAssistantConfig; import org.apache.bigtop.manager.ai.core.enums.PlatformType; @@ -50,15 +49,14 @@ public class GeneralAssistantFactory extends AbstractAIAssistantFactory { private ChatMemoryStoreProvider chatMemoryStoreProvider; private void configureSystemPrompt(AIAssistant.Builder builder, SystemPrompt systemPrompt, String locale) { - LocSystemPromptProvider locSystemPromptProvider = (LocSystemPromptProvider) systemPromptProvider; List systemPrompts = new ArrayList<>(); if (systemPrompt != null) { - systemPrompts.add(locSystemPromptProvider.getSystemMessage(systemPrompt)); + systemPrompts.add(systemPromptProvider.getSystemMessage(systemPrompt)); } if (locale != null) { - systemPrompts.add(locSystemPromptProvider.getLanguagePrompt(locale)); + systemPrompts.add(systemPromptProvider.getLanguagePrompt(locale)); } - builder.withSystemPrompt(locSystemPromptProvider.getSystemMessages(systemPrompts)); + builder.withSystemPrompt(systemPromptProvider.getSystemMessages(systemPrompts)); } private AIAssistant.Builder initializeBuilder(PlatformType platformType) {