jarvis-ai-assistant 0.1.188__py3-none-any.whl → 0.1.190__py3-none-any.whl

jarvis/__init__.py

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

jarvis/jarvis_agent/__init__.py

@@ -4,8 +4,6 @@ import datetime
 import platform
 from typing import Any, Callable, Dict, List, Optional, Protocol, Tuple, Union
 
-from jarvis.jarvis_platform.base import BasePlatform
-
 # 第三方库导入
 from yaspin import yaspin  # type: ignore
 
@@ -224,7 +222,11 @@ class Agent:
 
         # 如果有上传文件，自动禁用方法论
         self.use_methodology = (
-            False if files else (use_methodology if use_methodology is not None else is_use_methodology())
+            False
+            if files
+            else (
+                use_methodology if use_methodology is not None else is_use_methodology()
+            )
         )
         self.use_analysis = (
             use_analysis if use_analysis is not None else is_use_analysis()
@@ -440,24 +442,15 @@ class Agent:
         self.conversation_length += get_context_token_count(response)
         return response
 
-    def _summarize_and_clear_history(self) -> str:
-        """总结当前对话并清理历史记录
-
-        该方法将:
-        1. 生成关键信息摘要
-        2. 清除对话历史
-        3. 保留系统消息
-        4. 添加摘要作为新上下文
-        5. 重置对话长度计数器
+    def generate_summary(self) -> str:
+        """生成对话历史摘要
 
         返回:
             str: 包含对话摘要的字符串
 
         注意:
-            当上下文长度超过最大值时使用
+            仅生成摘要，不修改对话状态
         """
-        # Create a new model instance to summarize, avoid affecting the main conversation
-
         with yaspin(text="正在总结对话历史...", color="cyan") as spinner:
             summary_prompt = """
 <summary_request>
@@ -482,36 +475,48 @@ class Agent:
 </summary_request>
 """
 
-            try:
-                with spinner.hidden():
-                    summary = self.model.chat_until_success(self.prompt + "\n" + summary_prompt)  # type: ignore
+        try:
+            with spinner.hidden():
+                summary = self.model.chat_until_success(self.prompt + "\n" + summary_prompt)  # type: ignore
+            spinner.text = "总结对话历史完成"
+            spinner.ok("✅")
+            return summary
+        except Exception as e:
+            spinner.text = "总结对话历史失败"
+            spinner.fail("❌")
+            return ""
 
-                self.model.reset()  # type: ignore
+    def _summarize_and_clear_history(self) -> str:
+        """总结当前对话并清理历史记录
 
-                # 清空当前对话历史，但保留系统消息
-                self.conversation_length = 0  # Reset conversation length
+        该方法将:
+        1. 调用_generate_summary生成摘要
+        2. 清除对话历史
+        3. 保留系统消息
+        4. 添加摘要作为新上下文
+        5. 重置对话长度计数器
 
-                # 添加总结作为新的上下文
-                spinner.text = "总结对话历史完成"
-                spinner.ok("✅")
-                return f"""<summary>
-<header>
+        返回:
+            str: 包含对话摘要的字符串
+
+        注意:
+            当上下文长度超过最大值时使用
+        """
+        summary = self.generate_summary()
+        self.clear_history()  # type: ignore
+
+        if not summary:
+            return ""
+
+        return f"""
 以下是之前对话的关键信息总结：
-</header>
 
 <content>
 {summary}
 </content>
 
-<instructions>
 请基于以上信息继续完成任务。请注意，这是之前对话的摘要，上下文长度已超过限制而被重置。请直接继续任务，无需重复已完成的步骤。如有需要，可以询问用户以获取更多信息。
-</instructions>
-</summary>
 """
-            except Exception as e:
-                spinner.text = "总结对话历史失败"
-                spinner.fail("❌")
-                return ""
 
     def _call_tools(self, response: str) -> Tuple[bool, Any]:
         """调用工具执行响应
@@ -770,32 +775,31 @@ arguments:
             3. 包含错误处理和恢复逻辑
             4. 自动加载相关方法论(如果是首次运行)
         """
+
+        self.prompt = f"{user_input}"
         try:
             set_agent(self.name, self)
 
-            self.prompt = f"{user_input}"
-
-            if self.first:
-                # 如果有上传文件，先上传文件
-                if self.files and isinstance(self.model, BasePlatform) and hasattr(self.model, "upload_files"):
-                    self.model.upload_files(self.files)
-                    self.prompt = f"{user_input}"
-                
-                # 如果启用方法论且没有上传文件，上传方法论
-                elif self.use_methodology:
-                    platform = self.model if hasattr(self.model, "upload_files") else None
-                    if platform and upload_methodology(platform):
-                        self.prompt = f"{user_input}"
-                    else:
-                        # 上传失败则回退到本地加载
-                        msg = user_input
-                        for handler in self.input_handler:
-                            msg, _ = handler(msg, self)
-                        self.prompt = f"{user_input}\n\n以下是历史类似问题的执行经验，可参考：\n{load_methodology(msg, self.get_tool_registry())}"
-
-                self.first = False
-
             while True:
+                if self.first:
+                    # 如果有上传文件，先上传文件
+                    if (
+                        self.files
+                        and self.model
+                        and self.model.support_upload_files()
+                    ):
+                        self.model.upload_files(self.files)
+
+                    # 如果启用方法论且没有上传文件，上传方法论
+                    elif self.use_methodology:
+                        if not self.model or not upload_methodology(self.model):
+                            # 上传失败则回退到本地加载
+                            msg = self.prompt
+                            for handler in self.input_handler:
+                                msg, _ = handler(msg, self)
+                            self.prompt = f"{self.prompt}\n\n以下是历史类似问题的执行经验，可参考：\n{load_methodology(msg, self.get_tool_registry())}"
+
+                    self.first = False
                 try:
                     current_response = self._call_model(self.prompt, True)
                     self.prompt = ""
@@ -836,7 +840,7 @@ arguments:
             PrettyOutput.print(f"任务失败: {str(e)}", OutputType.ERROR)
             return f"Task failed: {str(e)}"
 
-    def _clear_history(self):
+    def clear_history(self):
         """清空对话历史但保留系统提示
 
         该方法将：

jarvis/jarvis_code_analysis/code_review.py

@@ -588,25 +588,33 @@ class CodeReviewTool:
                     temp_file.flush()
                 
                 try:
-                    upload_success = False
                     # Check if content is too large
                     is_large_content = is_context_overflow(diff_output)
                     
                     # Upload the file to the agent's model
-                    if is_large_content and agent.model and hasattr(agent.model, 'upload_files'):
+                    if is_large_content:
+                        if not agent.model or not agent.model.support_upload_files():
+                            return {
+                                "success": False,
+                                "stdout": "",
+                                "stderr": "代码差异太大，无法处理"
+                            }
+                        
                         with yaspin(text="正在上传代码差异文件...", color="cyan") as spinner:
                             upload_success = agent.model.upload_files([temp_file_path])
                             if upload_success:
                                 spinner.ok("✅")
                                 PrettyOutput.print(f"已成功上传代码差异文件", OutputType.SUCCESS)
                             else:
-                                spinner.fail("❌")
-                                PrettyOutput.print(f"上传代码差异文件失败，将使用分块处理", OutputType.WARNING)
-                    else:
-                        upload_success = False
+                                return {
+                                    "success": False,
+                                    "stdout": "",
+                                    "stderr": "上传代码差异文件失败"
+                                }
+                        
                     
                     # Prepare the prompt based on upload status
-                    if upload_success:
+                    if is_large_content:
                         # When file is uploaded, reference it in the prompt
                         complete_prompt = user_prompt + f"""
 
@@ -619,13 +627,6 @@ class CodeReviewTool:
                         # Run the agent with the prompt
                         result = agent.run(complete_prompt)
                     else:
-                        if is_large_content:
-                            return {
-                                "success": False,
-                                "stdout": "",
-                                "stderr": "错误：上传代码差异文件失败"
-                            }
-                        # Include the diff directly in the prompt for smaller content
                         complete_prompt = user_prompt + "\n\n代码差异内容:\n```diff\n" + diff_output + "\n```"
                         result = agent.run(complete_prompt)
                 finally:

jarvis/jarvis_data/config_schema.json

@@ -179,7 +179,7 @@
     "JARVIS_MAX_BIG_CONTENT_SIZE": {
       "type": "number",
       "description": "最大大内容尺寸",
-      "default": 1024000
+      "default": 160000
     },
     "JARVIS_PRETTY_OUTPUT": {
       "type": "boolean",

jarvis/jarvis_dev/main.py

@@ -1160,42 +1160,42 @@ def create_dev_team() -> MultiAgent:
             description="Project Manager - Coordinates team and manages project delivery",
             system_prompt=PM_PROMPT_WITH_TOOLS,
             output_handler=[PM_output_handler],
-            platform=PlatformRegistry().get_thinking_platform(),
+            platform=PlatformRegistry().get_normal_platform(),
         ),
         dict(
             name="BA",
             description="Business Analyst - Analyzes and documents requirements",
             system_prompt=BA_PROMPT_WITH_TOOLS,
             output_handler=[BA_output_handler],
-            platform=PlatformRegistry().get_thinking_platform(),
+            platform=PlatformRegistry().get_normal_platform(),
         ),
         dict(
             name="SA",
             description="Solution Architect - Designs technical solutions",
             system_prompt=SA_PROMPT_WITH_TOOLS,
             output_handler=[SA_output_handler],
-            platform=PlatformRegistry().get_thinking_platform(),
+            platform=PlatformRegistry().get_normal_platform(),
         ),
         dict(
             name="TL",
             description="Technical Lead - Leads development team and ensures technical quality",
             system_prompt=TL_PROMPT_WITH_TOOLS,
             output_handler=[TL_output_handler],
-            platform=PlatformRegistry().get_thinking_platform(),
+            platform=PlatformRegistry().get_normal_platform(),
         ),
         dict(
             name="DEV",
             description="Developer - Implements features and writes code",
             system_prompt=DEV_PROMPT_WITH_TOOLS,
             output_handler=[DEV_output_handler],
-            platform=PlatformRegistry().get_thinking_platform(),
+            platform=PlatformRegistry().get_normal_platform(),
         ),
         dict(
             name="QA",
             description="Quality Assurance - Ensures product quality through testing",
             system_prompt=QA_PROMPT_WITH_TOOLS,
             output_handler=[QA_output_handler],
-            platform=PlatformRegistry().get_thinking_platform(),
+            platform=PlatformRegistry().get_normal_platform(),
         )
     ]
 

jarvis/jarvis_git_details/main.py

@@ -207,7 +207,7 @@ class GitCommitAnalyzer:
 [检查代码是否符合行业最佳实践和项目规范]
 {ct("REPORT")}""",
             output_handler=[tool_registry],
-            platform=PlatformRegistry().get_thinking_platform(),
+            platform=PlatformRegistry().get_normal_platform(),
             auto_complete=True
         )