jarvis-ai-assistant 0.1.194__tar.gz → 0.1.196__tar.gz

{jarvis_ai_assistant-0.1.194/src/jarvis_ai_assistant.egg-info → jarvis_ai_assistant-0.1.196}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jarvis-ai-assistant
-Version: 0.1.194
+Version: 0.1.196
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire
@@ -117,6 +117,63 @@ pip3 install jarvis-ai-assistant
 | `jarvis-git-details` | - | 使用git details功能 |
 | `jarvis-methodology` | - | 使用方法论功能 |
 
+### Jarvis功能 (jarvis)
+
+`jarvis` 是Jarvis的通用代理工具，提供开箱即用的AI助手功能。
+
+#### 1. 核心功能
+- 任务分析与规划
+- 代码分析与修改
+- 系统交互与操作
+- 方法论应用与优化
+- 多代理协作
+
+#### 2. 使用方式
+```bash
+# 基本用法
+jarvis
+
+# 带参数使用
+jarvis -p <平台> -m <模型> -t "初始任务"
+```
+
+#### 3. 命令行参数
+| 参数 | 描述 |
+|------|------|
+| `-p/--platform` | 指定AI平台 (yuanbao/kimi/tongyi/openai) |
+| `-m/--model` | 指定模型名称 |
+| `-t/--task` | 指定初始任务 |
+
+#### 4. 工作流程
+1. 初始化环境
+2. 加载默认配置
+3. 创建代理实例
+4. 执行初始任务（如果指定）
+5. 进入交互式模式（如果没有初始任务）
+6. 根据用户输入执行任务
+
+#### 5. 任务执行特点
+- 自动应用最佳方法论
+- 智能任务分解
+- 多工具协同工作
+- 实时进度反馈
+- 自动生成任务总结
+
+#### 6. 示例
+```bash
+# 基本使用
+jarvis
+
+# 指定平台和模型
+jarvis -p yuanbao -m deep_seek_v3
+
+# 直接执行任务
+jarvis -t "分析项目结构并生成架构图"
+
+# 组合使用
+jarvis -p kimi -m k1 -t "优化项目性能"
+```
+
 ### 代码代理功能 (jarvis-code-agent)
 
 `jarvis-code-agent` 是Jarvis的代码分析与修改工具，专注于代码工程任务。
@@ -153,16 +210,7 @@ jarvis-code-agent -p <平台> -m <模型> -r "需求描述"
 4. 自动处理git提交
 5. 显示修改结果
 
-#### 5. 使用规范
-代码代理遵循以下规范进行代码修改：
-1. 每次响应仅执行一步操作，先分析再修改，避免一步多改
-2. 充分利用工具理解用户需求和现有代码，禁止凭空假设
-3. 如果不清楚要修改的文件，必须先分析并找出需要修改的文件
-4. 代码编辑任务优先使用 edit_file 工具，确保搜索文本在目标文件中有且仅有一次精确匹配
-5. 如需大范围重写，才可使用 rewrite_file 工具
-6. 如遇信息不明，优先调用工具补充分析，不要主观臆断
-
-#### 6. 示例
+#### 5. 示例
 ```bash
 # 使用默认平台分析代码
 jca
@@ -221,7 +269,7 @@ jarvis-git-commit --root-dir <目录> --prefix "前缀" --suffix "后缀"
 - `test`: 测试相关
 - `chore`: 其他修改
 
-格式规则：
+默认格式规则（可配置）：
 1. 类型必须使用上述预定义类型
 2. 范围表示变更的模块或组件（例如：auth, database, ui）
 3. 主题行不超过72个字符，不以句号结尾，使用祈使语气
@@ -250,7 +298,7 @@ jgc --prefix "[紧急修复]"
 jgc --suffix "相关issue: #123"
 ```
 
-### 通用代理功能 (jarvis-agent)
+### 自定义代理功能 (jarvis-agent)
 
 `jarvis-agent` 是Jarvis的核心代理工具，提供通用的AI助手功能，支持自定义配置和任务执行。
 
@@ -490,6 +538,7 @@ OPENAI_API_BASE: https://api.openai.com/v1
 | `JARVIS_PRINT_PROMPT` | false | 是否打印提示 |
 | `JARVIS_USE_METHODOLOGY` | true | 是否启用方法论功能 |
 | `JARVIS_USE_ANALYSIS` | true | 是否启用任务分析功能 |
