pycoze 0.1.196__tar.gz → 0.1.198__tar.gz

{pycoze-0.1.196 → pycoze-0.1.198}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycoze
-Version: 0.1.196
+Version: 0.1.198
 Summary: Package for pycoze only!
 Author: Yuan Jie Xiong
 Author-email: aiqqqqqqq@qq.com

{pycoze-0.1.196 → pycoze-0.1.198}/pycoze/automation/browser/edge_driver_manager.py

@@ -9,7 +9,7 @@ from selenium import webdriver
 def is_network_available():
     try:
         # Attempt to access the download URL for Edge Chromium driver to check network connection
-        response = requests.get("https://msedgedriver.azureedge.net/", timeout=5)
+        response = requests.get("https://msedgedriver.azureedge.net/", timeout=3)
         return response.status_code == 200
     except requests.ConnectionError:
         return False

{pycoze-0.1.196 → pycoze-0.1.198}/pycoze/bot/bot.py

@@ -6,6 +6,7 @@ import threading
 import queue
 import time
 import asyncio
+import multiprocessing
 
 
 class PeekableQueue(queue.Queue):
@@ -15,6 +16,14 @@ class PeekableQueue(queue.Queue):
         except queue.Empty:
             return None
 
+
+def run_agent_chat(bot_setting_file, history, should_exit):
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+    loop.run_until_complete(agent_chat(bot_setting_file, history, should_exit))
+    loop.close()
+
+
 def chat(bot_setting_file: str):
     history = []
     input_queue = PeekableQueue()
@@ -42,13 +51,13 @@ def chat(bot_setting_file: str):
                     history.append(HumanMessage(message["content"]))
                     clear_chat_data()
                     
-                    # 为每个线程创建一个独立的退出标志
-                    should_exit = threading.Event()
-                    agent_chat_thread = threading.Thread(target=asyncio.run, args=(agent_chat(bot_setting_file, history, should_exit),))
-                    agent_chat_thread.start()
+                    # 创建进程退出标志
+                    should_exit = multiprocessing.Event()
+                    agent_chat_process = multiprocessing.Process(target=run_agent_chat, args=(bot_setting_file, history, should_exit)) # 需要是独立进程，不然无法创建新进程
+                    agent_chat_process.start()
                     
                     next_input_text = ""
-                    while agent_chat_thread.is_alive():
+                    while agent_chat_process.is_alive():
                         if not input_queue.empty():
                             next_input_text = input_queue.peek()
                             if next_input_text == INTERRUPT_MESSAGE:
@@ -56,6 +65,7 @@ def chat(bot_setting_file: str):
                                 should_exit.set()  # 设置退出标志
                                 break
                         time.sleep(0.1)  # 每隔 0.1 秒检查一次
+                        time.sleep(0.1)  # 每隔 0.1 秒检查一次
                     if next_input_text != INTERRUPT_MESSAGE:
                         history.append(AIMessage(content=CHAT_DATA["output"]))
                 except json.JSONDecodeError:
@@ -72,6 +82,6 @@ def chat(bot_setting_file: str):
 def get_chat_response(bot_setting_file: str, input_text: str):
     history = [HumanMessage(input_text)]
     clear_chat_data()
-    should_exit = threading.Event()
+    should_exit = multiprocessing.Event()
     asyncio.run(agent_chat(bot_setting_file, history, should_exit))
     return CHAT_DATA["output"]

{pycoze-0.1.196 → pycoze-0.1.198}/pycoze.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycoze
-Version: 0.1.196
+Version: 0.1.198
 Summary: Package for pycoze only!
 Author: Yuan Jie Xiong
 Author-email: aiqqqqqqq@qq.com

{pycoze-0.1.196 → pycoze-0.1.198}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="pycoze",
-    version="0.1.196",
+    version="0.1.198",
     packages=find_packages(),
     install_requires=[],
     author="Yuan Jie Xiong",