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

jarvis/jarvis_treesitter/symbol.py

@@ -0,0 +1,31 @@
+"""Symbol-related classes for the tree-sitter code database."""
+
+from dataclasses import dataclass
+from enum import Enum
+from typing import Optional
+
+class SymbolType(Enum):
+    """Types of symbols that can be stored in the database."""
+    FUNCTION = "function"
+    CLASS = "class"
+    VARIABLE = "variable"
+    REFERENCE = "reference"
+    FUNCTION_CALL = "function_call"
+
+@dataclass
+class SymbolLocation:
+    """Location information for a symbol in source code."""
+    file_path: str
+    start_line: int
+    start_column: int
+    end_line: int
+    end_column: int
+
+@dataclass
+class Symbol:
+    """Represents a symbol in the code database."""
+    name: str
+    type: SymbolType
+    location: SymbolLocation
+    scope: Optional[str] = None  # Optional scope information
+    parent: Optional['Symbol'] = None  # Optional parent symbol (e.g., class for methods) 

jarvis/jarvis_treesitter/tools_usage.md

@@ -0,0 +1,121 @@
+# Tree-sitter 分析工具使用指南
+
+本文档说明如何使用 Jarvis 的 Tree-sitter 分析工具来查找代码中的符号定义、引用和调用关系。
+
+## 概述
+
+Tree-sitter 分析工具 (`treesitter_analyzer`) 可以帮助你快速分析代码库，查找符号的定义位置、引用位置和调用关系，支持多种编程语言，包括：
+
+- Python
+- C
+- C++
+- Go
+- Rust
+
+## 功能特点
+
+- **自动下载语法文件**：工具会自动下载和编译需要的语法文件，保存在 `~/.jarvis/treesitter` 目录
+- **多语言支持**：支持多种常用编程语言
+- **符号查找**：查找函数、类、变量等符号的定义位置
+- **引用查找**：查找符号的所有引用位置
+- **调用关系**：查找函数的所有调用位置
+- **自动索引**：每次操作前会自动索引指定目录的代码
+
+## 使用方法
+
+在 Jarvis 中，可以通过工具调用格式来使用这个工具：
+
+```yaml
+<TOOL_CALL>
+name: treesitter_analyzer
+arguments:
+    action: find_symbol
+    symbol_name: main
+    directory: /path/to/code
+    extensions: [".c", ".h"]
+    max_results: 10
+</TOOL_CALL>
+```
+
+## 支持的操作
+
+Tree-sitter 分析工具支持以下操作类型：
+
+1. **查找符号** (`find_symbol`)：查找符号的定义位置
+2. **查找引用** (`find_references`)：查找符号的引用位置
+3. **查找调用者** (`find_callers`)：查找函数的调用位置
+
+**注意**：每个操作前都会自动先索引指定目录下的代码文件，不需要单独执行索引步骤。
+
+## 参数说明
+
+工具接受以下参数：
+
+- `action`：**必需**，操作类型，可选值：`find_symbol`、`find_references`、`find_callers`
+- `symbol_name`：**必需**，符号名称，如函数名、类名、变量名
+- `directory`：**必需**，要索引的代码目录，默认为当前目录
+- `extensions`：可选，要索引的文件扩展名列表，如 `[".py", ".c"]`，不指定则索引所有支持的文件类型
+- `max_results`：可选，最大返回结果数量，默认为 20
+
+## 使用示例
+
+### 查找符号定义
+
+```yaml
+<TOOL_CALL>
+name: treesitter_analyzer
+arguments:
+    action: find_symbol
+    symbol_name: main
+    directory: /path/to/code
+</TOOL_CALL>
+```
+
+### 查找符号引用
+
+```yaml
+<TOOL_CALL>
+name: treesitter_analyzer
+arguments:
+    action: find_references
+    symbol_name: add_user
+    directory: /path/to/code
+    extensions: [".py"]
+</TOOL_CALL>
+```
+
+### 查找函数调用者
+
+```yaml
+<TOOL_CALL>
+name: treesitter_analyzer
+arguments:
+    action: find_callers
+    symbol_name: process_data
+    directory: /path/to/code
+    max_results: 50
+</TOOL_CALL>
+```
+
+## 返回结果
+
+工具返回的结果包含以下字段：
+
+- `success`：操作是否成功
+- `stdout`：操作的输出信息，包括索引摘要和查询结果
+- `stderr`：操作的错误信息（如果有）
+
+此外，不同操作还会返回其他特定字段：
+
+- `find_symbol`：返回 `symbols` 列表
+- `find_references`：返回 `symbol` 和 `references` 列表
+- `find_callers`：返回 `function` 和 `callers` 列表
+- 所有操作都会包含 `indexed_files` 和 `skipped_files` 的信息
+
+## 注意事项
+
+1. 首次使用时，工具会自动下载和编译语法文件，可能需要一些时间
+2. 大型代码库索引可能较慢，请耐心等待
+3. 为提高性能，可以使用 `extensions` 参数限制索引的文件类型
+4. 工具会缓存索引结果，后续查询同一代码库速度会更快
+5. 每次操作都会重新索引目录，确保分析结果基于最新的代码 

