jarvis-ai-assistant 0.3.9__tar.gz → 0.3.11__tar.gz

{jarvis_ai_assistant-0.3.9/src/jarvis_ai_assistant.egg-info → jarvis_ai_assistant-0.3.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jarvis-ai-assistant
-Version: 0.3.9
+Version: 0.3.11
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire

{jarvis_ai_assistant-0.3.9 → jarvis_ai_assistant-0.3.11}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "jarvis-ai-assistant"
-version = "0.3.9"
+version = "0.3.11"
 description = "Jarvis: An AI assistant that uses tools to interact with the system"
 readme = "README.md"
 authors = [{ name = "skyfire", email = "skyfireitdiy@hotmail.com" }]
@@ -106,7 +106,7 @@ jst = "jarvis.jarvis_stats.cli:main"
 index-strategy = "unsafe-first-match"
 
 [[tool.uv.index]]
-url = "https://artsz.zte.com.cn/artifactory/api/pypi/public-pypi-virtual/simple"
+url = "https://mirrors.cloud.tencent.com/pypi/simple"
 default = true
 
 [[tool.uv.index]]

{jarvis_ai_assistant-0.3.9 → jarvis_ai_assistant-0.3.11}/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup, find_packages  # type: ignore
 
 setup(
     name="jarvis-ai-assistant",
-    version="0.3.9",
+    version="0.3.11",
     author="skyfire",
     author_email="skyfireitdiy@hotmail.com",
     description="An AI assistant that uses various tools to interact with the system",

{jarvis_ai_assistant-0.3.9 → jarvis_ai_assistant-0.3.11}/src/jarvis/__init__.py

@@ -1,4 +1,4 @@
 # -*- coding: utf-8 -*-
 """Jarvis AI Assistant"""
 
-__version__ = "0.3.9"
+__version__ = "0.3.11"

{jarvis_ai_assistant-0.3.9 → jarvis_ai_assistant-0.3.11}/src/jarvis/jarvis_agent/jarvis.py

@@ -19,17 +19,17 @@ def run_cli(
     ctx: typer.Context,
     llm_type: str = typer.Option(
         "normal",
-        "-t", "--llm_type",
+        "-t", "--llm-type",
         help="使用的LLM类型，可选值：'normal'（普通）或 'thinking'（思考模式）",
     ),
     task: Optional[str] = typer.Option(
         None, "-T", "--task", help="从命令行直接输入任务内容"
     ),
     model_group: Optional[str] = typer.Option(
-        None, "-g", "--llm_group", help="使用的模型组，覆盖配置文件中的设置"
+        None, "-g", "--llm-group", help="使用的模型组，覆盖配置文件中的设置"
     ),
     tool_group: Optional[str] = typer.Option(
-        None, "-G", "--tool_group", help="使用的工具组，覆盖配置文件中的设置"
+        None, "-G", "--tool-group", help="使用的工具组，覆盖配置文件中的设置"
     ),
     config_file: Optional[str] = typer.Option(
         None, "-f", "--config", help="自定义配置文件路径"

{jarvis_ai_assistant-0.3.9 → jarvis_ai_assistant-0.3.11}/src/jarvis/jarvis_agent/main.py

@@ -43,18 +43,18 @@ def cli(
         None, "-f", "--config", help="代理配置文件路径"
     ),
     agent_definition: Optional[str] = typer.Option(
-        None, "-c", "--agent_definition", help="代理定义文件路径"
+        None, "-c", "--agent-definition", help="代理定义文件路径"
     ),
     task: Optional[str] = typer.Option(
         None, "-t", "--task", help="初始任务内容"
     ),
     llm_type: str = typer.Option(
         "normal",
-        "-t", "--llm_type",
+        "-t", "--llm-type",
         help="使用的LLM类型，覆盖配置文件中的设置",
     ),
     model_group: Optional[str] = typer.Option(
-        None, "-g", "--llm_group", help="使用的模型组，覆盖配置文件中的设置"
+        None, "-g", "--llm-group", help="使用的模型组，覆盖配置文件中的设置"
     ),):
     """Main entry point for Jarvis agent"""
     # Initialize environment

{jarvis_ai_assistant-0.3.9 → jarvis_ai_assistant-0.3.11}/src/jarvis/jarvis_code_agent/code_agent.py

@@ -51,24 +51,31 @@ class CodeAgent:
         llm_type: str = "normal",
         model_group: Optional[str] = None,
         need_summary: bool = True,
+        append_tools: Optional[str] = None,
     ):
         self.root_dir = os.getcwd()
 
         # 检测 git username 和 email 是否已设置
         self._check_git_config()
         tool_registry = ToolRegistry()  # type: ignore
-        tool_registry.use_tools(
-            [
-                "execute_script",
-                "search_web",
-                "ask_user",
-                "read_code",
-                "rewrite_file",
-                "save_memory",
-                "retrieve_memory",
-                "clear_memory",
-            ]
-        )
+        base_tools = [
+            "execute_script",
+            "search_web",
+            "ask_user",
+            "read_code",
+            "rewrite_file",
+            "save_memory",
+            "retrieve_memory",
+            "clear_memory",
+        ]
+        
+        if append_tools:
+            additional_tools = [tool.strip() for tool in append_tools.split(",")]
+            base_tools.extend(additional_tools)
+            # 去重
+            base_tools = list(dict.fromkeys(base_tools))
+            
+        tool_registry.use_tools(base_tools)
         code_system_prompt = self._get_system_prompt()
         self.agent = Agent(
             system_prompt=code_system_prompt,
@@ -639,15 +646,18 @@ def cli(
     llm_type: str = typer.Option(
         "normal",
         "-t",
-        "--llm_type",
+        "--llm-type",
         help="使用的LLM类型，可选值：'normal'（普通）或 'thinking'（思考模式）",
     ),
     model_group: Optional[str] = typer.Option(
-        None, "-g", "--llm_group", help="使用的模型组，覆盖配置文件中的设置"
+        None, "-g", "--llm-group", help="使用的模型组，覆盖配置文件中的设置"
     ),
     requirement: Optional[str] = typer.Option(
         None, "-r", "--requirement", help="要处理的需求描述"
     ),
+    append_tools: Optional[str] = typer.Option(
+        None, "--append-tools", help="要追加的工具列表，用逗号分隔"
+    ),
     restore_session: bool = typer.Option(
         False,
         "--restore-session",
@@ -698,6 +708,7 @@ def cli(
             llm_type=llm_type,
             model_group=model_group,
             need_summary=False,
+            append_tools=append_tools,
         )
 
         # 尝试恢复会话

{jarvis_ai_assistant-0.3.9 → jarvis_ai_assistant-0.3.11}/src/jarvis/jarvis_code_analysis/code_review.py

@@ -826,11 +826,11 @@ def review_commit(
     llm_type: str = typer.Option(
         "normal",
         "-t",
-        "--llm_type",
+        "--llm-type",
         help="使用的LLM类型，可选值：'normal'（普通）或 'thinking'（思考模式）",
     ),
     model_group: Optional[str] = typer.Option(
-        None, "-g", "--llm_group", help="使用的模型组，覆盖配置文件中的设置"
+        None, "-g", "--llm-group", help="使用的模型组，覆盖配置文件中的设置"
     ),
 ):
     """审查指定的提交"""
@@ -857,11 +857,11 @@ def review_current(
     llm_type: str = typer.Option(
         "normal",
         "-t",
-        "--llm_type",
+        "--llm-type",
         help="使用的LLM类型，可选值：'normal'（普通）或 'thinking'（思考模式）",
     ),
     model_group: Optional[str] = typer.Option(
-        None, "-g", "--llm_group", help="使用的模型组，覆盖配置文件中的设置"
+        None, "-g", "--llm-group", help="使用的模型组，覆盖配置文件中的设置"
     ),
 ):
     """审查当前的变更"""
@@ -889,11 +889,11 @@ def review_range(
     llm_type: str = typer.Option(
         "normal",
         "-t",
-        "--llm_type",
+        "--llm-type",
         help="使用的LLM类型，可选值：'normal'（普通）或 'thinking'（思考模式）",
     ),
     model_group: Optional[str] = typer.Option(
-        None, "-g", "--llm_group", help="使用的模型组，覆盖配置文件中的设置"
+        None, "-g", "--llm-group", help="使用的模型组，覆盖配置文件中的设置"
     ),
 ):
     """审查提交范围"""
@@ -922,11 +922,11 @@ def review_file(
     llm_type: str = typer.Option(
         "normal",
         "-t",
-        "--llm_type",
+        "--llm-type",
         help="使用的LLM类型，可选值：'normal'（普通）或 'thinking'（思考模式）",
     ),
     model_group: Optional[str] = typer.Option(
-        None, "-g", "--llm_group", help="使用的模型组，覆盖配置文件中的设置"
+        None, "-g", "--llm-group", help="使用的模型组，覆盖配置文件中的设置"
     ),
 ):
     """审查指定的文件"""

{jarvis_ai_assistant-0.3.9 → jarvis_ai_assistant-0.3.11}/src/jarvis/jarvis_data/config_schema.json

@@ -136,6 +136,16 @@
       "description": "思考操作模型名称",
       "default": "deep_seek_v3"
     },
+    "JARVIS_WEB_SEARCH_PLATFORM": {
+      "type": "string",
+      "description": "Web搜索使用的平台名称",
+      "default": ""
+    },
+    "JARVIS_WEB_SEARCH_MODEL": {
+      "type": "string",
+      "description": "Web搜索使用的模型名称",
+      "default": ""
+    },
     "JARVIS_LLM_GROUP": {
       "type": "string",
       "description": "选择一个预定义的模型组",

{jarvis_ai_assistant-0.3.9 → jarvis_ai_assistant-0.3.11}/src/jarvis/jarvis_git_utils/git_commiter.py

@@ -389,11 +389,11 @@ def cli(
     llm_type: str = typer.Option(
         "normal",
         "-t",
-        "--llm_type",
+        "--llm-type",
         help="使用的LLM类型，可选值：'normal'（普通）或 'thinking'（思考模式）",
     ),
     model_group: Optional[str] = typer.Option(
-        None, "-g", "--llm_group", help="使用的模型组，覆盖配置文件中的设置"
+        None, "-g", "--llm-group", help="使用的模型组，覆盖配置文件中的设置"
     ),
 ):
     init_env("欢迎使用 Jarvis-GitCommitTool，您的Git提交助手已准备就绪！")

{jarvis_ai_assistant-0.3.9 → jarvis_ai_assistant-0.3.11}/src/jarvis/jarvis_memory_organizer/memory_organizer.py

@@ -590,12 +590,12 @@ def organize(
         help="模拟运行，只显示将要进行的操作但不实际执行",
     ),
     llm_group: Optional[str] = typer.Option(
-        None, "-g", "--llm_group", help="使用的模型组，覆盖配置文件中的设置"
+        None, "-g", "--llm-group", help="使用的模型组，覆盖配置文件中的设置"
     ),
     llm_type: Optional[str] = typer.Option(
         "normal",
         "-t",
-        "--llm_type",
+        "--llm-type",
         help="使用的LLM类型，可选值：'normal'（普通）或 'thinking'（思考模式）",
     ),
 ):

{jarvis_ai_assistant-0.3.9 → jarvis_ai_assistant-0.3.11}/src/jarvis/jarvis_platform_manager/main.py

@@ -363,11 +363,11 @@ def chat_command(
     llm_type: str = typer.Option(
         "normal",
         "-t",
-        "--llm_type",
+        "--llm-type",
         help="使用的LLM类型，可选值：'normal'（普通）或 'thinking'（思考模式）",
     ),
     llm_group: Optional[str] = typer.Option(
-        None, "-g", "--llm_group", help="使用的模型组，覆盖配置文件中的设置"
+        None, "-g", "--llm-group", help="使用的模型组，覆盖配置文件中的设置"
     ),
 ) -> None:
     """与指定平台和模型聊天。"""
@@ -447,11 +447,11 @@ def role_command(
     llm_type: Optional[str] = typer.Option(
         None,
         "-t",
-        "--llm_type",
+        "--llm-type",
         help="使用的LLM类型，可选值：'normal'（普通）或 'thinking'（思考模式），覆盖角色配置",
     ),
     llm_group: Optional[str] = typer.Option(
-        None, "-g", "--llm_group", help="使用的模型组，覆盖配置文件中的设置"
+        None, "-g", "--llm-group", help="使用的模型组，覆盖配置文件中的设置"
     ),
 ) -> None:
     """加载角色配置文件并开始对话。"""

{jarvis_ai_assistant-0.3.9 → jarvis_ai_assistant-0.3.11}/src/jarvis/jarvis_tools/search_web.py

@@ -12,6 +12,10 @@ from ddgs import DDGS  # type: ignore[import-not-found]
 
 from jarvis.jarvis_agent import Agent
 from jarvis.jarvis_platform.registry import PlatformRegistry
+from jarvis.jarvis_utils.config import (
+    get_web_search_platform_name,
+    get_web_search_model_name,
+)
 from jarvis.jarvis_utils.http import get as http_get
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
@@ -136,6 +140,25 @@ class SearchWebTool:
                 "success": False,
             }
 
+        # 检查是否配置了专门的 Web 搜索平台和模型
+        web_search_platform = get_web_search_platform_name()
+        web_search_model = get_web_search_model_name()
+
+        # 如果配置了专门的 Web 搜索平台和模型，优先使用
+        if web_search_platform and web_search_model:
+            model = PlatformRegistry().create_platform(web_search_platform)
+            if model:
+                model.set_model_name(web_search_model)
+                if model.support_web():
+                    model.set_web(True)
+                    model.set_suppress_output(False)
+                    return {
+                        "stdout": model.chat_until_success(query),
+                        "stderr": "",
+                        "success": True,
+                    }
+
+        # 否则使用现有的模型选择流程
         if agent.model.support_web():
             model = PlatformRegistry().create_platform(agent.model.platform_name())
             if not model:

{jarvis_ai_assistant-0.3.9 → jarvis_ai_assistant-0.3.11}/src/jarvis/jarvis_utils/config.py

@@ -499,6 +499,31 @@ def get_rag_use_rerank() -> bool:
     return config.get("use_rerank", True) is True
 
 
+# ==============================================================================
+# Web Search Configuration
+# ==============================================================================
+
+
+def get_web_search_platform_name() -> Optional[str]:
+    """
+    获取Web搜索使用的平台名称。
+
+    返回:
+        Optional[str]: 平台名称，如果未配置则返回None
+    """
+    return GLOBAL_CONFIG_DATA.get("JARVIS_WEB_SEARCH_PLATFORM")
+
+
+def get_web_search_model_name() -> Optional[str]:
+    """
+    获取Web搜索使用的模型名称。
+
+    返回:
+        Optional[str]: 模型名称，如果未配置则返回None
+    """
+    return GLOBAL_CONFIG_DATA.get("JARVIS_WEB_SEARCH_MODEL")
+
+
 # ==============================================================================
 # Tool Configuration
 # ==============================================================================

{jarvis_ai_assistant-0.3.9 → jarvis_ai_assistant-0.3.11/src/jarvis_ai_assistant.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jarvis-ai-assistant
-Version: 0.3.9
+Version: 0.3.11
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire