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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jarvis-ai-assistant
-Version: 0.1.219
+Version: 0.1.221
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire
@@ -33,11 +33,11 @@ Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: POSIX :: Linux
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
-Requires-Python: >=3.8
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: requests==2.32.3
@@ -56,9 +56,9 @@ Requires-Dist: tabulate==0.9.0
 Requires-Dist: pyte==0.8.2
 Requires-Dist: httpx>=0.28.1
 Requires-Dist: pyyaml>=5.3.1
-Requires-Dist: ddgs==6.1.8
-Requires-Dist: beautifulsoup4==4.12.3
-Requires-Dist: lxml==5.2.2
+Requires-Dist: ddgs==9.0.2
+Requires-Dist: beautifulsoup4==4.13.4
+Requires-Dist: lxml==6.0.0
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: black; extra == "dev"
@@ -66,6 +66,17 @@ Requires-Dist: isort; extra == "dev"
 Requires-Dist: mypy; extra == "dev"
 Requires-Dist: build; extra == "dev"
 Requires-Dist: twine; extra == "dev"
+Provides-Extra: rag
+Requires-Dist: langchain; extra == "rag"
+Requires-Dist: langchain-community; extra == "rag"
+Requires-Dist: langchain-huggingface; extra == "rag"
+Requires-Dist: chromadb; extra == "rag"
+Requires-Dist: diskcache; extra == "rag"
+Requires-Dist: sentence-transformers==2.7.0; extra == "rag"
+Requires-Dist: torch>=2.6; extra == "rag"
+Requires-Dist: typer; extra == "rag"
+Requires-Dist: unstructured[md]; extra == "rag"
+Requires-Dist: rank-bm25; extra == "rag"
 Dynamic: author
 Dynamic: home-page
 Dynamic: license-file
@@ -536,6 +547,7 @@ OPENAI_API_BASE: https://api.openai.com/v1
 | `JARVIS_USE_METHODOLOGY` | true | 是否启用方法论功能 |
 | `JARVIS_USE_ANALYSIS` | true | 是否启用任务分析功能 |
 | `JARVIS_DATA_PATH` | ~/.jarvis | Jarvis数据存储目录路径 |
+| `JARVIS_RAG` | `{"embedding_model": "BAAI/bge-base-zh-v1.5"}` | RAG框架的配置 |
 
 ## 🛠️ 工具说明 <a id="tools"></a>
 
@@ -553,7 +565,77 @@ OPENAI_API_BASE: https://api.openai.com/v1
 | search_web | 使用互联网搜索 |
 | virtual_tty | 控制虚拟终端执行操作 |
 
-### 2. 命令替换功能
+### 2. 🧠 RAG增强知识库 (`jarvis-rag`)
+
+`jarvis-rag` 是一个强大的命令行工具，用于构建、管理和查询本地化的RAG（检索增强生成）知识库。它允许您将自己的文档（代码、笔记、文章等）作为外部知识源，让AI能够基于这些具体内容进行回答，而不是仅仅依赖其通用训练数据。
+
+#### 核心功能
+
+- **本地化知识库**：所有文档索引和数据都存储在本地，确保数据隐私和安全。
+- **智能文档加载**：自动识别并加载多种文本文件，无需关心文件后缀。
+- **灵活的查询**：支持使用项目默认的“思考”模型或动态指定任意已注册的LLM进行查询。
+- **配置驱动**：通过 `config.toml` 或 `.jarvis/config.yaml` 进行集中配置。
+
+#### 子命令说明
+
+##### 2.1 添加文档到知识库 (`add`)
+
+此命令用于将文件、目录或符合通配符模式的文档添加到知识库中。
+
+```bash
+# 基本用法
+jarvis-rag add <文件路径/目录路径/通配符模式>...
+
+# 示例
+# 1. 添加单个文件
+jarvis-rag add ./docs/my_document.md
+
+# 2. 添加整个目录（将递归扫描所有文本文件）
+jarvis-rag add ./src/
+
+# 3. 使用通配符添加所有Python文件
+jarvis-rag add 'src/**/*.py'
+
+# 4. 混合添加
+jarvis-rag add README.md ./docs/ 'src/jarvis/jarvis_rag/*.py'
+```
+
+**参数与选项:**
+
+| 参数/选项 | 描述 |
+|---|---|
+| `paths` | **[必需]** 一个或多个文件路径、目录路径或用引号包裹的通配符模式。 |
+| `--collection` | 指定知识库的集合名称（默认为 `jarvis_rag_collection`）。 |
+| `--embedding-model` | 覆盖全局配置，强制使用特定的嵌入模型名称。 |
+| `--db-path` | 覆盖全局配置，指定向量数据库的存储路径。 |
+
+##### 2.2 查询知识库 (`query`)
+
+此命令用于向已建立的知识库提出问题。
+
+```bash
+# 基本用法
+jarvis-rag query "你的问题"
+
+# 示例
+# 1. 使用默认配置进行查询
+jarvis-rag query "请总结一下我添加的文档的核心内容"
+
+# 2. 指定使用Kimi模型进行查询
+jarvis-rag query "代码中的 'PlatformRegistry' 类是做什么用的？" --platform kimi --model moonshot-v1-8k
+```
+
+**参数与选项:**
+
+| 参数/选项 | 描述 |
+|---|---|
+| `question` | **[必需]** 你要向知识库提出的问题。 |
+| `--collection` | 指定要查询的知识库集合名称。 |
+| `--platform` | 指定一个平台名称来回答问题，覆盖默认的“思考”模型。 |
+| `--model` | 指定一个模型名称来回答问题，需要与 `--platform` 同时使用。 |
+| `--embedding-model` | 覆盖全局配置，强制使用特定的嵌入模型名称。 |
+
+### 3. 命令替换功能
 支持使用特殊标记`'<tag>'`触发命令替换功能：
 
 | 标记 | 功能 |
@@ -568,7 +650,7 @@ OPENAI_API_BASE: https://api.openai.com/v1
 | `'Check'` | 执行静态代码检查，包括错误和风格问题 |
 | `'SaveSession'` | 保存当前会话并退出 |
 
-### 3. 自定义替换配置
+### 4. 自定义替换配置
 在`~/.jarvis/config.yaml`中添加：
 ```yaml
 JARVIS_REPLACE_MAP:

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

@@ -463,6 +463,7 @@ OPENAI_API_BASE: https://api.openai.com/v1
 | `JARVIS_USE_METHODOLOGY` | true | 是否启用方法论功能 |
 | `JARVIS_USE_ANALYSIS` | true | 是否启用任务分析功能 |
 | `JARVIS_DATA_PATH` | ~/.jarvis | Jarvis数据存储目录路径 |
+| `JARVIS_RAG` | `{"embedding_model": "BAAI/bge-base-zh-v1.5"}` | RAG框架的配置 |
 
 ## 🛠️ 工具说明 <a id="tools"></a>
 
@@ -480,7 +481,77 @@ OPENAI_API_BASE: https://api.openai.com/v1
 | search_web | 使用互联网搜索 |
 | virtual_tty | 控制虚拟终端执行操作 |
 
-### 2. 命令替换功能
+### 2. 🧠 RAG增强知识库 (`jarvis-rag`)
+
+`jarvis-rag` 是一个强大的命令行工具，用于构建、管理和查询本地化的RAG（检索增强生成）知识库。它允许您将自己的文档（代码、笔记、文章等）作为外部知识源，让AI能够基于这些具体内容进行回答，而不是仅仅依赖其通用训练数据。
+
+#### 核心功能
+
+- **本地化知识库**：所有文档索引和数据都存储在本地，确保数据隐私和安全。
+- **智能文档加载**：自动识别并加载多种文本文件，无需关心文件后缀。
+- **灵活的查询**：支持使用项目默认的“思考”模型或动态指定任意已注册的LLM进行查询。
+- **配置驱动**：通过 `config.toml` 或 `.jarvis/config.yaml` 进行集中配置。
+
+#### 子命令说明
+
+##### 2.1 添加文档到知识库 (`add`)
+
+此命令用于将文件、目录或符合通配符模式的文档添加到知识库中。
+
+```bash
+# 基本用法
+jarvis-rag add <文件路径/目录路径/通配符模式>...
+
+# 示例
+# 1. 添加单个文件
+jarvis-rag add ./docs/my_document.md
+
+# 2. 添加整个目录（将递归扫描所有文本文件）
+jarvis-rag add ./src/
+
+# 3. 使用通配符添加所有Python文件
+jarvis-rag add 'src/**/*.py'
+
+# 4. 混合添加
+jarvis-rag add README.md ./docs/ 'src/jarvis/jarvis_rag/*.py'
+```
+
+**参数与选项:**
+
+| 参数/选项 | 描述 |
+|---|---|
+| `paths` | **[必需]** 一个或多个文件路径、目录路径或用引号包裹的通配符模式。 |
+| `--collection` | 指定知识库的集合名称（默认为 `jarvis_rag_collection`）。 |
+| `--embedding-model` | 覆盖全局配置，强制使用特定的嵌入模型名称。 |
+| `--db-path` | 覆盖全局配置，指定向量数据库的存储路径。 |
+
+##### 2.2 查询知识库 (`query`)
+
+此命令用于向已建立的知识库提出问题。
+
+```bash
+# 基本用法
+jarvis-rag query "你的问题"
+
+# 示例
+# 1. 使用默认配置进行查询
+jarvis-rag query "请总结一下我添加的文档的核心内容"
+
+# 2. 指定使用Kimi模型进行查询
+jarvis-rag query "代码中的 'PlatformRegistry' 类是做什么用的？" --platform kimi --model moonshot-v1-8k
+```
+
+**参数与选项:**
+
+| 参数/选项 | 描述 |
+|---|---|
+| `question` | **[必需]** 你要向知识库提出的问题。 |
+| `--collection` | 指定要查询的知识库集合名称。 |
+| `--platform` | 指定一个平台名称来回答问题，覆盖默认的“思考”模型。 |
+| `--model` | 指定一个模型名称来回答问题，需要与 `--platform` 同时使用。 |
+| `--embedding-model` | 覆盖全局配置，强制使用特定的嵌入模型名称。 |
+
+### 3. 命令替换功能
 支持使用特殊标记`'<tag>'`触发命令替换功能：
 
 | 标记 | 功能 |
@@ -495,7 +566,7 @@ OPENAI_API_BASE: https://api.openai.com/v1
 | `'Check'` | 执行静态代码检查，包括错误和风格问题 |
 | `'SaveSession'` | 保存当前会话并退出 |
 
-### 3. 自定义替换配置
+### 4. 自定义替换配置
 在`~/.jarvis/config.yaml`中添加：
 ```yaml
 JARVIS_REPLACE_MAP:

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

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "jarvis-ai-assistant"
-version = "0.1.219"
+version = "0.1.221"
 description = "Jarvis: An AI assistant that uses tools to interact with the system"
 readme = "README.md"
 authors = [{ name = "skyfire", email = "skyfireitdiy@hotmail.com" }]
@@ -15,10 +15,10 @@ classifiers = [
     "License :: OSI Approved :: MIT License",
     "Operating System :: POSIX :: Linux",
     "Programming Language :: Python :: 3",
-    "Programming Language :: Python :: 3.8",
     "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
 ]
 keywords = ["jarvis", "ai", "assistant", "tools", "automation"]
 dependencies = [
@@ -38,14 +38,26 @@ dependencies = [
     "pyte==0.8.2",
     "httpx>=0.28.1",
     "pyyaml>=5.3.1",
-    "ddgs==6.1.8",
-    "beautifulsoup4==4.12.3",
-    "lxml==5.2.2",
+    "ddgs==9.0.2",
+    "beautifulsoup4==4.13.4",
+    "lxml==6.0.0",
 ]
-requires-python = ">=3.8"
+requires-python = ">=3.9"
 
 [project.optional-dependencies]
 dev = ["pytest", "black", "isort", "mypy", "build", "twine"]
+rag = [
+    "langchain",
+    "langchain-community",
+    "langchain-huggingface",
+    "chromadb",
+    "diskcache",
+    "sentence-transformers==2.7.0",
+    "torch>=2.6",
+    "typer",
+    "unstructured[md]",
+    "rank-bm25",
+]
 
 [project.urls]
 Homepage = "https://github.com/skyfireitdiy/Jarvis"
@@ -67,3 +79,4 @@ jarvis-agent = "jarvis.jarvis_agent.main:main"
 jarvis-tool = "jarvis.jarvis_tools.cli.main:main"
 jarvis-git-details = "jarvis.jarvis_git_details.main:main"
 jarvis-methodology = "jarvis.jarvis_methodology.main:main"
+jarvis-rag = "jarvis.jarvis_rag.cli:main"

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

@@ -3,7 +3,7 @@ from setuptools import setup, find_packages  # type: ignore
 
 setup(
     name="jarvis-ai-assistant",
-    version="0.1.219",
+    version="0.1.221",
     author="skyfire",
     author_email="skyfireitdiy@hotmail.com",
     description="An AI assistant that uses various tools to interact with the system",
@@ -33,12 +33,25 @@ setup(
         "pyte==0.8.2",
         "httpx>=0.28.1",
         "pyyaml>=5.3.1",
-        "ddgs==6.1.8",
-        "beautifulsoup4==4.12.3",
-        "lxml==5.2.2",
+        "ddgs==9.0.2",
+        "beautifulsoup4==4.13.4",
+        "lxml==6.0.0",
     ],
-    extras_require={"dev": ["pytest", "black",
-                            "isort", "mypy", "build", "twine"]},
+    extras_require={
+        "dev": ["pytest", "black", "isort", "mypy", "build", "twine"],
+        "rag": [
+            "langchain",
+            "langchain-community",
+            "langchain-huggingface",
+            "chromadb",
+            "diskcache",
+            "sentence-transformers==2.7.0",
+            "torch>=2.6",
+            "typer",
+            "unstructured[md]",
+            "rank-bm25",
+        ],
+    },
     entry_points={
         "console_scripts": [
             "jarvis=jarvis.jarvis_agent.jarvis:main",
@@ -57,18 +70,19 @@ setup(
             "jarvis-tool=jarvis.jarvis_tools.cli.main:main",
             "jarvis-git-details=jarvis.jarvis_git_details.main:main",
             "jarvis-methodology=jarvis.jarvis_methodology.main:main",
+            "jarvis-rag=jarvis.jarvis_rag.cli:main",
         ],
     },
-    python_requires=">=3.8",
+    python_requires=">=3.9",
     classifiers=[
         "Development Status :: 3 - Alpha",
         "Intended Audience :: Developers",
         "License :: OSI Approved :: MIT License",
         "Operating System :: POSIX :: Linux",
         "Programming Language :: Python :: 3",
-        "Programming Language :: Python :: 3.8",
         "Programming Language :: Python :: 3.9",
         "Programming Language :: Python :: 3.10",
         "Programming Language :: Python :: 3.11",
+        "Programming Language :: Python :: 3.12",
     ],
 )

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

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