jarvis/jarvis_utils/git_utils.py

@@ -14,9 +14,17 @@ import subprocess
 from typing import List, Tuple, Dict
 from jarvis.jarvis_utils.output import PrettyOutput, OutputType
 def find_git_root(start_dir="."):
-    """切换到给定路径的Git根目录"""
+    """切换到给定路径的Git根目录，如果不是Git仓库则初始化"""
     os.chdir(start_dir)
-    git_root = os.popen("git rev-parse --show-toplevel").read().strip()
+    try:
+        git_root = os.popen("git rev-parse --show-toplevel").read().strip()
+        if not git_root:
+            subprocess.run(["git", "init"], check=True)
+            git_root = os.path.abspath(".")
+    except subprocess.CalledProcessError:
+        # 如果不是Git仓库，初始化一个新的
+        subprocess.run(["git", "init"], check=True)
+        git_root = os.path.abspath(".")
     os.chdir(git_root)
     return git_root
 def has_uncommitted_changes():

jarvis/jarvis_utils/input.py

@@ -77,7 +77,9 @@ class FileCompleter(Completer):
             default_suggestions = [
                 ('<CodeBase>', '查询代码库'),
                 ('<Web>', '网页搜索'),
-                ('<RAG>', '知识库检索')
+                ('<RAG>', '知识库检索'),
+                ('<Summary>', '总结'),
+                ('<Clear>', '清除历史'),
             ]
             for name, desc in default_suggestions:
                 yield Completion(

jarvis/jarvis_utils/methodology.py

@@ -72,7 +72,7 @@ def load_methodology(user_input: str) -> str:
     
     try:
         with yaspin(text="加载方法论文件...", color="yellow") as spinner:
-            with open(user_jarvis_methodology, "r", encoding="utf-8") as f:
+            with open(user_jarvis_methodology, "r", encoding="utf-8", errors="ignore") as f:
                 data = yaml.safe_load(f)
             if dont_use_local_model():
                 spinner.text = "加载方法论文件完成"

jarvis/jarvis_utils/utils.py

@@ -22,7 +22,7 @@ def init_env():
         jarvis_dir.mkdir(parents=True)
     if env_file.exists():
         try:
-            with open(env_file, "r", encoding="utf-8") as f:
+            with open(env_file, "r", encoding="utf-8", errors="ignore") as f:
                 for line in f:
                     line = line.strip()
                     if line and not line.startswith(("#", ";")):
@@ -83,7 +83,7 @@ def get_file_line_count(filename: str) -> int:
         int: Number of lines in the file, 0 if file cannot be read
     """
     try:
-        return len(open(filename, "r", encoding="utf-8").readlines())
+        return len(open(filename, "r", encoding="utf-8", errors="ignore").readlines())
     except Exception as e:
         return 0
 def init_gpu_config() -> Dict:
@@ -137,7 +137,7 @@ def is_long_context(files: list) -> bool:
     
     for file_path in files:
         try:
-            with open(file_path, 'r', encoding='utf-8') as f:
+            with open(file_path, 'r', encoding='utf-8', errors="ignore") as f:
                 content = f.read()
                 total_tokens += get_context_token_count(content)
                 

{jarvis_ai_assistant-0.1.130.dist-info → jarvis_ai_assistant-0.1.131.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: jarvis-ai-assistant
-Version: 0.1.130
+Version: 0.1.131
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire
@@ -64,6 +64,7 @@ Requires-Dist: jedi>=0.17.2
 Requires-Dist: psutil>=7.0.0
 Requires-Dist: fastapi>=0.115.4
 Requires-Dist: uvicorn>=0.33.0
+Requires-Dist: tree-sitter>=0.20.0
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: black; extra == "dev"
@@ -179,9 +180,6 @@ jarvis-git-squash --help
 | create_sub_agent | 创建子代理 |
 | lsp_find_definition | 查找符号定义 |
 | lsp_find_references | 查找符号引用 |
-| lsp_get_diagnostics | 获取代码诊断信息 |
-| lsp_get_document_symbols | 获取文档符号 |
-| lsp_prepare_rename | 准备符号重命名 |
 | lsp_validate_edit | 验证代码编辑 |
 | rag | 文档检索和问答 |
 | select_code_files | 选择代码文件 |

jarvis_ai_assistant-0.1.131.dist-info/RECORD

@@ -0,0 +1,85 @@
+jarvis/__init__.py,sha256=LkxZHlCG5zW1xr1tlLt_s3LgitGFyjcKmI_offIk0Fc,51
+jarvis/jarvis_agent/__init__.py,sha256=dMGl8iAmhIl38YO_nS5-2u23HOBUcQYL5YsmbJVLCB4,23304
+jarvis/jarvis_agent/builtin_input_handler.py,sha256=Bm4aVRrjrkjgpudest58X707wvhz_Z_ufTEPw0s4DEw,2414
+jarvis/jarvis_agent/file_input_handler.py,sha256=55HwQGq3HWKmDSeJqJ0eOktoq-5Mbc5UpFqmSUrhxBY,3355
+jarvis/jarvis_agent/main.py,sha256=TRGDYi53BaDkaMgLZ4HFaM0WdGP5nX4Jcfaqfwy3lqg,2532
+jarvis/jarvis_agent/output_handler.py,sha256=kJeFTjjSu0K_2p0wyhq2veSZuhRXoaFC_8wVaoBKX0w,401
+jarvis/jarvis_agent/patch.py,sha256=1yBNdORlt_jAEHTZaiHpncWvTVekqpmHSp0K0i1Kh-k,17359
+jarvis/jarvis_agent/shell_input_handler.py,sha256=AgJTeZd3wPbwCQXvWS1TcJvcTib8cmYAwuaBzDm16FA,1006
+jarvis/jarvis_code_agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jarvis/jarvis_code_agent/code_agent.py,sha256=UQmNhToFCAZuYf6ka4lmicNm0Utx2m8q2BShfa69HBc,8798
+jarvis/jarvis_code_agent/file_select.py,sha256=2nSy1FW-kK-wvtz0YbbgSbd4ZwXMlA7sBP0nC80FzLI,8160
+jarvis/jarvis_codebase/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jarvis/jarvis_codebase/main.py,sha256=aZMKpRcBtLD3TKBBUVUR2WoT390vm3KeaV9rp5Gw1Sc,41410
+jarvis/jarvis_dev/main.py,sha256=--heacV6w2FRnQNQGd4ub8ihr5zEaeSjNpdcbgEJyog,30385
+jarvis/jarvis_git_squash/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jarvis/jarvis_git_squash/main.py,sha256=d3OaDO2DtT7G-13rAb5bDU4ISE_x2VAgeaACKwIqtyA,2799
+jarvis/jarvis_lsp/base.py,sha256=b5aIy_FoPHPb9DjHKCFGI5KNp6jDQYuClsUDuLq5PRI,3412
+jarvis/jarvis_lsp/cpp.py,sha256=7J6F2k1b7nWpuq_9dmfkojDrK3xyixrSEcJY1uTf4Yk,4011
+jarvis/jarvis_lsp/go.py,sha256=sS6XVQSr7ac1mkYxWswwvdW1BHJRxgsh4vxgiBfU_BA,4328
+jarvis/jarvis_lsp/python.py,sha256=-901gYdLQDGModUZ0s-QWPShI4zTnY05d4JhQbEiDKs,2693
+jarvis/jarvis_lsp/registry.py,sha256=f7_RmilgqfaBDNaEejK8K099fbo0mO-vTiBzLmgEeg8,9680
+jarvis/jarvis_lsp/rust.py,sha256=6Ge-_M5G29dw9a3kaG5uO020Nnas5R2Pg2tlqU4JYVQ,4564
+jarvis/jarvis_multi_agent/__init__.py,sha256=jOL0e9uq0Sefchq0l0UK-eXrCNHlEsankEIvvCVl5Os,6864
+jarvis/jarvis_platform/__init__.py,sha256=mrOt67nselz_H1gX9wdAO4y2DY5WPXzABqJbr5Des8k,63
+jarvis/jarvis_platform/ai8.py,sha256=xjgoiF7QHx_5FHj-npTFVvZFYg8RBzkqTGAOQ-MJiL0,11972
+jarvis/jarvis_platform/base.py,sha256=2chHt0pMx4rr0OFTeDpZcVqFUxF_q4YslUt30E5Augk,3277
+jarvis/jarvis_platform/kimi.py,sha256=Qwb81flbKPvj-qZyJAMS_u0lQatRFQztYxUGDbj1keM,15713
+jarvis/jarvis_platform/ollama.py,sha256=NHQMJSpC91WtSFuHKJuwD8qO-z4yDTF9mZX6BzWTKRU,5658
+jarvis/jarvis_platform/openai.py,sha256=GSxTB69WitXJS3pL0sxCoB2F0EVHmvhrwLBC_JT8s34,4470
+jarvis/jarvis_platform/oyi.py,sha256=pa72TtBYlhs3KPpqO4Y78a1Jvx4mN0pojBklu8X3F-k,15024
+jarvis/jarvis_platform/registry.py,sha256=JlFXnPK_lmKWscQJItD-eEVgFe4WspsKfaFja2XLhOU,8540
+jarvis/jarvis_platform_manager/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jarvis/jarvis_platform_manager/main.py,sha256=iNeaGVXYJJUHo-euQRo7sMesQ2Y5EyNDCyx0koKyeiI,20949
+jarvis/jarvis_platform_manager/openai_test.py,sha256=bkcVG6o2nNazj4zjkENgA4yOEzdTI9Qbm5dco-2MGYM,5190
+jarvis/jarvis_rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jarvis/jarvis_rag/main.py,sha256=2NkPDvelyk-qERhk45_uLFfw9NicvW4BM67ZI7N2AlE,35872
+jarvis/jarvis_smart_shell/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jarvis/jarvis_smart_shell/main.py,sha256=5gyBKgxuaTrDToE7Xy4DhsJPDOEviPi7ktm1vOZoIno,4618
+jarvis/jarvis_tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jarvis/jarvis_tools/ask_codebase.py,sha256=dDiFQd23ZZlM2Un2huwHNsWOId7Ux83UwMXE28Bh2f4,4124
+jarvis/jarvis_tools/ask_user.py,sha256=kU6BOhFz_mKfqABd0l_00TL7Mqp_dhV3r0qkKLy8yRM,1745
+jarvis/jarvis_tools/base.py,sha256=8gIgIx9LJAKOxdpPu7zMDItWqwXmXVTWAGfUkpQonzw,1196
+jarvis/jarvis_tools/chdir.py,sha256=GLbH2fOKESUn3XYhNg0uOI5axTO2csC0B9HUL0bMZ5U,2790
+jarvis/jarvis_tools/code_review.py,sha256=hIr8B7G6SFevkfG79u0JGzlQhBeGY3ojR_5BrJkVD8I,11520
+jarvis/jarvis_tools/create_code_agent.py,sha256=2BSxbb-5e5sQNpPLlIy49UXqhf92UEMMvxmOiagMSVY,4319
+jarvis/jarvis_tools/create_sub_agent.py,sha256=JheHbVJCOpruXWpqmzf7UHFmwdsJB7BRUP_2ysjYzMM,3039
+jarvis/jarvis_tools/execute_shell.py,sha256=ujyWCBsw4NUHpVCnXrHFQ1mrWhCOqLM6BWoXsWD-g3A,3899
+jarvis/jarvis_tools/execute_shell_script.py,sha256=sKau5m516xPCOYA4tGOBdUpf1kCJM2-6KdqrLbGVSmg,2047
+jarvis/jarvis_tools/file_operation.py,sha256=mw9t5AWchiwePlw378CYQ1ld45dGlm2kTkeGDS8BcvU,7007
+jarvis/jarvis_tools/git_commiter.py,sha256=2PqySQwan-bqKcbaM-rOqgPpUzL9KsrT79SmIyopNmY,6565
+jarvis/jarvis_tools/lsp_find_definition.py,sha256=RG_2VFaR0vRRTdoBvOKwJvq3PlZmytQZmfRhI21caAc,5565
+jarvis/jarvis_tools/lsp_find_references.py,sha256=mXTWLUoU6Tg5025hWkobtkwVAaQLaDvYI8KZQ9RCufc,4734
+jarvis/jarvis_tools/lsp_get_diagnostics.py,sha256=GVehn0V44m9eem3P87egqWOmDCu7W7k1Jewj-DkPZig,5551
+jarvis/jarvis_tools/methodology.py,sha256=7bLxXgQf__j9Cq0qy8izhAHnTVBojMhpNI6TwMkkjnI,5838
+jarvis/jarvis_tools/rag.py,sha256=WuTlyGV5rgZTRxts9eJPC3QABrsoNYKratdO8UzUFDw,5132
+jarvis/jarvis_tools/read_code.py,sha256=hSeWGgWu3Dnj87lKOQEjSnMh7T3FrY4_IKTQAc4_xn0,5447
+jarvis/jarvis_tools/read_webpage.py,sha256=TkVNgirvcjns8-MHaDXOmliOKWCYcq3WzcbVXBi0IxY,4173
+jarvis/jarvis_tools/registry.py,sha256=pYyjdyjYgvHfdglhp0I7YXfVIV8_jTSncoSgIG6HFUw,13719
+jarvis/jarvis_tools/search_web.py,sha256=FOtWHs7FbKNaEVYEkWs7OaxWzP__h3ktH5E7u1yOywg,12644
+jarvis/jarvis_tools/select_code_files.py,sha256=xCqHTjIGju9Pb1Yh2C1Y7l6uT_3pfVB6ARU0VQmeRNI,1879
+jarvis/jarvis_tools/tool_generator.py,sha256=DcQrt1uY0NuzwbHz7qqsvLMBODEK9yLTQO_tMKN2IoY,7739
+jarvis/jarvis_tools/treesitter_analyzer.py,sha256=v1CNH-EhnQsRZx55krZ36cx33O_sOwnVJkloTSfcMho,12875
+jarvis/jarvis_treesitter/README.md,sha256=LKrziIARGc70GPZzHEnbirtfDXkgsLLFBhmA92p2wuk,3378
+jarvis/jarvis_treesitter/__init__.py,sha256=Qkqps0PXBUKX19WzMJGISKhNwhvfaiGKMNtDSrk7vIw,526
+jarvis/jarvis_treesitter/database.py,sha256=cTyytDVKGYfjybw0rEgdqUCHnmpOJc1qyMAIBQ_iXLs,10903
+jarvis/jarvis_treesitter/example.py,sha256=oYHIDMYjheE7lMUAnhxXwY7onvHBjGEyY7RGOcezfQY,4225
+jarvis/jarvis_treesitter/grammar_builder.py,sha256=cw9y0RKAJQBP87f_ocNH2NV6PEn8w8IMw1zfgDVA9c0,6048
+jarvis/jarvis_treesitter/language.py,sha256=6OKO90PrivAwd4xKlF9JiLzvqjg2beakLU_b6RADib4,3744
+jarvis/jarvis_treesitter/symbol.py,sha256=ln3xmc0LugSaqyQ_27rXAoI4zFzHdEXZ-NQCv0wV76c,860
+jarvis/jarvis_treesitter/tools_usage.md,sha256=cmrYNGh3n-eI3dyVP0yP7bQcscsn-JSRuwP4Si9KMNc,3558
+jarvis/jarvis_utils/__init__.py,sha256=THxqKNkLvhnuoZRSQHCJ2-0epoZRrtrekJPQ3savarM,824
+jarvis/jarvis_utils/config.py,sha256=j1Hqqluxo7W6BUpScjBtKUJpYKRw-DU0qlQSa8vvUtQ,4999
+jarvis/jarvis_utils/embedding.py,sha256=vaY4lTE9Yw0joQVWTX1FxV0ZnnOz6W9vjQE1NASzXIk,6110
+jarvis/jarvis_utils/git_utils.py,sha256=CmQ_SG2aIA-oO80D7plKXo4Z1lHo828YMvh47fQsGJw,4831
+jarvis/jarvis_utils/globals.py,sha256=nvykWxBnqfAFqAIyJfxP5Y1yYIXIQXjippVE5i2Bubg,2269
+jarvis/jarvis_utils/input.py,sha256=iY9_18Mb3WUzta7dbQW0DvWPAGq2iKPsRcKuPY3uN5k,6553
+jarvis/jarvis_utils/methodology.py,sha256=AoiBgBKBbq-XdSouL10zLBYTPLRYzDtBZCQiU2vE_k4,6368
+jarvis/jarvis_utils/output.py,sha256=1X4cd_M2gwL9-W2M44BwGNtmeFwFXmtC1_RvbwC_AR4,8373
+jarvis/jarvis_utils/utils.py,sha256=zosfI__QSdNIJvVsqc4zPwOcaZ3JsXieXVlWDRLr8eY,5615
+jarvis_ai_assistant-0.1.131.dist-info/LICENSE,sha256=AGgVgQmTqFvaztRtCAXsAMryUymB18gZif7_l2e1XOg,1063
+jarvis_ai_assistant-0.1.131.dist-info/METADATA,sha256=84PWHYfvxswi7XKb4Tt9DojByCrJ_3eJGokjrlaWQmU,10406
+jarvis_ai_assistant-0.1.131.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+jarvis_ai_assistant-0.1.131.dist-info/entry_points.txt,sha256=npKEpBACgZAF97wZgHSJNnturrh9DP33usD0kzjMXPo,771
+jarvis_ai_assistant-0.1.131.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
+jarvis_ai_assistant-0.1.131.dist-info/RECORD,,