+| `JARVIS_USE_HISTORY` | false | 是否启用历史记录功能 |
 | `JARVIS_DATA_PATH` | ~/.jarvis | Jarvis数据存储目录路径 |
 
 ## 🛠️ 工具说明 <a id="tools"></a>

{jarvis_ai_assistant-0.1.194 → jarvis_ai_assistant-0.1.196}/README.md

@@ -47,6 +47,63 @@ pip3 install jarvis-ai-assistant
 | `jarvis-git-details` | - | 使用git details功能 |
 | `jarvis-methodology` | - | 使用方法论功能 |
 
+### Jarvis功能 (jarvis)
+
+`jarvis` 是Jarvis的通用代理工具，提供开箱即用的AI助手功能。
+
+#### 1. 核心功能
+- 任务分析与规划
+- 代码分析与修改
+- 系统交互与操作
+- 方法论应用与优化
+- 多代理协作
+
+#### 2. 使用方式
+```bash
+# 基本用法
+jarvis
+
+# 带参数使用
+jarvis -p <平台> -m <模型> -t "初始任务"
+```
+
+#### 3. 命令行参数
+| 参数 | 描述 |
+|------|------|
+| `-p/--platform` | 指定AI平台 (yuanbao/kimi/tongyi/openai) |
+| `-m/--model` | 指定模型名称 |
+| `-t/--task` | 指定初始任务 |
+
+#### 4. 工作流程
+1. 初始化环境
+2. 加载默认配置
+3. 创建代理实例
+4. 执行初始任务（如果指定）
+5. 进入交互式模式（如果没有初始任务）
+6. 根据用户输入执行任务
+
+#### 5. 任务执行特点
+- 自动应用最佳方法论
+- 智能任务分解
+- 多工具协同工作
+- 实时进度反馈
+- 自动生成任务总结
+
+#### 6. 示例
+```bash
+# 基本使用
+jarvis
+
+# 指定平台和模型
+jarvis -p yuanbao -m deep_seek_v3
+
+# 直接执行任务
+jarvis -t "分析项目结构并生成架构图"
+
+# 组合使用
+jarvis -p kimi -m k1 -t "优化项目性能"
+```
+
 ### 代码代理功能 (jarvis-code-agent)
 
 `jarvis-code-agent` 是Jarvis的代码分析与修改工具，专注于代码工程任务。
@@ -83,16 +140,7 @@ jarvis-code-agent -p <平台> -m <模型> -r "需求描述"
 4. 自动处理git提交
 5. 显示修改结果
 
-#### 5. 使用规范
-代码代理遵循以下规范进行代码修改：
-1. 每次响应仅执行一步操作，先分析再修改，避免一步多改
-2. 充分利用工具理解用户需求和现有代码，禁止凭空假设
-3. 如果不清楚要修改的文件，必须先分析并找出需要修改的文件
-4. 代码编辑任务优先使用 edit_file 工具，确保搜索文本在目标文件中有且仅有一次精确匹配
-5. 如需大范围重写，才可使用 rewrite_file 工具
-6. 如遇信息不明，优先调用工具补充分析，不要主观臆断
-
-#### 6. 示例
+#### 5. 示例
 ```bash
 # 使用默认平台分析代码
 jca
@@ -151,7 +199,7 @@ jarvis-git-commit --root-dir <目录> --prefix "前缀" --suffix "后缀"
 - `test`: 测试相关
 - `chore`: 其他修改
 
-格式规则：
+默认格式规则（可配置）：
 1. 类型必须使用上述预定义类型
 2. 范围表示变更的模块或组件（例如：auth, database, ui）
 3. 主题行不超过72个字符，不以句号结尾，使用祈使语气
@@ -180,7 +228,7 @@ jgc --prefix "[紧急修复]"
 jgc --suffix "相关issue: #123"
 ```
 
-### 通用代理功能 (jarvis-agent)
+### 自定义代理功能 (jarvis-agent)
 
 `jarvis-agent` 是Jarvis的核心代理工具，提供通用的AI助手功能，支持自定义配置和任务执行。
 
