|
|
@@ -94,9 +94,13 @@ class LinkAIBot(Bot, OpenAIImage): |
|
|
|
response = res.json() |
|
|
|
reply_content = response["choices"][0]["message"]["content"] |
|
|
|
total_tokens = response["usage"]["total_tokens"] |
|
|
|
suffix = self._fecth_knowledge_search_suffix(response) |
|
|
|
if suffix: |
|
|
|
reply_content += suffix |
|
|
|
agent_suffix = self._fetch_agent_suffix(response) |
|
|
|
if agent_suffix: |
|
|
|
reply_content += agent_suffix |
|
|
|
if not agent_suffix: |
|
|
|
knowledge_suffix = self._fetch_knowledge_search_suffix(response) |
|
|
|
if knowledge_suffix: |
|
|
|
reply_content += knowledge_suffix |
|
|
|
logger.info(f"[LINKAI] reply={reply_content}, total_tokens={total_tokens}") |
|
|
|
self.sessions.session_reply(reply_content, session_id, total_tokens) |
|
|
|
return Reply(ReplyType.TEXT, reply_content) |
|
|
@@ -188,14 +192,14 @@ class LinkAIBot(Bot, OpenAIImage): |
|
|
|
return self.reply_text(session, app_code, retry_count + 1) |
|
|
|
|
|
|
|
|
|
|
|
def _fecth_knowledge_search_suffix(self, response) -> str: |
|
|
|
def _fetch_knowledge_search_suffix(self, response) -> str: |
|
|
|
try: |
|
|
|
if response.get("knowledge_base"): |
|
|
|
search_hit = response.get("knowledge_base").get("search_hit") |
|
|
|
first_similarity = response.get("knowledge_base").get("first_similarity") |
|
|
|
logger.info(f"[LINKAI] knowledge base, search_hit={search_hit}, first_similarity={first_similarity}") |
|
|
|
plugin_config = pconf("linkai") |
|
|
|
if plugin_config.get("knowledge_base") and plugin_config.get("knowledge_base").get("search_miss_text_enabled"): |
|
|
|
if plugin_config and plugin_config.get("knowledge_base") and plugin_config.get("knowledge_base").get("search_miss_text_enabled"): |
|
|
|
search_miss_similarity = plugin_config.get("knowledge_base").get("search_miss_similarity") |
|
|
|
search_miss_text = plugin_config.get("knowledge_base").get("search_miss_suffix") |
|
|
|
if not search_hit: |
|
|
@@ -204,3 +208,20 @@ class LinkAIBot(Bot, OpenAIImage): |
|
|
|
return search_miss_text |
|
|
|
except Exception as e: |
|
|
|
logger.exception(e) |
|
|
|
|
|
|
|
def _fetch_agent_suffix(self, response): |
|
|
|
try: |
|
|
|
if response.get("agent") and response.get("agent").get("chain"): |
|
|
|
chain = response.get("agent").get("chain") |
|
|
|
suffix = "\n\n---------\n🧠思考过程" |
|
|
|
for turn in chain: |
|
|
|
suffix += "\n\n" |
|
|
|
if turn.get("thought"): |
|
|
|
suffix += f"{turn.get('thought')}" |
|
|
|
if turn.get('plugin_name'): |
|
|
|
suffix += f"\n{turn.get('plugin_icon')} 使用 {turn.get('plugin_name')}" |
|
|
|
if turn.get('plugin_input'): |
|
|
|
suffix += f",输入 {turn.get('plugin_input')}" |
|
|
|
return suffix |
|
|
|
except Exception as e: |
|
|
|
logger.exception(e) |