pycoze 0.1.196__py3-none-any.whl → 0.1.198__py3-none-any.whl

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/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.dist-info → pycoze-0.1.198.dist-info}/METADATA

@@ -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.dist-info → pycoze-0.1.198.dist-info}/RECORD

@@ -3,10 +3,10 @@ pycoze/ai/__init__.py,sha256=odM2lgYSnApxw4AzLV_5AyaxL5_MLfydOSB1VmLGFyA,75
 pycoze/ai/vram_reserve.py,sha256=brgXP42yj3yaZRgW8pfgc4Jg9EivAhcbp5W4igVHcow,4256
 pycoze/automation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycoze/automation/browser/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pycoze/automation/browser/edge_driver_manager.py,sha256=tJEGXab1XTk2LPjhpQ6NfAbRCdrsuyCDexMvYxZPUEM,3022
+pycoze/automation/browser/edge_driver_manager.py,sha256=gpgseunph5owZH6EskSYthuhey2SU3UP204gY0yIcuI,3022
 pycoze/bot/__init__.py,sha256=6HHMxDQVOyZM9dtSjQm9tjGnhj4h7CixD0JOvEwTi48,41
 pycoze/bot/agent_chat.py,sha256=hr0jGTd-lvLSe4vCcvaN2rKhB7BMBcVBtI1FpVa6ngs,3016
-pycoze/bot/bot.py,sha256=CCnBmxtmpmz7aLR_MIIiOuKR9IACo5P4uSkQXb5d0KU,3061
+pycoze/bot/bot.py,sha256=vRFLa3ABIqWHaWeuUhsTWzdas0SGpCRoe95D9C9IJ8c,3435
 pycoze/bot/agent/__init__.py,sha256=3wE8_FFQS8j2BY-g9Cr-onV0POEvDRZaw_NCzpqrNus,265
 pycoze/bot/agent/agent.py,sha256=7dyAsHSVnBmFUK9GR_BumkjSgrvQA0KLj0e3aDN_JU4,3592
 pycoze/bot/agent/assistant.py,sha256=3iLxnRvf_ia0cP-FHK5Fv4ylltlnzPq1KscRCFYqjkc,1147
@@ -28,8 +28,8 @@ pycoze/utils/__init__.py,sha256=Gi5EnrWZGMD2JRejgV4c_VLCXyvA2wwBFI_niDF5MUE,110
 pycoze/utils/arg.py,sha256=GtfGbMTMdaK75Fwh6MpUe1pCA5X6Ep4LFG7a72YrzjI,525
 pycoze/utils/env.py,sha256=W04lhvTHhAAC6EldP6kk2xrctqtu8K6kl1vDLZDNeh8,561
 pycoze/utils/text_or_file.py,sha256=gpxZVWt2DW6YiEg_MnMuwg36VNf3TX383QD_1oZNB0Y,551
-pycoze-0.1.196.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
-pycoze-0.1.196.dist-info/METADATA,sha256=3rOdsQze_AFFRpZgos633xfHeoWHYoFp6kLrb6LdXTg,726
-pycoze-0.1.196.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
-pycoze-0.1.196.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
-pycoze-0.1.196.dist-info/RECORD,,
+pycoze-0.1.198.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
+pycoze-0.1.198.dist-info/METADATA,sha256=SKG3vhboHl-tWEMD9GAE57F47Lw-_NcjvVMCdVard0s,726
+pycoze-0.1.198.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
+pycoze-0.1.198.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
+pycoze-0.1.198.dist-info/RECORD,,