jarvis-ai-assistant 0.1.190__tar.gz → 0.1.191__tar.gz

{jarvis_ai_assistant-0.1.190/src/jarvis_ai_assistant.egg-info → jarvis_ai_assistant-0.1.191}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jarvis-ai-assistant
-Version: 0.1.190
+Version: 0.1.191
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire
@@ -230,6 +230,54 @@ OPENAI_API_BASE: https://api.openai.com/v1  # 可选，默认为官方API地址
 | `jarvis-git-details` | - | 使用git details功能 |
 | `jarvis-methodology` | - | 使用方法论功能 |
 
+## 🏗️ 平台管理功能
+
+`jarvis-platform-manager` 提供以下子命令来管理AI平台和模型：
+
+### 1. 列出支持的平台和模型
+```bash
+jarvis-platform-manager info
+```
+显示所有支持的AI平台及其可用模型列表。
+
+### 2. 与指定平台和模型聊天
+```bash
+jarvis-platform-manager chat -p <平台名称> -m <模型名称>
+```
+启动交互式聊天会话，支持以下命令：
+- `/bye` - 退出聊天
+- `/clear` - 清除当前会话
+- `/upload <文件路径>` - 上传文件到当前会话
+- `/shell <命令>` - 执行shell命令
+- `/save <文件名>` - 保存最后一条消息
+- `/saveall <文件名>` - 保存完整对话历史
+
+### 3. 启动OpenAI兼容的API服务
+```bash
+jarvis-platform-manager service --host <IP地址> --port <端口号> -p <平台名称> -m <模型名称>
+```
+启动一个兼容OpenAI API的服务，可用于其他应用程序集成。
+
+### 4. 加载角色配置文件
+```bash
+jarvis-platform-manager role -c <配置文件路径>
+```
+从YAML配置文件加载预定义角色进行对话。配置文件格式示例：
+```yaml
+roles:
+  - name: "代码助手"
+    description: "专注于代码分析和生成的AI助手"
+    platform: "yuanbao"
+    model: "deep_seek_v3"
+    system_prompt: "你是一个专业的代码助手，专注于分析和生成高质量的代码"
+  - name: "文档撰写"
+    description: "帮助撰写技术文档的AI助手"
+    platform: "kimi"
+    model: "k1"
+    system_prompt: "你是一个技术文档撰写专家，擅长将复杂技术概念转化为清晰易懂的文字"
+```
+
+
 ---
 
 ## ⚙️ 配置说明 <a id="configuration"></a>
@@ -428,7 +476,7 @@ class CustomPlatform(BasePlatform):
         # 设置模型名称
         pass
 
-    def set_system_message(self, message: str):
+    def set_system_prompt(self, message: str):
         # 设置系统消息
         pass
 

{jarvis_ai_assistant-0.1.190 → jarvis_ai_assistant-0.1.191}/README.md

@@ -160,6 +160,54 @@ OPENAI_API_BASE: https://api.openai.com/v1  # 可选，默认为官方API地址
 | `jarvis-git-details` | - | 使用git details功能 |
 | `jarvis-methodology` | - | 使用方法论功能 |
 
+## 🏗️ 平台管理功能
+
+`jarvis-platform-manager` 提供以下子命令来管理AI平台和模型：
+
+### 1. 列出支持的平台和模型
+```bash
+jarvis-platform-manager info
+```
+显示所有支持的AI平台及其可用模型列表。
+
+### 2. 与指定平台和模型聊天
+```bash
+jarvis-platform-manager chat -p <平台名称> -m <模型名称>
+```
+启动交互式聊天会话，支持以下命令：
+- `/bye` - 退出聊天
+- `/clear` - 清除当前会话
+- `/upload <文件路径>` - 上传文件到当前会话
+- `/shell <命令>` - 执行shell命令
+- `/save <文件名>` - 保存最后一条消息
+- `/saveall <文件名>` - 保存完整对话历史
+
+### 3. 启动OpenAI兼容的API服务
+```bash
+jarvis-platform-manager service --host <IP地址> --port <端口号> -p <平台名称> -m <模型名称>
+```
+启动一个兼容OpenAI API的服务，可用于其他应用程序集成。
+
+### 4. 加载角色配置文件
+```bash
+jarvis-platform-manager role -c <配置文件路径>
+```
+从YAML配置文件加载预定义角色进行对话。配置文件格式示例：
+```yaml
+roles:
+  - name: "代码助手"
+    description: "专注于代码分析和生成的AI助手"
+    platform: "yuanbao"
+    model: "deep_seek_v3"
+    system_prompt: "你是一个专业的代码助手，专注于分析和生成高质量的代码"
+  - name: "文档撰写"
+    description: "帮助撰写技术文档的AI助手"
+    platform: "kimi"
+    model: "k1"
+    system_prompt: "你是一个技术文档撰写专家，擅长将复杂技术概念转化为清晰易懂的文字"
+```
+
+
 ---
 
 ## ⚙️ 配置说明 <a id="configuration"></a>
@@ -358,7 +406,7 @@ class CustomPlatform(BasePlatform):
         # 设置模型名称
         pass
 
-    def set_system_message(self, message: str):
+    def set_system_prompt(self, message: str):
         # 设置系统消息
         pass
 

{jarvis_ai_assistant-0.1.190 → jarvis_ai_assistant-0.1.191}/pyproject.toml

@@ -8,7 +8,7 @@ default = true
 
 [project]
 name = "jarvis-ai-assistant"
-version = "0.1.190"
+version = "0.1.191"
 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.190 → jarvis_ai_assistant-0.1.191}/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="jarvis-ai-assistant",
-    version="0.1.190",
+    version="0.1.191",
     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.190 → jarvis_ai_assistant-0.1.191}/src/jarvis/__init__.py

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

{jarvis_ai_assistant-0.1.190 → jarvis_ai_assistant-0.1.191}/src/jarvis/jarvis_agent/__init__.py

@@ -19,8 +19,8 @@ from jarvis.jarvis_utils.config import (get_max_token_count,
                                         is_execute_tool_confirm,
                                         is_use_analysis, is_use_methodology)
 from jarvis.jarvis_utils.embedding import get_context_token_count
-from jarvis.jarvis_utils.globals import (delete_agent, make_agent_name,
-                                         set_agent)
+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)
@@ -319,7 +319,7 @@ class Agent:
 </actions>
 """
 
-        self.model.set_system_message(
+        self.model.set_system_prompt(
             f"""
 {self.system_prompt}
 
@@ -371,40 +371,6 @@ class Agent:
                 return handler
         return None
 
-    def make_default_addon_prompt(self, need_complete: bool) -> str:
-        """生成附加提示。
-
-        参数:
-            need_complete: 是否需要完成任务
-
-        """
-        # 结构化系统指令
-        action_handlers = ", ".join([handler.name() for handler in self.output_handler])
-
-        # 任务完成提示
-        complete_prompt = (
-            f"- 输出{ot('!!!COMPLETE!!!')}"
-            if need_complete and self.auto_complete
-            else ""
-        )
-
-        addon_prompt = f"""
-<system_prompt>
-    请判断是否已经完成任务，如果已经完成：
-    - 直接输出完成原因，不需要再有新的操作，不要输出{ot("TOOL_CALL")}标签
-    {complete_prompt}
-    如果没有完成，请进行下一步操作：
-    - 仅包含一个操作
-    - 如果信息不明确，请请求用户补充
-    - 如果执行过程中连续失败5次，请使用ask_user询问用户操作
-    - 操作列表：{action_handlers}
-</system_prompt>
-
-请继续。
-"""
-
-        return addon_prompt
-
     def _call_model(self, message: str, need_complete: bool = False) -> str:
         """调用AI模型并实现重试逻辑
 
@@ -429,8 +395,6 @@ class Agent:
         if self.addon_prompt:
             message += f"\n\n{self.addon_prompt}"
             self.addon_prompt = ""
-        else:
-            message += f"\n\n{self.make_default_addon_prompt(need_complete)}"
 
         # 累加对话长度
         self.conversation_length += get_context_token_count(message)
@@ -812,6 +776,14 @@ arguments:
                     if self.after_tool_call_cb:
                         self.after_tool_call_cb(self)
 
+                    if get_interrupt():
+                        set_interrupt(False)
+                        user_input = self.multiline_inputer(
+                            f"模型交互期间被中断，请输入用户干预信息："
+                        )
+                        if user_input:
+                            self.prompt += f"\n\n用户干预信息：{user_input}"
+
                     if self.prompt or self.addon_prompt:
                         continue
 

{jarvis_ai_assistant-0.1.190 → jarvis_ai_assistant-0.1.191}/src/jarvis/jarvis_agent/jarvis.py

@@ -126,8 +126,6 @@ def main() -> None:
             need_summary=False
         )
 
-        agent.set_addon_prompt(agent.make_default_addon_prompt(False) + "\n如果使用脚本可以完成任务，优先execute_script工具")
-
         # 优先处理命令行直接传入的任务
         if args.task:
             agent.run(args.task)

{jarvis_ai_assistant-0.1.190 → jarvis_ai_assistant-0.1.191}/src/jarvis/jarvis_platform/base.py

@@ -12,6 +12,7 @@ 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.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,
@@ -128,10 +129,14 @@ class BasePlatform(ABC):
     
     def chat_until_success(self, message: str) -> str:
         """Chat with model until successful response"""
-        if not self.suppress_output and is_print_prompt():
-            PrettyOutput.print(f"{message}", OutputType.USER)
-        result: str = while_true(lambda: while_success(lambda: self._chat(message), 5), 5)
-        return result
+        try:
+            set_in_chat(True)
+            if not self.suppress_output and is_print_prompt():
+                PrettyOutput.print(f"{message}", OutputType.USER)
+            result: str = while_true(lambda: while_success(lambda: self._chat(message), 5), 5)
+            return result
+        finally:
+            set_in_chat(False)
 
     @abstractmethod
     def name(self) -> str:
@@ -144,9 +149,9 @@ class BasePlatform(ABC):
         raise NotImplementedError("delete_chat is not implemented")
 
     @abstractmethod
-    def set_system_message(self, message: str):
+    def set_system_prompt(self, message: str):
         """Set system message"""
-        raise NotImplementedError("set_system_message is not implemented")
+        raise NotImplementedError("set_system_prompt is not implemented")
 
     @abstractmethod
     def get_model_list(self) -> List[Tuple[str, str]]:

{jarvis_ai_assistant-0.1.190 → jarvis_ai_assistant-0.1.191}/src/jarvis/jarvis_platform/human.py

@@ -25,7 +25,7 @@ class HumanPlatform(BasePlatform):
         self.system_message = ""  # 系统消息，用于初始化对话
         self.first_message = True
 
-    def set_system_message(self, message: str):
+    def set_system_prompt(self, message: str):
         """设置系统消息"""
         self.system_message = message
 

{jarvis_ai_assistant-0.1.190 → jarvis_ai_assistant-0.1.191}/src/jarvis/jarvis_platform/kimi.py

@@ -41,7 +41,7 @@ class KimiModel(BasePlatform):
         self.system_message = ""
         self.model_name = "kimi"
 
-    def set_system_message(self, message: str):
+    def set_system_prompt(self, message: str):
         """Set system message"""
         self.system_message = message
 

{jarvis_ai_assistant-0.1.190 → jarvis_ai_assistant-0.1.191}/src/jarvis/jarvis_platform/openai.py

@@ -74,7 +74,7 @@ class OpenAIModel(BasePlatform):
 
         self.model_name = model_name
 
-    def set_system_message(self, message: str):
+    def set_system_prompt(self, message: str):
         """
         设置系统消息(角色设定)
         

{jarvis_ai_assistant-0.1.190 → jarvis_ai_assistant-0.1.191}/src/jarvis/jarvis_platform/registry.py

@@ -16,7 +16,7 @@ REQUIRED_METHODS = [
     ('chat', ['message']),           # 方法名和参数列表
     ('name', []),
     ('delete_chat', []),
-    ('set_system_message', ['message']),
+    ('set_system_prompt', ['message']),
     ('set_model_name', ['model_name']),
     ('get_model_list', []),
     ('upload_files', ['file_list']),

{jarvis_ai_assistant-0.1.190 → jarvis_ai_assistant-0.1.191}/src/jarvis/jarvis_platform/tongyi.py

@@ -493,7 +493,7 @@ class TongyiPlatform(BasePlatform):
             PrettyOutput.print(f"Error deleting chat: {str(e)}", OutputType.ERROR)
             return False
 
-    def set_system_message(self, message: str):
+    def set_system_prompt(self, message: str):
         """Set system message
 
         Args:

{jarvis_ai_assistant-0.1.190 → jarvis_ai_assistant-0.1.191}/src/jarvis/jarvis_platform/yuanbao.py

@@ -45,7 +45,7 @@ class YuanbaoPlatform(BasePlatform):
         self.model_name = "deep_seek_v3"  # 默认模型名称，使用下划线保持一致
         self.multimedia = []
 
-    def set_system_message(self, message: str):
+    def set_system_prompt(self, message: str):
         """设置系统消息"""
         self.system_message = message