auto-coder 0.1.354__py3-none-any.whl → 0.1.356__py3-none-any.whl

autocoder/chat_auto_coder_lang.py

@@ -375,8 +375,28 @@ MESSAGES = {
         "zh": "管理模型配置，仅在lite模式下可用"
     },
     "models_usage": {
-        "en": "Usage: /models <command>\nAvailable subcommands:\n  /list - List all models\n  /add <name> <api_key> - Add a built-in model\n  /add_model - Add a custom model\n  /remove <name> - Remove a model\n  /input_price <name> <value> - Set model input price\n  /output_price <name> <value> - Set model output price\n  /speed <name> <value> - Set model speed\n  /speed-test - Test models speed\n  /speed-test-long - Test models speed with long context",
-        "zh": "用法: /models <命令>\n可用的子命令:\n  /list - 列出所有模型\n  /add <名称> <API密钥> - 添加内置模型\n  /add_model - 添加自定义模型\n  /remove <名称> - 移除模型\n  /input_price <名称> <价格> - 设置模型输入价格\n  /output_price <名称> <价格> - 设置模型输出价格\n  /speed <名称> <速度> - 设置模型速度\n  /speed-test - 测试模型速度\n  /speed-test-long - 使用长文本上下文测试模型速度"
+        "en": '''Usage: /models <command>
+Available subcommands:
+  /list               - List all configured models (built-in + custom).
+  /add <name> <api_key> - Add or activate a built-in model and set its API key.
+  /add_model          - Add a custom model. Provide parameters in 'key=value' format, e.g., name=my_model model_name=gpt-4 base_url=... api_key=...
+  /remove <name>      - Remove a configured model by its name.
+  /input_price <name> <value> - Set the input price for a model (unit: Million tokens).
+  /output_price <name> <value> - Set the output price for a model (unit: Million tokens).
+  /speed <name> <value> - Set the average speed for a model (unit: seconds per request).
+  /speed-test [<rounds>] - Test the speed of configured models. Optionally specify the number of rounds.
+  /speed-test /long_context [<rounds>] - Test model speed using a long context. Optionally specify the number of rounds.''',
+        "zh": '''用法: /models <命令>
+可用的子命令:
+  /list               - 列出所有已配置的模型 (包括内置和自定义)。
+  /add <名称> <API密钥> - 添加或激活一个内置模型，并设置其 API 密钥。
+  /add_model          - 添加一个自定义模型。参数使用 'key=value' 格式提供，例如：name=my_model model_name=gpt-4 base_url=... api_key=...
+  /remove <名称>      - 根据名称移除一个已配置的模型。
+  /input_price <名称> <价格> - 设置指定模型的输入价格 (单位: 百万 Token)。
+  /output_price <名称> <价格> - 设置指定模型的输出价格 (单位: 百万 Token)。
+  /speed <名称> <速度> - 设置指定模型的平均速度 (单位: 秒/请求)。
+  /speed-test [<轮数>] - 测试已配置模型的速度。可以指定测试轮数 (可选)。
+  /speed-test /long_context [<轮数>] - 使用长文本上下文测试模型速度。可以指定测试轮数 (可选)。'''
     },
     "models_added": {
         "en": "Added/Updated model '{{name}}' successfully.",
@@ -458,6 +478,10 @@ MESSAGES = {
         "en": "No models found.",
         "zh": "未找到任何模型。"
     },
+    "models_no_models_matching_pattern": {
+        "en": "No models found matching pattern: {{pattern}}",
+        "zh": "未找到匹配模式的模型: {{pattern}}"
+    },
     "models_lite_only": {
         "en": "The /models command is only available in lite mode",
         "zh": "/models 命令仅在 lite 模式下可用"
@@ -569,6 +593,136 @@ MESSAGES = {
     "marketplace_add_error": {
         "en": "Error adding marketplace item: {{name}} - {{error}}",
         "zh": "添加市场项目时出错：{{name}} - {{error}}"
+    },
+    "rules_desc": {
+        "en": "Analyze current files with rules and create code learning notes",
+        "zh": "使用规则分析当前文件并创建代码学习笔记"
+    },
+    "rules_file_list_title": {
+        "en": "Rules Files List (Pattern: {{pattern}})",
+        "zh": "规则文件列表 (匹配: {{pattern}})"
+    },
+    "rules_file_path": {
+        "en": "File Path",
+        "zh": "文件路径"
+    },
+    "rules_content_length": {
+        "en": "Content Length",
+        "zh": "内容长度"
+    },
+    "rules_help_subtitle": {
+        "en": "Use '/rules help' for more information",
+        "zh": "使用 '/rules help' 获取更多帮助"
+    },
+    "rules_no_files_found": {
+        "en": "No rules files found. Use '/rules /help' to learn how to add rules.",
+        "zh": "未找到任何规则文件。请使用 '/rules /help' 了解如何添加规则。"
+    },
+    "rules_no_matching_files": {
+        "en": "No rules files found matching pattern '{{pattern}}'.",
+        "zh": "没有找到匹配模式 '{{pattern}}' 的规则文件。"
+    },
+    "rules_remove_param_required": {
+        "en": "Error: 'remove' command requires a parameter (file pattern). Usage: /rules /remove <pattern>",
+        "zh": "错误: 'remove' 命令需要一个参数（文件匹配模式）。用法: /rules /remove <匹配模式>"
+    },
+    "rules_no_files_to_remove": {
+        "en": "No rules files found matching pattern '{{pattern}}'.",
+        "zh": "没有找到匹配模式 '{{pattern}}' 的规则文件。"
+    },
+    "rules_delete_error": {
+        "en": "Error deleting file '{{file_path}}': {{error}}",
+        "zh": "删除文件 '{{file_path}}' 时出错: {{error}}"
+    },
+    "rules_delete_success": {
+        "en": "Successfully deleted {{count}} rules files.",
+        "zh": "成功删除了 {{count}} 个规则文件。"
+    },
+    "rules_no_active_files": {
+        "en": "Error: No files selected for analysis. Please use 'add_files' command to add files first.",
+        "zh": "错误: 没有选择任何文件进行分析。请先使用 'add_files' 命令添加文件。"
+    },
+    "rules_file_read_error": {
+        "en": "Error reading file '{{file_path}}': {{error}}",
+        "zh": "读取文件 '{{file_path}}' 时出错: {{error}}"
+    },
+    "rules_analysis_error": {
+        "en": "Error analyzing code: {{error}}",
+        "zh": "分析代码时出错: {{error}}"
+    },
+    "rules_help_text": {
+        "en": """
+/rules command usage:
+  /rules [query]       - Analyze current added files, optionally provide specific query content.  
+  /rules /list [pattern] - List all rules files. Optionally provide wildcard pattern (e.g. *.md).
+  /rules /remove <pattern> - Delete rules files matching the specified pattern.
+  /rules /get <pattern> - Display the content of rules files matching the specified pattern.
+  /rules /help or /rules help - Show this help message.
+
+Rules file usage:
+  Rules files are stored in the project's .autocoderrules/ directory, in Markdown format.
+  The system automatically monitors changes to this directory and updates rules.
+        """,
+        "zh": """
+/rules 命令用法:
+  /rules [查询内容]    - 分析当前已添加的文件，可选提供具体查询内容。  
+  /rules /list [匹配模式] - 列出所有规则文件。可选提供通配符匹配模式 (例如: *.md).
+  /rules /remove <匹配模式> - 删除匹配指定模式的规则文件。
+  /rules /get <匹配模式> - 显示匹配指定模式的规则文件内容。
+  /rules /help 或 /rules help - 显示此帮助信息。
+
+规则文件用法:
+  规则文件存储在项目的 .autocoderrules/ 目录下，为 Markdown 格式。
+  系统会自动监控该目录的变化并更新规则。
+        """
+    },
+    "rules_unknown_command": {
+        "en": "Unknown subcommand '/rules {{subcommand}}'. Use '/rules /help' for help.",
+        "zh": "未知的子命令 '/rules {{subcommand}}'。请使用 '/rules /help' 获取帮助。"
+    },
+    "rules_command_error": {
+        "en": "Error executing '/rules {{subcommand}}': {{error}}",
+        "zh": "执行 '/rules {{subcommand}}' 时发生错误: {{error}}"
+    },
+    "rules_get_param_required": {
+        "en": "Error: 'get' command requires a parameter (file pattern). Usage: /rules /get <pattern>",
+        "zh": "错误: 'get' 命令需要一个参数（文件匹配模式）。用法: /rules /get <匹配模式>"
+    },
+    "rules_get_no_matching_files": {
+        "en": "No rules files found matching pattern '{{pattern}}'.",
+        "zh": "没有找到匹配模式 '{{pattern}}' 的规则文件。"
+    },
+    "rules_get_file_title": {
+        "en": "Rule File: {{file_path}}",
+        "zh": "规则文件: {{file_path}}"
+    },
+    "rules_get_read_error": {
+        "en": "Error reading file '{{file_path}}': {{error}}",
+        "zh": "读取文件 '{{file_path}}' 时出错: {{error}}"
+    },
+    "rules_commit_param_required": {
+        "en": "Error: 'commit' command requires parameters. Usage: /rules /commit <commit_id> /query <query>",
+        "zh": "错误: 'commit' 命令需要参数。用法: /rules /commit <commit_id> /query <查询内容>"
+    },
+    "rules_commit_format_error": {
+        "en": "Error: Command format must be '/rules /commit <commit_id> /query <query>'",
+        "zh": "错误：命令格式必须为 '/rules /commit <commit_id> /query <你的需求>'"
+    },
+    "rules_commit_id_required": {
+        "en": "Error: Commit ID must be provided",
+        "zh": "错误：必须提供 commit ID"
+    },
+    "rules_query_required": {
+        "en": "Error: Query content must be provided",
+        "zh": "错误：必须提供查询内容"
+    },
+    "rules_commit_success": {
+        "en": "Successfully analyzed commit {{commit_id}}, query: {{query}}",
+        "zh": "成功分析 commit {{commit_id}}，查询：{{query}}"
+    },
+    "rules_commit_error": {
+        "en": "Error analyzing commit {{commit_id}}: {{error}}",
+        "zh": "分析 commit {{commit_id}} 时出错：{{error}}"
     }
 }
 

autocoder/commands/auto_command.py

@@ -440,7 +440,7 @@ class CommandAutoTuner:
         result_manager = ResultManager()
 
         while True:
-            global_cancel.check_and_raise()
+            global_cancel.check_and_raise(token=self.args.event_file)
             # 执行命令
             command = response.suggestions[0].command
             parameters = response.suggestions[0].parameters

autocoder/commands/auto_web.py

@@ -794,7 +794,7 @@ class AutoWebTuner:
             logger.info(f"开始执行迭代 {iterations}/{max_iterations}")
 
             # 检查是否需要取消操作
-            global_cancel.check_and_raise()
+            global_cancel.check_and_raise(token=self.args.event_file)
 
             # 如果没有更多操作，认为任务完成
             if not plan.actions:

autocoder/common/__init__.py

@@ -304,6 +304,8 @@ class AutoCoderArgs(pydantic.BaseModel):
     rag_duckdb_query_similarity: Optional[float] = 0.1  # DuckDB 向量化检索 相似度 阈值
     rag_duckdb_query_top_k: Optional[int] = 10000  # DuckDB 向量化检索 返回 TopK个结果(且大于相似度)
     rag_index_build_workers: Optional[int] = 10
+    rag_emb_dim: Optional[int] = 1024
+    rag_emb_text_size: Optional[int] = 1024
     # rag 本地图床地址
     local_image_host: Optional[str] = ""
     rag_recall_max_queries: Optional[int] = 5

autocoder/common/auto_coder_lang.py

@@ -106,6 +106,10 @@ MESSAGES = {
         "en": "DeepSeek Official",
         "zh": "DeepSeek官方"
     },
+    "model_provider_openrouter": {
+        "en": "OpenRouter",
+        "zh": "OpenRouter"
+    },
     "model_provider_api_key_title": {
         "en": "API Key",
         "zh": "API密钥"
@@ -114,6 +118,10 @@ MESSAGES = {
         "en": "Please enter your Volcano Engine API key:",
         "zh": "请输入您的火山方舟API密钥："
     },
+    "model_provider_openrouter_api_key_text": {
+        "en": "Please enter your OpenRouter API key:",
+        "zh": "请输入您的OpenRouter API密钥："
+    },
     "model_provider_volcano_r1_text": {
         "en": "Please enter your Volcano Engine R1 endpoint (format: ep-20250204215011-vzbsg):",
         "zh": "请输入您的火山方舟 R1 推理点(格式如: ep-20250204215011-vzbsg)："
@@ -129,7 +137,7 @@ MESSAGES = {
     "model_provider_deepseek_api_key_text": {
         "en": "Please enter your DeepSeek API key:",
         "zh": "请输入您的DeepSeek API密钥："
-    },
+    },    
     "model_provider_selected": {
         "en": "Provider configuration completed successfully! You can use /models command to view, add and modify all models later.",
         "zh": "供应商配置已成功完成！后续你可以使用 /models 命令，查看，新增和修改所有模型"

autocoder/common/command_completer.py

@@ -1,10 +1,12 @@
+import os
+import pydantic
+from typing import Callable,Dict,Any # Added import
+from pydantic import BaseModel,SkipValidation # Added import
 from prompt_toolkit.completion import Completer, Completion, CompleteEvent
 from prompt_toolkit.document import Document
-import pydantic
-from typing import Callable,Dict,Any
-from pydantic import BaseModel,SkipValidation
-from autocoder.common import AutoCoderArgs
-import os
+
+from autocoder import models as models_module # Import models module
+from autocoder.common import AutoCoderArgs # Ensure AutoCoderArgs is imported
 
 COMMANDS = {
     "/add_files": {
@@ -78,7 +80,7 @@ class MemoryConfig(BaseModel):
     """
     A model to encapsulate memory configuration and operations.
     """
-    memory: Dict[str, Any]
+    get_memory_func: SkipValidation[Callable]
     save_memory_func: SkipValidation[Callable]
 
     class Config:
@@ -366,10 +368,13 @@ class CommandCompleter(Completer):
         self.file_system_model = file_system_model
         self.memory_model = memory_model
         self.all_file_names = file_system_model.get_all_file_names_in_project()
-        self.all_files = file_system_model.get_all_file_in_project()
-        self.all_dir_names = file_system_model.get_all_dir_names_in_project()
-        self.all_files_with_dot = file_system_model.get_all_file_in_project_with_dot()
-        self.symbol_list = file_system_model.get_symbol_list()
+        self.all_files = self.file_system_model.get_all_file_in_project()
+        self.all_dir_names = self.file_system_model.get_all_dir_names_in_project()
+        self.all_files_with_dot = self.file_system_model.get_all_file_in_project_with_dot()
+        self.symbol_list = self.file_system_model.get_symbol_list()
+        # Refresh model names if they can change dynamically (optional)
+        self.all_model_names = [m['name'] for m in models_module.load_models()]
+        self.all_model_names = [m['name'] for m in models_module.load_models()] # Load model names
         self.current_file_names = []
 
     def get_completions(self, document, complete_event):
@@ -544,9 +549,49 @@ class CommandCompleter(Completer):
                         for field_name in AutoCoderArgs.model_fields.keys()
                         if field_name.startswith(current_word)
                     ]
-
+                # /conf key:[cursor] or /conf key:v[cursor]
+                elif ":" in text: # Check if colon exists anywhere after /conf
+                    parts = text[len("/conf"):].strip().split(":", 1)
+                    if len(parts) == 2:
+                        key = parts[0].strip()
+                        value_part = parts[1] # This is the part after the colon
+
+                        # Determine the word being completed (part after colon)
+                        # Find the start of the current word after the colon
+                        space_pos = value_part.rfind(" ")
+                        if space_pos != -1:
+                            current_value_word = value_part[space_pos + 1:]
+                        else:
+                            current_value_word = value_part
+
+                        # Model name completion
+                        if "model" in key:
+                            for model_name in self.all_model_names:
+                                if model_name.startswith(current_value_word):
+                                    yield Completion(model_name, start_position=-len(current_value_word))
+                            # Prioritize model completion if key matches
+                            return # Exit after providing model completions
+
+                        # Boolean completion
+                        bool_keys = {name for name, field in AutoCoderArgs.model_fields.items() if field.annotation == bool}
+                        if key in bool_keys:
+                            if "true".startswith(current_value_word):
+                                yield Completion("true", start_position=-len(current_value_word))
+                            if "false".startswith(current_value_word):
+                                yield Completion("false", start_position=-len(current_value_word))
+                            # Prioritize boolean completion if key matches
+                            return # Exit after providing boolean completions
+
+                        # Add other value completions based on key if needed here
+
+                        # No specific value completions matched, so no yield from here
+                        return # Exit if we were trying value completion
+
+                # Default completion for keys or /drop if no value completion logic was triggered above
                 for completion in completions:
-                    yield Completion(completion, start_position=-len(current_word))
+                     # Adjust start_position based on whether we are completing key or /drop
+                     yield Completion(completion, start_position=-len(current_word))
+
             elif words[0] in ["/chat", "/coding","/auto"]:
                 image_extensions = (
                     ".png",
@@ -895,6 +940,7 @@ class CommandCompleter(Completer):
     def update_current_files(self, files):
         self.current_file_names = [f for f in files]
 
+
     def refresh_files(self):
         self.all_file_names = self.file_system_model.get_all_file_names_in_project()
         self.all_files = self.file_system_model.get_all_file_in_project()