pycoze 0.1.308__py3-none-any.whl → 0.1.309__py3-none-any.whl

pycoze/bot/agent_chat.py

@@ -1,110 +0,0 @@
-import json
-from langchain_openai import ChatOpenAI
-from .agent import run_agent, Runnable,CHAT_DATA, clear_chat_data
-import asyncio
-from pycoze import utils
-from pycoze.reference.bot import ref_bot
-from pycoze.reference.tool import ref_tools
-from pycoze.reference.workflow import ref_workflow
-from langchain_core.utils.function_calling import convert_to_openai_tool
-import os
-
-cfg = utils.read_json_file("llm.json")
-
-def load_bot_setting(bot_setting_file: str):
-    with open(bot_setting_file, "r", encoding="utf-8") as f:
-        return json.load(f)
-
-
-def load_abilities(bot_setting_file: str):
-    with open(bot_setting_file, "r", encoding="utf-8") as f:
-        bot_setting = json.load(f)
-
-    abilities = []
-    for bot_id in bot_setting["bots"]:
-        bot = ref_bot(bot_id, as_agent_tool=True)
-        if bot:
-            abilities.append(bot)
-    for tool_id in bot_setting["tools"]:
-        abilities.extend(ref_tools(tool_id, as_agent_tool=True))
-    for workflow_id in bot_setting["workflows"]:
-        workflow = ref_workflow(workflow_id, as_agent_tool=True)
-        if workflow:
-            abilities.append(workflow)
-    return abilities
-
-
-async def check_interrupt_file(interval, interrupt_file,agent_task):
-    while True:
-        await asyncio.sleep(interval)
-        if os.path.exists(interrupt_file):
-            os.remove(interrupt_file)
-            agent_task.cancel()
-            break
-
-async def run_with_interrupt_check(agent, history, tool_compatibility_mode, interrupt_file, check_interval=1):
-    clear_chat_data()
-    try:
-        agent_task = asyncio.create_task(run_agent(agent, history, tool_compatibility_mode))
-        check_task = asyncio.create_task(check_interrupt_file(check_interval, interrupt_file, agent_task))
-        result = await agent_task
-        return result
-    except asyncio.CancelledError:
-        return CHAT_DATA['info']
-    except Exception as e:
-        import traceback
-        print(traceback.format_exc())
-        return None  # 返回 None 或者处理异常后的结果
-    finally:
-        if not agent_task.done():
-            agent_task.cancel()
-        # 确保即使发生异常也会取消检查任务
-        if not check_task.done():
-            check_task.cancel()
-            try:
-                await check_task
-            except asyncio.CancelledError:
-                pass  # 忽略取消错误
-
-async def agent_chat(bot_setting_file, history):
-    bot_setting = load_bot_setting(bot_setting_file)
-    abilities = load_abilities(bot_setting_file)
-
-    chat = ChatOpenAI(
-        api_key=cfg["apiKey"],
-        base_url=cfg["baseURL"],
-        model=cfg["model"],
-        temperature=bot_setting["temperature"],
-        stop_sequences=[
-            "tool▁calls▁end",
-            "tool▁call▁end",
-        ],  # 不然会虚构工具调用过程和结果
-    )
-    prompt = bot_setting["prompt"]
-    if cfg["toolCompatibilityMode"] and len(abilities) > 0:
-        prompt += """
-作为一个AI，你如果不确定结果，请务必使用工具查询。
-你可以通过下面的方式使用工具，并耐心等待工具返回结果。
-如果你需要调用工具，请使用以正确markdown中的json代码格式进行结尾（务必保证json格式正确，不要出现反斜杠未转义等问题）：
-```json
-{"name": 函数名, "parameters": 参数词典}
-```
-"""
-        prompt += "\nAvailable functions:\n"
-        for t in abilities:
-            prompt += f"\n```json\n{json.dumps(convert_to_openai_tool(t))}\n```"
-    agent = Runnable(
-        agent_execution_mode="FuncCall",
-        tools=abilities,
-        llm=chat,
-        assistant_message=prompt,
-        tool_compatibility_mode=cfg["toolCompatibilityMode"],
-    )
-    params = utils.params
-    if "interruptFile" in params:
-        interrupt_file_path = params["interruptFile"]
-        result = await run_with_interrupt_check(agent, history, cfg["toolCompatibilityMode"], interrupt_file_path)
-    else:
-        result = await run_agent(agent, history, cfg["toolCompatibilityMode"])
-    return result
-

pycoze/bot/bot.py

@@ -1,23 +0,0 @@
-from langchain_core.messages import HumanMessage, AIMessage
-import json
-from .agent import INPUT_MESSAGE, output, CHAT_DATA, clear_chat_data
-from .agent_chat import agent_chat
-import asyncio
-
-
-def chat(bot_setting_file: str):
-    history = []
-    while True:
-        input_text = input()
-        if not input_text.startswith(INPUT_MESSAGE):
-            raise ValueError("Invalid message")
-        message = json.loads(input_text[len(INPUT_MESSAGE) :])
-        history.append(HumanMessage(message["content"]))
-        result = asyncio.run(agent_chat(bot_setting_file, history))
-        output("assistant", result)
-        history.append(AIMessage(result))
-
-
-def get_chat_response(bot_setting_file: str, input_text: str):
-    result = asyncio.run(agent_chat(bot_setting_file, [HumanMessage(input_text)]))
-    return result