jarvis-ai-assistant 0.2.3__py3-none-any.whl → 0.2.5__py3-none-any.whl

jarvis/jarvis_utils/config.py

@@ -81,10 +81,10 @@ def get_max_token_count(model_group_override: Optional[str] = None) -> int:
     获取模型允许的最大token数量。
 
     返回:
-        int: 模型能处理的最大token数量。
+        int: 模型能处理的最大token数量，为最大输入token数量的100倍。
     """
-    config = _get_resolved_model_config(model_group_override)
-    return int(config.get("JARVIS_MAX_TOKEN_COUNT", "960000"))
+    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:
@@ -151,9 +151,7 @@ def _get_resolved_model_config(
         "JARVIS_MODEL",
         "JARVIS_THINKING_PLATFORM",
         "JARVIS_THINKING_MODEL",
-        "JARVIS_MAX_TOKEN_COUNT",
         "JARVIS_MAX_INPUT_TOKEN_COUNT",
-        "JARVIS_MAX_BIG_CONTENT_SIZE",
     ]:
         if key in GLOBAL_CONFIG_DATA:
             resolved_config[key] = GLOBAL_CONFIG_DATA[key]
@@ -249,10 +247,10 @@ def get_max_big_content_size(model_group_override: Optional[str] = None) -> int:
     获取最大大内容大小。
 
     返回：
-        int: 最大大内容大小
+        int: 最大大内容大小，为最大输入token数量的5倍
     """
-    config = _get_resolved_model_config(model_group_override)
-    return int(config.get("JARVIS_MAX_BIG_CONTENT_SIZE", "160000"))
+    max_input_tokens = get_max_input_token_count(model_group_override)
+    return max_input_tokens * 5
 
 
 def get_pretty_output() -> bool:
@@ -262,6 +260,12 @@ def get_pretty_output() -> bool:
     返回：
         bool: 如果启用PrettyOutput则返回True，默认为True
     """
+    import platform
+
+    # Windows系统强制设置为False
+    if platform.system() == "Windows":
+        return False
+
     return GLOBAL_CONFIG_DATA.get("JARVIS_PRETTY_OUTPUT", False) == True
 
 

jarvis/jarvis_utils/git_utils.py

@@ -214,9 +214,7 @@ def handle_commit_workflow() -> bool:
     Returns:
         bool: 提交是否成功
     """
-    if is_confirm_before_apply_patch() and not user_confirm(
-        "是否要提交代码？", default=True
-    ):
+    if is_confirm_before_apply_patch() and not user_confirm("是否要提交代码？", default=True):
         revert_change()
         return False
 
@@ -429,9 +427,7 @@ def check_and_update_git_repo(repo_path: str) -> bool:
                 if not in_venv and (
                     "Permission denied" in error_msg or "not writeable" in error_msg
                 ):
-                    if user_confirm(
-                        "检测到权限问题，是否尝试用户级安装(--user)？", True
-                    ):
+                    if user_confirm("检测到权限问题，是否尝试用户级安装(--user)？", True):
                         user_result = subprocess.run(
                             install_cmd + ["--user"],
                             cwd=git_root,
@@ -446,9 +442,7 @@ def check_and_update_git_repo(repo_path: str) -> bool:
                 PrettyOutput.print(f"代码安装失败: {error_msg}", OutputType.ERROR)
                 return False
             except Exception as e:
-                PrettyOutput.print(
-                    f"安装过程中发生意外错误: {str(e)}", OutputType.ERROR
-                )
+                PrettyOutput.print(f"安装过程中发生意外错误: {str(e)}", OutputType.ERROR)
                 return False
         # 更新检查日期文件
         with open(last_check_file, "w") as f:
@@ -482,9 +476,7 @@ def get_diff_file_list() -> List[str]:
         subprocess.run(["git", "reset"], check=True)
 
         if result.returncode != 0:
-            PrettyOutput.print(
-                f"获取差异文件列表失败: {result.stderr}", OutputType.ERROR
-            )
+            PrettyOutput.print(f"获取差异文件列表失败: {result.stderr}", OutputType.ERROR)
             return []
 
         return [f for f in result.stdout.splitlines() if f]
@@ -533,8 +525,10 @@ def get_recent_commits_with_files() -> List[Dict[str, Any]]:
             ],
             capture_output=True,
             text=True,
+            encoding="utf-8",
+            errors="replace",
         )
-        if result.returncode != 0:
+        if result.returncode != 0 or result.stdout is None:
             return []
 
         # 解析提交信息
@@ -632,9 +626,7 @@ def confirm_add_new_files() -> None:
             need_confirm = True
 
         if binary_files:
-            output_lines.append(
-                f"检测到{len(binary_files)}个二进制文件(选择N将重新检测)"
-            )
+            output_lines.append(f"检测到{len(binary_files)}个二进制文件(选择N将重新检测)")
             output_lines.append("二进制文件列表:")
             output_lines.extend(f"  - {file}" for file in binary_files)
             need_confirm = True

jarvis/jarvis_utils/methodology.py

@@ -149,6 +149,7 @@ def load_methodology(user_input: str, tool_registery: Optional[Any] = None) -> s
 
     参数：
         user_input: 用户输入文本，用于提示大模型
+        tool_registery: 工具注册表，用于获取工具列表
 
     返回：
         str: 相关的方法论提示，如果未找到方法论则返回空字符串
@@ -170,70 +171,87 @@ def load_methodology(user_input: str, tool_registery: Optional[Any] = None) -> s
             return ""
         print(f"✅ 加载方法论文件完成 (共 {len(methodologies)} 个)")
 
