pycoze 0.1.387__tar.gz → 0.1.389__tar.gz

{pycoze-0.1.387 → pycoze-0.1.389}/PKG-INFO

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

{pycoze-0.1.387 → pycoze-0.1.389}/pycoze/api/lib/window.py

@@ -39,7 +39,6 @@ class WindowCls:
     def close_window(self):
         socket.post("closeWindow", {})
 
-
     def get_slected_text(self) -> str:
         result = socket.post_and_recv_result("get-selected-text", {})
         return result

{pycoze-0.1.387 → pycoze-0.1.389}/pycoze/bot/chat.py

@@ -12,14 +12,27 @@ from pathlib import Path
 
 
 def eclipse_tool_result(text, threshold=100):
+    if not isinstance(threshold, int) or threshold < 0:
+        return text
+        
     pattern = r'\[Tool Result Begin\](.*?)\[Tool Result End\]'
     
+    # 检查标签是否匹配
+    begin_count = text.count('[Tool Result Begin]')
+    end_count = text.count('[Tool Result End]')
+    if begin_count != end_count:
+        return text
+    
     def replace_match(match):
         content = match.group(1)
+        # 检查嵌套标签
+        if '[Tool Result Begin]' in content or '[Tool Result End]' in content:
+            return match.group(0)  # 返回原始匹配内容，不替换
+            
         if len(content) < threshold:
             return match.group(0)  # 返回原始匹配内容，不替换
         else:
-            return '[Tool Result Begin]...[Tool Result End]'
+            return "[Tool Result Begin]too long to remember...[Tool Result End]"
     
     replaced_text = re.sub(pattern, replace_match, text, flags=re.DOTALL)
     return replaced_text

{pycoze-0.1.387 → pycoze-0.1.389}/pycoze/bot/chat_base.py

@@ -19,26 +19,34 @@ def guess_files_in_message(cwd: str, user_message: str) -> List[str]:
         return []
 
 
-def user_task_prompt(conversation_history, cwd, user_input: str, programmer_mode: bool):
+def content_str(cwd, potential_paths, conversation_history):
+    content = []
+    for file_path in potential_paths:
+        file_path = resolve_relative_path(cwd, file_path)
+        if os.path.isfile(file_path):
+            file_marker = f"[[{file_path}]]'"
+            file_content = read_local_file(file_path)
+            if not any(
+                file_marker in msg["content"] for msg in conversation_history
+            ):
+                content.append(f"{file_marker}\n{file_content}")
+    return (
+        "Partial contents of files are as follows:" + "\n".join(content)
+        if content
+        else ""
+    )
+
+def user_task_prompt(conversation_history, cwd, user_input: str, programmer_mode: bool, opened_files: List[str]):
     if programmer_mode:
         potential_paths = guess_files_in_message(cwd, user_input)
 
         exist_files = get_formatted_filelist_str(cwd, True, 200)
-        content = []
-        for file_path in potential_paths:
-            file_path = resolve_relative_path(cwd, file_path)
-            if os.path.isfile(file_path):
-                file_marker = f"[[{file_path}]]'"
-                file_content = read_local_file(file_path)
-                if not any(
-                    file_marker in msg["content"] for msg in conversation_history
-                ):
-                    content.append(f"{file_marker}\n{file_content}")
-        content_str = (
-            "Partial contents of files are as follows:" + "\n".join(content)
-            if content
-            else ""
+        exist_files_str = f"Files in current directory:\n{exist_files}" if exist_files else ""
+
+        opened_files_str = (
+            f"Opened files:\n{'\n'.join(opened_files)}" if opened_files else ""
         )
+       
         return f"""<task>
 {user_input}
 </task>
@@ -46,10 +54,11 @@ def user_task_prompt(conversation_history, cwd, user_input: str, programmer_mode
 <environment_details>
 Current working directory: {cwd}
 
-List of files under path:
-{exist_files}
+{exist_files_str}
 
-{content_str}
+{opened_files_str}
+
+{content_str(cwd, potential_paths, conversation_history)}
 
 </environment_details>
     """
@@ -144,6 +153,7 @@ async def handle_user_inputs(
     no_exit_if_incomplete = bot_setting["systemAbility"]["no_exit_if_incomplete"]
     programmer_mode = bot_setting["programmerMode"]
     show_tool_results = bot_setting["showToolResults"]
+    opened_files = bot_setting["openedFiles"]
 
     start_new_stream = {
         "value": False
@@ -157,7 +167,7 @@ async def handle_user_inputs(
         {
             "role": "user",
             "content": user_task_prompt(
-                conversation_history, cwd, user_input, programmer_mode
+                conversation_history, cwd, user_input, programmer_mode, opened_files
             ),
         }
     )
@@ -219,18 +229,18 @@ async def handle_user_inputs(
                     )
                     
                     assistant_content = (
-                        "Executing tool: "
+                        "Executing tool: \n"
                         + dumps_markdown_json(tool_request)
                         + "\n\n[Tool Result Begin]\n"
                         + result
                         + "\n[Tool Result End]\n"
                     )
                     lang = "json" if is_json_dumps else "text"
-                    status_str = "✅\n" if ok else "❌\n"
+                    
                     if show_tool_results:
+                        status_str = "✅\n" if ok else "❌\n"
                         info("assistant", status_str + f"\n```{lang}\n" + result + "\n```\n\n")
-                    else:
-                        info("assistant", status_str)
+
                     conversation_history.append(
                         {"role": "assistant", "content": assistant_content}
                     )

{pycoze-0.1.387 → pycoze-0.1.389}/pycoze/bot/lib.py

@@ -241,4 +241,3 @@ def format_filelist_str(items, limit):
 def get_formatted_filelist_str(root: str, recursive: bool, limit=200):
     items = get_files_and_folders(root, recursive)
     return format_filelist_str(items, limit=limit)
-

{pycoze-0.1.387 → pycoze-0.1.389}/pycoze.egg-info/PKG-INFO

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

{pycoze-0.1.387 → pycoze-0.1.389}/setup.py

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