pycoze 0.1.412__tar.gz → 0.1.414__tar.gz

{pycoze-0.1.412 → pycoze-0.1.414}/PKG-INFO

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

{pycoze-0.1.412 → pycoze-0.1.414}/pycoze/bot/chat_base.py

@@ -5,6 +5,7 @@ from .message import info
 from pycoze.ai import chat_stream_async, extract
 from .tools import ToolExecutor
 from typing import List
+import re
 
 def guess_files_in_message(cwd: str, user_message: str) -> List[str]:
     try:
@@ -164,26 +165,36 @@ async def stream_openai_response(conversation_history, start_new_stream):
             # 捕获其他异常（如网络错误）
             print(f"Error: {e}", style="bold red")
             break
-
-    # 处理 buffer 中剩余的内容
+    
     if buffer:
         if in_json_block:
             buffer = buffer.split("```")[0]
             json_block_content += buffer
+            yield ("json", json_block_content.strip())
         else:
             text_content += buffer
-
-    # 处理流结束后的剩余内容
-    if in_json_block:
-        # 如果仍在 JSON 代码块模式，处理剩余的 JSON 内容
-        if json_block_content.strip():
-            print("yield json", json_block_content.strip())
-            yield ("json", json_block_content.strip())
-    else:
-        # 如果不在 JSON 代码块模式，处理剩余的文本内容
-        if text_content.strip():
-            print("yield json", text_content.strip())
-            yield ("text", text_content.strip())
+            if text_content:
+                yield ("text", text_content.strip())
+
+    # # 处理 buffer 中剩余的内容
+    # if buffer:
+    #     if in_json_block:
+    #         buffer = buffer.split("```")[0]
+    #         json_block_content += buffer
+    #     else:
+    #         text_content += buffer
+
+    # # 处理流结束后的剩余内容
+    # if in_json_block:
+    #     # 如果仍在 JSON 代码块模式，处理剩余的 JSON 内容
+    #     if json_block_content.strip():
+    #         print("yield json", json_block_content.strip())
+    #         yield ("json", json_block_content.strip())
+    # else:
+    #     # 如果不在 JSON 代码块模式，处理剩余的文本内容
+    #     if text_content.strip():
+    #         print("yield json", text_content.strip())
+    #         yield ("text", text_content.strip())
 
 
 

{pycoze-0.1.412 → pycoze-0.1.414}/pycoze.egg-info/PKG-INFO

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

{pycoze-0.1.412 → pycoze-0.1.414}/setup.py

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