-        # 获取当前平台
-        agent = get_agent(current_agent_name)
-        if agent:
-            platform = agent.model
-            model_group = agent.model.model_group
-        else:
-            platform = PlatformRegistry().get_normal_platform()
-            model_group = None
-        platform.set_suppress_output(False)
-        if not platform:
-            return ""
+        platform = PlatformRegistry().get_normal_platform()
+        platform.set_suppress_output(True)
+
+        # 步骤1：获取所有方法论的标题
+        methodology_titles = list(methodologies.keys())
 
-        # 构建基础提示信息
-        base_prompt = f"""以下是所有可用的方法论内容：
+        # 步骤2：让大模型选择相关性高的方法论
+        selection_prompt = f"""以下是所有可用的方法论标题：
 
 """
-        # 构建完整内容
-        full_content = base_prompt
-        for problem_type, content in methodologies.items():
-            full_content += f"## {problem_type}\n\n{content}\n\n---\n\n"
+        for i, title in enumerate(methodology_titles, 1):
+            selection_prompt += f"{i}. {title}\n"
 
-        full_content += f"以下是所有可用的工具内容：\n\n"
-        full_content += prompt
+        selection_prompt += f"""
+以下是可用的工具列表：
+{prompt}
 
-        # 添加用户输入和输出要求
-        full_content += f"""
-请根据以上方法论和可调用的工具内容，规划/总结出以下用户需求的执行步骤: {user_input}
+用户需求：{user_input}
 
-请按以下格式回复：
-### 与该任务/需求相关的方法论
-1. [方法论名字]
-2. [方法论名字]
-### 根据以上方法论，规划/总结出执行步骤
-1. [步骤1]
-2. [步骤2]
-3. [步骤3]
+请分析用户需求，从上述方法论中选择出与需求相关性较高的方法论（可以选择多个）。
 
-如果没有匹配的方法论，请输出：没有历史方法论可参考
-除以上要求外，不要输出任何内容
+请严格按照以下格式返回序号：
+<NUM>序号1,序号2,序号3</NUM>
+
+例如：<NUM>1,3,5</NUM>
+
+如果没有相关的方法论，请返回：<NUM>none</NUM>
+
+注意：只返回<NUM>标签内的内容，不要有其他任何输出。
 """
 
-        # 检查内容是否过大
-        is_large_content = is_context_overflow(full_content, model_group)
-        temp_file_path = None
+        # 获取大模型选择的方法论序号
+        response = platform.chat_until_success(selection_prompt).strip()
+
+        # 重置平台，恢复输出
+        platform.reset()
+        platform.set_suppress_output(False)
+
+        # 从响应中提取<NUM>标签内的内容
+        import re
+        num_match = re.search(r'<NUM>(.*?)</NUM>', response, re.DOTALL)
+        
+        if not num_match:
+            # 如果没有找到<NUM>标签，尝试直接解析响应
+            selected_indices_str = response
+        else:
+            selected_indices_str = num_match.group(1).strip()
+
+        if selected_indices_str.lower() == "none":
+            return "没有历史方法论可参考"
 
+        # 解析选择的序号
+        selected_methodologies = {}
         try:
-            if is_large_content:
-                # 创建临时文件
-                print(f"📝 创建方法论临时文件...")
-                temp_file_path = _create_methodology_temp_file(methodologies)
-                if not temp_file_path:
-                    print(f"❌ 创建方法论临时文件失败")
-                    return ""
-                print(f"✅ 创建方法论临时文件完成")
-
-                # 尝试上传文件
-                upload_success = platform.upload_files([temp_file_path])
-
-                if upload_success:
-                    # 使用上传的文件生成摘要
-                    return platform.chat_until_success(
-                        base_prompt
-                        + f"""
-请根据已上传的方法论和可调用的工具文件内容，规划/总结出以下用户需求的执行步骤: {user_input}
+            if selected_indices_str:
+                indices = [int(idx.strip()) for idx in selected_indices_str.split(",") if idx.strip().isdigit()]
+                for idx in indices:
+                    if 1 <= idx <= len(methodology_titles):
+                        title = methodology_titles[idx - 1]
+                        selected_methodologies[title] = methodologies[title]
+        except Exception:
+            # 如果解析失败，返回空结果
+            return "没有历史方法论可参考"
+
+        if not selected_methodologies:
+            return "没有历史方法论可参考"
+
+        # 步骤3：将选择出来的方法论内容提供给大模型生成步骤
+        final_prompt = f"""以下是与用户需求相关的方法论内容：
+
+"""
+        for problem_type, content in selected_methodologies.items():
+            final_prompt += f"## {problem_type}\n\n{content}\n\n---\n\n"
+
+        final_prompt += f"""以下是所有可用的工具内容：
+
+{prompt}
+
+用户需求：{user_input}
+
+请根据以上方法论和可调用的工具内容，规划/总结出执行步骤。
 
 请按以下格式回复：
 ### 与该任务/需求相关的方法论
@@ -244,22 +262,11 @@ def load_methodology(user_input: str, tool_registery: Optional[Any] = None) -> s
 2. [步骤2]
 3. [步骤3]
 
-如果没有匹配的方法论，请输出：没有历史方法论可参考
 除以上要求外，不要输出任何内容
 """
-                    )
-                else:
-                    return "没有历史方法论可参考"
-            # 如果内容不大或上传失败，直接使用chat_until_success
-            return platform.chat_until_success(full_content)
-
-        finally:
-            # 清理临时文件
-            if temp_file_path and os.path.exists(temp_file_path):
-                try:
-                    os.remove(temp_file_path)
-                except Exception:
-                    pass
+
+        # 如果内容不大，直接使用chat_until_success
+        return platform.chat_until_success(final_prompt)
 
     except Exception as e:
         PrettyOutput.print(f"加载方法论失败: {str(e)}", OutputType.ERROR)