@@ -420,6 +468,7 @@ OPENAI_API_BASE: https://api.openai.com/v1
 | `JARVIS_PRINT_PROMPT` | false | 是否打印提示 |
 | `JARVIS_USE_METHODOLOGY` | true | 是否启用方法论功能 |
 | `JARVIS_USE_ANALYSIS` | true | 是否启用任务分析功能 |
+| `JARVIS_USE_HISTORY` | false | 是否启用历史记录功能 |
 | `JARVIS_DATA_PATH` | ~/.jarvis | Jarvis数据存储目录路径 |
 
 ## 🛠️ 工具说明 <a id="tools"></a>

{jarvis_ai_assistant-0.1.194 → jarvis_ai_assistant-0.1.196}/pyproject.toml

@@ -8,7 +8,7 @@ default = true
 
 [project]
 name = "jarvis-ai-assistant"
-version = "0.1.194"
+version = "0.1.196"
 description = "Jarvis: An AI assistant that uses tools to interact with the system"
 readme = "README.md"
 authors = [{ name = "skyfire", email = "skyfireitdiy@hotmail.com" }]

{jarvis_ai_assistant-0.1.194 → jarvis_ai_assistant-0.1.196}/setup.py

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

{jarvis_ai_assistant-0.1.194 → jarvis_ai_assistant-0.1.196}/src/jarvis/__init__.py

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

{jarvis_ai_assistant-0.1.194 → jarvis_ai_assistant-0.1.196}/src/jarvis/jarvis_agent/__init__.py

@@ -1,6 +1,8 @@
 # -*- coding: utf-8 -*-
 # 标准库导入
 import datetime
+import os
+from pathlib import Path
 import platform
 from typing import Any, Callable, Dict, List, Optional, Protocol, Tuple, Union
 
@@ -12,26 +14,20 @@ from yaspin import yaspin  # type: ignore
 # jarvis_platform 相关
 from jarvis.jarvis_platform.base import BasePlatform
 from jarvis.jarvis_platform.registry import PlatformRegistry
-
 # jarvis_utils 相关
-from jarvis.jarvis_utils.config import (
-    get_max_token_count,
-    get_max_tool_call_count,
-    is_auto_complete,
-    is_execute_tool_confirm,
-    is_use_analysis,
-    is_use_methodology,
-)
+from jarvis.jarvis_utils.config import (get_data_dir, get_max_token_count,
+                                        get_max_tool_call_count,
+                                        is_auto_complete,
+                                        is_execute_tool_confirm,
+                                        is_use_analysis, get_history_count, is_use_methodology)
 from jarvis.jarvis_utils.embedding import get_context_token_count
-from jarvis.jarvis_utils.globals import (
-    delete_agent,
-    get_interrupt,
-    make_agent_name,
-    set_agent,
-    set_interrupt,
-)
+from jarvis.jarvis_utils.globals import (delete_agent, get_interrupt,
+                                         make_agent_name, set_agent,
+                                         set_interrupt)
 from jarvis.jarvis_utils.input import get_multiline_input
-from jarvis.jarvis_utils.methodology import load_methodology, upload_methodology
+from jarvis.jarvis_utils.jarvis_history import JarvisHistory
+from jarvis.jarvis_utils.methodology import (load_methodology,
+                                             upload_methodology)
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.tag import ct, ot
 from jarvis.jarvis_utils.utils import user_confirm
@@ -143,6 +139,7 @@ class Agent:
         self.prompt = ""
 
     def __del__(self):
+        self.history.stop_record()
         delete_agent(self.name)
 
     def __init__(
@@ -163,6 +160,7 @@ class Agent:
         use_methodology: Optional[bool] = None,
         use_analysis: Optional[bool] = None,
         files: List[str] = [],
+        history_count: Optional[int] = None,
     ):
         self.files = files
         """初始化Jarvis Agent实例
@@ -241,6 +239,13 @@ class Agent:
         self.max_tool_call_count = get_max_tool_call_count()
         self.after_tool_call_cb: Optional[Callable[[Agent], None]] = None
 
+        self.history = JarvisHistory()
+        self.history_dir = str(Path(get_data_dir())/"history")
+        self.history.start_record(self.history_dir)
+
+        self.history_count = history_count if history_count is not None else get_history_count()
+
+
         self.execute_tool_confirm = (
             execute_tool_confirm
             if execute_tool_confirm is not None
@@ -402,8 +407,10 @@ class Agent:
         if self.conversation_length > self.max_token_count:
             message = self._summarize_and_clear_history() + "\n\n" + message
             self.conversation_length += get_context_token_count(message)
+        self.history.append_msg("user", message)
         response = self.model.chat_until_success(message)  # type: ignore
         self.conversation_length += get_context_token_count(response)
+        self.history.append_msg("assistant", response)
         return response
 
     def generate_summary(self) -> str:
@@ -781,21 +788,9 @@ arguments:
             set_agent(self.name, self)
 
             while True:
+                history_md = ""
                 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
+                    history_md = self._first_run()
                 try:
                     current_response = self._call_model(self.prompt, True)
                     self.prompt = ""
@@ -838,6 +833,8 @@ arguments:
                         return self._complete_task()
 
                 except Exception as e:
+                    if history_md:
+                        os.remove(history_md)
                     PrettyOutput.print(f"任务失败: {str(e)}", OutputType.ERROR)
                     return f"Task failed: {str(e)}"
 
@@ -845,6 +842,41 @@ arguments:
             PrettyOutput.print(f"任务失败: {str(e)}", OutputType.ERROR)
             return f"Task failed: {str(e)}"
 
+    def _first_run(self):
+        history_md = ""
+        if self.history_count > 0 and self.model and self.model.support_upload_files():
+            import tempfile
+            timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+            history_md = str(Path(tempfile.gettempdir())/f"{self.name}_history_{timestamp}.md")
+            self.history.export_history_to_markdown(tempfile.gettempdir(), history_md, max_files=self.history_count)
+            self.files.append(history_md)
+
+                        # 如果有上传文件，先上传文件
+        if self.model and self.model.support_upload_files():
+            if self.use_methodology:
+                if not upload_methodology(self.model, other_files=self.files):
+                    if self.files:
+                        PrettyOutput.print("文件上传失败，将忽略文件列表", OutputType.WARNING)
+                                    # 上传失败则回退到本地加载
+                    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())}"
+            elif self.files:
+                if not self.model.upload_files(self.files):
+                    PrettyOutput.print("文件上传失败，将忽略文件列表", OutputType.WARNING)
+        else:
+            if self.files:
+                PrettyOutput.print("不支持上传文件，将忽略文件列表", OutputType.WARNING)
+            if self.use_methodology:
+                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
+        return history_md
+
     def clear_history(self):
         """清空对话历史但保留系统提示
 

{jarvis_ai_assistant-0.1.194 → jarvis_ai_assistant-0.1.196}/src/jarvis/jarvis_agent/jarvis.py

@@ -8,14 +8,9 @@ import yaml
 from prompt_toolkit import prompt
 from yaspin import yaspin
 
-from jarvis.jarvis_agent import (
-    Agent,
-    OutputType,
-    PrettyOutput,
-    get_multiline_input,
-    origin_agent_system_prompt,
-    user_confirm,
-)
+from jarvis.jarvis_agent import (Agent, OutputType, PrettyOutput,
+                                 get_multiline_input,
+                                 origin_agent_system_prompt, user_confirm)
 from jarvis.jarvis_agent.builtin_input_handler import builtin_input_handler
 from jarvis.jarvis_agent.shell_input_handler import shell_input_handler
 from jarvis.jarvis_tools.registry import ToolRegistry

{jarvis_ai_assistant-0.1.194 → jarvis_ai_assistant-0.1.196}/src/jarvis/jarvis_code_agent/code_agent.py

@@ -10,30 +10,26 @@ import subprocess
 import sys
 from typing import List, Optional, Tuple
 
+from yaspin import yaspin  # type: ignore
+
 from jarvis.jarvis_agent import Agent
 from jarvis.jarvis_agent.builtin_input_handler import builtin_input_handler
 from jarvis.jarvis_agent.shell_input_handler import shell_input_handler
-
 # 忽略yaspin的类型检查
 from jarvis.jarvis_code_agent.lint import get_lint_tools
 from jarvis.jarvis_git_utils.git_commiter import GitCommitTool
 from jarvis.jarvis_platform.registry import PlatformRegistry
 from jarvis.jarvis_tools.registry import ToolRegistry
 from jarvis.jarvis_utils.config import is_confirm_before_apply_patch
