mirror of
https://github.com/xinnan-tech/xiaozhi-esp32-server.git
synced 2026-07-23 23:53:55 +08:00
2b662d3a14
* 优化function call消息处理 * fix:用户说话重复记录bug * 2025-3-10-优化llm intent (#254) Co-authored-by: 欣南科技 <huangrongzhuang@xin-nan.com> * update:回复上版提示词，无需再二次识别歌曲名 --------- Co-authored-by: 玄凤科技 <eric230308@gmail.com> Co-authored-by: hrz <1710360675@qq.com> Co-authored-by: Jiao Haoyang <108573524+XuSenfeng@users.noreply.github.com>
35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
from abc import ABC, abstractmethod
|
|
from typing import List, Dict
|
|
from config.logger import setup_logging
|
|
|
|
TAG = __name__
|
|
logger = setup_logging()
|
|
|
|
|
|
class IntentProviderBase(ABC):
|
|
def __init__(self, config):
|
|
self.config = config
|
|
self.intent_options = config.get("intent_options", {
|
|
"continue_chat": "继续聊天",
|
|
"end_chat": "结束聊天",
|
|
"play_music": "播放音乐"
|
|
})
|
|
|
|
def set_llm(self, llm):
|
|
self.llm = llm
|
|
logger.bind(tag=TAG).debug("Set LLM for intent provider")
|
|
|
|
@abstractmethod
|
|
async def detect_intent(self, dialogue_history: List[Dict], text: str) -> str:
|
|
"""
|
|
检测用户最后一句话的意图
|
|
Args:
|
|
dialogue_history: 对话历史记录列表，每条记录包含role和content
|
|
Returns:
|
|
返回识别出的意图，格式为:
|
|
- "继续聊天"
|
|
- "结束聊天"
|
|
- "播放音乐 歌名" 或 "随机播放音乐"
|
|
"""
|
|
pass
|