auto-coder 0.1.347__tar.gz → 0.1.348__tar.gz

{auto_coder-0.1.347 → auto_coder-0.1.348}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.347
+Version: 0.1.348
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Programming Language :: Python :: 3.10

{auto_coder-0.1.347 → auto_coder-0.1.348}/src/auto_coder.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.347
+Version: 0.1.348
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Programming Language :: Python :: 3.10

{auto_coder-0.1.347 → auto_coder-0.1.348}/src/auto_coder.egg-info/SOURCES.txt

@@ -233,6 +233,7 @@ src/autocoder/rag/cache/simple_cache.py
 src/autocoder/rag/loaders/__init__.py
 src/autocoder/rag/loaders/docx_loader.py
 src/autocoder/rag/loaders/excel_loader.py
+src/autocoder/rag/loaders/image_loader.py
 src/autocoder/rag/loaders/pdf_loader.py
 src/autocoder/rag/loaders/ppt_loader.py
 src/autocoder/rag/stream_event/__init__.py

{auto_coder-0.1.347 → auto_coder-0.1.348}/src/autocoder/auto_coder_runner.py

@@ -2838,11 +2838,12 @@ def auto_command(query: str,extra_args: Dict[str,Any]={}):
                             memory_config=MemoryConfig(memory=memory, 
                             save_memory_func=save_memory), command_config=CommandConfig,
                             conversation_name="current"
-                            )
-        if get_run_context() == RunMode.TERMINAL:
-            agent.run_in_terminal(AgenticEditRequest(user_input=query))
-        else:
+                            )           
+        if get_run_context().mode == RunMode.WEB:
             agent.run_with_events(AgenticEditRequest(user_input=query))
+        else:
+            agent.run_in_terminal(AgenticEditRequest(user_input=query))
+            
         return
         
     args = get_final_config()  

{auto_coder-0.1.347 → auto_coder-0.1.348}/src/autocoder/common/auto_coder_lang.py

@@ -836,11 +836,11 @@ MESSAGES = {
     },
     "/agent/edit/apply_pre_changes":{
         "en":"Commit user changes",
-        "zh":"提交用户修改"
+        "zh":"检查用户是否有手动修改(如有，会自动提交)..."
     },
     "/agent/edit/apply_changes":{
-        "en":"Commit changes",
-        "zh":"提交修改"
+        "en":"Commit the changes in preview steps",
+        "zh":"提交前面步骤的修改"
     }
 }
 

{auto_coder-0.1.347 → auto_coder-0.1.348}/src/autocoder/common/v2/agent/agentic_edit.py

@@ -212,18 +212,7 @@ class AgenticEdit:
     @byzerllm.prompt()
     def _analyze(self, request: AgenticEditRequest) -> str:
         """        
-        You are a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices.
-
-        ====
-
-        FILES CONTEXT
-
-        The following files are provided to you as context for the user's task. You can use these files to understand the project structure and codebase, and to make informed decisions about which files to modify.
-        If you need to read more files, you can use the tools to find and read more files.
-
-        <files>
-        {{files}}
-        </files>
+        You are a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices.        
 
         ====        
 
@@ -690,8 +679,10 @@ class AgenticEdit:
         The following rules are provided by the user, and you must follow them strictly.
 
         {% for key, value in extra_docs.items() %}
-        ### {{ key }}
+        <user_rule>
+        ##File: {{ key }}
         {{ value }}
+        </user_rule>
         {% endfor %}        
         {% endif %}
         """
@@ -706,7 +697,7 @@ class AgenticEdit:
                     try:
                         with open(fpath, "r", encoding="utf-8") as f:
                             content = f.read()
-                            key = os.path.splitext(fname)[0]
+                            key = fpath
                             extra_docs[key] = content
                     except Exception:
                         continue
@@ -784,11 +775,25 @@ class AgenticEdit:
         # print(system_prompt)
         conversations = [
             {"role": "system", "content": system_prompt},
-        ] + self.conversation_manager.get_history()
+        ] 
+        
+        conversations.append({
+            "role":"user","content":f'''
+Below are some files the user is focused on, and the content is up to date. These entries show the file paths along with their full text content, which can help you better understand the user's needs. If the information is insufficient, you can use tools such as read_file to retrieve more details.
+<files>
+{self.files.to_str()}
+</files>'''
+        })
+
+        conversations.append({
+            "role":"assistant","content":"Ok"
+        })
+        conversations.extend(self.conversation_manager.get_history())        
         conversations.append({
             "role": "user", "content": request.user_input
         })
         self.conversation_manager.add_user_message(request.user_input)
+        
         logger.debug(
             f"Initial conversation history size: {len(conversations)}")
 
@@ -1013,9 +1018,7 @@ class AgenticEdit:
 
         for content_chunk, metadata in generator:
             global_cancel.check_and_raise()
-
-            meta_holder.meta = metadata
-            logger.info(f"metadata: {metadata.input_tokens_count}")
+            meta_holder.meta = metadata            
             if not content_chunk:
                 continue
             buffer += content_chunk

{auto_coder-0.1.347 → auto_coder-0.1.348}/src/autocoder/plugins/__init__.py

@@ -256,6 +256,7 @@ class PluginManager:
             "/plugins/dirs /remove",
             "/plugins/dirs /clear",
         ]
+        self._wrapped_functions: Dict[str, Callable] = {}
 
     @property
     def cached_discover_plugins(self) -> List[Type[Plugin]]:
@@ -561,6 +562,7 @@ class PluginManager:
                 return result
             return None
 
+        self._wrapped_functions[func_name] = wrapped
         return wrapped
 
     def register_function_interception(self, plugin_name: str, func_name: str) -> None:
@@ -1107,6 +1109,24 @@ class PluginManager:
 
         return processed_completions
 
+    def get_wrapped_functions(self) -> Dict[str, Callable]:
+        """获取所有已包装的函数。
+
+        Returns:
+            Dict[str, Callable]: 包含所有已包装函数的字典，键为函数名，值为包装后的函数
+        """
+        return self._wrapped_functions
+
+    def get_wrapped_function(self, func_name: str) -> Callable:
+        """获取已包装的函数。
+
+        Args:
+            func_name: 函数名
+
+        Returns:
+            已包装的函数，如果未找到则返回 None
+        """
+        return self._wrapped_functions.get(func_name)
 
 def register_global_plugin_dir(plugin_dir: str) -> None:
     """注册一个全局插件目录。