auto-coder 0.1.179__py3-none-any.whl → 0.1.181__py3-none-any.whl

{auto_coder-0.1.179.dist-info → auto_coder-0.1.181.dist-info}/METADATA

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

{auto_coder-0.1.179.dist-info → auto_coder-0.1.181.dist-info}/RECORD

@@ -3,11 +3,11 @@ autocoder/auto_coder.py,sha256=HmgKa_ZApFlCsqo6BvuVeCPuncBT_Dh29ayZxxGR6lo,32216
 autocoder/auto_coder_lang.py,sha256=4qIS1tbEI8mpbtt6ThppTwKOM6MLuJTWJdgs5jIDGE0,2301
 autocoder/auto_coder_rag.py,sha256=V82EyeslAO2Z8qkMrwkyC11f1Cz6Ccjo9c867f0J_x8,11455
 autocoder/auto_coder_server.py,sha256=qRY88mkBnqSGFDcwYE5gwpe2WPhIw1nEH6LdbjCQhQk,20306
-autocoder/chat_auto_coder.py,sha256=i5xIuWlTqF0pJz8kXoa-_bW3Ic3SfCFvU2WJIMxrUHU,81798
+autocoder/chat_auto_coder.py,sha256=Yh9QRYsNe8_aQS9_R_K25WrNzLFS57sMDQTK6QKWR1U,81579
 autocoder/chat_auto_coder_lang.py,sha256=QYtu5gWEQmWKVovR_qUZ8plySZarNFX_Onk-1vN9IiA,8524
 autocoder/command_args.py,sha256=ftWw6HnFUZPiQPt1oV-SfpHQe69XN3knaFy1lpROBcU,26854
 autocoder/lang.py,sha256=e-07rYTgimpxS8sm-AxKSmH4kKQX4N05YFHJBg9trVs,12598
-autocoder/version.py,sha256=2dEBMYUzBVdhW5N4Nqr07NH9Uz48P3Yl417KfXorcI4,23
+autocoder/version.py,sha256=ZdxQFgfwMgcLstS2tHqv93Tm2T7lXzvvmJLNB0GKS_c,23
 autocoder/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/agent/auto_tool.py,sha256=DBzip-P_T6ZtT2eHexPcusmKYD0h7ufzp7TLwXAY10E,11554
 autocoder/agent/coder.py,sha256=dnITYHqkcOip8zV4lywbkYNH9w7Q3qyYaUArJ4WPrTs,866
@@ -36,7 +36,7 @@ autocoder/common/command_templates.py,sha256=zhM0pveqwjNhvvUrPYm11Jjn7D-q2C_T20o
 autocoder/common/const.py,sha256=eTjhjh4Aj4CUzviJ81jaf3Y5cwqsLATySn2wJxaS6RQ,2911
 autocoder/common/git_utils.py,sha256=UdsbUC8SpQdWt2VZ-6px6GQBWRvSJotD4IuDM4J7NBk,6598
 autocoder/common/image_to_page.py,sha256=O0cNO_vHHUP-fP4GXiVojShmNqkPnZXeIyiY1MRLpKg,13936
-autocoder/common/interpreter.py,sha256=cUqM49-d1RzgCXR7QFxqrxWqclS9ldL-T7p02NP1ANw,2014
+autocoder/common/interpreter.py,sha256=62-dIakOunYB4yjmX8SHC0Gdy2h8NtxdgbpdqRZJ5vk,2833
 autocoder/common/llm_rerank.py,sha256=FbvtCzaR661Mt2wn0qsuiEL1Y3puD6jeIJS4zg_e7Bs,3260
 autocoder/common/screenshots.py,sha256=_gA-z1HxGjPShBrtgkdideq58MG6rqFB2qMUJKjrycs,3769
 autocoder/common/search.py,sha256=_ZX03ph89rDPGMY1OrfqaDfxsDR-flh6YEHixherjwM,16616
@@ -59,10 +59,10 @@ autocoder/index/symbols_utils.py,sha256=CjcjUVajmJZB75Ty3a7kMv1BZphrm-tIBAdOJv6u
 autocoder/pyproject/__init__.py,sha256=-2-ImQVw6e3NQZQOyDlHEP5b4xVs5ur2G5izB-JCa-A,13160
 autocoder/rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/rag/api_server.py,sha256=zokIlDJlk7ucRorSLQm80uICO1mecfmn4J2zVqEBskE,6786
-autocoder/rag/doc_filter.py,sha256=LqU8Wi6klwpY9WTHVtkioSHpmo9IWhRz39dzV1gvp6E,9315
-autocoder/rag/document_retriever.py,sha256=Kz79hl9Iro1TTznn486WuzULmEatQqtAExcEIIFbL9k,23420
+autocoder/rag/doc_filter.py,sha256=Ha0Yae_G_hF72YzvrO7NoDZcG18K4hRcqGAEqfrIwAs,9330
+autocoder/rag/document_retriever.py,sha256=i_ilMTQiMg8gLK8YibJuRDc_7CVxER6bNR6-ZlcIgVI,23420
 autocoder/rag/llm_wrapper.py,sha256=xRbTBpLUH43Ah5jplL8WWWU-kjKfNgEJoUntLGBq5F4,2484
-autocoder/rag/long_context_rag.py,sha256=FW4Qv7YpNXOMrNbe28p3hY-baes6nwyVmZ7Qq73J-uc,20173
+autocoder/rag/long_context_rag.py,sha256=V8Z1LNekm6s1XO950yab1QEj9C4vLJv3qpGqhsRqmBo,20356
 autocoder/rag/rag_config.py,sha256=8LwFcTd8OJWWwi1_WY4IzjqgtT6RyE2j4PjxS5cCTDE,802
 autocoder/rag/rag_entry.py,sha256=V1RJ8RGqM30DNPmzymv64rZjNRGWn6kfc8sRy_LECg0,2451
 autocoder/rag/raw_rag.py,sha256=yS2Ur6kG0IRjhCj2_VonwxjY_xls_E62jO5Gz5j2nqE,2952
@@ -95,9 +95,9 @@ autocoder/utils/request_event_queue.py,sha256=r3lo5qGsB1dIjzVQ05dnr0z_9Z3zOkBdP1
 autocoder/utils/request_queue.py,sha256=nwp6PMtgTCiuwJI24p8OLNZjUiprC-TsefQrhMI-yPE,3889
 autocoder/utils/rest.py,sha256=3tXA8KZG6jKz_tddHNLGx77Icee88WcUeesfNsgPno4,8790
 autocoder/utils/tests.py,sha256=BqphrwyycGAvs-5mhH8pKtMZdObwhFtJ5MC_ZAOiLq8,1340
-auto_coder-0.1.179.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.179.dist-info/METADATA,sha256=Fs9mfHZCIadHFKZ7kT4yPmT_cFLvBRMtbSjySL1QT8o,2352
-auto_coder-0.1.179.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-auto_coder-0.1.179.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.179.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.179.dist-info/RECORD,,
+auto_coder-0.1.181.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.181.dist-info/METADATA,sha256=6o-R-qsou7m4vPmbbw_bnEkSvPpUYAkYq0vE0SvWvTk,2352
+auto_coder-0.1.181.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.181.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.181.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.181.dist-info/RECORD,,

{auto_coder-0.1.179.dist-info → auto_coder-0.1.181.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: setuptools (75.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

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))

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
-

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之间的数字，数字越大表示相关度越高。
     """
 

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
 
@@ -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 []
 

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]
                 )

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.179"
+__version__ = "0.1.181"