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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (121) hide show
  1. {jarvis_ai_assistant-0.1.220/src/jarvis_ai_assistant.egg-info → jarvis_ai_assistant-0.1.221}/PKG-INFO +15 -12
  2. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/README.md +3 -1
  3. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/pyproject.toml +7 -6
  4. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/setup.py +16 -13
  5. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/__init__.py +1 -1
  6. jarvis_ai_assistant-0.1.221/src/jarvis/jarvis_agent/__init__.py +578 -0
  7. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_agent/edit_file_handler.py +32 -185
  8. jarvis_ai_assistant-0.1.221/src/jarvis/jarvis_agent/prompt_builder.py +57 -0
  9. jarvis_ai_assistant-0.1.221/src/jarvis/jarvis_agent/prompts.py +188 -0
  10. jarvis_ai_assistant-0.1.221/src/jarvis/jarvis_agent/protocols.py +30 -0
  11. jarvis_ai_assistant-0.1.221/src/jarvis/jarvis_agent/session_manager.py +84 -0
  12. jarvis_ai_assistant-0.1.221/src/jarvis/jarvis_agent/tool_executor.py +49 -0
  13. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_agent/code_agent.py +4 -4
  14. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_data/config_schema.json +8 -18
  15. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_rag/__init__.py +2 -2
  16. jarvis_ai_assistant-0.1.221/src/jarvis/jarvis_rag/cache.py +85 -0
  17. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_rag/cli.py +141 -52
  18. jarvis_ai_assistant-0.1.221/src/jarvis/jarvis_rag/embedding_manager.py +95 -0
  19. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_rag/llm_interface.py +32 -34
  20. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_rag/query_rewriter.py +11 -12
  21. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_rag/rag_pipeline.py +40 -43
  22. jarvis_ai_assistant-0.1.221/src/jarvis/jarvis_rag/reranker.py +56 -0
  23. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_rag/retriever.py +29 -29
  24. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_tools/edit_file.py +11 -36
  25. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_utils/config.py +10 -25
  26. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221/src/jarvis_ai_assistant.egg-info}/PKG-INFO +15 -12
  27. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis_ai_assistant.egg-info/SOURCES.txt +5 -0
  28. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis_ai_assistant.egg-info/requires.txt +10 -8
  29. jarvis_ai_assistant-0.1.220/src/jarvis/jarvis_agent/__init__.py +0 -867
  30. jarvis_ai_assistant-0.1.220/src/jarvis/jarvis_rag/cache.py +0 -87
  31. jarvis_ai_assistant-0.1.220/src/jarvis/jarvis_rag/embedding_manager.py +0 -109
  32. jarvis_ai_assistant-0.1.220/src/jarvis/jarvis_rag/reranker.py +0 -56
  33. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/LICENSE +0 -0
  34. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/MANIFEST.in +0 -0
  35. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/setup.cfg +0 -0
  36. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_agent/builtin_input_handler.py +0 -0
  37. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_agent/jarvis.py +0 -0
  38. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_agent/main.py +0 -0
  39. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_agent/output_handler.py +0 -0
  40. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_agent/shell_input_handler.py +0 -0
  41. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_agent/__init__.py +0 -0
  42. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_agent/lint.py +0 -0
  43. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/__init__.py +0 -0
  44. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/c_cpp.py +0 -0
  45. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/csharp.py +0 -0
  46. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/data_format.py +0 -0
  47. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/devops.py +0 -0
  48. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/docs.py +0 -0
  49. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/go.py +0 -0
  50. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/infrastructure.py +0 -0
  51. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/java.py +0 -0
  52. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/javascript.py +0 -0
  53. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/kotlin.py +0 -0
  54. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/loader.py +0 -0
  55. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/php.py +0 -0
  56. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/python.py +0 -0
  57. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/ruby.py +0 -0
  58. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/rust.py +0 -0
  59. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/shell.py +0 -0
  60. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/sql.py +0 -0
  61. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/swift.py +0 -0
  62. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/checklists/web.py +0 -0
  63. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_code_analysis/code_review.py +0 -0
  64. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_data/tiktoken/9b5ad71b2ce5302211f9c61530b329a4922fc6a4 +0 -0
  65. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_git_details/__init__.py +0 -0
  66. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_git_details/main.py +0 -0
  67. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_git_squash/__init__.py +0 -0
  68. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_git_squash/main.py +0 -0
  69. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_git_utils/git_commiter.py +0 -0
  70. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_mcp/__init__.py +0 -0
  71. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_mcp/sse_mcp_client.py +0 -0
  72. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_mcp/stdio_mcp_client.py +0 -0
  73. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_mcp/streamable_mcp_client.py +0 -0
  74. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_methodology/main.py +0 -0
  75. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_multi_agent/__init__.py +0 -0
  76. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_multi_agent/main.py +0 -0
  77. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_platform/__init__.py +0 -0
  78. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_platform/ai8.py +0 -0
  79. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_platform/base.py +0 -0
  80. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_platform/human.py +0 -0
  81. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_platform/kimi.py +0 -0
  82. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_platform/openai.py +0 -0
  83. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_platform/oyi.py +0 -0
  84. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_platform/registry.py +0 -0
  85. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_platform/tongyi.py +0 -0
  86. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_platform/yuanbao.py +0 -0
  87. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_platform_manager/__init__.py +0 -0
  88. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_platform_manager/main.py +0 -0
  89. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_platform_manager/service.py +0 -0
  90. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_smart_shell/__init__.py +0 -0
  91. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_smart_shell/main.py +0 -0
  92. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_tools/__init__.py +0 -0
  93. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_tools/ask_user.py +0 -0
  94. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_tools/base.py +0 -0
  95. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_tools/cli/__init__.py +0 -0
  96. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_tools/cli/main.py +0 -0
  97. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_tools/execute_script.py +0 -0
  98. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_tools/file_analyzer.py +0 -0
  99. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_tools/generate_new_tool.py +0 -0
  100. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_tools/methodology.py +0 -0
  101. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_tools/read_code.py +0 -0
  102. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_tools/read_webpage.py +0 -0
  103. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_tools/registry.py +0 -0
  104. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_tools/rewrite_file.py +0 -0
  105. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_tools/search_web.py +0 -0
  106. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_tools/virtual_tty.py +0 -0
  107. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_utils/__init__.py +0 -0
  108. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_utils/builtin_replace_map.py +0 -0
  109. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_utils/embedding.py +0 -0
  110. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_utils/file_processors.py +0 -0
  111. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_utils/git_utils.py +0 -0
  112. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_utils/globals.py +0 -0
  113. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_utils/http.py +0 -0
  114. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_utils/input.py +0 -0
  115. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_utils/methodology.py +0 -0
  116. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_utils/output.py +0 -0
  117. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_utils/tag.py +0 -0
  118. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis/jarvis_utils/utils.py +0 -0
  119. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis_ai_assistant.egg-info/dependency_links.txt +0 -0
  120. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis_ai_assistant.egg-info/entry_points.txt +0 -0
  121. {jarvis_ai_assistant-0.1.220 → jarvis_ai_assistant-0.1.221}/src/jarvis_ai_assistant.egg-info/top_level.txt +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: jarvis-ai-assistant
3
- Version: 0.1.220
3
+ Version: 0.1.221
4
4
  Summary: Jarvis: An AI assistant that uses tools to interact with the system
5
5
  Home-page: https://github.com/skyfireitdiy/Jarvis
6
6
  Author: skyfire
@@ -59,16 +59,6 @@ Requires-Dist: pyyaml>=5.3.1
59
59
  Requires-Dist: ddgs==9.0.2
60
60
  Requires-Dist: beautifulsoup4==4.13.4
61
61
  Requires-Dist: lxml==6.0.0
62
- Requires-Dist: langchain
63
- Requires-Dist: langchain-community
64
- Requires-Dist: langchain-huggingface
65
- Requires-Dist: chromadb
66
- Requires-Dist: diskcache
67
- Requires-Dist: sentence-transformers==2.7.0
68
- Requires-Dist: torch>=2.6
69
- Requires-Dist: typer
70
- Requires-Dist: unstructured[md]
71
- Requires-Dist: rank-bm25
72
62
  Provides-Extra: dev
73
63
  Requires-Dist: pytest; extra == "dev"
74
64
  Requires-Dist: black; extra == "dev"
@@ -76,6 +66,17 @@ Requires-Dist: isort; extra == "dev"
76
66
  Requires-Dist: mypy; extra == "dev"
