jarvis-ai-assistant 0.1.221__tar.gz → 0.1.223__tar.gz

{jarvis_ai_assistant-0.1.221/src/jarvis_ai_assistant.egg-info → jarvis_ai_assistant-0.1.223}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jarvis-ai-assistant
-Version: 0.1.221
+Version: 0.1.223
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire
@@ -117,18 +117,19 @@ pip3 install jarvis-ai-assistant
 ### 基本使用
 | 命令 | 快捷方式 | 功能描述 |
 |------|----------|----------|
-| `jarvis` | - | 使用通用代理 |
+| `jarvis` | `jvs` | 使用通用代理 |
 | `jarvis-code-agent` | `jca` | 使用代码代理 |
 | `jarvis-smart-shell` | `jss` | 使用智能shell功能 |
 | `jarvis-platform-manager` | `jpm` | 使用平台管理功能 |
-| `jarvis-code-review` | - | 使用代码审查功能 |
+| `jarvis-code-review` | `jcr` | 使用代码审查功能 |
 | `jarvis-git-commit` | `jgc` | 使用自动化git commit功能 |
-| `jarvis-git-squash` | - | 使用git squash功能 |
-| `jarvis-multi-agent` | - | 使用多代理功能 |
-| `jarvis-agent` | - | 使用agent功能 |
-| `jarvis-tool` | - | 使用工具功能 |
-| `jarvis-git-details` | - | 使用git details功能 |
-| `jarvis-methodology` | - | 使用方法论功能 |
+| `jarvis-git-squash` | `jgs` | 使用git squash功能 |
+| `jarvis-multi-agent` | `jma` | 使用多代理功能 |
+| `jarvis-agent` | `ja` | 使用agent功能 |
+| `jarvis-tool` | `jt` | 使用工具功能 |
+| `jarvis-git-details` | `jgd` | 使用git details功能 |
+| `jarvis-methodology` | `jm` | 使用方法论功能 |
+| `jarvis-rag` | `jrg` | 使用RAG功能 |
 
 ### Jarvis功能 (jarvis)
 
@@ -147,15 +148,16 @@ pip3 install jarvis-ai-assistant
 jarvis
 
 # 带参数使用
-jarvis -p <平台> -m <模型> -t "初始任务"
+jarvis --llm_type thinking -t "初始任务"
 ```
 
 #### 3. 命令行参数
 | 参数 | 描述 |
 |------|------|
-| `-p/--platform` | 指定AI平台 (yuanbao/kimi/tongyi/openai) |
-| `-m/--model` | 指定模型名称 |
-| `-t/--task` | 指定初始任务 |
+| `--llm_type` | 使用的LLM类型 (normal/thinking) |
+| `-t/--task` | 直接从命令行输入任务内容 |
+| `-f/--config` | 自定义配置文件的路径 |
+| `--restore-session` | 从 .jarvis/saved_session.json 恢复会话 |
 
 #### 4. 工作流程
 1. 初始化环境
@@ -177,14 +179,14 @@ jarvis -p <平台> -m <模型> -t "初始任务"
 # 基本使用
 jarvis
 
-# 指定平台和模型
-jarvis -p yuanbao -m deep_seek_v3
+# 指定LLM类型
+jarvis --llm_type thinking
 
 # 直接执行任务
 jarvis -t "分析项目结构并生成架构图"
 
 # 组合使用
-jarvis -p kimi -m k1 -t "优化项目性能"
+jarvis --llm_type thinking -t "优化项目性能"
 ```
 
 ### 代码代理功能 (jarvis-code-agent)
@@ -206,15 +208,15 @@ jarvis-code-agent
 jca
 
 # 带参数使用
-jarvis-code-agent -p <平台> -m <模型> -r "需求描述"
+jarvis-code-agent --llm_type thinking -r "需求描述"
 ```
 
 #### 3. 命令行参数
 | 参数 | 描述 |
 |------|------|
-| `-p/--platform` | 指定AI平台 (yuanbao/kimi/tongyi/openai) |
-| `-m/--model` | 指定模型名称 |
-| `-r/--requirement` | 直接指定需求描述 |
+| `--llm_type` | 使用的LLM类型 (normal/thinking) |
+| `-r/--requirement` | 要处理的需求 |
+| `--restore-session` | 从 .jarvis/saved_session.json 恢复会话 |
 
 #### 4. 工作流程
 1. 初始化环境（查找git根目录，检查未提交修改）
@@ -228,8 +230,8 @@ jarvis-code-agent -p <平台> -m <模型> -r "需求描述"
 # 使用默认平台分析代码
 jca
 
-# 指定平台和模型
-jca -p yuanbao -m deep_seek_v3
+# 指定LLM类型
+jca --llm_type thinking
 
 # 直接处理需求
 jca -r "修复src/example.py中的内存泄漏问题"
@@ -337,6 +339,7 @@ jarvis-agent -f <配置文件> -c <代理定义文件> -t "初始任务"
 | `-f/--config` | 指定配置文件路径（可选） |
 | `-c/--agent_definition` | 指定代理定义文件路径（可选） |
 | `-t/--task` | 指定初始任务（可选） |
+| `--llm_type` | 使用的LLM类型 (normal/thinking)，会覆盖配置文件中的设置 |
 
 #### 4. 配置文件格式
 代理定义文件使用YAML格式：
@@ -412,6 +415,8 @@ jarvis-platform-manager chat -p <平台名称> -m <模型名称>
 - `/shell <命令>` - 执行shell命令
 - `/save <文件名>` - 保存最后一条消息
 - `/saveall <文件名>` - 保存完整对话历史
+- `/save_session <文件名>` - 保存当前会话状态
+- `/load_session <文件名>` - 加载会话状态
 
 ##### 2.3 启动API服务
 ```bash
@@ -429,6 +434,7 @@ jarvis-platform-manager service --host <IP地址> --port <端口号> -p <平台
 ```bash
 # 加载角色配置文件并开始对话
 jarvis-platform-manager role -c <配置文件路径>
+# 配置文件默认为 ~/.jarvis/roles.yaml
 ```
 
 角色配置文件格式（YAML）：
@@ -546,6 +552,7 @@ OPENAI_API_BASE: https://api.openai.com/v1
 | `JARVIS_PRINT_PROMPT` | false | 是否打印提示 |
 | `JARVIS_USE_METHODOLOGY` | true | 是否启用方法论功能 |
 | `JARVIS_USE_ANALYSIS` | true | 是否启用任务分析功能 |
+| `JARVIS_ENABLE_STATIC_ANALYSIS` | true | 是否启用静态代码分析 |
 | `JARVIS_DATA_PATH` | ~/.jarvis | Jarvis数据存储目录路径 |
 | `JARVIS_RAG` | `{"embedding_model": "BAAI/bge-base-zh-v1.5"}` | RAG框架的配置 |
 
@@ -602,14 +609,36 @@ jarvis-rag add README.md ./docs/ 'src/jarvis/jarvis_rag/*.py'
 
 **参数与选项:**
 
+| 参数/选项 | 描述 |
+|---|---|
 | 参数/选项 | 描述 |
 |---|---|
 | `paths` | **[必需]** 一个或多个文件路径、目录路径或用引号包裹的通配符模式。 |
-| `--collection` | 指定知识库的集合名称（默认为 `jarvis_rag_collection`）。 |
-| `--embedding-model` | 覆盖全局配置，强制使用特定的嵌入模型名称。 |
+| `--collection` / `-c` | 指定知识库的集合名称（默认为 `jarvis_rag_collection`）。 |
+| `--embedding-model` / `-e` | 覆盖全局配置，强制使用特定的嵌入模型名称。 |
 | `--db-path` | 覆盖全局配置，指定向量数据库的存储路径。 |
+| `--batch-size` / `-b` | 单个批次中要处理的文档数（默认为 500）。 |
+
+##### 2.2 列出知识库中的文档 (`list-docs`)
 
-##### 2.2 查询知识库 (`query`)
+此命令用于列出知识库中所有唯一的文档来源。
+
+```bash
+# 基本用法
+jarvis-rag list-docs
+
+# 指定集合
+jarvis-rag list-docs --collection my_collection
+```
+
+**参数与选项:**
+
+| 参数/选项 | 描述 |
+|---|---|
+| `--collection` / `-c` | 指定要查询的知识库集合名称。 |
+| `--db-path` | 覆盖全局配置，指定向量数据库的存储路径。 |
+
+##### 2.3 查询知识库 (`query`)
 
 此命令用于向已建立的知识库提出问题。
 
@@ -622,18 +651,20 @@ jarvis-rag query "你的问题"
 jarvis-rag query "请总结一下我添加的文档的核心内容"
 
 # 2. 指定使用Kimi模型进行查询
-jarvis-rag query "代码中的 'PlatformRegistry' 类是做什么用的？" --platform kimi --model moonshot-v1-8k
+jarvis-rag query "代码中的 'PlatformRegistry' 类是做什么用的？" --platform kimi --model kimi
 ```
 
 **参数与选项:**
 
+| 参数/选项 | 描述 |
+|---|---|
 | 参数/选项 | 描述 |
 |---|---|
 | `question` | **[必需]** 你要向知识库提出的问题。 |
-| `--collection` | 指定要查询的知识库集合名称。 |
-| `--platform` | 指定一个平台名称来回答问题，覆盖默认的“思考”模型。 |
-| `--model` | 指定一个模型名称来回答问题，需要与 `--platform` 同时使用。 |
-| `--embedding-model` | 覆盖全局配置，强制使用特定的嵌入模型名称。 |
+| `--collection` / `-c` | 指定要查询的知识库集合名称。 |
+| `--platform` / `-p` | 指定一个平台名称来回答问题，覆盖默认的“思考”模型。 |
+| `--model` / `-m` | 指定一个模型名称来回答问题，需要与 `--platform` 同时使用。 |
+| `--embedding-model` / `-e` | 覆盖全局配置，强制使用特定的嵌入模型名称。 |
 
 ### 3. 命令替换功能
 支持使用特殊标记`'<tag>'`触发命令替换功能：

{jarvis_ai_assistant-0.1.221 → jarvis_ai_assistant-0.1.223}/README.md

@@ -33,18 +33,19 @@ pip3 install jarvis-ai-assistant
 ### 基本使用
 | 命令 | 快捷方式 | 功能描述 |
 |------|----------|----------|
-| `jarvis` | - | 使用通用代理 |
+| `jarvis` | `jvs` | 使用通用代理 |
 | `jarvis-code-agent` | `jca` | 使用代码代理 |
 | `jarvis-smart-shell` | `jss` | 使用智能shell功能 |
 | `jarvis-platform-manager` | `jpm` | 使用平台管理功能 |
-| `jarvis-code-review` | - | 使用代码审查功能 |
+| `jarvis-code-review` | `jcr` | 使用代码审查功能 |
 | `jarvis-git-commit` | `jgc` | 使用自动化git commit功能 |
-| `jarvis-git-squash` | - | 使用git squash功能 |
-| `jarvis-multi-agent` | - | 使用多代理功能 |
-| `jarvis-agent` | - | 使用agent功能 |
-| `jarvis-tool` | - | 使用工具功能 |
-| `jarvis-git-details` | - | 使用git details功能 |
-| `jarvis-methodology` | - | 使用方法论功能 |
+| `jarvis-git-squash` | `jgs` | 使用git squash功能 |
+| `jarvis-multi-agent` | `jma` | 使用多代理功能 |
+| `jarvis-agent` | `ja` | 使用agent功能 |
+| `jarvis-tool` | `jt` | 使用工具功能 |
+| `jarvis-git-details` | `jgd` | 使用git details功能 |
+| `jarvis-methodology` | `jm` | 使用方法论功能 |
+| `jarvis-rag` | `jrg` | 使用RAG功能 |
 
 ### Jarvis功能 (jarvis)
 
@@ -63,15 +64,16 @@ pip3 install jarvis-ai-assistant
 jarvis
 
 # 带参数使用
-jarvis -p <平台> -m <模型> -t "初始任务"
+jarvis --llm_type thinking -t "初始任务"
 ```
 
 #### 3. 命令行参数
 | 参数 | 描述 |
 |------|------|
-| `-p/--platform` | 指定AI平台 (yuanbao/kimi/tongyi/openai) |
-| `-m/--model` | 指定模型名称 |
-| `-t/--task` | 指定初始任务 |
+| `--llm_type` | 使用的LLM类型 (normal/thinking) |
+| `-t/--task` | 直接从命令行输入任务内容 |
+| `-f/--config` | 自定义配置文件的路径 |
+| `--restore-session` | 从 .jarvis/saved_session.json 恢复会话 |
 
 #### 4. 工作流程
 1. 初始化环境
@@ -93,14 +95,14 @@ jarvis -p <平台> -m <模型> -t "初始任务"
 # 基本使用
 jarvis
 
-# 指定平台和模型
-jarvis -p yuanbao -m deep_seek_v3
+# 指定LLM类型
+jarvis --llm_type thinking
 
 # 直接执行任务
 jarvis -t "分析项目结构并生成架构图"
 
 # 组合使用
-jarvis -p kimi -m k1 -t "优化项目性能"
+jarvis --llm_type thinking -t "优化项目性能"
 ```
 
 ### 代码代理功能 (jarvis-code-agent)
@@ -122,15 +124,15 @@ jarvis-code-agent
 jca
 
 # 带参数使用
-jarvis-code-agent -p <平台> -m <模型> -r "需求描述"
+jarvis-code-agent --llm_type thinking -r "需求描述"
 ```
 
 #### 3. 命令行参数
 | 参数 | 描述 |
 |------|------|
-| `-p/--platform` | 指定AI平台 (yuanbao/kimi/tongyi/openai) |
-| `-m/--model` | 指定模型名称 |
-| `-r/--requirement` | 直接指定需求描述 |
+| `--llm_type` | 使用的LLM类型 (normal/thinking) |
+| `-r/--requirement` | 要处理的需求 |
+| `--restore-session` | 从 .jarvis/saved_session.json 恢复会话 |
 
 #### 4. 工作流程
 1. 初始化环境（查找git根目录，检查未提交修改）
@@ -144,8 +146,8 @@ jarvis-code-agent -p <平台> -m <模型> -r "需求描述"
 # 使用默认平台分析代码
 jca
 
-# 指定平台和模型
-jca -p yuanbao -m deep_seek_v3
+# 指定LLM类型
+jca --llm_type thinking
 
 # 直接处理需求
 jca -r "修复src/example.py中的内存泄漏问题"
@@ -253,6 +255,7 @@ jarvis-agent -f <配置文件> -c <代理定义文件> -t "初始任务"
 | `-f/--config` | 指定配置文件路径（可选） |
 | `-c/--agent_definition` | 指定代理定义文件路径（可选） |
 | `-t/--task` | 指定初始任务（可选） |
+| `--llm_type` | 使用的LLM类型 (normal/thinking)，会覆盖配置文件中的设置 |
 
 #### 4. 配置文件格式
 代理定义文件使用YAML格式：
@@ -328,6 +331,8 @@ jarvis-platform-manager chat -p <平台名称> -m <模型名称>
 - `/shell <命令>` - 执行shell命令
 - `/save <文件名>` - 保存最后一条消息
 - `/saveall <文件名>` - 保存完整对话历史
+- `/save_session <文件名>` - 保存当前会话状态
+- `/load_session <文件名>` - 加载会话状态
 
 ##### 2.3 启动API服务
 ```bash
@@ -345,6 +350,7 @@ jarvis-platform-manager service --host <IP地址> --port <端口号> -p <平台
 ```bash
 # 加载角色配置文件并开始对话
 jarvis-platform-manager role -c <配置文件路径>
+# 配置文件默认为 ~/.jarvis/roles.yaml
 ```
 
 角色配置文件格式（YAML）：
@@ -462,6 +468,7 @@ OPENAI_API_BASE: https://api.openai.com/v1
 | `JARVIS_PRINT_PROMPT` | false | 是否打印提示 |
 | `JARVIS_USE_METHODOLOGY` | true | 是否启用方法论功能 |
 | `JARVIS_USE_ANALYSIS` | true | 是否启用任务分析功能 |
+| `JARVIS_ENABLE_STATIC_ANALYSIS` | true | 是否启用静态代码分析 |
 | `JARVIS_DATA_PATH` | ~/.jarvis | Jarvis数据存储目录路径 |
 | `JARVIS_RAG` | `{"embedding_model": "BAAI/bge-base-zh-v1.5"}` | RAG框架的配置 |
 
@@ -518,14 +525,36 @@ jarvis-rag add README.md ./docs/ 'src/jarvis/jarvis_rag/*.py'
 
 **参数与选项:**
 
+| 参数/选项 | 描述 |
+|---|---|
 | 参数/选项 | 描述 |
 |---|---|
 | `paths` | **[必需]** 一个或多个文件路径、目录路径或用引号包裹的通配符模式。 |
-| `--collection` | 指定知识库的集合名称（默认为 `jarvis_rag_collection`）。 |
-| `--embedding-model` | 覆盖全局配置，强制使用特定的嵌入模型名称。 |
+| `--collection` / `-c` | 指定知识库的集合名称（默认为 `jarvis_rag_collection`）。 |
+| `--embedding-model` / `-e` | 覆盖全局配置，强制使用特定的嵌入模型名称。 |
 | `--db-path` | 覆盖全局配置，指定向量数据库的存储路径。 |
+| `--batch-size` / `-b` | 单个批次中要处理的文档数（默认为 500）。 |
+
+##### 2.2 列出知识库中的文档 (`list-docs`)
 
-##### 2.2 查询知识库 (`query`)
+此命令用于列出知识库中所有唯一的文档来源。
+
+```bash
+# 基本用法
+jarvis-rag list-docs
+
+# 指定集合
+jarvis-rag list-docs --collection my_collection
+```
+
+**参数与选项:**
+
+| 参数/选项 | 描述 |
+|---|---|
+| `--collection` / `-c` | 指定要查询的知识库集合名称。 |
+| `--db-path` | 覆盖全局配置，指定向量数据库的存储路径。 |
+
+##### 2.3 查询知识库 (`query`)
 
 此命令用于向已建立的知识库提出问题。
 
@@ -538,18 +567,20 @@ jarvis-rag query "你的问题"
 jarvis-rag query "请总结一下我添加的文档的核心内容"
 
 # 2. 指定使用Kimi模型进行查询
-jarvis-rag query "代码中的 'PlatformRegistry' 类是做什么用的？" --platform kimi --model moonshot-v1-8k
+jarvis-rag query "代码中的 'PlatformRegistry' 类是做什么用的？" --platform kimi --model kimi
 ```
 
 **参数与选项:**
 
+| 参数/选项 | 描述 |
+|---|---|
 | 参数/选项 | 描述 |
 |---|---|
 | `question` | **[必需]** 你要向知识库提出的问题。 |
-| `--collection` | 指定要查询的知识库集合名称。 |
-| `--platform` | 指定一个平台名称来回答问题，覆盖默认的“思考”模型。 |
-| `--model` | 指定一个模型名称来回答问题，需要与 `--platform` 同时使用。 |
-| `--embedding-model` | 覆盖全局配置，强制使用特定的嵌入模型名称。 |
+| `--collection` / `-c` | 指定要查询的知识库集合名称。 |
+| `--platform` / `-p` | 指定一个平台名称来回答问题，覆盖默认的“思考”模型。 |
+| `--model` / `-m` | 指定一个模型名称来回答问题，需要与 `--platform` 同时使用。 |
+| `--embedding-model` / `-e` | 覆盖全局配置，强制使用特定的嵌入模型名称。 |
 
 ### 3. 命令替换功能
 支持使用特殊标记`'<tag>'`触发命令替换功能：

{jarvis_ai_assistant-0.1.221 → jarvis_ai_assistant-0.1.223}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "jarvis-ai-assistant"
-version = "0.1.221"
+version = "0.1.223"
 description = "Jarvis: An AI assistant that uses tools to interact with the system"
 readme = "README.md"
 authors = [{ name = "skyfire", email = "skyfireitdiy@hotmail.com" }]
@@ -64,6 +64,7 @@ Homepage = "https://github.com/skyfireitdiy/Jarvis"
 
 [project.scripts]
 jarvis = "jarvis.jarvis_agent.jarvis:main"
+jvs = "jarvis.jarvis_agent.jarvis:main"
 jarvis-code-agent = "jarvis.jarvis_code_agent.code_agent:main"
 jca = "jarvis.jarvis_code_agent.code_agent:main"
 jarvis-smart-shell = "jarvis.jarvis_smart_shell.main:main"
@@ -72,11 +73,19 @@ jarvis-platform-manager = "jarvis.jarvis_platform_manager.main:main"
 jpm = "jarvis.jarvis_platform_manager.main:main"
 jarvis-git-commit = "jarvis.jarvis_git_utils.git_commiter:main"
 jarvis-code-review = "jarvis.jarvis_code_analysis.code_review:main"
+jcr = "jarvis.jarvis_code_analysis.code_review:main"
 jgc = "jarvis.jarvis_git_utils.git_commiter:main"
 jarvis-git-squash = "jarvis.jarvis_git_squash.main:main"
+jgs = "jarvis.jarvis_git_squash.main:main"
 jarvis-multi-agent = "jarvis.jarvis_multi_agent.main:main"
+jma = "jarvis.jarvis_multi_agent.main:main"
 jarvis-agent = "jarvis.jarvis_agent.main:main"
+ja = "jarvis.jarvis_agent.main:main"
 jarvis-tool = "jarvis.jarvis_tools.cli.main:main"
+jt = "jarvis.jarvis_tools.cli.main:main"
 jarvis-git-details = "jarvis.jarvis_git_details.main:main"
+jgd = "jarvis.jarvis_git_details.main:main"
 jarvis-methodology = "jarvis.jarvis_methodology.main:main"
+jm = "jarvis.jarvis_methodology.main:main"
 jarvis-rag = "jarvis.jarvis_rag.cli:main"
+jrg = "jarvis.jarvis_rag.cli:main"

