auto-coder 0.1.233__tar.gz → 0.1.237__tar.gz

{auto_coder-0.1.233 → auto_coder-0.1.237}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.233
+Version: 0.1.237
 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.151
+Requires-Dist: byzerllm[saas]>=0.1.153
 Requires-Dist: patch
 Requires-Dist: diff_match_patch
 Requires-Dist: GitPython
@@ -39,7 +39,6 @@ Requires-Dist: real_agent
 Requires-Dist: python-docx
 Requires-Dist: docx2txt
 Requires-Dist: pdf2image
-Requires-Dist: Spire.Doc
 Requires-Dist: docx2pdf
 Requires-Dist: pyperclip
 Requires-Dist: colorama

{auto_coder-0.1.233 → auto_coder-0.1.237}/src/auto_coder.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.233
+Version: 0.1.237
 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.151
+Requires-Dist: byzerllm[saas]>=0.1.153
 Requires-Dist: patch
 Requires-Dist: diff_match_patch
 Requires-Dist: GitPython
@@ -39,7 +39,6 @@ Requires-Dist: real_agent
 Requires-Dist: python-docx
 Requires-Dist: docx2txt
 Requires-Dist: pdf2image
-Requires-Dist: Spire.Doc
 Requires-Dist: docx2pdf
 Requires-Dist: pyperclip
 Requires-Dist: colorama

{auto_coder-0.1.233 → auto_coder-0.1.237}/src/auto_coder.egg-info/SOURCES.txt

@@ -9,7 +9,6 @@ src/auto_coder.egg-info/requires.txt
 src/auto_coder.egg-info/top_level.txt
 src/autocoder/__init__.py
 src/autocoder/auto_coder.py
-src/autocoder/auto_coder_lang.py
 src/autocoder/auto_coder_rag.py
 src/autocoder/auto_coder_rag_client_mcp.py
 src/autocoder/auto_coder_rag_mcp.py
@@ -37,6 +36,7 @@ src/autocoder/common/__init__.py
 src/autocoder/common/anything2images.py
 src/autocoder/common/anything2img.py
 src/autocoder/common/audio.py
+src/autocoder/common/auto_coder_lang.py
 src/autocoder/common/buildin_tokenizer.py
 src/autocoder/common/chunk_validation.py
 src/autocoder/common/cleaner.py
@@ -63,6 +63,7 @@ src/autocoder/common/mcp_hub.py
 src/autocoder/common/mcp_server.py
 src/autocoder/common/mcp_tools.py
 src/autocoder/common/memory_manager.py
+src/autocoder/common/printer.py
 src/autocoder/common/recall_validation.py
 src/autocoder/common/screenshots.py
 src/autocoder/common/search.py
@@ -141,6 +142,7 @@ src/autocoder/utils/request_event_queue.py
 src/autocoder/utils/request_queue.py
 src/autocoder/utils/rest.py
 src/autocoder/utils/tests.py
+src/autocoder/utils/types.py
 src/autocoder/utils/auto_coder_utils/__init__.py
 src/autocoder/utils/auto_coder_utils/chat_stream_out.py
 src/autocoder/utils/chat_auto_coder_utils/__init__.py

{auto_coder-0.1.233 → auto_coder-0.1.237}/src/auto_coder.egg-info/requires.txt

@@ -16,7 +16,7 @@ tabulate
 jupyter_client
 prompt-toolkit
 tokenizers
-byzerllm[saas]>=0.1.151
+byzerllm[saas]>=0.1.153
 patch
 diff_match_patch
 GitPython
@@ -29,7 +29,6 @@ real_agent
 python-docx
 docx2txt
 pdf2image
-Spire.Doc
 docx2pdf
 pyperclip
 colorama

{auto_coder-0.1.233 → auto_coder-0.1.237}/src/autocoder/auto_coder.py

@@ -39,12 +39,12 @@ from rich.console import Console
 from rich.panel import Panel
 from rich.markdown import Markdown
 from rich.live import Live
-from autocoder.auto_coder_lang import get_message
+from autocoder.common.auto_coder_lang import get_message
 from autocoder.common.memory_manager import save_to_memory_file
 from autocoder import models as models_module
 from autocoder.common.utils_code_auto_generate import stream_chat_with_continue
 from autocoder.utils.auto_coder_utils.chat_stream_out import stream_out
-
+from autocoder.common.printer import Printer
 console = Console()
 
 
@@ -69,8 +69,7 @@ def load_include_files(config, base_path, max_depth=10, current_depth=0):
             include_files = [include_files]
 
         for include_file in include_files:
-            abs_include_path = resolve_include_path(base_path, include_file)
-            logger.info(f"Loading include file: {abs_include_path}")
+            abs_include_path = resolve_include_path(base_path, include_file)            
             with open(abs_include_path, "r") as f:
                 include_config = yaml.safe_load(f)
                 if not include_config:
@@ -131,15 +130,15 @@ def main(input_args: Optional[List[str]] = None):
     if not os.path.isabs(args.source_dir):
         args.source_dir = os.path.abspath(args.source_dir)
 
