jarvis-ai-assistant 0.1.150__py3-none-any.whl → 0.1.152__py3-none-any.whl

jarvis/jarvis_mcp/{local_mcp_client.py → stdio_mcp_client.py}

@@ -6,7 +6,7 @@ from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from . import McpClient
 
 
-class LocalMcpClient(McpClient):
+class StdioMcpClient(McpClient):
     """本地MCP客户端实现
     
     参数:
@@ -228,6 +228,91 @@ class LocalMcpClient(McpClient):
                 'stderr': str(e)
             }
 
+    def get_resource_list(self) -> List[Dict[str, Any]]:
+        """获取资源列表
+        
+        返回:
+            List[Dict[str, Any]]: 资源列表，每个资源包含以下字段：
+                - uri: str - 资源的唯一标识符
+                - name: str - 资源的名称
+                - description: str - 资源的描述（可选）
+                - mimeType: str - 资源的MIME类型（可选）
+        """
+        try:
+            response = self._send_request('resources/list', {})
+            if 'result' in response and 'resources' in response['result']:
+                return response['result']['resources']
+            else:
+                error_msg = "获取资源列表失败"
+                if 'error' in response:
+                    error_msg += f": {response['error']}"
+                else:
+                    error_msg += ": 未知错误"
+                PrettyOutput.print(error_msg, OutputType.ERROR)
+                return []
+        except Exception as e:
+            PrettyOutput.print(f"获取资源列表失败: {str(e)}", OutputType.ERROR)
+            return []
+
+    def get_resource(self, uri: str) -> Dict[str, Any]:
+        """获取指定资源的内容
+        
+        参数:
+            uri: str - 资源的URI标识符
+            
+        返回:
+            Dict[str, Any]: 执行结果，包含以下字段：
+                - success: bool - 是否执行成功
+                - stdout: str - 资源内容（文本或base64编码的二进制内容）
+                - stderr: str - 错误信息
+        """
+        try:
+            response = self._send_request('resources/read', {
+                'uri': uri
+            })
+            if 'result' in response and 'contents' in response['result']:
+                contents = response['result']['contents']
+                if contents:
+                    content = contents[0]  # 获取第一个资源内容
+                    # 根据资源类型返回内容
+                    if 'text' in content:
+                        return {
+                            'success': True,
+                            'stdout': content['text'],
+                            'stderr': ''
+                        }
+                    elif 'blob' in content:
+                        return {
+                            'success': True,
+                            'stdout': content['blob'],
+                            'stderr': ''
+                        }
+                return {
+                    'success': False,
+                    'stdout': '',
+                    'stderr': '资源内容为空'
+                }
+            else:
+                error_msg = "获取资源内容失败"
+                if 'error' in response:
+                    error_msg += f": {response['error']}"
+                else:
+                    error_msg += ": 未知错误"
+                PrettyOutput.print(error_msg, OutputType.ERROR)
+                return {
+                    'success': False,
+                    'stdout': '',
+                    'stderr': error_msg
+                }
+        except Exception as e:
+            error_msg = f"获取资源内容失败: {str(e)}"
+            PrettyOutput.print(error_msg, OutputType.ERROR)
+            return {
+                'success': False,
+                'stdout': '',
+                'stderr': error_msg
+            }
+
     def __del__(self):
         """清理资源"""
         if self.process:

jarvis/jarvis_tools/ask_codebase.py

@@ -156,10 +156,9 @@ class AskCodebaseTool:
 2. 使用fd命令查找可能相关的文件
 3. 使用rg命令搜索关键词和代码模式
 4. 使用read_code工具直接读取和分析相关文件内容
-5. 只有在fd、rg和read_code都无法解决问题时才考虑使用RAG工具
-6. 根据文件内容提供具体、准确的回答
-7. 确保分析的完整性，收集充分的信息后再得出结论，不要在只掌握部分信息就得出结论
-8. 优先查阅README文件、文档目录和项目文档
+5. 根据文件内容提供具体、准确的回答
+6. 确保分析的完整性，收集充分的信息后再得出结论，不要在只掌握部分信息就得出结论
+7. 优先查阅README文件、文档目录和项目文档
 
 ## 分析步骤
 1. **确定项目的编程语言**:
@@ -186,12 +185,6 @@ class AskCodebaseTool:
    - 提供基于直接分析代码的具体回答
    - 引用具体文件和代码片段作为依据
 
-## 关于RAG工具使用
-- RAG工具应作为最后选择，仅在fd、rg和read_code都无法解决问题时使用
-- 必须通过查看实际代码文件验证RAG返回的每条重要信息
-- 对于关键发现，始终使用`read_code`工具查看原始文件内容进行求证
-- 如发现RAG结果与实际代码不符，以实际代码为准
-
 ## 输出要求
 - 提供准确、具体的回答，避免模糊不清的描述
 - 引用具体文件路径和代码片段作为依据