mirror of
https://github.com/xinnan-tech/xiaozhi-esp32-server.git
synced 2026-07-27 17:43:55 +08:00
Update base.py
This commit is contained in:
@@ -12,12 +12,22 @@ class IntentProviderBase(ABC):
|
|||||||
self.intent_options = config.get("intent_options", {
|
self.intent_options = config.get("intent_options", {
|
||||||
"continue_chat": "继续聊天",
|
"continue_chat": "继续聊天",
|
||||||
"end_chat": "结束聊天",
|
"end_chat": "结束聊天",
|
||||||
"play_music": "播放音乐"
|
"play_music": "播放音乐",
|
||||||
|
"get_weather": "查询天气",
|
||||||
|
"get_news": "查询新闻"
|
||||||
})
|
})
|
||||||
|
|
||||||
def set_llm(self, llm):
|
def set_llm(self, llm):
|
||||||
self.llm = 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
|
@abstractmethod
|
||||||
async def detect_intent(self, conn, dialogue_history: List[Dict], text: str) -> str:
|
async def detect_intent(self, conn, dialogue_history: List[Dict], text: str) -> str:
|
||||||
@@ -30,5 +40,6 @@ class IntentProviderBase(ABC):
|
|||||||
- "继续聊天"
|
- "继续聊天"
|
||||||
- "结束聊天"
|
- "结束聊天"
|
||||||
- "播放音乐 歌名" 或 "随机播放音乐"
|
- "播放音乐 歌名" 或 "随机播放音乐"
|
||||||
|
- "查询天气 地点名" 或 "查询天气 [当前位置]"
|
||||||
"""
|
"""
|
||||||
pass
|
pass
|
||||||
|
|||||||
Reference in New Issue
Block a user