pycoze 0.1.203__tar.gz → 0.1.204__tar.gz

{pycoze-0.1.203 → pycoze-0.1.204}/PKG-INFO

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

pycoze-0.1.204/pycoze/bot/bot.py

@@ -0,0 +1,76 @@
+import signal
+import sys
+import json
+from .agent import INPUT_MESSAGE, output, CHAT_DATA, clear_chat_data
+from .agent_chat import agent_chat
+from multiprocessing import Process, Event
+import asyncio
+
+# 用于标记中断请求
+interrupt_flag = False
+
+# 信号处理函数
+def handle_interrupt(signum, frame):
+    global interrupt_flag
+    interrupt_flag = True
+    print("Interrupt signal received. Waiting for the current operation to complete...")
+
+# 设置信号处理器
+signal.signal(signal.SIGINT, handle_interrupt)
+
+def read_input():
+    while True:
+        try:
+            input_text = input()
+            if input_text.startswith(INPUT_MESSAGE):
+                yield input_text
+            else:
+                raise ValueError("Invalid message")
+        except EOFError:  # 如果输入流结束，则退出循环
+            break
+
+def chat(bot_setting_file: str):
+    history = []
+    for input_text in read_input():
+        try:
+            message = json.loads(input_text[len(INPUT_MESSAGE):])
+            history.append(HumanMessage(message["content"]))
+            clear_chat_data()
+
+            # 创建一个事件来控制子进程的退出
+            should_exit = Event()
+            
+            # 使用进程来运行 agent_chat
+            agent_chat_process = Process(target=asyncio.run, args=(agent_chat(bot_setting_file, history, should_exit),))
+            agent_chat_process.start()
+
+            # 检查是否收到了中断信号
+            while agent_chat_process.is_alive():
+                if interrupt_flag:
+                    should_exit.set()  # 设置退出标志
+                    break
+                time.sleep(0.1)  # 每隔 0.1 秒检查一次
+
+            # 确保子进程已经退出
+            agent_chat_process.join()
+
+            # 如果没有收到中断信号，记录输出
+            if not interrupt_flag:
+                history.append(AIMessage(content=CHAT_DATA["output"]))
+            
+            # 重置中断标志
+            interrupt_flag = False
+        except json.JSONDecodeError:
+            print("Invalid JSON format in input message.")
+        except KeyError:
+            print("Missing 'content' key in input message.")
+        except Exception as e:
+            print(f"An error occurred: {e}")
+
+
+def get_chat_response(bot_setting_file: str, input_text: str):
+    history = [HumanMessage(input_text)]
+    clear_chat_data()
+    should_exit = threading.Event()
+    asyncio.run(agent_chat(bot_setting_file, history, should_exit))
+    return CHAT_DATA["output"]

{pycoze-0.1.203 → pycoze-0.1.204}/pycoze.egg-info/PKG-INFO

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

{pycoze-0.1.203 → pycoze-0.1.204}/setup.py

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

pycoze-0.1.203/pycoze/bot/bot.py

@@ -1,77 +0,0 @@
-from langchain_core.messages import HumanMessage, AIMessage
-from .agent import INPUT_MESSAGE, INTERRUPT_MESSAGE, output, CHAT_DATA, clear_chat_data
-from .agent_chat import agent_chat
-import json
-import threading
-import queue
-import time
-import asyncio
-
-
-class PeekableQueue(queue.Queue):
-    def peek(self):
-        try:
-            return self._get()
-        except queue.Empty:
-            return None
-
-def chat(bot_setting_file: str):
-    history = []
-    input_queue = PeekableQueue()
-
-    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")
-            
-
-    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()
-                if input_text == INTERRUPT_MESSAGE:
-                    continue
-                try:
-                    message = json.loads(input_text[len(INPUT_MESSAGE):])
-                    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()
-                    
-                    next_input_text = ""
-                    while agent_chat_thread.is_alive():
-                        if not input_queue.empty():
-                            next_input_text = input_queue.peek()
-                            if next_input_text == INTERRUPT_MESSAGE:
-                                history.append(AIMessage(content=CHAT_DATA["info"]))
-                                should_exit.set()  # 设置退出标志
-                                break
-                        time.sleep(0.1)  # 每隔 0.1 秒检查一次
-                    if next_input_text != INTERRUPT_MESSAGE:
-                        history.append(AIMessage(content=CHAT_DATA["output"]))
-                except json.JSONDecodeError:
-                    print("Invalid JSON format in input message.")
-                except KeyError:
-                    print("Missing 'content' key in input message.")
-                except Exception as e:
-                    print(f"An error occurred: {e}")
-    finally:
-        input_thread_instance.join()
-
-
-
-def get_chat_response(bot_setting_file: str, input_text: str):
-    history = [HumanMessage(input_text)]
-    clear_chat_data()
-    should_exit = threading.Event()
-    asyncio.run(agent_chat(bot_setting_file, history, should_exit))
-    return CHAT_DATA["output"]