github/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-04-17 17:43:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: show thought and plugin in agent process

Browse Source
This commit is contained in:
zhayujie
2023-10-27 12:28:34 +08:00
parent 701daedf49
commit 34e06fcbf8
2 changed files with 18 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
bot/linkai/link_ai_bot.py
View File

@@ -211,17 +211,27 @@ class LinkAIBot(Bot, OpenAIImage):
def _fetch_agent_suffix(self, response): def _fetch_agent_suffix(self, response):
try: try:
if response.get("agent") and response.get("agent").get("chain"): plugin_list = []
logger.debug(f"[LinkAgent] res={response}")
if response.get("agent") and response.get("agent").get("chain") and response.get("agent").get("need_show_plugin"):
chain = response.get("agent").get("chain") chain = response.get("agent").get("chain")
suffix = "\n\n---------\n🧠思考过程" suffix = "\n\n- - - - - - - - - - - -"
i = 0
for turn in chain: for turn in chain:
suffix += "\n\n" plugin_name = turn.get('plugin_name')
if turn.get("thought"): suffix += "\n"
suffix += f"{turn.get('thought')}" need_show_thought = response.get("agent").get("need_show_thought")
if turn.get('plugin_name'): if turn.get("thought") and plugin_name and need_show_thought:
suffix += f"\n{turn.get('plugin_icon')} 使用 {turn.get('plugin_name')}" suffix += f"{turn.get('thought')}\n"
if plugin_name:
plugin_list.append(turn.get('plugin_name'))
suffix += f"{turn.get('plugin_icon')} {turn.get('plugin_name')}"
if turn.get('plugin_input'): if turn.get('plugin_input'):
suffix += f"，输入 {turn.get('plugin_input')}" suffix += f"：{turn.get('plugin_input')}"
if i < len(chain) - 1:
suffix += "\n"
i += 1
logger.info(f"[LinkAgent] use plugins: {plugin_list}")
return suffix return suffix
except Exception as e: except Exception as e:
logger.exception(e) logger.exception(e)

5
plugins/linkai/config.json.template
View File

@@ -15,10 +15,5 @@
"enabled": true, "enabled": true,
"group_enabled": true, "group_enabled": true,
"max_file_size": 5000 "max_file_size": 5000
},
"knowledge_base": {
"search_miss_text_enabled": false,
"search_miss_similarity": 0.8,
"search_miss_suffix": "\n-------------\n回复仅供参考，请以官方文档为准"
} }
} }