{jarvis_ai_assistant-0.1.221 → jarvis_ai_assistant-0.1.223}/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup, find_packages  # type: ignore
 
 setup(
     name="jarvis-ai-assistant",
-    version="0.1.221",
+    version="0.1.223",
     author="skyfire",
     author_email="skyfireitdiy@hotmail.com",
     description="An AI assistant that uses various tools to interact with the system",
@@ -55,22 +55,31 @@ setup(
     entry_points={
         "console_scripts": [
             "jarvis=jarvis.jarvis_agent.jarvis:main",
+            "jvs=jarvis.jarvis_agent.jarvis:main",
             "jarvis-code-agent=jarvis.jarvis_code_agent.code_agent:main",
             "jca=jarvis.jarvis_code_agent.code_agent:main",
             "jarvis-smart-shell=jarvis.jarvis_smart_shell.main:main",
             "jss=jarvis.jarvis_smart_shell.main:main",
             "jarvis-platform-manager=jarvis.jarvis_platform_manager.main:main",
             "jpm=jarvis.jarvis_platform_manager.main:main",
-            "jarvis-code-review=jarvis.jarvis_code_analysis.code_review:main",
             "jarvis-git-commit=jarvis.jarvis_git_utils.git_commiter:main",
             "jgc=jarvis.jarvis_git_utils.git_commiter:main",
+            "jarvis-code-review=jarvis.jarvis_code_analysis.code_review:main",
+            "jcr=jarvis.jarvis_code_analysis.code_review:main",
             "jarvis-git-squash=jarvis.jarvis_git_squash.main:main",
+            "jgs=jarvis.jarvis_git_squash.main:main",
             "jarvis-multi-agent=jarvis.jarvis_multi_agent.main:main",
+            "jma=jarvis.jarvis_multi_agent.main:main",
             "jarvis-agent=jarvis.jarvis_agent.main:main",
+            "ja=jarvis.jarvis_agent.main:main",
             "jarvis-tool=jarvis.jarvis_tools.cli.main:main",
+            "jt=jarvis.jarvis_tools.cli.main:main",
             "jarvis-git-details=jarvis.jarvis_git_details.main:main",
+            "jgd=jarvis.jarvis_git_details.main:main",
             "jarvis-methodology=jarvis.jarvis_methodology.main:main",
+            "jm=jarvis.jarvis_methodology.main:main",
             "jarvis-rag=jarvis.jarvis_rag.cli:main",
+            "jrg=jarvis.jarvis_rag.cli:main",
         ],
     },
     python_requires=">=3.9",

{jarvis_ai_assistant-0.1.221 → jarvis_ai_assistant-0.1.223}/src/jarvis/__init__.py

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

{jarvis_ai_assistant-0.1.221 → jarvis_ai_assistant-0.1.223}/src/jarvis/jarvis_agent/__init__.py

@@ -26,6 +26,10 @@ from jarvis.jarvis_platform.registry import PlatformRegistry
 # jarvis_utils 相关
 from jarvis.jarvis_utils.config import (
     get_max_token_count,
+    get_normal_model_name,
+    get_normal_platform_name,
+    get_thinking_model_name,
+    get_thinking_platform_name,
     is_execute_tool_confirm,
     is_use_analysis,
     is_use_methodology,
@@ -93,8 +97,7 @@ class Agent:
         system_prompt: str,
         name: str = "Jarvis",
         description: str = "",
-        platform: Union[Optional[BasePlatform], Optional[str]] = None,
-        model_name: Optional[str] = None,
+        llm_type: str = "normal",
         summary_prompt: Optional[str] = None,
         auto_complete: bool = False,
         output_handler: List[OutputHandlerProtocol] = [],
@@ -114,8 +117,7 @@ class Agent:
             system_prompt: 系统提示词，定义Agent的行为准则
             name: Agent名称，默认为"Jarvis"
             description: Agent描述信息
-            platform: 平台实例或平台名称字符串
-            model_name: 使用的模型名称
+            llm_type: LLM类型，可以是 'normal' 或 'thinking'
             summary_prompt: 任务总结提示模板
             auto_complete: 是否自动完成任务
             output_handler: 输出处理器列表
@@ -130,22 +132,21 @@ class Agent:
         self.name = make_agent_name(name)
         self.description = description
         # 初始化平台和模型
-        if platform is not None:
-            if isinstance(platform, str):
-                self.model = PlatformRegistry().create_platform(platform)
-                if self.model is None:
-                    PrettyOutput.print(
-                        f"平台 {platform} 不存在，将使用普通模型", OutputType.WARNING
-                    )
-                    self.model = PlatformRegistry().get_normal_platform()
-            else:
-                self.model = platform
-        else:
-            self.model = (
-                PlatformRegistry.get_global_platform_registry().get_normal_platform()
+        if llm_type == "thinking":
+            platform_name = get_thinking_platform_name()
+            model_name = get_thinking_model_name()
+        else:  # 默认为 normal
+            platform_name = get_normal_platform_name()
+            model_name = get_normal_model_name()
+
+        self.model = PlatformRegistry().create_platform(platform_name)
+        if self.model is None:
+            PrettyOutput.print(
+                f"平台 {platform_name} 不存在，将使用普通模型", OutputType.WARNING
             )
+            self.model = PlatformRegistry().get_normal_platform()
 
-        if model_name is not None:
+        if model_name:
             self.model.set_model_name(model_name)
 
         self.user_data: Dict[str, Any] = {}
@@ -198,7 +199,7 @@ class Agent:
 
         PrettyOutput.print(welcome_message, OutputType.SYSTEM)
 
-        action_prompt = build_action_prompt(self.output_handler) # type: ignore
+        action_prompt = build_action_prompt(self.output_handler)  # type: ignore
 
         self.model.set_system_prompt(
             f"""
@@ -245,7 +246,10 @@ class Agent:
 
     def restore_session(self) -> bool:
         """Restores the session state by delegating to the session manager."""
-        return self.session.restore_session()
+        if self.session.restore_session():
+            self.first = False
+            return True
+        return False
 
     def get_tool_registry(self) -> Optional[Any]:
         """获取工具注册表实例"""

{jarvis_ai_assistant-0.1.221 → jarvis_ai_assistant-0.1.223}/src/jarvis/jarvis_agent/builtin_input_handler.py

@@ -39,12 +39,9 @@ def builtin_input_handler(user_input: str, agent_: Any) -> Tuple[str, bool]:
             return "", True
         elif tag == "ToolUsage":
             from jarvis.jarvis_tools.registry import ToolRegistry
-
-            tool_registry_ = agent.get_tool_registry()
-            tool_registry: ToolRegistry = (
-                tool_registry_ if tool_registry_ else ToolRegistry()
-            )
-            agent.set_addon_prompt(tool_registry.prompt())
+            from jarvis.jarvis_agent.prompt_builder import build_action_prompt
+            action_prompt = build_action_prompt(agent.output_handler) # type: ignore
+            agent.set_addon_prompt(action_prompt)
             return "", False
         elif tag == "ReloadConfig":
             from jarvis.jarvis_utils.utils import load_config

{jarvis_ai_assistant-0.1.221 → jarvis_ai_assistant-0.1.223}/src/jarvis/jarvis_agent/jarvis.py

@@ -109,8 +109,13 @@ def _select_task(tasks: Dict[str, str]) -> str:
 def main() -> None:
 
     parser = argparse.ArgumentParser(description="Jarvis AI assistant")
-    parser.add_argument("-p", "--platform", type=str, help="Platform to use")
-    parser.add_argument("-m", "--model", type=str, help="Model to use")
+    parser.add_argument(
+        "--llm_type",
+        type=str,
+        default="normal",
+        choices=["normal", "thinking"],
+        help="LLM type to use",
+    )
     parser.add_argument(
         "-t",
         "--task",
@@ -132,8 +137,7 @@ def main() -> None:
     try:
         agent = Agent(
             system_prompt=origin_agent_system_prompt,
-            platform=args.platform,
-            model_name=args.model,
+            llm_type=args.llm_type,
             input_handler=[shell_input_handler, builtin_input_handler],
             output_handler=[ToolRegistry()],
             need_summary=False,
@@ -151,11 +155,12 @@ def main() -> None:
             agent.run(args.task)
             sys.exit(0)
 
-        tasks = _load_tasks()
-        if tasks and (selected_task := _select_task(tasks)):
-            PrettyOutput.print(f"开始执行任务: \n{selected_task}", OutputType.INFO)
-            agent.run(selected_task)
-            sys.exit(0)
+        if agent.first:
+            tasks = _load_tasks()
+            if tasks and (selected_task := _select_task(tasks)):
+                PrettyOutput.print(f"开始执行任务: \n{selected_task}", OutputType.INFO)
+                agent.run(selected_task)
+                sys.exit(0)
 
         user_input = get_multiline_input("请输入你的任务（输入空行退出）:")
         if user_input: