auto-coder 0.1.180__tar.gz → 0.1.182__tar.gz

{auto_coder-0.1.180 → auto_coder-0.1.182}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.180
+Version: 0.1.182
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -26,7 +26,7 @@ Requires-Dist: tabulate
 Requires-Dist: jupyter_client
 Requires-Dist: prompt-toolkit
 Requires-Dist: tokenizers
-Requires-Dist: byzerllm[saas]>=0.1.134
+Requires-Dist: byzerllm[saas]>=0.1.135
 Requires-Dist: patch
 Requires-Dist: diff_match_patch
 Requires-Dist: GitPython

{auto_coder-0.1.180 → auto_coder-0.1.182}/src/auto_coder.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.180
+Version: 0.1.182
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -26,7 +26,7 @@ Requires-Dist: tabulate
 Requires-Dist: jupyter_client
 Requires-Dist: prompt-toolkit
 Requires-Dist: tokenizers
-Requires-Dist: byzerllm[saas]>=0.1.134
+Requires-Dist: byzerllm[saas]>=0.1.135
 Requires-Dist: patch
 Requires-Dist: diff_match_patch
 Requires-Dist: GitPython

{auto_coder-0.1.180 → auto_coder-0.1.182}/src/auto_coder.egg-info/requires.txt

@@ -16,7 +16,7 @@ tabulate
 jupyter_client
 prompt-toolkit
 tokenizers
-byzerllm[saas]>=0.1.134
+byzerllm[saas]>=0.1.135
 patch
 diff_match_patch
 GitPython

{auto_coder-0.1.180 → auto_coder-0.1.182}/src/autocoder/chat_auto_coder.py

@@ -931,12 +931,7 @@ class CommandCompleter(Completer):
                         field_name + ":"
                         for field_name in AutoCoderArgs.model_fields.keys()
                         if field_name.startswith(current_word)
-                    ]
-                    completions += [
-                        field_name + ":"
-                        for field_name in AutoCoderArgs.model_fields.keys()
-                        if field_name.startswith(current_word)
-                    ]
+                    ]                    
 
                 for completion in completions:
                     yield Completion(completion, start_position=-len(current_word))

{auto_coder-0.1.180 → auto_coder-0.1.182}/src/autocoder/common/interpreter.py

@@ -1,30 +1,39 @@
 from autocoder.common.JupyterClient import JupyterNotebook
 from autocoder.common.ShellClient import ShellClient
 from autocoder.common import ExecuteSteps
-from typing import List,Dict,Any,Optional,Union
+from typing import List, Dict, Any, Optional, Callable
+
 
 class Interpreter:
-    def __init__(self,cwd:Optional[str]=None):
+    def __init__(self, cwd: Optional[str] = None):
         self.jupyter_client = JupyterNotebook()
         self.shell_client = ShellClient(working_dir=cwd)
 
     def close(self):
         self.jupyter_client.close()
-        self.shell_client.close()    
+        self.shell_client.close()
 
-    def execute_steps(self, steps: ExecuteSteps) -> str:
+    def execute_steps(
+        self, steps: ExecuteSteps, output_callback: Optional[Callable] = None
+    ) -> str:
         jupyter_client = self.jupyter_client
         shell_client = self.shell_client
 
         if steps.steps[0].cwd:
-            self.shell_client.working_dir = steps.steps[0].cwd        
+            self.shell_client.working_dir = steps.steps[0].cwd
         try:
             output = ""
             for step in steps.steps:
                 if step.lang and step.lang.lower() in ["python"]:
                     output += f"Python Code:\n{step.code}\n"
+                    if output_callback:
+                        output_callback("code", step.code)
                     output += "Output:\n"
                     result, error = jupyter_client.add_and_run(step.code)
+                    if output_callback:
+                        output_callback("result", result)
+                        if error:
+                            output_callback("error", error)
                     output += result + "\n"
                     if error:
                         output += f"Error: {str(error)}\n"
@@ -40,16 +49,25 @@ class Interpreter:
                 ]:
                     output += f"Shell Command:\n{step.code}\n"
                     output += "Output:\n"
+                    if output_callback:
+                        output_callback("code", step.code)
                     stdout, stderr = shell_client.add_and_run(step.code)
+                    if output_callback:
+                        output_callback("result", stdout)
+                        if stderr:
+                            output_callback("error", stderr)
                     output += stdout + "\n"
                     if stderr:
                         output += f"Error: {stderr}\n"
+                        if output_callback:
+                            output_callback("error", stderr)
                 else:
                     output += f"Unknown step type: {step.lang}\n"
 
                 output += "-" * 20 + "\n"
         except Exception as e:
-            output += f"Error: {str(e)}\n"        
+            output += f"Error: {str(e)}\n"
+            if output_callback:
+                output_callback("error", str(e))
 
         return output
-

{auto_coder-0.1.180 → auto_coder-0.1.182}/src/autocoder/rag/doc_filter.py

@@ -52,8 +52,8 @@ def _check_relevance_with_conversation(
     {{ filter_config }}
     {% endif %}
 
-    请结合提供的文档以及用户对话历史，判断提供的文档是不是能回答用户的最后一个问题。
-    如果该文档提供的知识能够回答问题，那么请回复"yes/<relevant>" 否则回复"no/<relevant>"。
+    请结合提供的文档以及用户对话历史，判断提供的文档是不是能和用户的最后一个问题相关。
+    如果该文档提供的知识能够和用户的问题相关，那么请回复"yes/<relevant>" 否则回复"no/<relevant>"。
     其中， <relevant> 是你认为文档中和问题的相关度，0-10之间的数字，数字越大表示相关度越高。
     """
 

{auto_coder-0.1.180 → auto_coder-0.1.182}/src/autocoder/rag/long_context_rag.py

@@ -31,8 +31,8 @@ from tokenizers import Tokenizer
 from autocoder.rag.variable_holder import VariableHolder
 
 try:
-    from autocoder_pro.rag.llm_compute import LLMComputeEngine    
-except ImportError:    
+    from autocoder_pro.rag.llm_compute import LLMComputeEngine
+except ImportError:
     logger.warning("Please install auto-coder-pro to enhance llm compute ability")
     LLMComputeEngine = None
 
@@ -289,7 +289,7 @@ class LongContextRAG:
                 llm_config=llm_config,
             )
         except Exception as e:
-            logger.error(f"Error in stream_chat_oai: {str(e)}")            
+            logger.error(f"Error in stream_chat_oai: {str(e)}")
             traceback.print_exc()
             return ["出现错误，请稍后再试。"], []
 
@@ -491,7 +491,9 @@ class LongContextRAG:
             )
 
             if LLMComputeEngine is not None:
-                llm_compute_engine = LLMComputeEngine(llm=self.llm)
+                llm_compute_engine = LLMComputeEngine(
+                    llm=self.llm, inference_enhance=True, debug=False
+                )
                 new_conversations = llm_compute_engine.process_conversation(
                     conversations, query, [doc.source_code for doc in relevant_docs]
                 )

auto_coder-0.1.182/src/autocoder/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.182"

auto_coder-0.1.180/src/autocoder/version.py

@@ -1 +0,0 @@
-__version__ = "0.1.180"