-    if not args.silence:
-        print("Command Line Arguments:")
-        print("-" * 50)
-        for arg, value in vars(args).items():
-            if arg == "context" and value:
-                print(f"{arg:20}: {value[:30]}...")
-            else:
-                print(f"{arg:20}: {value}")
-        print("-" * 50)
+    # if not args.silence:
+    #     print("Command Line Arguments:")
+    #     print("-" * 50)
+    #     for arg, value in vars(args).items():
+    #         if arg == "context" and value:
+    #             print(f"{arg:20}: {value[:30]}...")
+    #         else:
+    #             print(f"{arg:20}: {value}")
+    #     print("-" * 50)
 
     # init store
     store = Store(os.path.join(args.source_dir, ".auto-coder", "metadata.db"))
@@ -244,10 +243,7 @@ def main(input_args: Optional[List[str]] = None):
                 new_file = os.path.join(
                     actions_dir, f"{new_seq}_{raw_args.name}.yml")
                 with open(new_file, "w") as f:
-                    f.write(content)
-
-        print(f"Successfully created new action file: {new_file}")
-
+                    f.write(content)        
         # open_yaml_file_in_editor(new_file)
         return
 
@@ -281,6 +277,22 @@ def main(input_args: Optional[List[str]] = None):
                 byzerllm.connect_cluster(address=args.ray_address)
 
             llm = byzerllm.ByzerLLM(verbose=args.print_request)
+
+            code_model = byzerllm.ByzerLLM()
+            code_model.setup_default_model_name("deepseek_chat")
+            llm.setup_sub_client("code_model", code_model)
+
+            index_filter_model = byzerllm.ByzerLLM()
+            index_filter_model.setup_default_model_name("deepseek_r1_chat")
+            llm.setup_sub_client("index_filter_model", index_filter_model)
+
+            generate_rerank_model = byzerllm.ByzerLLM()
+            generate_rerank_model.setup_default_model_name("deepseek_r1_chat")
+            llm.setup_sub_client("generate_rerank_model", generate_rerank_model)
+
+            chat_model = byzerllm.ByzerLLM()
+            chat_model.setup_default_model_name("deepseek_r1_chat")
+            llm.setup_sub_client("chat_model", chat_model)
         
         if args.product_mode == "lite":
             llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_chat")
@@ -344,9 +356,23 @@ def main(input_args: Optional[List[str]] = None):
                 }
             )
 
+            index_filter_llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_r1_chat")
+            index_filter_llm.deploy(
+                model_path="",
+                pretrained_model_type="saas/openai",
+                udf_name="deepseek_r1_chat",
+                infer_params={
+                    "saas.base_url": "https://api.deepseek.com/v1",
+                    "saas.api_key": api_key,
+                    "saas.model": "deepseek-reasoner",
+                    "saas.is_reasoning": True
+                }
+            )
+
             llm.setup_sub_client("code_model", code_llm)
             llm.setup_sub_client("chat_model", chat_llm)
             llm.setup_sub_client("generate_rerank_model", generate_rerank_llm)
+            llm.setup_sub_client("index_filter_model", index_filter_llm)
 
         if args.product_mode == "lite":                                    
             # Set up default models based on configuration
@@ -444,7 +470,7 @@ def main(input_args: Optional[List[str]] = None):
                         "saas.is_reasoning": model_info["is_reasoning"]
                     }
                 )
-                llm.setup_sub_client("inference_model", inference_model) 
+                llm.setup_sub_client("inference_model", inference_model)                 
 
             if args.index_filter_model:
                 model_info = models_module.get_model_by_name(args.index_filter_model)
@@ -1114,20 +1140,25 @@ def main(input_args: Optional[List[str]] = None):
                 chat_llm = llm
 
             source_count = 0
-            pre_conversations = []
-            if args.context:
-                context = json.loads(args.context)
-                if "file_content" in context:
-                    file_content = context["file_content"]
-                    pre_conversations.append(
-                        {
-                            "role": "user",
-                            "content": f"请阅读下面的代码和文档：\n\n <files>\n{file_content}\n</files>",
-                        },
-                    )
-                    pre_conversations.append(
-                        {"role": "assistant", "content": "read"})
-                    source_count += 1
+            pre_conversations = []            
+            context_content = args.context if args.context else ""
+            if args.context:                
+                try:
+                    context = json.loads(args.context)
+                    if "file_content" in context:
+                        context_content = context["file_content"]
+                except:
+                    pass                     
+
+                pre_conversations.append(
+                    {
+                        "role": "user",
+                        "content": f"请阅读下面的代码和文档：\n\n <files>\n{context_content}\n</files>",
+                    },
+                )
+                pre_conversations.append(
+                    {"role": "assistant", "content": "read"})
+                source_count += 1
 
             from autocoder.index.index import IndexManager
             from autocoder.index.entry import build_index_and_filter_files
@@ -1142,10 +1173,11 @@ def main(input_args: Optional[List[str]] = None):
             else:
                 pp = SuffixProject(args=args, llm=llm, file_filter=None)
             pp.run()
-            sources = pp.sources
+            sources = pp.sources                       
 
             s = build_index_and_filter_files(
-                llm=llm, args=args, sources=sources)
+                llm=llm, args=args, sources=sources)                        
+            
             if s:
                 pre_conversations.append(
                     {
@@ -1275,7 +1307,8 @@ def main(input_args: Optional[List[str]] = None):
                     save_to_memory_file(ask_conversation=chat_history["ask_conversation"],
                                         query=args.query,
                                         response=result)
-                    print("Saved to your memory")
+                    printer = Printer()
+                    printer.print_in_terminal("memory_save_success")
                 return {}
                         
             if "rag" in args.action: