pycoze 0.1.433__py3-none-any.whl → 0.1.435__py3-none-any.whl

pycoze/bot/chat_base.py

@@ -94,6 +94,12 @@ class BufferProcessor:
         self.json_block_content = ""
         self.text_content = ""
 
+    def can_process(self):
+        if not self.in_json_block:
+            return re.search(r"```json", self.buffer, re.IGNORECASE)
+        else:
+            return re.search(r"```", self.buffer)
+
     def process_buffer(self):
         """
         处理 buffer 中的内容，并返回生成的内容类型和内容。
@@ -188,7 +194,7 @@ async def stream_openai_response(conversation_history, start_new_stream):
                 info("assistant", chunk)
                 buffer_processor.buffer += chunk
                 
-                if "\n" in chunk: # 不能数据不足时就解析，不然会解析出错，例如```json就不能解析为```和json
+                if buffer_processor.can_process():
                     result = buffer_processor.process_buffer()
                     if result:
                         yield result
@@ -204,14 +210,12 @@ async def stream_openai_response(conversation_history, start_new_stream):
     while buffer_processor.buffer:
         result = buffer_processor.process_buffer()
         if result:
-            print("final process result", result)
             yield result
         else:
             break
 
     # 处理流结束后的剩余内容
     for result in buffer_processor.finalize():
-        print("final finalize result", result)
         yield result
 
 

{pycoze-0.1.433.dist-info → pycoze-0.1.435.dist-info}/METADATA

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

{pycoze-0.1.433.dist-info → pycoze-0.1.435.dist-info}/RECORD

@@ -11,7 +11,7 @@ pycoze/api/lib/web.py,sha256=gR2Ffaj3Pk4NfnE314UTLfZm7rW_rizZxWVDcZMeFgs,382
 pycoze/api/lib/window.py,sha256=dkzWfLwn5pE_L0DfQ38K8nx9tQyT5KO-GYyXi0rytFc,2073
 pycoze/bot/__init__.py,sha256=rL3Q-ycczRpSFfKn84fg3QBl5k22WpyeIU5qOEjEby8,79
 pycoze/bot/chat.py,sha256=qEuMxH0cVFU9QSU36FrOsjhRAxtsvOv7CQtuXvM3F6Y,6446
-pycoze/bot/chat_base.py,sha256=J19nZIQ_owKRSNRRxStUDB8_kKsJKLbmm1L3EoHJsyo,12599
+pycoze/bot/chat_base.py,sha256=uwAVhhvpKHQytJYMsIOHsTThkr45xRehbXF0-kDbvjM,12614
 pycoze/bot/lib.py,sha256=_bQ52mKsWgFGAogFHnmRBJbvK_tPOwsAJ8NqJNMR5K4,7210
 pycoze/bot/message.py,sha256=udnIi-h4QgGzkbr_5VcAsVGjoLp9wXJSfBCeuOz7_Bk,802
 pycoze/bot/prompt.md,sha256=t7NQdiiNe-jCDVfeVbvTPfq5WK5nF8CxFUQUFMyXJlo,13880
@@ -31,8 +31,8 @@ pycoze/utils/arg.py,sha256=jop1tBfe5hYkHW1NSpCeaZBEznkgguBscj_7M2dWfrs,503
 pycoze/utils/env.py,sha256=5pWlXfM1F5ZU9hhv1rHlDEanjEW5wf0nbyez9bNRqqA,559
 pycoze/utils/socket.py,sha256=bZbFFRH4mfThzRqt55BAAGQ6eICx_ja4x8UGGrUdAm8,2428
 pycoze/utils/text_or_file.py,sha256=gpxZVWt2DW6YiEg_MnMuwg36VNf3TX383QD_1oZNB0Y,551
-pycoze-0.1.433.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
-pycoze-0.1.433.dist-info/METADATA,sha256=LI4UaJuXK07NcoYg6L4RJlj34GHUSVhTn8lYzs-qwlg,854
-pycoze-0.1.433.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-pycoze-0.1.433.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
-pycoze-0.1.433.dist-info/RECORD,,
+pycoze-0.1.435.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
+pycoze-0.1.435.dist-info/METADATA,sha256=fUIAiLjTXrhXwW6G844HE0kv22S2hm0mYevE7Jk8IVA,854
+pycoze-0.1.435.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+pycoze-0.1.435.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
+pycoze-0.1.435.dist-info/RECORD,,