auto-coder 0.1.178__tar.gz → 0.1.181__tar.gz

{auto-coder-0.1.178 → auto_coder-0.1.181}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.178
+Version: 0.1.181
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence

{auto-coder-0.1.178 → auto_coder-0.1.181}/src/auto_coder.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.178
+Version: 0.1.181
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence

{auto-coder-0.1.178 → auto_coder-0.1.181}/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.178 → auto_coder-0.1.181}/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.178 → auto_coder-0.1.181}/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.178 → auto_coder-0.1.181}/src/autocoder/rag/document_retriever.py

@@ -2,6 +2,7 @@ import json
 import os
 import platform
 import time
+import traceback
 
 from watchfiles import Change, DefaultFilter, awatch, watch
 
@@ -24,9 +25,9 @@ from autocoder.rag.loaders import (
     extract_text_from_pdf,
     extract_text_from_ppt,
 )
-from autocoder.rag import variable_holder
 from autocoder.rag.token_counter import count_tokens_worker, count_tokens
 from uuid import uuid4
+from autocoder.rag.variable_holder import VariableHolder
 
 cache_lock = threading.Lock()
 
@@ -158,8 +159,7 @@ def process_file_local(file_path: str) -> List[SourceCode]:
         logger.info(f"Load file {file_path} in {time.time() - start_time}")
         return v
     except Exception as e:
-        logger.error(f"Error processing file {file_path}: {str(e)}")
-        import traceback
+        logger.error(f"Error processing file {file_path}: {str(e)}")        
         traceback.print_exc()
         return []
 
@@ -352,7 +352,7 @@ class AutoCoderRAGAsyncUpdateQueue:
             with Pool(
                 processes=os.cpu_count(),
                 initializer=initialize_tokenizer,
-                initargs=(variable_holder.TOKENIZER_PATH,),
+                initargs=(VariableHolder.TOKENIZER_PATH,),
             ) as pool:
                 results = pool.map(process_file_in_multi_process, files_to_process)
 

{auto-coder-0.1.178 → auto_coder-0.1.181}/src/autocoder/rag/long_context_rag.py

@@ -13,6 +13,7 @@ from rich.console import Console
 from rich.panel import Panel
 from rich.table import Table
 import statistics
+import traceback
 
 from autocoder.common import AutoCoderArgs, SourceCode
 from autocoder.rag.doc_filter import DocFilter
@@ -32,6 +33,7 @@ from autocoder.rag.variable_holder import VariableHolder
 try:
     from autocoder_pro.rag.llm_compute import LLMComputeEngine
 except ImportError:
+    logger.warning("Please install auto-coder-pro to enhance llm compute ability")
     LLMComputeEngine = None
 
 
@@ -288,7 +290,7 @@ class LongContextRAG:
             )
         except Exception as e:
             logger.error(f"Error in stream_chat_oai: {str(e)}")
-            e.print_exc()
+            traceback.print_exc()
             return ["出现错误，请稍后再试。"], []
 
     def _stream_chat_oai(
@@ -489,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.181/src/autocoder/version.py

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

auto-coder-0.1.178/src/autocoder/version.py

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