import gradio as gr # from demo.gpt_asyn_http import gpt_call_fn # from gpt_syn_http import gpt_syn_http_fn import time import random from call_theta import call_theta_message from skillAgent import SkillAgent skill_agent = SkillAgent() messages = [{"role":"system","content":skill_agent.get_system_prompt()}] def on_radio1_select(choice, chat_history, log_text): tool_result = f"tool_result: 用户选择交易：{choice}" messages.append({"role": "user", "content": tool_result}) chat_history.append({"role": "user", "content": choice}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) while True: log_text += "\n" + "思考中...." res = skill_agent.call(messages) # print("*-" * 50) # print(messages, res) if res["type"] == "answer": log_text += "\n" + "回复生成中...." chat_history.append({"role": "assistant", "content": res["content"]}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) break elif res["type"] == "skill": messages.append({"role": "assistant", "content": res["raw_content"]}) log_text += "\n" + f"读取skill({res["content"]})信息中...." with open("skills/资金撤回类问题解答/Skill.md", "r", encoding='utf-8') as f: reference = f.read() reference = "Skill信息：" + reference messages.append({"role": "user", "content": reference}) elif res["type"] == "md": messages.append({"role": "assistant", "content": res["raw_content"]}) log_text += "\n" + f"读取markdown({res["content"]})信息中...." if "红包资金撤回" in res["content"]: with open("skills/资金撤回类问题解答/红包资金撤回问题解答思路.md", "r", encoding='utf-8') as f: reference = f.read() reference = "参考markdown信息：" + reference messages.append({"role": "user", "content": reference}) else: log_text += "\n" + f"读取失败" chat_history.append({"role": "assistant", "content": "很抱歉，关于这方面的内容我暂时无法回答您，您看还有别的问题吗？"}) reference = "参考markdown信息：读取失败，信息缺失" messages.append({"role": "user", "content": reference}) messages.append({"role": "assistant", "content": "很抱歉，关于这方面的内容我暂时无法回答您，您看还有别的问题吗？"}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) break elif res["type"] == "tool": log_text += "\n" + f"发送选项工具({res["content"]})" if "bill_search_tool" in res["content"]: chat_history.append({"role": "assistant", "content": "请选择你想咨询的交易"}) yield "", chat_history, log_text, gr.update(visible=True), gr.update(visible=False), gr.update(visible=False) break elif "red_envelope_search_tool" in res["content"]: chat_history.append({"role": "assistant", "content": "请选择你想撤回的红包"}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=True), gr.update(visible=False) break elif "red_envelope_withdraw_tool" in res["content"]: chat_history.append({"role": "assistant", "content": "请选择红包撤回是否成功"}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=True) break def on_radio2_select(choice, chat_history, log_text): tool_result = f"tool_result: 用户选择红包：{choice}" messages.append({"role": "user", "content": tool_result}) chat_history.append({"role": "user", "content": choice}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) while True: log_text += "\n" + "思考中...." res = skill_agent.call(messages) # print("*-" * 50) # print(messages, res) if res["type"] == "answer": log_text += "\n" + "回复生成中...." chat_history.append({"role": "assistant", "content": res["content"]}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) break elif res["type"] == "skill": messages.append({"role": "assistant", "content": res["raw_content"]}) log_text += "\n" + f"读取skill({res["content"]})信息中...." with open("skills/资金撤回类问题解答/Skill.md", "r", encoding='utf-8') as f: reference = f.read() reference = "Skill信息：" + reference messages.append({"role": "user", "content": reference}) elif res["type"] == "md": messages.append({"role": "assistant", "content": res["raw_content"]}) log_text += "\n" + f"读取markdown({res["content"]})信息中...." if "红包资金撤回" in res["content"]: with open("skills/资金撤回类问题解答/红包资金撤回问题解答思路.md", "r", encoding='utf-8') as f: reference = f.read() reference = "参考markdown信息：" + reference messages.append({"role": "user", "content": reference}) else: log_text += "\n" + f"读取失败" chat_history.append({"role": "assistant", "content": "很抱歉，关于这方面的内容我暂时无法回答您，您看还有别的问题吗？"}) reference = "参考markdown信息：读取失败，信息缺失" messages.append({"role": "user", "content": reference}) messages.append({"role": "assistant", "content": "很抱歉，关于这方面的内容我暂时无法回答您，您看还有别的问题吗？"}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) break elif res["type"] == "tool": log_text += "\n" + f"发送选项工具({res["content"]})" if "bill_search_tool" in res["content"]: chat_history.append({"role": "assistant", "content": "请选择你想咨询的交易"}) yield "", chat_history, log_text, gr.update(visible=True), gr.update(visible=False), gr.update(visible=False) break elif "red_envelope_search_tool" in res["content"]: chat_history.append({"role": "assistant", "content": "请选择你想撤回的红包"}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=True), gr.update(visible=False) break elif "red_envelope_withdraw_tool" in res["content"]: chat_history.append({"role": "assistant", "content": "请选择红包撤回是否成功"}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=True) break def on_radio3_select(choice, chat_history, log_text): tool_result = f"tool_result: 用户红包撤回结果：{choice}， 当前用户还有10个红包可撤回" messages.append({"role": "user", "content": tool_result}) chat_history.append({"role": "user", "content": choice}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) while True: log_text += "\n" + "思考中...." res = skill_agent.call(messages) # print("*-" * 50) # print(messages, res) if res["type"] == "answer": log_text += "\n" + "回复生成中...." chat_history.append({"role": "assistant", "content": res["content"]}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) break elif res["type"] == "skill": messages.append({"role": "assistant", "content": res["raw_content"]}) log_text += "\n" + f"读取skill({res["content"]})信息中...." with open("skills/资金撤回类问题解答/Skill.md", "r", encoding='utf-8') as f: reference = f.read() reference = "Skill信息：" + reference messages.append({"role": "user", "content": reference}) elif res["type"] == "md": messages.append({"role": "assistant", "content": res["raw_content"]}) log_text += "\n" + f"读取markdown({res["content"]})信息中...." if "红包资金撤回" in res["content"]: with open("skills/资金撤回类问题解答/红包资金撤回问题解答思路.md", "r", encoding='utf-8') as f: reference = f.read() reference = "参考markdown信息：" + reference messages.append({"role": "user", "content": reference}) else: log_text += "\n" + f"读取失败" chat_history.append({"role": "assistant", "content": "很抱歉，关于这方面的内容我暂时无法回答您，您看还有别的问题吗？"}) reference = "参考markdown信息：读取失败，信息缺失" messages.append({"role": "user", "content": reference}) messages.append({"role": "assistant", "content": "很抱歉，关于这方面的内容我暂时无法回答您，您看还有别的问题吗？"}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) break elif res["type"] == "tool": log_text += "\n" + f"发送选项工具({res["content"]})" if "bill_search_tool" in res["content"]: chat_history.append({"role": "assistant", "content": "请选择你想咨询的交易"}) yield "", chat_history, log_text, gr.update(visible=True), gr.update(visible=False), gr.update(visible=False) break elif "red_envelope_search_tool" in res["content"]: chat_history.append({"role": "assistant", "content": "请选择你想撤回的红包"}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=True), gr.update(visible=False) break elif "red_envelope_withdraw_tool" in res["content"]: chat_history.append({"role": "assistant", "content": "请选择红包撤回是否成功"}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=True) break def clear_msg(): global messages messages = [{"role":"system","content":skill_agent.get_system_prompt()}] with gr.Blocks() as demo: with gr.Row(): with gr.Column(scale=2): chatbot = gr.Chatbot() msg = gr.Textbox() clear = gr.ClearButton([msg, chatbot]) radio1 = gr.Radio(choices=["红包交易1", "收钱码交易2", "余额提现交易3"], interactive=True, visible=False) radio2 = gr.Radio(choices=["红包交易1，已撤回", "红包交易2，未撤回", "红包交易3，已领取"], interactive=True, visible=False) radio3 = gr.Radio(choices=["红包撤回成功", "红包撤回失败"], interactive=True, visible=False) with gr.Column(scale=1): log_display = gr.Textbox( label="执行日志", interactive=False, max_lines=20, lines=20 ) def respond(message, chat_history, log_text): chat_history.append({"role": "user", "content": message}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) messages.append({"role":"user","content":message}) while True: log_text += "\n" + "思考中...." res = skill_agent.call(messages) # print("*-" * 50) print(messages, res) if res["type"] == "answer": log_text += "\n" + "回复生成中...." chat_history.append({"role": "assistant", "content": res["content"]}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) break elif res["type"] == "skill": messages.append({"role": "assistant", "content": res["raw_content"]}) log_text += "\n" + f"读取skill({res["content"]})信息中...." with open("skills/资金撤回类问题解答/Skill.md", "r", encoding='utf-8') as f: reference = f.read() reference = "Skill信息：" + reference messages.append({"role": "user", "content": reference}) elif res["type"] == "md": messages.append({"role": "assistant", "content": res["raw_content"]}) log_text += "\n" + f"读取markdown({res["content"]})信息中...." if "红包资金撤回" in res["content"]: with open("skills/资金撤回类问题解答/红包资金撤回问题解答思路.md", "r", encoding='utf-8') as f: reference = f.read() reference = "参考markdown信息：" + reference messages.append({"role": "user", "content": reference}) else: log_text += "\n" + f"读取失败" chat_history.append({"role": "assistant", "content": "很抱歉，关于这方面的内容我暂时无法回答您，您看还有别的问题吗？"}) reference = "参考markdown信息：读取失败，信息缺失" messages.append({"role": "user", "content": reference}) messages.append({"role": "assistant", "content": "很抱歉，关于这方面的内容我暂时无法回答您，您看还有别的问题吗？"}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) break elif res["type"] == "tool": log_text += "\n" + f"发送选项工具({res["content"]})" if "bill_search_tool" in res["content"]: chat_history.append({"role": "assistant", "content": "请选择你想咨询的交易"}) yield "", chat_history, log_text, gr.update(visible=True), gr.update(visible=False), gr.update(visible=False) break elif "red_envelope_search_tool" in res["content"]: chat_history.append({"role": "assistant", "content": "请选择你想撤回的红包"}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=True), gr.update(visible=False) break elif "red_envelope_withdraw_tool" in res["content"]: chat_history.append({"role": "assistant", "content": "请选择红包撤回是否成功"}) yield "", chat_history, log_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=True) break # return "", chat_history, log_text, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False) msg.submit(respond, [msg, chatbot, log_display], [msg, chatbot, log_display, radio1, radio2, radio3]) clear.click(clear_msg) radio1.change( on_radio1_select, [radio1, chatbot, log_display], [msg, chatbot, log_display, radio1, radio2, radio3], queue=True ) radio2.change( on_radio2_select, [radio2, chatbot, log_display], [msg, chatbot, log_display, radio1, radio2, radio3], # queue=True ) radio3.change( on_radio3_select, [radio3, chatbot, log_display], [msg, chatbot, log_display, radio1, radio2, radio3], # queue=True ) demo.launch(share=True)