pycoze 0.1.414__tar.gz → 0.1.415__tar.gz

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

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

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

@@ -112,7 +112,7 @@ async def stream_openai_response(conversation_history, start_new_stream):
         # 使用 async for 迭代异步生成器
         try:
             async for chunk in stream:
-
+                print("chunk", chunk)
                 # 检查是否需要重新创建流
                 if start_new_stream["value"]:
                     break  # 退出当前的 async for 循环，进入下一次 while 循环
@@ -166,35 +166,26 @@ 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 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())
+
+    # 处理流结束后的剩余内容
+    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.414 → pycoze-0.1.415}/pycoze.egg-info/PKG-INFO

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

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

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