pycoze 0.1.370__tar.gz → 0.1.372__tar.gz

{pycoze-0.1.370 → pycoze-0.1.372}/PKG-INFO

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

{pycoze-0.1.370 → pycoze-0.1.372}/pycoze/bot/chat.py

@@ -8,6 +8,7 @@ from pycoze import utils
 import tempfile
 import re
 import datetime
+import pprint
 
 def eclipse_tool_result(text):
     # 使用正则表达式匹配 [Tool Result Begin] 和 [Tool Result End] 之间的内容
@@ -16,6 +17,11 @@ def eclipse_tool_result(text):
     replaced_text = re.sub(pattern, '[Tool Result Begin]...[Tool Result End]', text, flags=re.DOTALL)
     return replaced_text
 
+def output_chat_history(file_path, chat_history):
+    with open(file_path, 'w', encoding='utf-8') as f:
+        for chat in chat_history:
+            f.write(chat["role"] + ":\n" + chat["content"] + "\n\n")
+        
 
 async def check_interrupt_file(interval, interrupt_file, chat_task):
     while True:
@@ -75,7 +81,7 @@ def chat(bot_setting_file: str):
     current_time_str = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     chat_history_folder = os.path.join(os.path.dirname(bot_setting_file), "chatHistory")
     os.makedirs(chat_history_folder, exist_ok=True)
-    chat_history_file = os.path.join(chat_history_folder, current_time_str + ".json")
+    chat_history_file = os.path.join(chat_history_folder, current_time_str + ".txt")
 
     with open(bot_setting_file, encoding="utf-8") as f:
         bot_setting = json.load(f)
@@ -124,8 +130,7 @@ def chat(bot_setting_file: str):
         output("assistant", CHAT_DATA["info"])
         is_save_fail = False
         try:
-            with open(chat_history_file, "w", encoding="utf-8") as f:
-                f.write(json.dumps(conversation_history, ensure_ascii=False, indent=4) + "\n")
+            output_chat_history(conversation_history, chat_history_file)
         except Exception as e:
             print(f"Error writing chat history to file: {e}")
             is_save_fail = True
@@ -135,8 +140,7 @@ def chat(bot_setting_file: str):
         if is_save_fail:
             # 去除工具结果后，重新保存
             try:
-                with open(chat_history_file, "w", encoding="utf-8") as f:
-                    f.write(json.dumps(eclipse_conversation_history, ensure_ascii=False, indent=4) + "\n")
+                output_chat_history(eclipse_conversation_history, chat_history_file)
             except Exception as e:
                 print(f"Error writing chat history to file: {e}")
             

{pycoze-0.1.370 → pycoze-0.1.372}/pycoze.egg-info/PKG-INFO

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

{pycoze-0.1.370 → pycoze-0.1.372}/setup.py

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