-from jarvis.jarvis_utils.git_utils import (
-    find_git_root,
-    get_commits_between,
-    get_diff,
-    get_diff_file_list,
-    get_latest_commit_hash,
-    get_recent_commits_with_files,
-    handle_commit_workflow,
-    has_uncommitted_changes,
-)
+from jarvis.jarvis_utils.git_utils import (find_git_root, get_commits_between,
+                                           get_diff, get_diff_file_list,
+                                           get_latest_commit_hash,
+                                           get_recent_commits_with_files,
+                                           handle_commit_workflow,
+                                           has_uncommitted_changes)
 from jarvis.jarvis_utils.input import get_multiline_input
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.utils import get_loc_stats, init_env, user_confirm
-from yaspin import yaspin  # type: ignore
 
 
 class CodeAgent:

{jarvis_ai_assistant-0.1.194 → jarvis_ai_assistant-0.1.196}/src/jarvis/jarvis_code_analysis/checklists/loader.py

@@ -5,26 +5,12 @@ Utility module for loading language-specific code review checklists.
 from typing import Dict, Optional
 
 # Import checklist modules
-from jarvis.jarvis_code_analysis.checklists import (
-    c_cpp,
-    csharp,
-    data_format,
-    devops,
-    docs,
-    go,
-    infrastructure,
-    java,
-    javascript,
-    kotlin,
-    php,
-    python,
-    ruby,
-    rust,
-    shell,
-    sql,
-    swift,
-    web,
-)
+from jarvis.jarvis_code_analysis.checklists import (c_cpp, csharp, data_format,
+                                                    devops, docs, go,
+                                                    infrastructure, java,
+                                                    javascript, kotlin, php,
+                                                    python, ruby, rust, shell,
+                                                    sql, swift, web)
 
 # Map of language identifiers to their checklist content
 CHECKLIST_MAP = {

{jarvis_ai_assistant-0.1.194 → jarvis_ai_assistant-0.1.196}/src/jarvis/jarvis_code_analysis/code_review.py

@@ -8,7 +8,8 @@ from typing import Any, Dict, List
 from yaspin import yaspin
 
 from jarvis.jarvis_agent import Agent
-from jarvis.jarvis_code_analysis.checklists.loader import get_language_checklist
+from jarvis.jarvis_code_analysis.checklists.loader import \
+    get_language_checklist
 from jarvis.jarvis_platform.registry import PlatformRegistry
 from jarvis.jarvis_tools.read_code import ReadCodeTool
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput

{jarvis_ai_assistant-0.1.194 → jarvis_ai_assistant-0.1.196}/src/jarvis/jarvis_data/config_schema.json

@@ -201,6 +201,11 @@
       "description": "是否打印提示",
       "default": false
     },
+    "JARVIS_USE_HISTORY_COUNT": {
+      "type": "number",
+      "description": "使用的历史记录数量",
+      "default": 0
+    },
     "JARVIS_REPLACE_MAP": {
       "type": "object",
       "description": "自定义替换映射表配置",

{jarvis_ai_assistant-0.1.194 → jarvis_ai_assistant-0.1.196}/src/jarvis/jarvis_git_utils/git_commiter.py

@@ -12,7 +12,8 @@ from yaspin import yaspin
 
 from jarvis.jarvis_platform.registry import PlatformRegistry
 from jarvis.jarvis_utils.config import get_git_commit_prompt
-from jarvis.jarvis_utils.git_utils import find_git_root, has_uncommitted_changes
+from jarvis.jarvis_utils.git_utils import (find_git_root,
+                                           has_uncommitted_changes)
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.tag import ct, ot
 from jarvis.jarvis_utils.utils import init_env, is_context_overflow
@@ -117,7 +118,7 @@ class GitCommitTool:
                     stdout=subprocess.PIPE,
                     stderr=subprocess.PIPE,
                 )
-                diff = process.communicate()[0].decode()
+                diff = process.communicate()[0].decode(errors="ignore")
                 spinner.write(f"✅ 获取差异 ({file_count} 个文件)")
                 try:
                     temp_diff_file_path = None

{jarvis_ai_assistant-0.1.194 → jarvis_ai_assistant-0.1.196}/src/jarvis/jarvis_methodology/main.py

@@ -17,10 +17,8 @@ import yaml  # type: ignore
 from yaspin import yaspin  # type: ignore
 
 from jarvis.jarvis_platform.registry import PlatformRegistry
