jarvis-ai-assistant 0.1.131__py3-none-any.whl → 0.1.134__py3-none-any.whl

jarvis/jarvis_git_squash/main.py

@@ -1,3 +1,4 @@
+import re
 import sys
 import argparse
 from typing import Dict, Any
@@ -26,39 +27,23 @@ class GitSquashTool:
         except Exception:
             return False
     
-    def execute(self, args: Dict) -> Dict[str, Any]:
+    def execute(self, args: Dict):
         """Execute the squash operation"""
         try:
             if not self._confirm_squash():
-                return {
-                    "success": False,
-                    "stdout": "Operation cancelled",
-                    "stderr": ""
-                }
+                PrettyOutput.print("操作已取消", OutputType.WARNING)
+                return
             
             if not self._reset_to_commit(args['commit_hash']):
-                return {
-                    "success": False,
-                    "stdout": "",
-                    "stderr": "Failed to reset to specified commit"
-                }
+                PrettyOutput.print("重置到指定提交失败", OutputType.WARNING)
+                return
             
             # Use existing GitCommitTool for new commit
             commit_tool = GitCommitTool()
-            result = commit_tool.execute({"lang": args.get('lang', 'Chinese')})
-            
-            return {
-                "success": result['success'],
-                "stdout": result['stdout'],
-                "stderr": result['stderr']
-            }
-            
+            commit_tool.execute({"lang": args.get('lang', 'Chinese')})
         except Exception as e:
-            return {
-                "success": False,
-                "stdout": "",
-                "stderr": f"Squash failed: {str(e)}"
-            }
+            PrettyOutput.print(f"压缩提交失败: {str(e)}", OutputType.WARNING)
+            
 def main():
     init_env()
     parser = argparse.ArgumentParser(description='Git squash tool')
@@ -67,15 +52,9 @@ def main():
     args = parser.parse_args()
     
     tool = GitSquashTool()
-    result = tool.execute({
+    tool.execute({
         'commit_hash': args.commit_hash,
         'lang': args.lang
     })
-    
-    if not result['success']:
-        PrettyOutput.print(result['stderr'], OutputType.ERROR)
-        sys.exit(1)
-        
-    PrettyOutput.print(result['stdout'], OutputType.SUCCESS)
 if __name__ == "__main__":
     sys.exit(main())

jarvis/jarvis_lsp/base.py

@@ -25,48 +25,6 @@ class BaseLSP(ABC):
         """
         return False
     
-    @abstractmethod
-    def find_references(self, file_path: str, position: Tuple[int, int]) -> List[Dict[str, Any]]:
-        """Find all references of symbol at position.
-        
-        Args:
-            file_path: Path to the file
-            position: (line, character) tuple
-            
-        Returns:
-            List of references with location info:
-            [
-                {
-                    "uri": "file path",
-                    "range": {
-                        "start": {"line": int, "character": int},
-                        "end": {"line": int, "character": int}
-                    }
-                }
-            ]
-        """
-        return []
-    
-    @abstractmethod
-    def find_definition(self, file_path: str, position: Tuple[int, int]) -> Optional[Dict[str, Any]]:
-        """Find definition of symbol at position.
-        
-        Args:
-            file_path: Path to the file
-            position: (line, character) tuple
-            
-        Returns:
-            Location of definition:
-            {
-                "uri": "file path",
-                "range": {
-                    "start": {"line": int, "character": int},
-                    "end": {"line": int, "character": int}
-                }
-            }
-        """
-        return None
-    
     
     @abstractmethod
     def get_diagnostics(self, file_path: str) -> List[Dict[str, Any]]:

jarvis/jarvis_lsp/cpp.py

@@ -66,21 +66,6 @@ class CPPLSP(BaseLSP):
         except Exception:
             return None
     
-    def find_references(self, file_path: str, position: Tuple[int, int]) -> List[Dict[str, Any]]:
-        result = self._send_request("textDocument/references", {
-            "textDocument": {"uri": f"file://{file_path}"},
-            "position": {"line": position[0], "character": position[1]},
-            "context": {"includeDeclaration": True}
-        })
-        return result or [] # type: ignore
-    
-    def find_definition(self, file_path: str, position: Tuple[int, int]) -> Optional[Dict[str, Any]]:
-        result = self._send_request("textDocument/definition", {
-            "textDocument": {"uri": f"file://{file_path}"},
-            "position": {"line": position[0], "character": position[1]}
-        })
-        return result[0] if result else None
-
     
     def get_diagnostics(self, file_path: str) -> List[Dict[str, Any]]:
         # Send didOpen notification to trigger diagnostics

jarvis/jarvis_lsp/go.py

@@ -72,21 +72,6 @@ class GoLSP(BaseLSP):
         except Exception:
             return None
     
-    def find_references(self, file_path: str, position: Tuple[int, int]) -> List[Dict[str, Any]]:
-        result = self._send_request("textDocument/references", {
-            "textDocument": {"uri": f"file://{file_path}"},
-            "position": {"line": position[0], "character": position[1]},
-            "context": {"includeDeclaration": True}
-        })
-        return result or [] # type: ignore
-    
-    def find_definition(self, file_path: str, position: Tuple[int, int]) -> Optional[Dict[str, Any]]:
-        result = self._send_request("textDocument/definition", {
-            "textDocument": {"uri": f"file://{file_path}"},
-            "position": {"line": position[0], "character": position[1]}
-        })
-        return result[0] if result else None
-    
     def get_diagnostics(self, file_path: str) -> List[Dict[str, Any]]:
         # Send didOpen notification to trigger diagnostics
         self._send_request("textDocument/didOpen", {

jarvis/jarvis_lsp/python.py

@@ -25,25 +25,6 @@ class PythonLSP(BaseLSP):
                 return None
         return self.script_cache[file_path]
     
-    def find_references(self, file_path: str, position: Tuple[int, int]) -> List[Dict[str, Any]]:
-        script = self._get_script(file_path)
-        if not script:
-            return []
-        try:
-            refs = script.get_references(line=position[0] + 1, column=position[1])
-            return [self._location_to_dict(ref) for ref in refs]
-        except Exception:
-            return []
-    
-    def find_definition(self, file_path: str, position: Tuple[int, int]) -> Optional[Dict[str, Any]]:
-        script = self._get_script(file_path)
-        if not script:
-            return None
-        try:
-            defs = script.goto(line=position[0] + 1, column=position[1])
-            return self._location_to_dict(defs[0]) if defs else None
-        except Exception:
-            return None
     
     def _location_to_dict(self, location) -> Dict[str, Any]:
         return {

jarvis/jarvis_lsp/registry.py

@@ -9,8 +9,6 @@ from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
 REQUIRED_METHODS = [
     ('initialize', ['workspace_path']),
-    ('find_references', ['file_path', 'position']),
-    ('find_definition', ['file_path', 'position']),
     ('get_diagnostics', ['file_path']),
     ('shutdown', [])
 ]
@@ -167,63 +165,3 @@ class LSPRegistry:
         with open(file_path, 'r', errors="ignore") as file:
             lines = file.readlines()
             return lines[line]
-
-def main():
-    """CLI entry point for LSP testing."""
-    import argparse
-    
-    parser = argparse.ArgumentParser(description='LSP functionality testing')
-    parser.add_argument('--language', type=str, required=True, help='Programming language')
-    parser.add_argument('--file', type=str, required=True, help='File to analyze')
-    parser.add_argument('--action', choices=['symbols', 'diagnostics', 'references', 'definition'],
-                       required=True, help='Action to perform')
-    parser.add_argument('--line', type=int, help='Line number (0-based) for references/definition')
-    parser.add_argument('--character', type=int, help='Character position for references/definition')
-    
-    args = parser.parse_args()
-    
-    # Initialize LSP
-    registry = LSPRegistry.get_global_lsp_registry()
-    lsp = registry.create_lsp(args.language)
-    
-    if not lsp:
-        PrettyOutput.print(f"没有 LSP 支持的语言: {args.language}", OutputType.WARNING)
-        return 1
-        
-    if not lsp.initialize(os.path.abspath(os.getcwd())):
-        PrettyOutput.print("LSP 初始化失败", OutputType.WARNING)
-        return 1
-    
-    try:        
-        if args.action == 'diagnostics':
-            diagnostics = lsp.get_diagnostics(args.file)
-            for diag in diagnostics:
-                severity = ['Error', 'Warning', 'Info', 'Hint'][diag['severity'] - 1]
-                PrettyOutput.print(f"{severity} 在 {diag['range']['start']['line']}:{diag['range']['start']['character']}: {diag['message']}", OutputType.INFO)
-                
-        elif args.action in ('references', 'definition'):
-            if args.line is None or args.character is None:
-                PrettyOutput.print("需要行和字符位置用于 references/definition", OutputType.WARNING)
-                return 1
-                
-            if args.action == 'references':
-                refs = lsp.find_references(args.file, (args.line, args.character))
-                for ref in refs:
-                    PrettyOutput.print(f"引用在 {ref['uri']} 在 {ref['range']['start']['line']}:{ref['range']['start']['character']}\n行: {LSPRegistry.get_line_at_position(ref['uri'], ref['range']['start']['line'])}", OutputType.INFO)
-            else:
-                defn = lsp.find_definition(args.file, (args.line, args.character))
-                if defn:
-                    PrettyOutput.print(f"定义在 {defn['uri']} 在 {defn['range']['start']['line']}:{defn['range']['start']['character']}\n行: {LSPRegistry.get_line_at_position(defn['uri'], defn['range']['start']['line'])}", OutputType.INFO)
-                else:
-                    PrettyOutput.print("没有找到定义", OutputType.WARNING)
-                    
-    except Exception as e:
-        PrettyOutput.print(f"错误: {str(e)}", OutputType.ERROR)
-        return 1
-    finally:
-        lsp.shutdown()
-    
-    return 0
-
-if __name__ == "__main__":
-    exit(main())

jarvis/jarvis_lsp/rust.py

@@ -74,21 +74,6 @@ class RustLSP(BaseLSP):
         except Exception:
             return None
     
-    def find_references(self, file_path: str, position: Tuple[int, int]) -> List[Dict[str, Any]]:
-        result = self._send_request("textDocument/references", {
-            "textDocument": {"uri": f"file://{file_path}"},
-            "position": {"line": position[0], "character": position[1]},
-            "context": {"includeDeclaration": True}
-        })
-        return result or [] # type: ignore
-    
-    def find_definition(self, file_path: str, position: Tuple[int, int]) -> Optional[Dict[str, Any]]:
-        result = self._send_request("textDocument/definition", {
-            "textDocument": {"uri": f"file://{file_path}"},
-            "position": {"line": position[0], "character": position[1]}
-        })
-        return result[0] if result else None
-    
     
     def get_diagnostics(self, file_path: str) -> List[Dict[str, Any]]:
         # Send didOpen notification to trigger diagnostics

jarvis/jarvis_multi_agent/__init__.py

@@ -8,7 +8,7 @@ from jarvis.jarvis_agent.output_handler import OutputHandler
 from jarvis.jarvis_tools.registry import ToolRegistry
 from jarvis.jarvis_utils.input import get_multiline_input
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
-from jarvis.jarvis_utils.utils import init_env
+from jarvis.jarvis_utils.utils import ct, ot, init_env
 
 
 class MultiAgent(OutputHandler):
@@ -20,12 +20,7 @@ class MultiAgent(OutputHandler):
 
     def prompt(self) -> str:
         return f"""
-# 多智能体协作系统
-
-## 身份与角色定位
-- **核心职责**：作为多智能体系统的协调者，通过结构化消息实现高效协作
-- **关键能力**：消息路由、任务分发、结果整合、流程协调
-- **工作范围**：在多个专业智能体之间建立有效沟通渠道
+# 多智能体消息发送
 
 ## 交互原则与策略
 ### 消息处理规范
@@ -36,7 +31,7 @@ class MultiAgent(OutputHandler):
 
 ### 消息格式标准
 ```
-<SEND_MESSAGE>
+{ot("SEND_MESSAGE")}
 to: 智能体名称    # 目标智能体名称
 content: |
     # 消息主题
@@ -52,31 +47,27 @@ content: |
     
     ## 期望结果
     [描述期望的输出格式和内容]
-</SEND_MESSAGE>
+
+    ## 下一步计划
+    [描述下一步的计划和行动]
+{ct("SEND_MESSAGE")}
 ```
 
-## 协作流程规范
-### 任务分发流程
-1. **需求分析**：理解用户需求并确定最适合的智能体
-2. **任务分解**：将复杂任务分解为可管理的子任务
-3. **精准分发**：根据专长将任务分配给合适的智能体
-4. **结果整合**：收集各智能体的输出并整合为连贯结果
+或者：
 
-### 消息流控制
-1. **单向流动**：发送消息后等待响应，避免消息风暴
-2. **优先级管理**：处理紧急消息优先，保持任务顺序
-3. **状态跟踪**：记录每个任务的当前状态和处理进度
-4. **异常处理**：优雅处理超时、错误和意外响应
+```
+{ot("SEND_MESSAGE")}
+to: 智能体名称    # 目标智能体名称
+content: |
+    # 消息主题
+    
+    ## 任务结果
+    [任务完成结果，用于反馈]
+{ct("SEND_MESSAGE")}
+```
 
 ## 可用智能体资源
 {chr(10).join([f"- **{c['name']}**: {c.get('description', '')}" for c in self.agents_config])}
-
-## 最佳实践指南
-1. **任务明确化**：每个消息专注于单一、明确的任务
-2. **信息充分性**：提供足够信息让接收者能独立完成任务
-3. **反馈循环**：建立清晰的反馈机制，及时调整方向
-4. **知识共享**：确保关键信息在相关智能体间共享
-5. **协作效率**：避免不必要的消息传递，减少协调开销
 """
 
     def can_handle(self, response: str) -> bool:
@@ -102,7 +93,7 @@ content: |
         Args:
             content: The content containing send message
         """
-        data = re.findall(r'<SEND_MESSAGE>(.*?)</SEND_MESSAGE>', content, re.DOTALL)
+        data = re.findall(ot("SEND_MESSAGE")+r'\n(.*?)\n'+ct("SEND_MESSAGE"), content, re.DOTALL)
         ret = []
         for item in data:
             try:
@@ -147,45 +138,4 @@ content: {msg['content']}
                 last_agent = self.agents[msg['to']].name
                 msg = self.agents[msg['to']].run(prompt)
         return ""
-
-
-def main():
-    """从YAML配置文件初始化并运行多智能体系统
-    
-    Returns:
-        最终处理结果
-    """
-    init_env()
-    import argparse
-    parser = argparse.ArgumentParser(description="多智能体系统启动器")
-    parser.add_argument("--config", "-c", required=True, help="YAML配置文件路径")
-    parser.add_argument("--input", "-i", help="用户输入（可选）")
-    args = parser.parse_args()
-        
-    try:
-        with open(args.config, 'r', errors="ignore") as f:
-            config_data = yaml.safe_load(f)
-            
-        # 获取agents配置
-        agents_config = config_data.get('agents', [])
-        
-        main_agent_name = config_data.get('main_agent', '')
-        if not main_agent_name:
-            raise ValueError("必须指定main_agent作为主智能体")
-            
-        # 创建并运行多智能体系统
-        multi_agent = MultiAgent(agents_config, main_agent_name)
-        user_input = args.input if args.input is not None else get_multiline_input("请输入内容（输入空行结束）：")
-        if user_input == "":
-            return
-        return multi_agent.run(user_input)
-        
-    except yaml.YAMLError as e:
-        raise ValueError(f"YAML配置文件解析错误: {str(e)}")
-    except Exception as e:
-        raise RuntimeError(f"多智能体系统初始化失败: {str(e)}")
-
-
-if __name__ == "__main__":
-    result = main()
     

jarvis/jarvis_multi_agent/main.py

@@ -0,0 +1,43 @@
+import yaml
+from jarvis.jarvis_multi_agent import MultiAgent
+from jarvis.jarvis_utils.utils import init_env
+from jarvis.jarvis_utils.input import get_multiline_input
+
+def main():
+    """从YAML配置文件初始化并运行多智能体系统
+    
+    Returns:
+        最终处理结果
+    """
+    init_env()
+    import argparse
+    parser = argparse.ArgumentParser(description="多智能体系统启动器")
+    parser.add_argument("--config", "-c", required=True, help="YAML配置文件路径")
+    parser.add_argument("--input", "-i", help="用户输入（可选）")
+    args = parser.parse_args()
+        
+    try:
+        with open(args.config, 'r', errors="ignore") as f:
+            config_data = yaml.safe_load(f)
+            
+        # 获取agents配置
+        agents_config = config_data.get('agents', [])
+        
+        main_agent_name = config_data.get('main_agent', '')
+        if not main_agent_name:
+            raise ValueError("必须指定main_agent作为主智能体")
+            
+        # 创建并运行多智能体系统
+        multi_agent = MultiAgent(agents_config, main_agent_name)
+        user_input = args.input if args.input is not None else get_multiline_input("请输入内容（输入空行结束）：")
+        if user_input == "":
+            return
+        return multi_agent.run(user_input)
+        
+    except yaml.YAMLError as e:
+        raise ValueError(f"YAML配置文件解析错误: {str(e)}")
+    except Exception as e:
+        raise RuntimeError(f"多智能体系统初始化失败: {str(e)}")
+
+if __name__ == "__main__":
+    result = main()

jarvis/jarvis_platform/ai8.py

@@ -23,7 +23,6 @@ class AI8Model(BasePlatform):
         super().__init__()
         self.system_message = ""
         self.conversation = {}
-        self.files = []
         self.models = {}  # 存储模型信息
 
         self.token = os.getenv("AI8_API_KEY")
@@ -103,18 +102,6 @@ class AI8Model(BasePlatform):
         except Exception as e:
             PrettyOutput.print(f"创建会话失败: {str(e)}", OutputType.ERROR)
             return False
-        
-    def upload_files(self, file_list: List[str]) -> List[Dict]:
-        for file_path in file_list:
-            name = os.path.basename(file_path)
-            with open(file_path, 'rb') as f:
-                file_data = f.read()
-            base64_data = base64.b64encode(file_data).decode('utf-8')
-            self.files.append({
-                "name": name,
-                "data": f"data:image/png;base64,{base64_data}"
-            })
-        return self.files
     
     def set_system_message(self, message: str):
         """Set system message"""
@@ -145,14 +132,6 @@ class AI8Model(BasePlatform):
                 "files": []
             }
             
-            # 如果有文件需要发送
-            if self.files:
-                for file_data in self.files:
-                    payload["files"].append({
-                        "name": file_data["name"],
-                        "data": file_data["data"]
-                    })
-                self.files = []  # 清空已使用的文件
             
             response = requests.post(
                 f"{self.BASE_URL}/api/chat/completions",
@@ -200,7 +179,6 @@ class AI8Model(BasePlatform):
     def reset(self):
         """Reset model state"""
         self.conversation = None
-        self.files = []  # 清空文件列表
             
     def delete_chat(self) -> bool:
         """Delete current chat session"""
@@ -284,13 +262,6 @@ class AI8Model(BasePlatform):
                 # 添加标签
                 model_str = f"{model['label']}"
                 
-                # 添加标签和积分信息
-                attrs = []
-                if model['attr'].get('tag'):
-                    attrs.append(model['attr']['tag'])
-                if model['attr'].get('integral'):
-                    attrs.append(model['attr']['integral'])
-                    
                 # 添加特性标记
                 features = []
                 if model['attr'].get('multimodal'):
@@ -299,12 +270,16 @@ class AI8Model(BasePlatform):
                     features.append("Plugin support")
                 if model['attr'].get('onlyImg'):
                     features.append("Image support")
-                if features:
-                    model_str += f" [{'|'.join(features)}]"
-                    
+                if model['attr'].get('tag'):
+                    features.append(model['attr']['tag'])
+                if model['attr'].get('integral'):
+                    features.append(model['attr']['integral'])
                 # 添加备注
                 if model['attr'].get('note'):
                     model_str += f" - {model['attr']['note']}"
+                if features:
+                    model_str += f" [{'|'.join(features)}]"
+                    
                 model['desc'] = model_str
             
             return list(self.models.keys())

jarvis/jarvis_platform/base.py

@@ -2,7 +2,7 @@ from abc import ABC, abstractmethod
 import re
 from typing import Dict, List, Tuple
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
-from jarvis.jarvis_utils.utils import get_context_token_count, while_success, while_true
+from jarvis.jarvis_utils.utils import ct, ot, get_context_token_count, while_success, while_true
 
 
 class BasePlatform(ABC):
@@ -53,16 +53,11 @@ class BasePlatform(ABC):
                 )
 
             # Keep original think tag handling
-            response = re.sub(r'<think>.*?</think>', '', response, flags=re.DOTALL)
+            response = re.sub(ot("think")+r'.*?'+ct("think"), '', response, flags=re.DOTALL)
             return response
                 
         return while_true(lambda: while_success(lambda: _chat(), 5), 5)
 
-    @abstractmethod
-    def upload_files(self, file_list: List[str]) -> List[Dict]:
-        """Upload files"""
-        raise NotImplementedError("upload_files is not implemented")
-
     @abstractmethod
     def reset(self):
         """Reset model"""