pycoze 0.1.201__py3-none-any.whl → 0.1.202__py3-none-any.whl

pycoze/bot/bot.py

@@ -9,59 +9,40 @@ import asyncio
 import multiprocessing
 from multiprocessing import Queue, Event
 
-# 假设 PeekableQueue 类已经定义好了，如果没有，请自行实现或使用其他方式
-class PeekableQueue:
-    def __init__(self, maxsize=0):
-        self.queue = Queue(maxsize=maxsize)
-    
-    def put(self, item):
-        self.queue.put(item)
-    
-    def get(self):
-        return self.queue.get()
-    
-    def peek(self):
-        with self.queue.mutex:
-            if self.queue.empty():
-                return None
-            return self.queue.queue[0]
-    
-    def empty(self):
-        return self.queue.empty()
-
-def input_process(input_queue: Queue, interrupt_event: Event):
-    while not interrupt_event.is_set():
-        input_text = input()
-        if input_text == INTERRUPT_MESSAGE or input_text.startswith(INPUT_MESSAGE):
-            input_queue.put(input_text)
-        else:
-            raise ValueError("Invalid message")
 
-async def handle_input_queue(input_queue: Queue, should_exit: asyncio.Event, history):
-    while not should_exit.is_set():
+class PeekableQueue(queue.Queue):
+    def peek(self):
         try:
+            return self._get()
+        except queue.Empty:
+            return None
+        
+async def handle_input_queue(input_queue, should_exit, history):
+    while not should_exit.is_set():
+        if not input_queue.empty():
             next_input_text = input_queue.get_nowait()
             if next_input_text == INTERRUPT_MESSAGE:
                 history.append(AIMessage(content=CHAT_DATA["info"]))
                 should_exit.set()  # 设置退出标志
                 break
-        except queue.Empty:
-            pass
         await asyncio.sleep(0.1)  # 非阻塞地等待一段时间
 
 def chat(bot_setting_file: str):
     history = []
     input_queue = PeekableQueue()
-    interrupt_event = Event()  # 用于中断输入进程的事件
 
-    # 创建并启动输入进程
-    input_process_instance = multiprocessing.Process(target=input_process, args=(input_queue, interrupt_event))
-    input_process_instance.start()
+    def input_thread(input_queue: PeekableQueue):
+        while True:
+            input_text = input()
+            if input_text == INTERRUPT_MESSAGE or input_text.startswith(INPUT_MESSAGE):
+                input_queue.put(input_text)
+            else:
+                raise ValueError("Invalid message")
 
-    try:
-        loop = asyncio.new_event_loop()
-        asyncio.set_event_loop(loop)
+    input_thread_instance = threading.Thread(target=input_thread, args=(input_queue,))
+    input_thread_instance.start()
 
+    try:
         while True:
             if not input_queue.empty():
                 input_text = input_queue.get()
@@ -71,6 +52,10 @@ def chat(bot_setting_file: str):
                     message = json.loads(input_text[len(INPUT_MESSAGE):])
                     history.append(HumanMessage(message["content"]))
                     clear_chat_data()
+                    
+                    # 创建一个异步事件循环
+                    loop = asyncio.new_event_loop()
+                    asyncio.set_event_loop(loop)
 
                     # 创建一个 Event 对象用于中断
                     should_exit = asyncio.Event()
@@ -93,10 +78,7 @@ def chat(bot_setting_file: str):
                 except Exception as e:
                     print(f"An error occurred: {e}")
     finally:
-        # 设置中断事件以停止输入进程
-        interrupt_event.set()
-        input_process_instance.join()
-
+        input_thread_instance.join()
 
 
 def get_chat_response(bot_setting_file: str, input_text: str):

{pycoze-0.1.201.dist-info → pycoze-0.1.202.dist-info}/METADATA

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

{pycoze-0.1.201.dist-info → pycoze-0.1.202.dist-info}/RECORD

@@ -6,7 +6,7 @@ pycoze/automation/browser/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 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=9WtvKt3_kgXnBXZEXINoIm_Vz1Oj2C_onWxnS7rvY2k,4041
+pycoze/bot/bot.py,sha256=F4YI2RGruC31jX3MX1fLOTs6j81CRM_GZsU_xAyqKiY,3481
 pycoze/bot/agent/__init__.py,sha256=3wE8_FFQS8j2BY-g9Cr-onV0POEvDRZaw_NCzpqrNus,265
 pycoze/bot/agent/agent.py,sha256=Ueste4xWvkQFk5Giklv4c-kcxiPL-D-KafzZ4DONosY,3613
 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.201.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
-pycoze-0.1.201.dist-info/METADATA,sha256=NhdIlRwQ0ND-FWYkLsIv1xrVms2SVbF4w5hmBb15N9s,726
-pycoze-0.1.201.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
-pycoze-0.1.201.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
-pycoze-0.1.201.dist-info/RECORD,,
+pycoze-0.1.202.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
+pycoze-0.1.202.dist-info/METADATA,sha256=Eba5-C_KEnuUfCafG1RTxGyWGHiAIMaczsfgUeVQupo,726
+pycoze-0.1.202.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
+pycoze-0.1.202.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
+pycoze-0.1.202.dist-info/RECORD,,