jarvis-ai-assistant 0.3.30__py3-none-any.whl → 0.7.6__py3-none-any.whl

jarvis/jarvis_utils/config.py

@@ -1,9 +1,9 @@
 # -*- coding: utf-8 -*-
 import os
 from functools import lru_cache
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional, cast
 
-import yaml  # type: ignore
+import yaml
 
 from jarvis.jarvis_utils.builtin_replace_map import BUILTIN_REPLACE_MAP
 
@@ -37,7 +37,7 @@ def get_git_commit_prompt() -> str:
     返回:
         str: Git提交信息生成提示模板，如果未配置则返回空字符串
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_GIT_COMMIT_PROMPT", "")
+    return cast(str, GLOBAL_CONFIG_DATA.get("JARVIS_GIT_COMMIT_PROMPT", ""))
 
 
 # 输出窗口预留大小
@@ -63,30 +63,13 @@ def get_replace_map() -> dict:
     if not os.path.exists(replace_map_path):
         return BUILTIN_REPLACE_MAP.copy()
 
-    from jarvis.jarvis_utils.output import OutputType, PrettyOutput
-
-    PrettyOutput.print(
-        "警告：使用replace_map.yaml进行配置的方式已被弃用，将在未来版本中移除。"
-        "请迁移到使用GLOBAL_CONFIG_DATA中的JARVIS_REPLACE_MAP配置。",
-        output_type=OutputType.WARNING,
-    )
+    print("⚠️ 警告：使用replace_map.yaml进行配置的方式已被弃用，将在未来版本中移除。请迁移到使用GLOBAL_CONFIG_DATA中的JARVIS_REPLACE_MAP配置。")
 
     with open(replace_map_path, "r", encoding="utf-8", errors="ignore") as file:
         file_map = yaml.safe_load(file) or {}
         return {**BUILTIN_REPLACE_MAP, **file_map}
 
 
-def get_max_token_count(model_group_override: Optional[str] = None) -> int:
-    """
-    获取模型允许的最大token数量。
-
-    返回:
-        int: 模型能处理的最大token数量，为最大输入token数量的100倍。
-    """
-    max_input_tokens = get_max_input_token_count(model_group_override)
-    return max_input_tokens * 100
-
-
 def get_max_input_token_count(model_group_override: Optional[str] = None) -> int:
     """
     获取模型允许的最大输入token数量。
@@ -112,7 +95,7 @@ def get_shell_name() -> str:
     4. 如果都未配置，则默认返回bash
     """
     shell_path = GLOBAL_CONFIG_DATA.get("SHELL", os.getenv("SHELL", "/bin/bash"))
-    return os.path.basename(shell_path).lower()
+    return cast(str, os.path.basename(shell_path).lower())
 
 
 def _apply_llm_group_env_override(group_config: Dict[str, Any]) -> None:
