auto-coder 0.1.252__tar.gz → 0.1.253__tar.gz

{auto_coder-0.1.252 → auto_coder-0.1.253}/PKG-INFO

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

{auto_coder-0.1.252 → auto_coder-0.1.253}/src/auto_coder.egg-info/PKG-INFO

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

{auto_coder-0.1.252 → auto_coder-0.1.253}/src/autocoder/auto_coder.py

@@ -1109,8 +1109,7 @@ def main(input_args: Optional[List[str]] = None):
             return
 
         elif raw_args.agent_command == "chat":
-            from autocoder.rag.rag_entry import RAGFactory
-
+            
             memory_dir = os.path.join(args.source_dir, ".auto-coder", "memory")
             os.makedirs(memory_dir, exist_ok=True)
             memory_file = os.path.join(memory_dir, "chat_history.json")
@@ -1335,6 +1334,7 @@ def main(input_args: Optional[List[str]] = None):
             start_time = time.time()    
                         
             if "rag" in args.action:
+                from autocoder.rag.rag_entry import RAGFactory
                 args.enable_rag_search = True
                 args.enable_rag_context = False
                 rag = RAGFactory.get_rag(llm=chat_llm, args=args, path="")
@@ -1348,7 +1348,8 @@ def main(input_args: Optional[List[str]] = None):
                 response = mcp_server.send_request(
                     McpRequest(
                         query=args.query,
-                        model=args.inference_model or args.model
+                        model=args.inference_model or args.model,
+                        product_mode=args.product_mode
                     )
                 )
                 v = [[response.result,None]]

{auto_coder-0.1.252 → auto_coder-0.1.253}/src/autocoder/chat_auto_coder.py

@@ -49,7 +49,7 @@ from autocoder.common.mcp_server import get_mcp_server, McpRequest, McpInstallRe
 import byzerllm
 from byzerllm.utils import format_str_jinja2
 from autocoder.common.memory_manager import get_global_memory_file_paths 
-from autocoder import models
+from autocoder import models as models_module
 import shlex
 from autocoder.utils.llms import get_single_llm
 import pkg_resources
@@ -1539,11 +1539,12 @@ def mcp(query: str):
         if os.path.exists(temp_yaml):
             os.remove(temp_yaml)
 
-    mcp_server = get_mcp_server()
+    mcp_server = get_mcp_server()    
     response = mcp_server.send_request(
         McpRequest(
             query=query,
-            model=args.inference_model or args.model
+            model=args.inference_model or args.model,
+            product_mode=args.product_mode
         )
     )
 
@@ -1564,19 +1565,19 @@ def mcp(query: str):
         file_path = os.path.join(mcp_dir, f"{timestamp}.md")
         
         # Format response as markdown
-        markdown_content = f"# {printer.get_message_from_key('mcp_response_title')}\n\n{response.result}"
+        markdown_content = response.result
         
         # Save to file
         with open(file_path, "w", encoding="utf-8") as f:
             f.write(markdown_content)
             
-        # Print with markdown formatting
-        printer.print_panel(
-            Markdown(markdown_content),
-            text_options={"justify": "left"},
-            panel_options={
-                "border_style": "green"
-            }
+        console = Console()
+        console.print(
+            Panel(
+                Markdown(markdown_content, justify="left"),
+                title=printer.get_message_from_key('mcp_response_title'),
+                border_style="green"
+            )
         )
 
 
@@ -1741,7 +1742,7 @@ def commit(query: str):
             md5 = hashlib.md5(file_content.encode("utf-8")).hexdigest()
             file_name = os.path.basename(execute_file)
             commit_result = git_utils.commit_changes(
-                ".", f"auto_coder_{file_name}_{md5}"
+                ".", f"auto_coder_{file_name}_{md5}\n{commit_message}"
             )
             git_utils.print_commit_info(commit_result=commit_result)
             if commit_message:
@@ -2173,7 +2174,7 @@ def manage_models(params, query: str):
         printer.print_in_terminal("models_lite_only", style="red")
         return
         
-    models_data = models.load_models()
+    models_data = models_module.load_models()
     subcmd = ""
     if "/list" in query:
         subcmd = "/list"
@@ -2241,7 +2242,7 @@ def manage_models(params, query: str):
             name = args[0]
             try:
                 price = float(args[1])
-                if models.update_model_input_price(name, price):
+                if models_module.update_model_input_price(name, price):
                     printer.print_in_terminal("models_input_price_updated", style="green", name=name, price=price)
                 else:
                     printer.print_in_terminal("models_not_found", style="red", name=name)
@@ -2256,7 +2257,7 @@ def manage_models(params, query: str):
             name = args[0]
             try:
                 price = float(args[1])
-                if models.update_model_output_price(name, price):
+                if models_module.update_model_output_price(name, price):
                     printer.print_in_terminal("models_output_price_updated", style="green", name=name, price=price)
                 else:
                     printer.print_in_terminal("models_not_found", style="red", name=name)
@@ -2271,7 +2272,7 @@ def manage_models(params, query: str):
             name = args[0]
             try:
                 speed = float(args[1])
-                if models.update_model_speed(name, speed):
+                if models_module.update_model_speed(name, speed):
                     printer.print_in_terminal("models_speed_updated", style="green", name=name, speed=speed)
                 else:
                     printer.print_in_terminal("models_not_found", style="red", name=name)
@@ -2286,7 +2287,7 @@ def manage_models(params, query: str):
         if len(args) == 2:
             # Simplified: /models /add <name> <api_key>
             name, api_key = args[0], args[1]            
-            result = models.update_model_with_api_key(name, api_key)
+            result = models_module.update_model_with_api_key(name, api_key)
             if result:
                 printer.print_in_terminal("models_added", style="green", name=name)
             else:
@@ -2328,7 +2329,7 @@ def manage_models(params, query: str):
         }
 
         models_data.append(final_model)