-from jarvis.jarvis_utils.methodology import (
-    _get_methodology_directory,
-    _load_all_methodologies,
-)
+from jarvis.jarvis_utils.methodology import (_get_methodology_directory,
+                                             _load_all_methodologies)
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
 

{jarvis_ai_assistant-0.1.194 → jarvis_ai_assistant-0.1.196}/src/jarvis/jarvis_platform/base.py

@@ -9,16 +9,14 @@ from rich.panel import Panel
 from rich.text import Text
 from yaspin import yaspin
 
-from jarvis.jarvis_utils.config import (
-    get_max_input_token_count,
-    get_pretty_output,
-    is_print_prompt,
-)
+from jarvis.jarvis_utils.config import (get_max_input_token_count,
+                                        get_pretty_output, is_print_prompt)
 from jarvis.jarvis_utils.embedding import split_text_into_chunks
 from jarvis.jarvis_utils.globals import set_in_chat
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.tag import ct, ot
-from jarvis.jarvis_utils.utils import get_context_token_count, while_success, while_true
+from jarvis.jarvis_utils.utils import (get_context_token_count, while_success,
+                                       while_true)
 
 
 class BasePlatform(ABC):

{jarvis_ai_assistant-0.1.194 → jarvis_ai_assistant-0.1.196}/src/jarvis/jarvis_platform/human.py

@@ -1,4 +1,9 @@
 # -*- coding: utf-8 -*-
+
+# 人类交互平台实现模块
+
+# 提供与真实人类交互的模拟接口
+
 import random
 import string
 from typing import Generator, List, Tuple
@@ -72,4 +77,9 @@ class HumanPlatform(BasePlatform):
         return self.model_name
 
     def support_web(self) -> bool:
+        """是否支持网页浏览功能"""
+        return False
+
+    def support_upload_files(self) -> bool:
+        """是否支持文件上传功能"""
         return False

{jarvis_ai_assistant-0.1.194 → jarvis_ai_assistant-0.1.196}/src/jarvis/jarvis_platform/kimi.py

@@ -1,4 +1,6 @@
 # -*- coding: utf-8 -*-
+# Kimi 平台实现模块
+# 提供与 Moonshot AI 的 Kimi 大模型交互功能
 import json
 import mimetypes
 import os
@@ -13,7 +15,12 @@ from jarvis.jarvis_utils.utils import while_success
 
 
 class KimiModel(BasePlatform):
-    """Kimi model implementation"""
+    """Kimi 大模型平台实现类
+    封装了与 Kimi 大模型交互的所有功能，包括：
+    - 会话管理
+    - 文件上传
+    - 消息收发
+    """
 
     platform_name = "kimi"
 
@@ -29,16 +36,16 @@ class KimiModel(BasePlatform):
         Initialize Kimi model
         """
         super().__init__()
-        self.chat_id = ""
-        self.api_key = os.getenv("KIMI_API_KEY")
+        self.chat_id = ""  # 当前会话ID
+        self.api_key = os.getenv("KIMI_API_KEY")  # 从环境变量获取API密钥
         if not self.api_key:
             PrettyOutput.print("KIMI_API_KEY 未设置", OutputType.WARNING)
-        self.auth_header = f"Bearer {self.api_key}"
+        self.auth_header = f"Bearer {self.api_key}"  # 认证头信息
         self.uploaded_files = []  # 存储已上传文件的信息
-        self.chat_id = ""
-        self.first_chat = True  # 添加标记，用于判断是否是第一次对话
-        self.system_message = ""
-        self.model_name = "kimi"
+        self.chat_id = ""  # 当前会话ID
+        self.first_chat = True  # 标记是否是第一次对话
+        self.system_message = ""  # 系统提示消息
+        self.model_name = "kimi"  # 默认模型名称
 
     def set_system_prompt(self, message: str):
         """Set system message"""
@@ -246,7 +253,12 @@ class KimiModel(BasePlatform):
         return True
 
     def chat(self, message: str) -> Generator[str, None, None]:
-        """Send message and get response"""
+        """发送消息并获取响应流
+        参数:
+            message: 要发送的消息内容
+        返回:
+            生成器，逐块返回模型响应
+        """
         if not self.chat_id:
             if not self._create_chat():
                 raise Exception("Failed to create chat session")