From 63a3c33df93d09b252fce6cf53e37781b14d23c4 Mon Sep 17 00:00:00 2001 From: koalalgx <82762701+koalalgx@users.noreply.github.com> Date: Tue, 25 Mar 2025 14:27:10 +0800 Subject: [PATCH] Update base.py --- main/xiaozhi-server/core/providers/intent/base.py | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/main/xiaozhi-server/core/providers/intent/base.py b/main/xiaozhi-server/core/providers/intent/base.py index a691d6cb..877dfced 100644 --- a/main/xiaozhi-server/core/providers/intent/base.py +++ b/main/xiaozhi-server/core/providers/intent/base.py @@ -12,12 +12,22 @@ class IntentProviderBase(ABC): self.intent_options = config.get("intent_options", { "continue_chat": "继续聊天", "end_chat": "结束聊天", - "play_music": "播放音乐" + "play_music": "播放音乐", + "get_weather": "查询天气", + "get_news": "查询新闻" }) def set_llm(self, llm): self.llm = llm - logger.bind(tag=TAG).debug("Set LLM for intent provider") + # 获取模型名称和类型信息 + model_name = getattr(llm, 'model_name', str(llm.__class__.__name__)) + model_type = getattr(llm, 'type', 'unknown') + # 记录更详细的日志 + logger.bind(tag=TAG).info(f"意图识别设置LLM: {model_name}, 类型: {model_type}") + # 尝试获取模型基础URL + base_url = getattr(llm, 'base_url', 'N/A') + if base_url != 'N/A': + logger.bind(tag=TAG).debug(f"意图识别LLM基础URL: {base_url}") @abstractmethod async def detect_intent(self, conn, dialogue_history: List[Dict], text: str) -> str: @@ -30,5 +40,6 @@ class IntentProviderBase(ABC): - "继续聊天" - "结束聊天" - "播放音乐 歌名" 或 "随机播放音乐" + - "查询天气 地点名" 或 "查询天气 [当前位置]" """ pass