-        models.save_models(models_data)
+        models_module.save_models(models_data)
         printer.print_in_terminal("models_add_model_success", style="green", name=data_dict["name"])
 
     elif subcmd == "/remove":
@@ -2341,7 +2342,7 @@ def manage_models(params, query: str):
         if len(filtered_models) == len(models_data):
             printer.print_in_terminal("models_add_model_remove", style="yellow", name=name)
             return
-        models.save_models(filtered_models)
+        models_module.save_models(filtered_models)
         printer.print_in_terminal("models_add_model_removed", style="green", name=name)
 
     else:

{auto_coder-0.1.252 → auto_coder-0.1.253}/src/autocoder/common/__init__.py

@@ -359,6 +359,7 @@ class AutoCoderArgs(pydantic.BaseModel):
     product_mode: Optional[str] = "lite"
 
     keep_reasoning_content: Optional[bool] = False
+    keep_only_reasoning_content: Optional[bool] = False
 
     in_code_apply: bool = False
 

{auto_coder-0.1.252 → auto_coder-0.1.253}/src/autocoder/common/code_auto_merge.py

@@ -177,5 +177,5 @@ class CodeAutoMerge:
 
         self.printer.print_in_terminal("files_merged", total=total)
         if not force_skip_git:
-            commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}")
+            commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}\n{self.args.query}")
             git_utils.print_commit_info(commit_result=commit_result)

{auto_coder-0.1.252 → auto_coder-0.1.253}/src/autocoder/common/code_auto_merge_diff.py

@@ -523,5 +523,5 @@ class CodeAutoMergeDiff:
 
         self.printer.print_in_terminal("files_merged_total", total=total)
         if not force_skip_git:
-            commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}")
+            commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}\n{self.args.query}")
             git_utils.print_commit_info(commit_result=commit_result)

{auto_coder-0.1.252 → auto_coder-0.1.253}/src/autocoder/common/code_auto_merge_editblock.py

@@ -403,7 +403,7 @@ class CodeAutoMergeEditBlock:
             if not force_skip_git:
                 try:
                     commit_result = git_utils.commit_changes(
-                        self.args.source_dir, f"auto_coder_{file_name}_{md5}"
+                        self.args.source_dir, f"auto_coder_{file_name}_{md5}\n{self.args.query}"
                     )
                     git_utils.print_commit_info(commit_result=commit_result)
                 except Exception as e:

{auto_coder-0.1.252 → auto_coder-0.1.253}/src/autocoder/common/code_auto_merge_strict_diff.py

@@ -233,7 +233,7 @@ class CodeAutoMergeStrictDiff:
                             
         self.printer.print_in_terminal("files_merged_total", total=total)
         if not force_skip_git:
-            commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}")
+            commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}\n{self.args.query}")
             git_utils.print_commit_info(commit_result=commit_result)
 
     @byzerllm.prompt(render="jinja2")

{auto_coder-0.1.252 → auto_coder-0.1.253}/src/autocoder/common/mcp_server.py

@@ -3,7 +3,6 @@ from asyncio import Queue as AsyncQueue
 import threading
 from typing import List, Dict, Any, Optional
 from dataclasses import dataclass
-import byzerllm
 from autocoder.common.mcp_hub import McpHub
 from autocoder.common.mcp_tools import McpExecutor
 from autocoder.common.mcp_hub import MCP_BUILD_IN_SERVERS
@@ -13,12 +12,13 @@ import time
 from pydantic import BaseModel
 import sys
 from loguru import logger
+from autocoder.utils.llms import get_single_llm
 
 @dataclass
 class McpRequest:
     query: str
     model: Optional[str] = None
-
+    product_mode: Optional[str] = None
 
 @dataclass
 class McpInstallRequest:
@@ -279,8 +279,7 @@ class McpServer:
                         await self._response_queue.put(McpResponse(result="", error=f"Failed to refresh MCP servers: {str(e)}"))
 
                 else:
-                    llm = byzerllm.ByzerLLM.from_default_model(
-                        model=request.model)
+                    llm = get_single_llm(request.model,product_mode=request.product_mode)
                     mcp_executor = McpExecutor(hub, llm)
                     conversations = [
                         {"role": "user", "content": request.query}]

{auto_coder-0.1.252 → auto_coder-0.1.253}/src/autocoder/common/mcp_servers/mcp_server_perplexity.py

@@ -55,7 +55,7 @@ async def handle_list_tools() -> list[types.Tool]:
                             # Commenting out larger models,which have higher risks of timing out,
                             # until Claude Desktop can handle long-running tasks effectively.
                             # "llama-3.1-sonar-large-128k-online",
-                            "llama-3.1-sonar-huge-128k-online",
+                            "sonar-reasoning-pro",
                         ],
                     },
                     "messages": {

{auto_coder-0.1.252 → auto_coder-0.1.253}/src/autocoder/rag/doc_filter.py

@@ -148,7 +148,7 @@ class DocFilter:
                 )
                 if (
                     relevance
-                    and relevance.is_relevant
+                    # and relevance.is_relevant
                     and relevance.relevant_score >= self.relevant_score
                 ):
                     relevant_docs.append(

{auto_coder-0.1.252 → auto_coder-0.1.253}/src/autocoder/rag/long_context_rag.py

@@ -439,7 +439,7 @@ class LongContextRAG:
 
             logger.info(f"Query: {query} only_contexts: {only_contexts}")
             start_time = time.time()
-            relevant_docs: List[FilterDoc] = self._filter_docs(conversations)
+            relevant_docs: List[FilterDoc] = self._filter_docs(conversations)            
             filter_time = time.time() - start_time
 
             # Filter relevant_docs to only include those with is_relevant=True

{auto_coder-0.1.252 → auto_coder-0.1.253}/src/autocoder/utils/auto_coder_utils/chat_stream_out.py

@@ -169,6 +169,10 @@ def stream_out(
     keep_reasoning_content = True
     if args:
         keep_reasoning_content = args.keep_reasoning_content            
+    
+    keep_only_reasoning_content = False
+    if args:
+        keep_only_reasoning_content = args.keep_only_reasoning_content
 
     lines_buffer = []  # 存储历史行
     current_line = ""  # 当前行
@@ -190,7 +194,10 @@ def stream_out(
                     break
                 last_meta = res[1]                
                 content = res[0]
-                reasoning_content = last_meta.reasoning_content
+
+                reasoning_content = ""
+                if last_meta:
+                    reasoning_content = last_meta.reasoning_content
 
                 if reasoning_content == "" and content == "":
                     continue
@@ -198,21 +205,24 @@ def stream_out(
                 if first_token_time == 0.0:
                     first_token_time = time.time() - first_token_time_start
 
-                if keep_reasoning_content:
-                    # 处理思考内容
-                    if reasoning_content:
-                        if assistant_response == "":  # 首次遇到思考内容时添加开标签
-                            assistant_response = "<thinking>"
-                        assistant_response += reasoning_content
-                        
-                    # 处理正式内容
-                    if content:
-                        # 如果之前有思考内容,需要先关闭thinking标签
-                        if "</thinking>" not in assistant_response and "<thinking>" in assistant_response:
-                            assistant_response += "</thinking>"
+                if keep_only_reasoning_content:
+                    assistant_response += reasoning_content
+                else:    
+                    if keep_reasoning_content:
+                        # 处理思考内容
+                        if reasoning_content:
+                            if assistant_response == "":  # 首次遇到思考内容时添加开标签
+                                assistant_response = "<thinking>"
+                            assistant_response += reasoning_content
+                            
+                        # 处理正式内容
+                        if content:
+                            # 如果之前有思考内容,需要先关闭thinking标签
+                            if "</thinking>" not in assistant_response and "<thinking>" in assistant_response:
+                                assistant_response += "</thinking>"
+                            assistant_response += content
+                    else:
                         assistant_response += content
-                else:
-                    assistant_response += content
 
                 display_delta = reasoning_content if reasoning_content else content
                 
@@ -291,5 +301,7 @@ def stream_out(
                     status=RequestOption.COMPLETED
                 ),
             )
-    last_meta.first_token_time = first_token_time
+
+    if last_meta:
+        last_meta.first_token_time = first_token_time
     return assistant_response, last_meta

auto_coder-0.1.253/src/autocoder/version.py

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

auto_coder-0.1.252/src/autocoder/version.py

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