pycoze 0.1.421__tar.gz → 0.1.423__tar.gz

{pycoze-0.1.421 → pycoze-0.1.423}/PKG-INFO

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

{pycoze-0.1.421 → pycoze-0.1.423}/pycoze/bot/chat_base.py

@@ -164,38 +164,40 @@ async def stream_openai_response(conversation_history, start_new_stream):
             print(f"Error: {e}", style="bold red")
             break
     
-    if not in_json_block:
-        # 查找 JSON 代码块的起始标记
-        json_start_match = re.search(r"```json", buffer, re.IGNORECASE)
-        if json_start_match:
-            # 提取 JSON 代码块之前的文本
-            text_content += buffer[:json_start_match.start()]
-            # 如果 text_content 不为空，先 yield 文本
-            if text_content.strip():
-                yield ("text", text_content.strip())
-                text_content = ""
-            # 进入 JSON 代码块模式
-            in_json_block = True
-            buffer = buffer[json_start_match.end():]
+    if buffer:
+        if not in_json_block:
+            # 查找 JSON 代码块的起始标记
+            json_start_match = re.search(r"```json", buffer, re.IGNORECASE)
+            if json_start_match:
+                # 提取 JSON 代码块之前的文本
+                text_content += buffer[:json_start_match.start()]
+                # 如果 text_content 不为空，先 yield 文本
+                if text_content.strip():
+                    yield ("text", text_content.strip())
+                    text_content = ""
+                # 进入 JSON 代码块模式
+                in_json_block = True
+                buffer = buffer[json_start_match.end():]
+            else:
+                # 如果没有找到 JSON 代码块，继续累积到 text_content
+                text_content += buffer
+                buffer = ""
         else:
-            # 如果没有找到 JSON 代码块，继续累积到 text_content
-            text_content += buffer
-            buffer = ""
-    else:
-        # 查找 JSON 代码块的结束标记
-        json_end_match = re.search(r"```", buffer)
-        if json_end_match:
-            # 提取 JSON 代码块内容
-            json_block_content += buffer[:json_end_match.start()]
-            # yield JSON 代码块
-            yield ("json", json_block_content.strip())
-            json_block_content = ""
-            in_json_block = False
-            buffer = buffer[json_end_match.end():]
-        else:
-            # 如果没有找到结束标记，继续累积 JSON 内容
-            json_block_content += buffer
-            buffer = ""
+            # 查找 JSON 代码块的结束标记
+            json_end_match = re.search(r"```", buffer)
+            if json_end_match:
+                # 提取 JSON 代码块内容
+                json_block_content += buffer[:json_end_match.start()]
+                # yield JSON 代码块
+                yield ("json", json_block_content.strip())
+                json_block_content = ""
+                in_json_block = False
+                buffer = buffer[json_end_match.end():]
+            else:
+                # 如果没有找到结束标记，继续累积 JSON 内容
+                json_block_content += buffer
+                buffer = ""
+    
     # 处理流结束后的剩余内容
     if in_json_block:
         # 如果仍在 JSON 代码块模式，处理剩余的 JSON 内容

{pycoze-0.1.421 → pycoze-0.1.423}/pycoze.egg-info/PKG-INFO

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

{pycoze-0.1.421 → pycoze-0.1.423}/setup.py

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