jarvis-ai-assistant 0.3.27__py3-none-any.whl → 0.3.28__py3-none-any.whl

jarvis/jarvis_tools/sub_code_agent.py

@@ -8,12 +8,13 @@ sub_code_agent 工具
 - 不依赖父 Agent，所有配置使用系统默认与全局变量
 - 子Agent必须自动完成(auto_complete=True)且需要summary(need_summary=True)
 """
-from typing import Any, Dict, Optional
+from typing import Any, Dict
 import json
 
 from jarvis.jarvis_code_agent.code_agent import CodeAgent
 from jarvis.jarvis_utils.globals import delete_agent
 from jarvis.jarvis_utils.config import set_config, get_git_check_mode
+from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
 
 class SubCodeAgentTool:
@@ -171,6 +172,21 @@ class SubCodeAgentTool:
                             model_obj: Any = getattr(code_agent.agent, "model", None)
                             if model_obj is not None:
                                 model_obj.set_model_name(parent_model_name)
+                                # 模型有效性校验与回退，确保父Agent模型在子Agent平台上可用
+                                try:
+                                    available_models = model_obj.get_model_list()
+                                    if available_models:
+                                        available_names = [m for m, _ in available_models]
+                                        current_model_name = model_obj.name()
+                                        if current_model_name not in available_names:
+                                            PrettyOutput.print(
+                                                f"检测到子CodeAgent模型 {current_model_name} 不存在于平台 {model_obj.platform_name()} 的可用模型列表，将回退到 {available_names[0]}",
+                                                OutputType.WARNING,
+                                            )
+                                            model_obj.set_model_name(available_names[0])
+                                except Exception:
+                                    # 获取模型列表或设置模型失败时，保持原设置并继续，交由底层报错处理
+                                    pass
                     except Exception:
                         pass
             except Exception:

jarvis/jarvis_tools/virtual_tty.py

@@ -7,25 +7,14 @@ from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
 # 为了类型检查，总是导入这些模块
 if TYPE_CHECKING:
-    import fcntl
-    import pty
-    import select
-    import signal
-    import subprocess
-    import threading
-    import queue
+    pass
 
 # 平台相关的导入
 if sys.platform != "win32":
-    import fcntl
-    import pty
-    import select
-    import signal
+    pass
 else:
     # Windows平台的导入
-    import subprocess
-    import threading
-    import queue
+    pass
 
 
 class VirtualTTYTool:

jarvis/jarvis_utils/builtin_replace_map.py

@@ -31,25 +31,25 @@ arguments:
     },
     "FindRelatedFiles": {
         "append": False,
-        "template": f"""
+        "template": """
 请使用工具在当前目录下查找与以下功能相关的文件：
 """,
     },
     "Dev": {
         "append": False,
-        "template": f"""
+        "template": """
 请调用create_code_agent开发以下需求：
 """,
     },
     "Fix": {
         "append": False,
-        "template": f"""
+        "template": """
 请修复以下问题：
 """,
     },
     "Check": {
         "append": True,
-        "template": f"""
+        "template": """
 请使用静态检查工具检查当前代码，必须严格遵守工具调用格式。
 
 检查要求：

jarvis/jarvis_utils/config.py

@@ -130,9 +130,14 @@ def _get_resolved_model_config(
     解析并合并模型配置，处理模型组。
 
     优先级顺序:
-    1. 单独的环境变量 (JARVIS_PLATFORM, JARVIS_MODEL, etc.)
-    2. JARVIS_LLM_GROUP 中定义的组配置
-    3. 代码中的默认值
+    - 当通过 model_group_override（例如命令行 -g/--llm-group）指定组时：
+        1. JARVIS_LLM_GROUP 中定义的组配置
+        2. 仅当组未提供对应键时，回退到顶层环境变量 (JARVIS_PLATFORM, JARVIS_MODEL, JARVIS_MAX_INPUT_TOKEN_COUNT)
+        3. 代码中的默认值
+    - 当未显式指定组时（使用默认组或未设置）：
+        1. 顶层环境变量 (JARVIS_PLATFORM, JARVIS_MODEL, JARVIS_MAX_INPUT_TOKEN_COUNT)
+        2. JARVIS_LLM_GROUP 中定义的组配置
+        3. 代码中的默认值
 
     返回:
         Dict[str, Any]: 解析后的模型配置字典
@@ -155,14 +160,24 @@ def _get_resolved_model_config(
     # Start with group config
     resolved_config = group_config.copy()
 
-    # Override with specific settings from GLOBAL_CONFIG_DATA
-    for key in [
+    # 覆盖策略：
+    # - 若通过 CLI 传入了 model_group_override，则优先使用组内配置；
+    #   仅当组未提供对应键时，才回落到顶层 GLOBAL_CONFIG_DATA。
+    # - 若未传入 override（即使用默认组），保持原有行为：由顶层键覆盖组配置。
+    override_keys = [
         "JARVIS_PLATFORM",
         "JARVIS_MODEL",
         "JARVIS_MAX_INPUT_TOKEN_COUNT",
-    ]:
+    ]
+    for key in override_keys:
         if key in GLOBAL_CONFIG_DATA:
-            resolved_config[key] = GLOBAL_CONFIG_DATA[key]
+            if model_group_override is None:
+                # 未显式指定组：顶层覆盖组
+                resolved_config[key] = GLOBAL_CONFIG_DATA[key]
+            else:
+                # 显式指定组：仅在组未定义该键时回退到顶层
+                if key not in resolved_config:
+                    resolved_config[key] = GLOBAL_CONFIG_DATA[key]
 
     return resolved_config
 
@@ -196,7 +211,7 @@ def _deprecated_platform_name_v1(model_group_override: Optional[str] = None) ->
     返回：
         str: 平台名称，默认为正常操作平台
     """
-    config = _get_resolved_model_config(model_group_override)
+    _get_resolved_model_config(model_group_override)
     # Fallback to normal platform if thinking platform is not specified
     return get_normal_platform_name(model_group_override)
 
@@ -208,7 +223,7 @@ def _deprecated_model_name_v1(model_group_override: Optional[str] = None) -> str
     返回：
         str: 模型名称，默认为正常操作模型
     """
-    config = _get_resolved_model_config(model_group_override)
+    _get_resolved_model_config(model_group_override)
     # Fallback to normal model if thinking model is not specified
     return get_normal_model_name(model_group_override)
 
@@ -220,7 +235,7 @@ def is_execute_tool_confirm() -> bool:
     返回：
         bool: 如果需要确认则返回True，默认为False
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_EXECUTE_TOOL_CONFIRM", False) == True
+    return GLOBAL_CONFIG_DATA.get("JARVIS_EXECUTE_TOOL_CONFIRM", False)
 
 
 def is_confirm_before_apply_patch() -> bool:
@@ -230,7 +245,7 @@ def is_confirm_before_apply_patch() -> bool:
     返回：
         bool: 如果需要确认则返回True，默认为False
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_CONFIRM_BEFORE_APPLY_PATCH", False) == True
+    return GLOBAL_CONFIG_DATA.get("JARVIS_CONFIRM_BEFORE_APPLY_PATCH", False)
 
 
 def get_data_dir() -> str:
@@ -270,7 +285,7 @@ def get_pretty_output() -> bool:
     if platform.system() == "Windows":
         return False
 
-    return GLOBAL_CONFIG_DATA.get("JARVIS_PRETTY_OUTPUT", False) == True
+    return GLOBAL_CONFIG_DATA.get("JARVIS_PRETTY_OUTPUT", False)
 
 
 def is_use_methodology() -> bool:
@@ -280,7 +295,7 @@ def is_use_methodology() -> bool:
     返回：
         bool: 如果启用方法论则返回True，默认为True
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_USE_METHODOLOGY", True) == True
+    return GLOBAL_CONFIG_DATA.get("JARVIS_USE_METHODOLOGY", True)
 
 
 def is_use_analysis() -> bool:
@@ -290,7 +305,7 @@ def is_use_analysis() -> bool:
     返回：
         bool: 如果启用任务分析则返回True，默认为True
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_USE_ANALYSIS", True) == True
+    return GLOBAL_CONFIG_DATA.get("JARVIS_USE_ANALYSIS", True)
 
 
 def get_tool_load_dirs() -> List[str]:
@@ -390,7 +405,7 @@ def is_print_prompt() -> bool:
     返回：
         bool: 如果打印提示则返回True，默认为True
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_PRINT_PROMPT", False) == True
+    return GLOBAL_CONFIG_DATA.get("JARVIS_PRINT_PROMPT", False)
 
 
 def is_print_error_traceback() -> bool:

jarvis/jarvis_utils/git_utils.py

@@ -246,7 +246,7 @@ def handle_commit_workflow() -> bool:
             ["git", "commit", "-m", f"CheckPoint #{commit_count + 1}"], check=True
         )
         return True
-    except subprocess.CalledProcessError as e:
+    except subprocess.CalledProcessError:
         return False
 
 

jarvis/jarvis_utils/globals.py

@@ -10,8 +10,7 @@
 import os
 
 # 全局变量：保存消息历史
-from typing import Any, Dict, Set, List, Optional
-from datetime import datetime
+from typing import Any, Dict, List, Optional
 
 message_history: List[str] = []
 MAX_HISTORY_SIZE = 50

jarvis/jarvis_utils/methodology.py

@@ -10,7 +10,6 @@
 import json
 import os
 import tempfile
-from pathlib import Path
 from typing import Any, Dict, List, Optional
 
 from jarvis.jarvis_platform.base import BasePlatform
@@ -21,9 +20,8 @@ from jarvis.jarvis_utils.config import (
     get_central_methodology_repo,
     get_max_input_token_count,
 )
-from jarvis.jarvis_utils.globals import get_agent, current_agent_name
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
-from jarvis.jarvis_utils.utils import is_context_overflow, daily_check_git_updates
+from jarvis.jarvis_utils.utils import daily_check_git_updates
 from jarvis.jarvis_utils.embedding import get_context_token_count
 
 
@@ -234,7 +232,7 @@ def load_methodology(
         methodology_titles = list(methodologies.keys())
 
         # 步骤2：让大模型选择相关性高的方法论
-        selection_prompt = f"""以下是所有可用的方法论标题：
+        selection_prompt = """以下是所有可用的方法论标题：
 
 """
         for i, title in enumerate(methodology_titles, 1):
@@ -305,7 +303,7 @@ def load_methodology(
 
         # 步骤3：将选择出来的方法论内容提供给大模型生成步骤
         # 首先构建基础提示词部分
-        base_prompt = f"""以下是与用户需求相关的方法论内容：
+        base_prompt = """以下是与用户需求相关的方法论内容：
 
 """
         suffix_prompt = f"""以下是所有可用的工具内容：

jarvis/jarvis_utils/output.py

@@ -193,7 +193,7 @@ class ConsoleOutputSink(OutputSink):
             ),
         }
 
-        header = Text(
+        Text(
             PrettyOutput._format(event.output_type, event.timestamp),
             style=header_styles[event.output_type],
         )

jarvis/jarvis_utils/utils.py

@@ -2,7 +2,6 @@
 import hashlib
 import json
 import os
-import platform
 import signal
 import subprocess
 import sys
@@ -13,10 +12,7 @@ from datetime import datetime, date
 
 import yaml  # type: ignore
 from rich.align import Align
-from rich.console import Group, RenderableType
-from rich.panel import Panel
-from rich.table import Table
-from rich.text import Text
+from rich.console import RenderableType
 
 from jarvis import __version__
 from jarvis.jarvis_utils.config import (
@@ -121,7 +117,7 @@ def _check_pip_updates() -> bool:
     """检查pip安装的Jarvis是否有更新
 
     返回:
-        bool: 是否执行了更新（总是返回False，因为pip更新需要用户手动执行）
+        bool: 是否执行了更新（成功更新返回True以触发重启）
     """
     import urllib.request
     import urllib.error
@@ -146,7 +142,7 @@ def _check_pip_updates() -> bool:
             with urllib.request.urlopen(url, timeout=5) as response:
                 data = json.loads(response.read().decode())
                 latest_version = data["info"]["version"]
-        except (urllib.error.URLError, KeyError, json.JSONDecodeError) as e:
+        except (urllib.error.URLError, KeyError, json.JSONDecodeError):
             return False
 
         # 比较版本
@@ -166,6 +162,7 @@ def _check_pip_updates() -> bool:
 
             # 检测是否使用uv
             is_uv_env = False
+            uv_executable: Optional[str] = None
             if in_venv:
                 if sys.platform == "win32":
                     uv_path = Path(sys.prefix) / "Scripts" / "uv.exe"
@@ -173,6 +170,7 @@ def _check_pip_updates() -> bool:
                     uv_path = Path(sys.prefix) / "bin" / "uv"
                 if uv_path.exists():
                     is_uv_env = True
+                    uv_executable = str(uv_path)
 
             # 检测是否安装了 RAG 特性（更精确）
             from jarvis.jarvis_utils.utils import (
@@ -180,21 +178,60 @@ def _check_pip_updates() -> bool:
             )  # 延迟导入避免潜在循环依赖
             rag_installed = _is_rag_installed()
 
-            # 提供更新命令
+            # 更新命令
             package_spec = (
                 "jarvis-ai-assistant[rag]" if rag_installed else "jarvis-ai-assistant"
             )
-            if is_uv_env:
+            if is_uv_env and uv_executable:
+                cmd_list = [uv_executable, "pip", "install", "--upgrade", package_spec]
                 update_cmd = f"uv pip install --upgrade {package_spec}"
             else:
+                cmd_list = [
+                    sys.executable,
+                    "-m",
+                    "pip",
+                    "install",
+                    "--upgrade",
+                    package_spec,
+                ]
                 update_cmd = f"{sys.executable} -m pip install --upgrade {package_spec}"
 
-            PrettyOutput.print(f"请手动执行以下命令更新: {update_cmd}", OutputType.INFO)
+            # 自动尝试升级（失败时提供手动命令）
+            try:
+                PrettyOutput.print("正在自动更新 Jarvis，请稍候...", OutputType.INFO)
+                result = subprocess.run(
+                    cmd_list,
+                    capture_output=True,
+                    text=True,
+                    encoding="utf-8",
+                    errors="replace",
+                    timeout=600,
+                )
+                if result.returncode == 0:
+                    PrettyOutput.print("更新成功，正在重启以应用新版本...", OutputType.SUCCESS)
+                    # 更新检查日期，避免重复触发
+                    last_check_file.write_text(today_str)
+                    return True
+                else:
+                    err = (result.stderr or result.stdout or "").strip()
+                    if err:
+                        PrettyOutput.print(
+                            f"自动更新失败，错误信息（已截断）: {err[:500]}",
+                            OutputType.WARNING,
+                        )
+                    PrettyOutput.print(
+                        f"请手动执行以下命令更新: {update_cmd}", OutputType.INFO
+                    )
+            except Exception:
+                PrettyOutput.print("自动更新出现异常，已切换为手动更新方式。", OutputType.WARNING)
+                PrettyOutput.print(
+                    f"请手动执行以下命令更新: {update_cmd}", OutputType.INFO
+                )
 
         # 更新检查日期
         last_check_file.write_text(today_str)
 
-    except Exception as e:
+    except Exception:
         # 静默处理错误，不影响正常使用
         pass
 
@@ -225,7 +262,6 @@ def _show_usage_stats(welcome_str: str) -> None:
     from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
     try:
-        from datetime import datetime
 
         from rich.console import Console, Group
         from rich.panel import Panel
@@ -709,7 +745,7 @@ def _interactive_config_setup(config_file_path: Path):
             guide_lines = ["", "配置获取方法:"]
             for key in required_keys:
                 if key in config_guide and config_guide[key]:
-                    guide_lines.append(f"")
+                    guide_lines.append("")
                     guide_lines.append(f"{key} 获取方法:")
                     guide_lines.append(str(config_guide[key]))
             PrettyOutput.print("\n".join(guide_lines), OutputType.INFO)
@@ -833,8 +869,6 @@ def load_config():
 
 from typing import Tuple
 
-from typing import Tuple
-
 
 def _load_config_file(config_file: str) -> Tuple[str, dict]:
     """读取并解析YAML格式的配置文件
@@ -1207,7 +1241,10 @@ def _collect_optional_config_interactively(
             except ValueError:
                 default_index = 0  # 默认为第一个选项
 
-            new_mode = get_choice(tip, choices, default_index)
+            new_mode = get_choice(
+                tip,
+                choices,
+            )
 
             if new_mode == current_mode:
                 return False
@@ -1329,7 +1366,6 @@ def _load_and_process_config(jarvis_dir: str, config_file: str) -> None:
         config_file: 配置文件路径
     """
     from jarvis.jarvis_utils.input import user_confirm as get_yes_no
-    from jarvis.jarvis_utils.input import get_single_line_input
 
     try:
         content, config_data = _load_config_file(config_file)
@@ -1533,7 +1569,7 @@ def _read_old_config_file(config_file):
         )
         set_global_env_data(config_data)
     PrettyOutput.print(
-        f"检测到旧格式配置文件，旧格式以后将不再支持，请尽快迁移到新格式",
+        "检测到旧格式配置文件，旧格式以后将不再支持，请尽快迁移到新格式",
         OutputType.WARNING,
     )
 
@@ -1558,7 +1594,10 @@ def while_success(func: Callable[[], Any], sleep_time: float = 0.1, max_retries:
         except Exception:
             retry_count += 1
             if retry_count < max_retries:
-                PrettyOutput.print(f"重试中 ({retry_count}/{max_retries})，等待 {sleep_time}s...", OutputType.WARNING)
+                PrettyOutput.print(
+                    f"发生异常，重试中 ({retry_count}/{max_retries})，等待 {sleep_time}s...",
+                    OutputType.WARNING,
+                )
                 time.sleep(sleep_time)
             continue
     return result
@@ -1587,7 +1626,10 @@ def while_true(func: Callable[[], bool], sleep_time: float = 0.1, max_retries: i
             break
         retry_count += 1
         if retry_count < max_retries:
-            PrettyOutput.print(f"重试中 ({retry_count}/{max_retries})，等待 {sleep_time}s...", OutputType.WARNING)
+            PrettyOutput.print(
+                f"返回空值，重试中 ({retry_count}/{max_retries})，等待 {sleep_time}s...",
+                OutputType.WARNING,
+            )
             time.sleep(sleep_time)
     return ret
 
@@ -1615,7 +1657,7 @@ def get_file_line_count(filename: str) -> int:
     """
     try:
         return len(open(filename, "r", encoding="utf-8", errors="ignore").readlines())
-    except Exception as e:
+    except Exception:
         return 0
 
 
@@ -1756,7 +1798,7 @@ def _pull_git_repo(repo_path: Path, repo_type: str):
             return
 
         # 执行 git pull
-        pull_result = subprocess.run(
+        subprocess.run(
             ["git", "pull"],
             cwd=repo_path,
             capture_output=True,

{jarvis_ai_assistant-0.3.27.dist-info → jarvis_ai_assistant-0.3.28.dist-info}/METADATA

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