@@ -150,10 +133,24 @@ def _get_resolved_model_config(
     model_groups = GLOBAL_CONFIG_DATA.get("JARVIS_LLM_GROUPS", [])
 
     if model_group_name and isinstance(model_groups, list):
+        found = False
         for group_item in model_groups:
             if isinstance(group_item, dict) and model_group_name in group_item:
                 group_config = group_item[model_group_name]
+                found = True
                 break
+        
+        # 当显式指定了模型组但未找到时，报错并退出
+        if model_group_override and not found:
+            print(f"❌ 错误：指定的模型组 '{model_group_name}' 不存在于配置中。")
+            print("ℹ️ 可用的模型组: " + 
+                ", ".join(
+                    list(group.keys())[0] 
+                    for group in model_groups 
+                    if isinstance(group, dict)
+                ) if model_groups else "无可用模型组")
+            import sys
+            sys.exit(1)
     
     _apply_llm_group_env_override(group_config)
 
@@ -168,6 +165,10 @@ def _get_resolved_model_config(
         "JARVIS_PLATFORM",
         "JARVIS_MODEL",
         "JARVIS_MAX_INPUT_TOKEN_COUNT",
+        "JARVIS_CHEAP_PLATFORM",
+        "JARVIS_CHEAP_MODEL",
+        "JARVIS_SMART_PLATFORM",
+        "JARVIS_SMART_MODEL",
     ]
     for key in override_keys:
         if key in GLOBAL_CONFIG_DATA:
@@ -187,10 +188,10 @@ def get_normal_platform_name(model_group_override: Optional[str] = None) -> str:
     获取正常操作的平台名称。
 
     返回：
-        str: 平台名称，默认为'yuanbao'
+        str: 平台名称，默认为'openai'
     """
     config = _get_resolved_model_config(model_group_override)
-    return config.get("JARVIS_PLATFORM", "yuanbao")
+    return cast(str, config.get("JARVIS_PLATFORM", "openai"))
 
 
 def get_normal_model_name(model_group_override: Optional[str] = None) -> str:
@@ -198,10 +199,10 @@ def get_normal_model_name(model_group_override: Optional[str] = None) -> str:
     获取正常操作的模型名称。
 
     返回：
-        str: 模型名称，默认为'deep_seek_v3'
+        str: 模型名称，默认为'gpt-5'
     """
     config = _get_resolved_model_config(model_group_override)
-    return config.get("JARVIS_MODEL", "deep_seek_v3")
+    return cast(str, config.get("JARVIS_MODEL", "gpt-5"))
 
 
 def _deprecated_platform_name_v1(model_group_override: Optional[str] = None) -> str:
@@ -228,6 +229,62 @@ def _deprecated_model_name_v1(model_group_override: Optional[str] = None) -> str
     return get_normal_model_name(model_group_override)
 
 
+def get_cheap_platform_name(model_group_override: Optional[str] = None) -> str:
+    """
+    获取廉价操作的平台名称。
+
+    返回：
+        str: 平台名称，如果未配置则回退到正常操作平台
+    """
+    config = _get_resolved_model_config(model_group_override)
+    cheap_platform = config.get("JARVIS_CHEAP_PLATFORM")
+    if cheap_platform:
+        return cast(str, cheap_platform)
+    return get_normal_platform_name(model_group_override)
+
+
+def get_cheap_model_name(model_group_override: Optional[str] = None) -> str:
+    """
+    获取廉价操作的模型名称。
+
+    返回：
+        str: 模型名称，如果未配置则回退到正常操作模型
+    """
+    config = _get_resolved_model_config(model_group_override)
+    cheap_model = config.get("JARVIS_CHEAP_MODEL")
+    if cheap_model:
+        return cast(str, cheap_model)
+    return get_normal_model_name(model_group_override)
+
+
+def get_smart_platform_name(model_group_override: Optional[str] = None) -> str:
+    """
+    获取智能操作的平台名称。
+
+    返回：
+        str: 平台名称，如果未配置则回退到正常操作平台
+    """
+    config = _get_resolved_model_config(model_group_override)
+    smart_platform = config.get("JARVIS_SMART_PLATFORM")
+    if smart_platform:
+        return cast(str, smart_platform)
+    return get_normal_platform_name(model_group_override)
+
+
+def get_smart_model_name(model_group_override: Optional[str] = None) -> str:
+    """
+    获取智能操作的模型名称。
+
+    返回：
+        str: 模型名称，如果未配置则回退到正常操作模型
+    """
+    config = _get_resolved_model_config(model_group_override)
+    smart_model = config.get("JARVIS_SMART_MODEL")
+    if smart_model:
+        return cast(str, smart_model)
+    return get_normal_model_name(model_group_override)
+
+
 def is_execute_tool_confirm() -> bool:
     """
     检查工具执行是否需要确认。
@@ -235,7 +292,7 @@ def is_execute_tool_confirm() -> bool:
     返回：
         bool: 如果需要确认则返回True，默认为False
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_EXECUTE_TOOL_CONFIRM", False)
+    return cast(bool, GLOBAL_CONFIG_DATA.get("JARVIS_EXECUTE_TOOL_CONFIRM", False))
 
 
 def is_confirm_before_apply_patch() -> bool:
@@ -245,19 +302,19 @@ def is_confirm_before_apply_patch() -> bool:
     返回：
         bool: 如果需要确认则返回True，默认为False
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_CONFIRM_BEFORE_APPLY_PATCH", False)
+    return cast(bool, GLOBAL_CONFIG_DATA.get("JARVIS_CONFIRM_BEFORE_APPLY_PATCH", False))
 
 
 def get_data_dir() -> str:
     """
     获取Jarvis数据存储目录路径。
 
-    返回:
+    返回: 
         str: 数据目录路径，优先从JARVIS_DATA_PATH环境变量获取，
              如果未设置或为空，则使用~/.jarvis作为默认值
     """
     return os.path.expanduser(
-        GLOBAL_CONFIG_DATA.get("JARVIS_DATA_PATH", "~/.jarvis").strip()
+        cast(str, GLOBAL_CONFIG_DATA.get("JARVIS_DATA_PATH", "~/.jarvis")).strip()
     )
 
 
@@ -285,7 +342,7 @@ def get_pretty_output() -> bool:
     if platform.system() == "Windows":
         return False
 
-    return GLOBAL_CONFIG_DATA.get("JARVIS_PRETTY_OUTPUT", True)
+    return cast(bool, GLOBAL_CONFIG_DATA.get("JARVIS_PRETTY_OUTPUT", True))
 
 
 def is_use_methodology() -> bool:
@@ -295,7 +352,7 @@ def is_use_methodology() -> bool:
     返回：
         bool: 如果启用方法论则返回True，默认为True
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_USE_METHODOLOGY", True)
+    return cast(bool, GLOBAL_CONFIG_DATA.get("JARVIS_USE_METHODOLOGY", True))
 
 
 def is_use_analysis() -> bool:
@@ -305,7 +362,7 @@ def is_use_analysis() -> bool:
     返回：
         bool: 如果启用任务分析则返回True，默认为True
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_USE_ANALYSIS", True)
+    return cast(bool, GLOBAL_CONFIG_DATA.get("JARVIS_USE_ANALYSIS", True))
 
 
 def get_tool_load_dirs() -> List[str]:
@@ -399,7 +456,7 @@ def get_central_methodology_repo() -> str:
     返回:
         str: 中心方法论Git仓库地址，如果未配置则返回空字符串
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_CENTRAL_METHODOLOGY_REPO", "")
+    return cast(str, GLOBAL_CONFIG_DATA.get("JARVIS_CENTRAL_METHODOLOGY_REPO", ""))
 
 
 def get_central_tool_repo() -> str:
@@ -409,7 +466,7 @@ def get_central_tool_repo() -> str:
     返回:
         str: 中心工具Git仓库地址，如果未配置则返回空字符串
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_CENTRAL_TOOL_REPO", "")
+    return cast(str, GLOBAL_CONFIG_DATA.get("JARVIS_CENTRAL_TOOL_REPO", ""))
 
 
 def is_print_prompt() -> bool:
@@ -419,7 +476,7 @@ def is_print_prompt() -> bool:
     返回：
         bool: 如果打印提示则返回True，默认为True
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_PRINT_PROMPT", False)
+    return cast(bool, GLOBAL_CONFIG_DATA.get("JARVIS_PRINT_PROMPT", False))
 
 
 def is_print_error_traceback() -> bool:
@@ -452,6 +509,36 @@ def is_enable_static_analysis() -> bool:
     return GLOBAL_CONFIG_DATA.get("JARVIS_ENABLE_STATIC_ANALYSIS", True) is True
 
 
+def is_enable_build_validation() -> bool:
+    """
+    获取是否启用构建验证。
+
+    返回：
+        bool: 如果启用构建验证则返回True，默认为True
+    """
+    return GLOBAL_CONFIG_DATA.get("JARVIS_ENABLE_BUILD_VALIDATION", True) is True
+
+
+def is_enable_impact_analysis() -> bool:
+    """
+    获取是否启用编辑影响范围分析。
+
+    返回：
+        bool: 如果启用影响范围分析则返回True，默认为True
+    """
+    return GLOBAL_CONFIG_DATA.get("JARVIS_ENABLE_IMPACT_ANALYSIS", True) is True
+
+
+def get_build_validation_timeout() -> int:
+    """
+    获取构建验证的超时时间（秒）。
+
+    返回：
+        int: 超时时间，默认为30秒
+    """
+    return int(GLOBAL_CONFIG_DATA.get("JARVIS_BUILD_VALIDATION_TIMEOUT", 30))
+
+
 def get_git_check_mode() -> str:
     """
     获取Git校验模式。
@@ -473,7 +560,7 @@ def get_mcp_config() -> List[Dict[str, Any]]:
     返回:
         List[Dict[str, Any]]: MCP配置项列表，如果未配置则返回空列表
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_MCP", [])
+    return cast(List[Dict[str, Any]], GLOBAL_CONFIG_DATA.get("JARVIS_MCP", []))
 
 
 # ==============================================================================
@@ -550,7 +637,7 @@ def get_rag_embedding_model() -> str:
         str: 嵌入模型的名称
     """
     config = _get_resolved_rag_config()
-    return config.get("embedding_model", "BAAI/bge-m3")
+    return cast(str, config.get("embedding_model", "BAAI/bge-m3"))
 
 
 def get_rag_rerank_model() -> str:
@@ -561,7 +648,7 @@ def get_rag_rerank_model() -> str:
         str: rerank模型的名称
     """
     config = _get_resolved_rag_config()
-    return config.get("rerank_model", "BAAI/bge-reranker-v2-m3")
+    return cast(str, config.get("rerank_model", "BAAI/bge-reranker-v2-m3"))
 
 
 def get_rag_embedding_cache_path() -> str:
@@ -671,7 +758,7 @@ def get_tool_use_list() -> List[str]:
         List[str]: 要使用的工具名称列表，空列表表示使用所有工具
     """
     config = _get_resolved_tool_config()
-    return config.get("use", [])
+    return cast(List[str], config.get("use", []))
 
 
 def get_tool_dont_use_list() -> List[str]:
@@ -682,7 +769,7 @@ def get_tool_dont_use_list() -> List[str]:
         List[str]: 不使用的工具名称列表
     """
     config = _get_resolved_tool_config()
-    return config.get("dont_use", [])
+    return cast(List[str], config.get("dont_use", []))
 
 
 def get_tool_filter_threshold() -> int:
@@ -695,27 +782,127 @@ def get_tool_filter_threshold() -> int:
     return int(GLOBAL_CONFIG_DATA.get("JARVIS_TOOL_FILTER_THRESHOLD", 30))
 
 
+
+
+
+
+def get_script_execution_timeout() -> int:
+    """
+    获取脚本执行的超时时间（秒）。
+
+    返回:
+        int: 超时时间，默认为300秒（5分钟）
+    """
+    return int(GLOBAL_CONFIG_DATA.get("JARVIS_SCRIPT_EXECUTION_TIMEOUT", 300))
+
+
 def is_enable_git_repo_jca_switch() -> bool:
     """
     是否启用：在初始化环境前检测Git仓库并提示可切换到代码开发模式（jca）
-    默认关闭
+    默认开启
     """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_ENABLE_GIT_JCA_SWITCH", False) is True
+    return GLOBAL_CONFIG_DATA.get("JARVIS_ENABLE_GIT_JCA_SWITCH", True) is True
 
 
 def is_enable_builtin_config_selector() -> bool:
     """
     是否启用：在进入默认通用代理前，列出可用配置（agent/multi_agent/roles）供选择
-    默认关闭
+    默认开启
     """
     return (
-        GLOBAL_CONFIG_DATA.get("JARVIS_ENABLE_STARTUP_CONFIG_SELECTOR", False) is True
+        GLOBAL_CONFIG_DATA.get("JARVIS_ENABLE_STARTUP_CONFIG_SELECTOR", True) is True
     )
 
 
+def is_save_session_history() -> bool:
+    """
+    是否保存会话记录。
+
+    返回:
+        bool: 如果要保存会话记录则返回True, 默认为False
+    """
+    return GLOBAL_CONFIG_DATA.get("JARVIS_SAVE_SESSION_HISTORY", False) is True
+
+
 def is_immediate_abort() -> bool:
     """
     是否启用立即中断：当在对话过程中检测到用户中断信号时，立即停止输出并返回。
     默认关闭
     """
     return GLOBAL_CONFIG_DATA.get("JARVIS_IMMEDIATE_ABORT", False) is True
+
+
+def is_non_interactive() -> bool:
+    """
+    获取是否启用非交互模式。
+    
+    返回：
+        bool: 如果启用非交互模式则返回True，默认为False
+    """
+    # 优先读取环境变量，确保 CLI 标志生效且不被配置覆盖
+    try:
+        import os
+        v = os.getenv("JARVIS_NON_INTERACTIVE")
+        if v is not None:
+            val = str(v).strip().lower()
+            if val in ("1", "true", "yes", "on"):
+                return True
+            if val in ("0", "false", "no", "off"):
+                return False
+    except Exception:
+        # 忽略环境变量解析异常，回退到配置
+        pass
+    return GLOBAL_CONFIG_DATA.get("JARVIS_NON_INTERACTIVE", False) is True
+
+
+def is_skip_predefined_tasks() -> bool:
+    """
+    是否跳过预定义任务加载。
+    
+    返回：
+        bool: 如果跳过预定义任务加载则返回True，默认为False
+    """
+    return GLOBAL_CONFIG_DATA.get("JARVIS_SKIP_PREDEFINED_TASKS", False) is True
+
+
+def get_addon_prompt_threshold() -> int:
+    """
+    获取附加提示的触发阈值（字符数）。
+    
+    当消息长度超过此阈值时，会自动添加默认的附加提示。
+    
+    返回:
+        int: 触发阈值，默认为1024
+    """
+    try:
+        return int(GLOBAL_CONFIG_DATA.get("JARVIS_ADDON_PROMPT_THRESHOLD", 1024))
+    except Exception:
+        return 1024
+
+
+def is_enable_intent_recognition() -> bool:
+    """
+    获取是否启用意图识别功能。
+    
+    返回:
+        bool: 是否启用意图识别，默认为True（可通过 GLOBAL_CONFIG_DATA['JARVIS_ENABLE_INTENT_RECOGNITION'] 配置）
+    """
+    return GLOBAL_CONFIG_DATA.get("JARVIS_ENABLE_INTENT_RECOGNITION", True) is True
+
+
+def is_enable_memory_organizer() -> bool:
+    """
+    获取是否启用自动记忆整理功能。
+    
+    返回:
+        bool: 是否启用自动记忆整理，默认为False（可通过 GLOBAL_CONFIG_DATA['JARVIS_ENABLE_MEMORY_ORGANIZER'] 配置）
+    """
+    return GLOBAL_CONFIG_DATA.get("JARVIS_ENABLE_MEMORY_ORGANIZER", False) is True
+def get_conversation_turn_threshold() -> int:
+    """
+    获取对话轮次阈值，用于触发总结。
+    
+    返回:
+        int: 对话轮次阈值，默认为50
+    """
+    return int(GLOBAL_CONFIG_DATA.get("JARVIS_CONVERSATION_TURN_THRESHOLD", 50))

jarvis/jarvis_utils/embedding.py

@@ -3,7 +3,6 @@ import os
 from pathlib import Path
 from typing import List
 
-from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
 # 设置tiktoken缓存目录
 script_dir = Path(__file__).parent
@@ -21,14 +20,18 @@ def get_context_token_count(text: str) -> int:
     返回：
         int: 文本中的token数量
     """
+    # 防御性检查：入参为 None 或空字符串时直接返回 0
+    if text is None or text == "":
+        return 0
     try:
         import tiktoken
 
         encoding = tiktoken.get_encoding("cl100k_base")
-        return len(encoding.encode(text))
+        # 调整token计算为原来的10/7倍
+        return int(len(encoding.encode(text)) * 10 / 7)
     except Exception as e:
-        PrettyOutput.print(f"计算token失败: {str(e)}", OutputType.WARNING)
-        return len(text) // 4  # 每个token大约4个字符的粗略估计
+        print(f"⚠️ 计算token失败: {str(e)}")
+        return int(len(text) // 4 * 10 / 7)  # 每个token大约4个字符的粗略估计，调整为10/7倍
 
 
 def split_text_into_chunks(
@@ -77,6 +80,6 @@ def split_text_into_chunks(
         return chunks
 
     except Exception as e:
-        PrettyOutput.print(f"文本分割失败: {str(e)}", OutputType.WARNING)
+        print(f"⚠️ 文本分割失败: {str(e)}")
         # 发生错误时回退到简单的字符分割
         return [text[i : i + max_length] for i in range(0, len(text), max_length)]

jarvis/jarvis_utils/fzf.py

@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-"""FZF selector utility."""
+"""FZF选择器工具。"""
 import shutil
 import subprocess
 from typing import List, Optional, Union, Dict, Any, cast
@@ -10,15 +10,15 @@ def fzf_select(
     key: Optional[str] = None,
 ) -> Optional[str]:
     """
-    Uses fzf to select an item from a list.
+    使用fzf从列表中选择一个项目。
 
-    Args:
-        options: A list of strings or dicts to choose from.
-        prompt: The prompt to display in fzf.
-        key: If options is a list of dicts, this is the key to display.
+    参数:
+        options: 可供选择的字符串或字典列表。
+        prompt: 在fzf中显示的提示信息。
+        key: 如果options是字典列表，则此参数指定要显示的键名。
 
-    Returns:
-        The selected item, or None if fzf is not available or the selection is cancelled.
+    返回:
+        选中的项目，如果fzf不可用或选择被取消则返回None。
     """
     if shutil.which("fzf") is None:
         return None