jarvis-ai-assistant 0.1.3__py3-none-any.whl → 0.1.6__py3-none-any.whl

jarvis/tools/rag.py

@@ -1,154 +0,0 @@
-from typing import Dict, Any, List
-import chromadb
-from chromadb.utils import embedding_functions
-import hashlib
-import re
-from ..utils import PrettyOutput, OutputType
-
-class RAGTool:
-    name = "rag_query"
-    description = """Execute RAG queries on documents.
-    Features:
-    1. Auto-creates document embeddings
-    2. Returns relevant passages
-    3. Maintains embedding database
-    """
-    parameters = {
-        "type": "object",
-        "properties": {
-            "files": {
-                "type": "array",
-                "items": {"type": "string"},
-                "description": "Files to process"
-            },
-            "query": {
-                "type": "string",
-                "description": "Query text"
-            },
-            "num_passages": {
-                "type": "integer",
-                "description": "Number of passages",
-                "default": 3
-            },
-            "chunk_size": {
-                "type": "integer",
-                "description": "Chunk size",
-                "default": 500
-            }
-        },
-        "required": ["files", "query"]
-    }
-
-    def _get_document_hash(self, content: str) -> str:
-        return hashlib.md5(content.encode()).hexdigest()
-
-    def _chunk_text(self, text: str, chunk_size: int = 500) -> List[str]:
-        """将文本分割成适当大小的块"""
-        # 按句子分割
-        sentences = re.split(r'(?<=[.!?])\s+', text)
-        chunks = []
-        current_chunk = []
-        current_length = 0
-        
-        for sentence in sentences:
-            sentence_length = len(sentence)
-            if current_length + sentence_length > chunk_size and current_chunk:
-                chunks.append(" ".join(current_chunk))
-                current_chunk = []
-                current_length = 0
-            current_chunk.append(sentence)
-            current_length += sentence_length
-            
-        if current_chunk:
-            chunks.append(" ".join(current_chunk))
-            
-        return chunks
-
-    def execute(self, args: Dict) -> Dict[str, Any]:
-        """执行RAG查询"""
-        try:
-            files = args["files"]
-            query = args["query"]
-            num_passages = args.get("num_passages", 3)
-            chunk_size = args.get("chunk_size", 500)
-            
-            # 初始化ChromaDB
-            chroma_client = chromadb.PersistentClient(path="./data/chromadb")
-            
-            # 使用sentence-transformers作为嵌入模型
-            embedding_function = embedding_functions.SentenceTransformerEmbeddingFunction(
-                model_name="all-MiniLM-L6-v2"
-            )
-            
-            # 获取或创建集合
-            collection = chroma_client.get_or_create_collection(
-                name="document_store",
-                embedding_function=embedding_function
-            )
-            
-            # 处理每个文件
-            for file_path in files:
-                try:
-                    with open(file_path, 'r', encoding='utf-8') as f:
-                        content = f.read()
-                    
-                    # 计算文档哈希值
-                    doc_hash = self._get_document_hash(content)
-                    
-                    # 检查文档是否已经存在且未更改
-                    existing_ids = collection.get(
-                        where={"doc_hash": doc_hash}
-                    )
-                    
-                    if not existing_ids["ids"]:
-                        # 分块处理文档
-                        chunks = self._chunk_text(content, chunk_size)
-                        
-                        # 为每个块生成唯一ID
-                        chunk_ids = [f"{doc_hash}_{i}" for i in range(len(chunks))]
-                        
-                        # 添加到数据库
-                        collection.add(
-                            documents=chunks,
-                            ids=chunk_ids,
-                            metadatas=[{
-                                "file_path": file_path,
-                                "doc_hash": doc_hash,
-                                "chunk_index": i
-                            } for i in range(len(chunks))]
-                        )
-                        
-                        PrettyOutput.print(f"已添加文档: {file_path}", OutputType.INFO)
-                    else:
-                        PrettyOutput.print(f"文档已存在且未更改: {file_path}", OutputType.INFO)
-                        
-                except Exception as e:
-                    PrettyOutput.print(f"处理文件 {file_path} 时出错: {str(e)}", OutputType.ERROR)
-            
-            # 执行查询
-            results = collection.query(
-                query_texts=[query],
-                n_results=num_passages
-            )
-            
-            # 格式化输出
-            output = [f"查询: {query}\n"]
-            output.append(f"找到 {len(results['documents'][0])} 个相关段落:\n")
-            
-            for i, (doc, metadata) in enumerate(zip(results['documents'][0], results['metadatas'][0]), 1):
-                output.append(f"\n段落 {i}:")
-                output.append(f"来源: {metadata['file_path']}")
-                output.append(f"相关内容:\n{doc}\n")
-                output.append("-" * 50)
-            
-            return {
-                "success": True,
-                "stdout": "\n".join(output),
-                "stderr": ""
-            }
-
-        except Exception as e:
-            return {
-                "success": False,
-                "error": f"RAG查询失败: {str(e)}"
-            } 

jarvis/tools/user_confirmation.py

@@ -1,58 +0,0 @@
-from typing import Dict, Any
-from ..utils import PrettyOutput, OutputType
-
-class UserConfirmationTool:
-    name = "ask_user_confirmation"
-    description = "Request confirmation from user, returns yes/no"
-    parameters = {
-        "type": "object",
-        "properties": {
-            "question": {
-                "type": "string",
-                "description": "Question to ask for confirmation"
-            },
-            "details": {
-                "type": "string",
-                "description": "Additional details or context",
-                "default": ""
-            }
-        },
-        "required": ["question"]
-    }
-
-    def execute(self, args: Dict) -> Dict[str, Any]:
-        """获取用户确认"""
-        try:
-            question = args["question"]
-            details = args.get("details", "")
-            
-            # 打印详细信息
-            if details:
-                PrettyOutput.print(details, OutputType.INFO)
-                PrettyOutput.print("", OutputType.INFO)  # 空行
-            
-            # 打印问题
-            PrettyOutput.print(f"{question} (y/n)", OutputType.INFO)
-            
-            while True:
-                response = input(">>> ").strip().lower()
-                if response in ['y', 'yes']:
-                    result = True
-                    break
-                elif response in ['n', 'no']:
-                    result = False
-                    break
-                else:
-                    PrettyOutput.print("请输入 y/yes 或 n/no", OutputType.ERROR)
-            
-            return {
-                "success": True,
-                "stdout": str(result),
-                "stderr": ""
-            }
-
-        except Exception as e:
-            return {
-                "success": False,
-                "error": f"获取用户确认失败: {str(e)}"
-            } 

jarvis/tools/user_interaction.py

@@ -1,86 +0,0 @@
-from typing import Dict, Any, List
-from ..utils import PrettyOutput, OutputType, get_multiline_input
-
-class UserInteractionTool:
-    name = "ask_user"
-    description = "Ask user for information, supports option selection and multiline input"
-    parameters = {
-        "type": "object",
-        "properties": {
-            "question": {
-                "type": "string",
-                "description": "Question to ask the user"
-            },
-            "options": {
-                "type": "array",
-                "items": {"type": "string"},
-                "description": "List of options for user to choose from",
-                "default": []
-            },
-            "multiline": {
-                "type": "boolean",
-                "description": "Allow multiline input",
-                "default": False
-            },
-            "description": {
-                "type": "string",
-                "description": "Additional description or context",
-                "default": ""
-            }
-        },
-        "required": ["question"]
-    }
-
-    def execute(self, args: Dict) -> Dict[str, Any]:
-        """获取用户输入"""
-        try:
-            question = args["question"]
-            options = args.get("options", [])
-            multiline = args.get("multiline", False)
-            description = args.get("description", "")
-            
-            # 打印问题描述
-            if description:
-                PrettyOutput.print(description, OutputType.INFO)
-            
-            # 如果有选项，显示选项列表
-            if options:
-                PrettyOutput.print("\n可选项:", OutputType.INFO)
-                for i, option in enumerate(options, 1):
-                    PrettyOutput.print(f"[{i}] {option}", OutputType.INFO)
-                
-                while True:
-                    try:
-                        choice = input("\n请选择 (输入数字): ").strip()
-                        if not choice:
-                            continue
-                        
-                        choice = int(choice)
-                        if 1 <= choice <= len(options):
-                            response = options[choice - 1]
-                            break
-                        else:
-                            PrettyOutput.print("无效选择，请重试", OutputType.ERROR)
-                    except ValueError:
-                        PrettyOutput.print("请输入有效数字", OutputType.ERROR)
-            
-            # 多行输入
-            elif multiline:
-                response = get_multiline_input(question)
-            
-            # 单行输入
-            else:
-                PrettyOutput.print(question, OutputType.INFO)
-                response = input(">>> ").strip()
-            
-            return {
-                "success": True,
-                "stdout": response,
-                "stderr": ""
-            }
-
-        except Exception as e:
-            return {
-                "success": False,
-                "error": f"获取用户输入失败: {str(e)}"
-            } 

jarvis_ai_assistant-0.1.3.dist-info/RECORD

@@ -1,36 +0,0 @@
-jarvis/.jarvis,sha256=S4ZMmLqlVLHAPmnwHFQ3vYt0gnDFp-KQRH4okBh8Hpw,209
-jarvis/__init__.py,sha256=rLUKzHgrbk82WAMvg4u4GlFDRal1Yw3XpMg_741mQXw,49
-jarvis/agent.py,sha256=hEatrIothG6C_RuOsygZz4ez1aF1EycnUPwY2Krwiwk,4092
-jarvis/main.py,sha256=wAeeGoRq8fFiTWUCQ31MYVeU9o6SZ48ORUow0jlMJJE,5259
-jarvis/models.py,sha256=ZQAyc39e_UsNmHkME6ATp6053safQ7Gog-oWwETsrMM,4415
-jarvis/utils.py,sha256=-fOPAgiKVPC6DevHM7A7IJvvIzvvFlGM0pHBKyBXDcA,3044
-jarvis/__pycache__/__init__.cpython-313.pyc,sha256=u64PLu4KyOPrirhIqNXZSkj9B6yB1L_ohFYhPX8p334,208
-jarvis/__pycache__/agent.cpython-313.pyc,sha256=I1JYUKC-zedURwuxQQjd9saj_yQKUhpdbb1KNWlY6Ss,4940
-jarvis/__pycache__/models.cpython-313.pyc,sha256=fot2VOV0lgODg4b_WuD-kkq-g7uZGLuZlYXiljlW13U,6350
-jarvis/__pycache__/tools.cpython-313.pyc,sha256=lAD4LrnnWzNZQmHXGfZ_2l7oskOpr2_2OC-gdFhxQY8,33933
-jarvis/__pycache__/utils.cpython-313.pyc,sha256=o8Fubq8Dcrvv3ATzQ9BnZoZjQmU3_FNPrfTt0ANdKdI,5804
-jarvis/tools/__init__.py,sha256=FNF5X32UzKvlT08-hPz7-7vcIyqqMngQYj36LXyNbV0,553
-jarvis/tools/base.py,sha256=Y1XInc2skUVZYZzLRPlCpH3JyY3pk1XNZgys6MSCudQ,6586
-jarvis/tools/file_ops.py,sha256=05Vc4u-NGMjLD15WI52eL_nt_RyYt-Z_cXJnnBepf-c,3781
-jarvis/tools/python_script.py,sha256=_eK8LqNs-Mz50zdcgwbjdd8-qAeOl6kJ_qRDvWawGMw,5006
-jarvis/tools/rag.py,sha256=eBrn1fdqy-nd2nR0b4oH1EQpthTYVen-sYDhC5Ypl38,5647
-jarvis/tools/search.py,sha256=dyJmeP_s2tWv5KQejOl-TuVF12B6SXViiXEyTemD1Wo,1412
-jarvis/tools/shell.py,sha256=VhEqEdoUGIRW6CT6F0YNMAlzJxpHcsKS9hSS0VC3ezw,2644
-jarvis/tools/user_confirmation.py,sha256=p0JverifHJfnALeIhtKtaVBBVEGkgSpUzT-PSycG4NY,1850
-jarvis/tools/user_interaction.py,sha256=xOaoYsCKTa9-K3mg5tsu1zegn1HKcO40k2aJ3VygZqM,3073
-jarvis/tools/webpage.py,sha256=UTEomu5j7jbOw8c5az2jsjv5E7LeokWKj1QahvZO7xc,3077
-jarvis/tools/__pycache__/__init__.cpython-313.pyc,sha256=wktmOMuGMwQjQ--eRzooUNsL5YjY0bS_CbcSFUWze54,674
-jarvis/tools/__pycache__/base.cpython-313.pyc,sha256=hqhRtxwQwWLnBEzNqZ1FqcWYuyX87iu89SgHxu10Gho,8192
-jarvis/tools/__pycache__/file_ops.cpython-313.pyc,sha256=bawv11xhWSj5wCtW0QeJLI-InhUBUXaSkogq6rZzvTQ,3636
-jarvis/tools/__pycache__/python_script.cpython-313.pyc,sha256=8JpryqTovEiTvBlWAK1KjZmPvHUuPc9GT9rTXBEQoJc,6693
-jarvis/tools/__pycache__/rag.cpython-313.pyc,sha256=JH6-PSZRMKAvTZqCwlRXJGClxYXNMs-vetU0q7hBLz0,6064
-jarvis/tools/__pycache__/search.cpython-313.pyc,sha256=VX9zztOwIsPCkYwev0A0XJGyu4Tr0PQcQkbbqx8vfB0,1870
-jarvis/tools/__pycache__/shell.cpython-313.pyc,sha256=Xtqt-LzMRfsCXeCE7QxFY8TR8XEeQG_jic73YLkhXQw,2302
-jarvis/tools/__pycache__/user_confirmation.cpython-313.pyc,sha256=wK3Ev10lHSUSRvoYmi7A0GzxYkzU-C4Wfhs5qW_HBqs,2271
-jarvis/tools/__pycache__/user_interaction.cpython-313.pyc,sha256=RuVZ-pmiPBDywY3efgXSfohMAciC1avMGPmBK5qlnew,3305
-jarvis/tools/__pycache__/webpage.cpython-313.pyc,sha256=VcpkaV8IyOOtebedXjn8ybjr8AglU-3-Cs80yzE4FYo,3628
-jarvis_ai_assistant-0.1.3.dist-info/METADATA,sha256=eVKO4jecgKkHOIRUXZGvfvi5KiR9CGaZuQLhDR9W3NY,3675
-jarvis_ai_assistant-0.1.3.dist-info/WHEEL,sha256=A3WOREP4zgxI0fKrHUG8DC8013e3dK3n7a6HDbcEIwE,91
-jarvis_ai_assistant-0.1.3.dist-info/entry_points.txt,sha256=iKu7OMfew9dtfGhW71gIMTg4wvafuPqKb4wyQOnMAGU,44
-jarvis_ai_assistant-0.1.3.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
-jarvis_ai_assistant-0.1.3.dist-info/RECORD,,