77
67
  Requires-Dist: build; extra == "dev"
78
68
  Requires-Dist: twine; extra == "dev"
69
+ Provides-Extra: rag
70
+ Requires-Dist: langchain; extra == "rag"
71
+ Requires-Dist: langchain-community; extra == "rag"
72
+ Requires-Dist: langchain-huggingface; extra == "rag"
73
+ Requires-Dist: chromadb; extra == "rag"
74
+ Requires-Dist: diskcache; extra == "rag"
75
+ Requires-Dist: sentence-transformers==2.7.0; extra == "rag"
76
+ Requires-Dist: torch>=2.6; extra == "rag"
77
+ Requires-Dist: typer; extra == "rag"
78
+ Requires-Dist: unstructured[md]; extra == "rag"
79
+ Requires-Dist: rank-bm25; extra == "rag"
79
80
  Dynamic: author
80
81
  Dynamic: home-page
81
82
  Dynamic: license-file
@@ -546,6 +547,7 @@ OPENAI_API_BASE: https://api.openai.com/v1
546
547
  | `JARVIS_USE_METHODOLOGY` | true | 是否启用方法论功能 |
547
548
  | `JARVIS_USE_ANALYSIS` | true | 是否启用任务分析功能 |
548
549
  | `JARVIS_DATA_PATH` | ~/.jarvis | Jarvis数据存储目录路径 |
550
+ | `JARVIS_RAG` | `{"embedding_model": "BAAI/bge-base-zh-v1.5"}` | RAG框架的配置 |
549
551
 
550
552
  ## 🛠️ 工具说明 <a id="tools"></a>
551
553
 
@@ -604,7 +606,7 @@ jarvis-rag add README.md ./docs/ 'src/jarvis/jarvis_rag/*.py'
604
606
  |---|---|
605
607
  | `paths` | **[必需]** 一个或多个文件路径、目录路径或用引号包裹的通配符模式。 |
606
608
  | `--collection` | 指定知识库的集合名称(默认为 `jarvis_rag_collection`)。 |
607
- | `--embedding-mode` | 覆盖全局配置,强制使用特定嵌入模式 (`performance` 或 `accuracy`)。 |
609
+ | `--embedding-model` | 覆盖全局配置,强制使用特定的嵌入模型名称。 |
608
610
  | `--db-path` | 覆盖全局配置,指定向量数据库的存储路径。 |
609
611
 
610
612
  ##### 2.2 查询知识库 (`query`)
@@ -631,6 +633,7 @@ jarvis-rag query "代码中的 'PlatformRegistry' 类是做什么用的?" --pl
631
633
  | `--collection` | 指定要查询的知识库集合名称。 |
632
634
  | `--platform` | 指定一个平台名称来回答问题,覆盖默认的“思考”模型。 |
633
635
  | `--model` | 指定一个模型名称来回答问题,需要与 `--platform` 同时使用。 |
636
+ | `--embedding-model` | 覆盖全局配置,强制使用特定的嵌入模型名称。 |
634
637
 
635
638
  ### 3. 命令替换功能
636
639
  支持使用特殊标记`'<tag>'`触发命令替换功能:
@@ -463,6 +463,7 @@ OPENAI_API_BASE: https://api.openai.com/v1
463
463
  | `JARVIS_USE_METHODOLOGY` | true | 是否启用方法论功能 |
464
464
  | `JARVIS_USE_ANALYSIS` | true | 是否启用任务分析功能 |
465
465
  | `JARVIS_DATA_PATH` | ~/.jarvis | Jarvis数据存储目录路径 |
466
+ | `JARVIS_RAG` | `{"embedding_model": "BAAI/bge-base-zh-v1.5"}` | RAG框架的配置 |
466
467
 
467
468
  ## 🛠️ 工具说明 <a id="tools"></a>
468
469
 
@@ -521,7 +522,7 @@ jarvis-rag add README.md ./docs/ 'src/jarvis/jarvis_rag/*.py'
521
522
  |---|---|
522
523
  | `paths` | **[必需]** 一个或多个文件路径、目录路径或用引号包裹的通配符模式。 |
523
524
  | `--collection` | 指定知识库的集合名称(默认为 `jarvis_rag_collection`)。 |
524
- | `--embedding-mode` | 覆盖全局配置,强制使用特定嵌入模式 (`performance` 或 `accuracy`)。 |
525
+ | `--embedding-model` | 覆盖全局配置,强制使用特定的嵌入模型名称。 |
525
526
  | `--db-path` | 覆盖全局配置,指定向量数据库的存储路径。 |
526
527
 
527
528
  ##### 2.2 查询知识库 (`query`)
@@ -548,6 +549,7 @@ jarvis-rag query "代码中的 'PlatformRegistry' 类是做什么用的?" --pl
548
549
  | `--collection` | 指定要查询的知识库集合名称。 |
549
550
  | `--platform` | 指定一个平台名称来回答问题,覆盖默认的“思考”模型。 |
550
551
  | `--model` | 指定一个模型名称来回答问题,需要与 `--platform` 同时使用。 |
552
+ | `--embedding-model` | 覆盖全局配置,强制使用特定的嵌入模型名称。 |
551
553
 
552
554
  ### 3. 命令替换功能
553
555
  支持使用特殊标记`'<tag>'`触发命令替换功能:
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "jarvis-ai-assistant"
7
- version = "0.1.220"
7
+ version = "0.1.221"
8
8
  description = "Jarvis: An AI assistant that uses tools to interact with the system"
9
9
  readme = "README.md"
10
10
  authors = [{ name = "skyfire", email = "skyfireitdiy@hotmail.com" }]
@@ -41,7 +41,12 @@ dependencies = [
41
41
  "ddgs==9.0.2",
42
42
  "beautifulsoup4==4.13.4",
43
43
  "lxml==6.0.0",
44
- # RAG dependencies
44
+ ]
45
+ requires-python = ">=3.9"
46
+
47
+ [project.optional-dependencies]
48
+ dev = ["pytest", "black", "isort", "mypy", "build", "twine"]
49
+ rag = [
45
50
  "langchain",
46
51
  "langchain-community",
47
52
  "langchain-huggingface",
@@ -53,10 +58,6 @@ dependencies = [
53
58
  "unstructured[md]",
54
59
  "rank-bm25",
55
60
  ]
56
- requires-python = ">=3.9"
57
-
58
- [project.optional-dependencies]
59
- dev = ["pytest", "black", "isort", "mypy", "build", "twine"]
60
61
 
61
62
  [project.urls]
62
63
  Homepage = "https://github.com/skyfireitdiy/Jarvis"
@@ -3,7 +3,7 @@ from setuptools import setup, find_packages # type: ignore
3
3
 
4
4
  setup(
5
5
  name="jarvis-ai-assistant",
6
- version="0.1.220",
6
+ version="0.1.221",
7
7
  author="skyfire",
8
8
  author_email="skyfireitdiy@hotmail.com",
9
9
  description="An AI assistant that uses various tools to interact with the system",
@@ -36,19 +36,22 @@ setup(
36
36
  "ddgs==9.0.2",
37
37
  "beautifulsoup4==4.13.4",
38
38
  "lxml==6.0.0",
39
- # RAG dependencies
40
- "langchain",
41
- "langchain-community",
42
- "langchain-huggingface",
43
- "chromadb",
44
- "diskcache",
45
- "sentence-transformers==2.7.0",
46
- "torch>=2.6",
47
- "typer",
48
- "unstructured[md]",
49
- "rank-bm25",
50
39
  ],
51
- extras_require={"dev": ["pytest", "black", "isort", "mypy", "build", "twine"]},
40
+ extras_require={
41
+ "dev": ["pytest", "black", "isort", "mypy", "build", "twine"],
42
+ "rag": [
43
+ "langchain",
44
+ "langchain-community",
45
+ "langchain-huggingface",
46
+ "chromadb",
47
+ "diskcache",
48
+ "sentence-transformers==2.7.0",
49
+ "torch>=2.6",
50
+ "typer",
51
+ "unstructured[md]",
52
+ "rank-bm25",
53
+ ],
54
+ },
52
55
  entry_points={
53
56
  "console_scripts": [
54
57
  "jarvis=jarvis.jarvis_agent.jarvis:main",
@@ -1,4 +1,4 @@
1
1
  # -*- coding: utf-8 -*-
2
2
  """Jarvis AI Assistant"""
3
3
 
4
- __version__ = "0.1.220"
4
+ __version__ = "0.1.221"