jarvis-ai-assistant 0.1.180__py3-none-any.whl → 0.1.181__py3-none-any.whl

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.

Potentially problematic release.


This version of jarvis-ai-assistant might be problematic. Click here for more details.

@@ -100,7 +100,7 @@ def is_auto_complete() -> bool:
100
100
  返回:
101
101
  bool: 如果启用了自动补全则返回True,默认为False
102
102
  """
103
- return GLOBAL_CONFIG_DATA.get('JARVIS_AUTO_COMPLETE', 'false') == 'true'
103
+ return GLOBAL_CONFIG_DATA.get('JARVIS_AUTO_COMPLETE', False) == True
104
104
 
105
105
 
106
106
  def get_shell_name() -> str:
@@ -154,7 +154,7 @@ def is_execute_tool_confirm() -> bool:
154
154
  返回:
155
155
  bool: 如果需要确认则返回True,默认为False
156
156
  """
157
- return GLOBAL_CONFIG_DATA.get('JARVIS_EXECUTE_TOOL_CONFIRM', 'false') == 'true'
157
+ return GLOBAL_CONFIG_DATA.get('JARVIS_EXECUTE_TOOL_CONFIRM', False) == True
158
158
  def is_confirm_before_apply_patch() -> bool:
159
159
  """
160
160
  检查应用补丁前是否需要确认。
@@ -162,7 +162,7 @@ def is_confirm_before_apply_patch() -> bool:
162
162
  返回:
163
163
  bool: 如果需要确认则返回True,默认为False
164
164
  """
165
- return GLOBAL_CONFIG_DATA.get('JARVIS_CONFIRM_BEFORE_APPLY_PATCH', 'true') == 'true'
165
+ return GLOBAL_CONFIG_DATA.get('JARVIS_CONFIRM_BEFORE_APPLY_PATCH', True) == True
166
166
 
167
167
  def get_max_tool_call_count() -> int:
168
168
  """
@@ -194,7 +194,7 @@ def get_auto_update() -> bool:
194
194
  返回:
195
195
  bool: 如果需要自动更新则返回True,默认为True
196
196
  """
197
- return GLOBAL_CONFIG_DATA.get('JARVIS_AUTO_UPDATE', 'true') == 'true'
197
+ return GLOBAL_CONFIG_DATA.get('JARVIS_AUTO_UPDATE', True) == True
198
198
 
199
199
  def get_max_big_content_size() -> int:
200
200
  """
@@ -212,7 +212,7 @@ def get_pretty_output() -> bool:
212
212
  返回:
213
213
  bool: 如果启用PrettyOutput则返回True,默认为True
214
214
  """
215
- return GLOBAL_CONFIG_DATA.get('JARVIS_PRETTY_OUTPUT', 'false') == 'true'
215
+ return GLOBAL_CONFIG_DATA.get('JARVIS_PRETTY_OUTPUT', False) == True
216
216
 
217
217
  def is_use_methodology() -> bool:
218
218
  """
@@ -221,7 +221,7 @@ def is_use_methodology() -> bool:
221
221
  返回:
222
222
  bool: 如果启用方法论则返回True,默认为True
223
223
  """
224
- return GLOBAL_CONFIG_DATA.get('JARVIS_USE_METHODOLOGY', 'false') == 'true'
224
+ return GLOBAL_CONFIG_DATA.get('JARVIS_USE_METHODOLOGY', True) == True
225
225
 
226
226
  def is_use_analysis() -> bool:
227
227
  """
@@ -230,7 +230,7 @@ def is_use_analysis() -> bool:
230
230
  返回:
231
231
  bool: 如果启用任务分析则返回True,默认为True
232
232
  """
233
- return GLOBAL_CONFIG_DATA.get('JARVIS_USE_ANALYSIS', 'false') == 'true'
233
+ return GLOBAL_CONFIG_DATA.get('JARVIS_USE_ANALYSIS', True) == True
234
234
 
235
235
  def is_print_prompt() -> bool:
236
236
  """
@@ -239,7 +239,7 @@ def is_print_prompt() -> bool:
239
239
  返回:
240
240
  bool: 如果打印提示则返回True,默认为True
241
241
  """
242
- return GLOBAL_CONFIG_DATA.get('JARVIS_PRINT_PROMPT', 'false') == 'true'
242
+ return GLOBAL_CONFIG_DATA.get('JARVIS_PRINT_PROMPT', False) == True
243
243
 
244
244
 
245
245
  def get_mcp_config() -> List[Dict[str, Any]]:
@@ -12,7 +12,7 @@ Git工具模块
12
12
  import os
13
13
  import re
14
14
  import subprocess
15
- from typing import Dict, List, Tuple
15
+ from typing import Any, Dict, List, Tuple
16
16
 
17
17
  from jarvis.jarvis_utils.config import (get_auto_update,
18
18
  is_confirm_before_apply_patch)
@@ -376,3 +376,63 @@ def get_diff_file_list() -> List[str]:
376
376
  except Exception as e:
377
377
  PrettyOutput.print(f"获取差异文件列表异常: {str(e)}", OutputType.ERROR)
378
378
  return []
379
+
380
+
381
+
382
+ def get_recent_commits_with_files() -> List[Dict[str, Any]]:
383
+ """获取最近5次提交的commit信息和文件清单
384
+
385
+ 返回:
386
+ List[Dict[str, Any]]: 包含commit信息和文件清单的字典列表,格式为:
387
+ [
388
+ {
389
+ 'hash': 提交hash,
390
+ 'message': 提交信息,
391
+ 'author': 作者,
392
+ 'date': 提交日期,
393
+ 'files': [修改的文件列表] (最多20个文件)
394
+ },
395
+ ...
396
+ ]
397
+ 失败时返回空列表
398
+ """
399
+ try:
400
+ # 获取最近5次提交的基本信息
401
+ result = subprocess.run(
402
+ ['git', 'log', '-5', '--pretty=format:%H%n%s%n%an%n%ad'],
403
+ capture_output=True,
404
+ text=True
405
+ )
406
+ if result.returncode != 0:
407
+ return []
408
+
409
+ # 解析提交信息
410
+ commits = []
411
+ lines = result.stdout.splitlines()
412
+ for i in range(0, len(lines), 4):
413
+ if i + 3 >= len(lines):
414
+ break
415
+ commit = {
416
+ 'hash': lines[i],
417
+ 'message': lines[i+1],
418
+ 'author': lines[i+2],
419
+ 'date': lines[i+3],
420
+ 'files': []
421
+ }
422
+ commits.append(commit)
423
+
424
+ # 获取每个提交的文件修改清单
425
+ for commit in commits:
426
+ files_result = subprocess.run(
427
+ ['git', 'show', '--name-only', '--pretty=format:', commit['hash']],
428
+ capture_output=True,
429
+ text=True
430
+ )
431
+ if files_result.returncode == 0:
432
+ files = list(set(filter(None, files_result.stdout.splitlines())))
433
+ commit['files'] = files[:20] # 限制最多20个文件
434
+
435
+ return commits
436
+
437
+ except subprocess.CalledProcessError:
438
+ return []
@@ -1,6 +1,7 @@
1
1
  # -*- coding: utf-8 -*-
2
2
  import hashlib
3
3
  import os
4
+ import subprocess
4
5
  import tarfile
5
6
  import time
6
7
  from pathlib import Path
@@ -133,7 +134,13 @@ def _read_old_config_file(config_file):
133
134
  if "=" in line and not line.startswith((" ", "\t")):
134
135
  # 处理之前收集的多行值
135
136
  if current_key is not None:
136
- config_data[current_key] = "\n".join(current_value).strip().strip("'").strip('"')
137
+ value = "\n".join(current_value).strip().strip("'").strip('"')
138
+ # 将字符串"true"/"false"转换为bool类型
139
+ if value.lower() == "true":
140
+ value = True
141
+ elif value.lower() == "false":
142
+ value = False
143
+ config_data[current_key] = value
137
144
  current_value = []
138
145
  # 解析新的键值对
139
146
  key, value = line.split("=", 1)
@@ -144,7 +151,13 @@ def _read_old_config_file(config_file):
144
151
  current_value.append(line.strip())
145
152
  # 处理最后一个键值对
146
153
  if current_key is not None:
147
- config_data[current_key] = "\n".join(current_value).strip().strip("'").strip('"')
154
+ value = "\n".join(current_value).strip().strip("'").strip('"')
155
+ # 将字符串"true"/"false"转换为bool类型
156
+ if value.lower() == "true":
157
+ value = True
158
+ elif value.lower() == "false":
159
+ value = False
160
+ config_data[current_key] = value
148
161
  os.environ.update({str(k): str(v) for k, v in config_data.items() if v is not None})
149
162
  set_global_env_data(config_data)
150
163
  PrettyOutput.print(f"检测到旧格式配置文件,旧格式以后将不再支持,请尽快迁移到新格式", OutputType.WARNING)
@@ -261,4 +274,20 @@ def count_cmd_usage() -> None:
261
274
 
262
275
  def is_context_overflow(content: str) -> bool:
263
276
  """判断文件内容是否超出上下文限制"""
264
- return get_context_token_count(content) > get_max_big_content_size()
277
+ return get_context_token_count(content) > get_max_big_content_size()
278
+
279
+ def get_loc_stats() -> str:
280
+ """使用loc命令获取当前目录的代码统计信息
281
+
282
+ 返回:
283
+ str: loc命令输出的原始字符串,失败时返回空字符串
284
+ """
285
+ try:
286
+ result = subprocess.run(
287
+ ['loc'],
288
+ capture_output=True,
289
+ text=True
290
+ )
291
+ return result.stdout if result.returncode == 0 else ""
292
+ except FileNotFoundError:
293
+ return ""
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: jarvis-ai-assistant
3
- Version: 0.1.180
3
+ Version: 0.1.181
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
@@ -55,6 +55,7 @@ Requires-Dist: python-Levenshtein==0.25.1
55
55
  Requires-Dist: pillow==10.2.0
56
56
  Requires-Dist: openai==1.78.1
57
57
  Requires-Dist: tabulate==0.9.0
58
+ Requires-Dist: pyte==0.8.2
58
59
  Provides-Extra: dev
59
60
  Requires-Dist: pytest; extra == "dev"
60
61
  Requires-Dist: black; extra == "dev"
@@ -212,8 +213,8 @@ OPENAI_API_BASE: https://api.openai.com/v1 # 可选,默认为官方API地址
212
213
  | `JARVIS_PRETTY_OUTPUT` | false | 是否启用PrettyOutput |
213
214
  | `JARVIS_GIT_COMMIT_PROMPT` | "" | 自定义git提交信息生成提示模板 |
214
215
  | `JARVIS_PRINT_PROMPT` | false | 是否打印提示 |
215
- | `JARVIS_USE_METHODOLOGY` | false | 是否启用方法论功能 |
216
- | `JARVIS_USE_ANALYSIS` | false | 是否启用任务分析功能 |
216
+ | `JARVIS_USE_METHODOLOGY` | true | 是否启用方法论功能 |
217
+ | `JARVIS_USE_ANALYSIS` | true | 是否启用任务分析功能 |
217
218
  | `JARVIS_DATA_PATH` | ~/.jarvis | Jarvis数据存储目录路径 |
218
219
 
219
220
  所有配置编写到`~/.jarvis/config.yaml`文件中即可生效。
@@ -232,7 +233,6 @@ ENV:
232
233
  ### 内置工具
233
234
  | 工具名称 | 描述 |
234
235
  |----------|------|
235
- | ask_codebase | 智能代码库查询和分析,用于定位功能所在文件和理解单点实现,适合查询特定功能位置和实现原理 |
236
236
  | ask_user | 交互式用户输入收集 |
237
237
  | chdir | 更改当前工作目录 |
238
238
  | rewrite_file | 文件重写工具,用于完全重写或创建文件,提供完整的文件内容替换 |
@@ -243,7 +243,6 @@ ENV:
243
243
  | execute_script | 执行脚本并返回结果,支持任意解释器。 |
244
244
  | file_analyzer | 分析文件内容并提取关键信息。支持的文件:文本文件、word文档、pdf文件、图片 |
245
245
  | file_operation | 文件批量操作工具,可批量读写多个文件,支持文本文件,适用于需要同时处理多个文件的场景(读取配置文件、保存生成内容等) |
246
- | find_methodology | 方法论查找工具,用于在执行过程中查看历史方法论辅助决策 |
247
246
  | methodology | 方法论管理工具,支持添加、更新和删除操作 |
248
247
  | read_code | 代码阅读与分析工具,用于读取源代码文件并添加行号,针对代码文件优化,提供更好的格式化输出和行号显示,适用于代码分析、审查和理解代码实现的场景 |
249
248
  | read_webpage | 读取网页内容并分析 |
@@ -1,4 +1,4 @@
1
- jarvis/__init__.py,sha256=VWahAg7gYnoP3F-d2naYtaf7HKC8ZuoYbeBZ5SH0GnQ,74
1
+ jarvis/__init__.py,sha256=dgQhg6Gd7pGul0ij7e_dHnY6CjyGHRhshKt8hag08qo,74
2
2
  jarvis/jarvis_agent/__init__.py,sha256=9vyME5JbA4koVoqJ_zv5N9eNdvu6iBkB4o4a1CmkVgA,30054
3
3
  jarvis/jarvis_agent/builtin_input_handler.py,sha256=f4DaEHPakXcAbgykFP-tiOQP6fh_yGFlZx_h91_j2tQ,1529
4
4
  jarvis/jarvis_agent/jarvis.py,sha256=UkNMVUlSNKV6y3v12eAhqc_gIDB6Obxrwk5f7-sQeiQ,6137
@@ -6,7 +6,7 @@ jarvis/jarvis_agent/main.py,sha256=GkjMTIbsd56nkVuRwD_tU_PZWyzixZZhMjVOCd0SzOA,2
6
6
  jarvis/jarvis_agent/output_handler.py,sha256=7qori-RGrQmdiFepoEe3oPPKJIvRt90l_JDmvCoa4zA,1219
7
7
  jarvis/jarvis_agent/shell_input_handler.py,sha256=pi3AtPKrkKc6K9e99S1djKXQ_XrxtP6FrSWebQmRT6E,1261
8
8
  jarvis/jarvis_code_agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
9
- jarvis/jarvis_code_agent/code_agent.py,sha256=jEJhPk4Eg-2hsDYeIa2NOfk2YnuvhuPrLCtVHm5hTPg,17879
9
+ jarvis/jarvis_code_agent/code_agent.py,sha256=IHUDqZ2RvJFoYIUuJAVCsbEOtYPHwXLfrmPXnpxKjvU,15721
10
10
  jarvis/jarvis_code_agent/lint.py,sha256=TZlhNbeaoLzO9DzExjN5GAjrt66owd8lyQV56LTfkrs,4370
11
11
  jarvis/jarvis_code_analysis/code_review.py,sha256=SEK5NOGDWgMNX3zmRMWrIOtq3Xr8cKt_asG51U8h6SA,30219
12
12
  jarvis/jarvis_code_analysis/checklists/__init__.py,sha256=cKQ_FOGy5TQgM-YkRCqORo-mUOZaPAJ9VDmZoFX58us,78
@@ -29,9 +29,9 @@ jarvis/jarvis_code_analysis/checklists/shell.py,sha256=IXQkWHwA-4GUQz3WUs7l6hEy7
29
29
  jarvis/jarvis_code_analysis/checklists/sql.py,sha256=ecKKT6wJAibn8R0NxGZDNlm4teYXvF3CAJvVk8mmX7w,2355
30
30
  jarvis/jarvis_code_analysis/checklists/swift.py,sha256=YcsYFxAitHqOtBZjG-RV9-KNM7X5lIcl6zlEI9XfmfM,2566
31
31
  jarvis/jarvis_code_analysis/checklists/web.py,sha256=-Pnj1FQTsGVZUQK7-4ptDsGd7a22Cs0585jRAPT2SdQ,3943
32
- jarvis/jarvis_data/config_schema.json,sha256=jAoVYIB4mBf6TEwpnsPCLIwsYmpwLyH_dAPB6-S6R6s,6853
32
+ jarvis/jarvis_data/config_schema.json,sha256=Jx9RPeVdkV7xgGro-5B8MFpVDLFmI80d_deLNMiTSgM,6398
33
33
  jarvis/jarvis_data/huggingface.tar.gz,sha256=dWKnc_tvyx-I_ZkXo91O0b38KxDmLW1ZbmJ3E6fCl_k,1120205
34
- jarvis/jarvis_dev/main.py,sha256=RJ7Z7_qX6ENKrlgjlz7tMfpBTh9g22mGqcasWCv5ygw,42710
34
+ jarvis/jarvis_dev/main.py,sha256=b-Sa5i5T3lcazg4GIMnECsZ3L9zTs7QCg_voWE4K0I8,40642
35
35
  jarvis/jarvis_event/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
36
36
  jarvis/jarvis_git_details/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
37
37
  jarvis/jarvis_git_details/main.py,sha256=MfR7feVVQ7Eo9eZk-wO2bFypnA6uRrYUQn6iTeoF0Os,9007
@@ -46,29 +46,27 @@ jarvis/jarvis_methodology/main.py,sha256=HhEArlKI5PCpGnBCwVrXMuDn2z84LgpgK7-aGSQ
46
46
  jarvis/jarvis_multi_agent/__init__.py,sha256=Xab5sFltJmX_9MoXqanmZs6FqKfUb2v_pG29Vk8ZXaw,4311
47
47
  jarvis/jarvis_multi_agent/main.py,sha256=KeGv8sdpSgTjW6VE4-tQ8BWDC_a0aE_4R3OqzPBd5N4,1646
48
48
  jarvis/jarvis_platform/__init__.py,sha256=0YnsUoM4JkIBOtImFdjfuDbrqQZT3dEaAwSJ62DrpCc,104
49
- jarvis/jarvis_platform/base.py,sha256=EyZOl98rpYUX1yISDFvj1ztrYfWIT2ozenwUIZ7TyyI,6972
49
+ jarvis/jarvis_platform/base.py,sha256=HbE7BVh8F5F38rr9K9281h6Q11XyWgDGzyPXe-e_Th0,7086
50
50
  jarvis/jarvis_platform/human.py,sha256=xwaTZ1zdrAYZZFXxkbHvUdECwCGsic0kgAFUncUr45g,2567
51
51
  jarvis/jarvis_platform/kimi.py,sha256=k0dYwuRf-snmJF206D7inahUcZUZG0VqOmhphj09NzQ,11969
52
52
  jarvis/jarvis_platform/openai.py,sha256=VyX3bR1rGxrJdWOtUBf8PgSL9n06KaNbOewL1urzOnk,4741
53
53
  jarvis/jarvis_platform/registry.py,sha256=CxAELjDrc-KKPPKdP71E_qaFisfQztvwc_tdf3WpOt8,7863
54
- jarvis/jarvis_platform/yuanbao.py,sha256=EABz4qLg_LJfbRfWn_8e_sEUZpD68IT98ax1q4pgaTg,20343
54
+ jarvis/jarvis_platform/yuanbao.py,sha256=vKb6oy5cTMQCwqcqpaVur7BFtQwX1Cv-mYnswP-L4mA,20291
55
55
  jarvis/jarvis_platform_manager/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
56
56
  jarvis/jarvis_platform_manager/main.py,sha256=OXWj18SqiV0Gl75YT6D9wspCCB4Nes04EY-ShI9kbpU,25677
57
57
  jarvis/jarvis_smart_shell/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
58
58
  jarvis/jarvis_smart_shell/main.py,sha256=k59o5UD7merbsPhJQzae95ThTmZY2EcNHB3Ov6kb0PA,5291
59
59
  jarvis/jarvis_tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
60
- jarvis/jarvis_tools/ask_codebase.py,sha256=FYlrbiN5PjxS4hGvCjN8idF6YnESDH6NLBgQguSnG48,10145
61
60
  jarvis/jarvis_tools/ask_user.py,sha256=qwxwJIL698rEWdi1txxlPgIr4UFuihfe--NqBEYhIQQ,2168
62
61
  jarvis/jarvis_tools/base.py,sha256=OdlvzUjYQBmZIMcAeBxAqIQo2urh126OerArK-wOPzU,1191
63
62
  jarvis/jarvis_tools/chdir.py,sha256=DNKVFrWqu6t_sZ2ipv99s6802QR4cSGlqKlmaI--arE,2707
64
- jarvis/jarvis_tools/code_plan.py,sha256=pfGOX3fH4-Gg2ECuJkjGpnRtHXEhlVWLV8ZVJAEKmwk,7755
63
+ jarvis/jarvis_tools/code_plan.py,sha256=gWR0lzY62x2PxWKoMRBqW6jq7zQuO8vhpjC4TcHSYjk,7685
65
64
  jarvis/jarvis_tools/create_code_agent.py,sha256=-nHfo5O5pDIG5IX3w1ClQafGvGcdI2_w75-KGrD-gUQ,3458
66
65
  jarvis/jarvis_tools/create_sub_agent.py,sha256=lyFrrg4V0yXULmU3vldwGp_euZjwZzJcRU6mJ20zejY,3023
67
66
  jarvis/jarvis_tools/edit_file.py,sha256=OiTP50yaUE-WTSKnp8axo96OfddHufWKpGuEa6YRG30,16269
68
- jarvis/jarvis_tools/execute_script.py,sha256=yQpc90t678UGbkr-ycGeB7jwqmY7GWWbtFtmcAI0-d4,5739
67
+ jarvis/jarvis_tools/execute_script.py,sha256=IA1SkcnwBB9PKG2voBNx5N9GXL303OC7OOtdqRfqWOk,6428
69
68
  jarvis/jarvis_tools/file_analyzer.py,sha256=7ILHkUFm8pPZn1y_s4uT0kaWHP-EmlHnpkovDdA1yRE,4872
70
69
  jarvis/jarvis_tools/file_operation.py,sha256=RcOKuMFUv01tvKoiOfu1ERCjvDVfJBvkT4oBpq-8vNQ,9036
71
- jarvis/jarvis_tools/find_methodology.py,sha256=P1IJU2x9yBoOK-X5TTvzC9Lfa1i6ES3eZn5kjEpWGdA,2440
72
70
  jarvis/jarvis_tools/generate_new_tool.py,sha256=k1Vt88kI1bYi1OwxvJqFKr3Ewwwv7lOegYNmZ-1F7x0,10283
73
71
  jarvis/jarvis_tools/methodology.py,sha256=m7cQmVhhQpUUl_uYTVvcW0JBovQLx5pWTXh_8K77HsU,5237
74
72
  jarvis/jarvis_tools/read_code.py,sha256=j4niDMOAKW_3rVxmOo3wxYBQ2T8vkxqEiTkBt28gUT8,5898
@@ -80,20 +78,20 @@ jarvis/jarvis_tools/virtual_tty.py,sha256=8E_n-eC-RRPTqYx6BI5Q2RnorY8dbhKFBfAjIi
80
78
  jarvis/jarvis_tools/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
81
79
  jarvis/jarvis_tools/cli/main.py,sha256=3UuU9tk5cQAS0rfNPXgdtnAd5uDB7v0Exo0_I9sJHRE,6355
82
80
  jarvis/jarvis_utils/__init__.py,sha256=x5lbQRH1uOulmWr1IEqNMLXNmDHbqQQot7d1uhKFg4M,851
83
- jarvis/jarvis_utils/builtin_replace_map.py,sha256=Cx0HFvMmDkkYR9htSS-PCCd8D0Np7yRm2g-C1J2-u08,4314
84
- jarvis/jarvis_utils/config.py,sha256=3m6v3ALWfnZ9yeMEIEDUFFyYuG6h2r8fEMv2yJAad_M,7139
81
+ jarvis/jarvis_utils/builtin_replace_map.py,sha256=n4gBUwAJDABUhQu9qIiIHWNfPk_T7chfNk5ygCiOPtE,2931
82
+ jarvis/jarvis_utils/config.py,sha256=2STTlI2Acy5TMHhLEZaz44MDsY4W9qSThI6tbwg7iSg,7105
85
83
  jarvis/jarvis_utils/embedding.py,sha256=J8YAqIEj16TJIPEG24uvUlPHeN-5zq0JW_hbNLizQug,3832
86
84
  jarvis/jarvis_utils/file_processors.py,sha256=G5kQI7vCGIDnjgAB5J1dYIR102u6WUv3IhcWFfDh_gs,2977
87
- jarvis/jarvis_utils/git_utils.py,sha256=WuVSoqSSrjqO4t3XAiG7SpmH9GduS8YVcfFnuLI_0ik,13172
85
+ jarvis/jarvis_utils/git_utils.py,sha256=k0rrMAbKwnD7hztmtegxtFFiCzyID4p2oHKTycE2Q-4,15070
88
86
  jarvis/jarvis_utils/globals.py,sha256=6JWtB1XoD-wEFiMzZNA790ixlZ_OsJEYUM_B8EwkOE8,2277
89
87
  jarvis/jarvis_utils/input.py,sha256=FkLW7MXL8awQUghFLQnW1r5F1wV8K3EZeVPwHFRHJTo,7458
90
88
  jarvis/jarvis_utils/methodology.py,sha256=A8pE8ZqNHvGKaDO4TFtg7Oz-hAXPBcQfhmSPWMr6vdg,8629
91
89
  jarvis/jarvis_utils/output.py,sha256=QboL42GtG_dnvd1O64sl8o72mEBhXNRADPXQMXgDE7Q,9661
92
90
  jarvis/jarvis_utils/tag.py,sha256=YJHmuedLb7_AiqvKQetHr4R1FxyzIh7HN0RRkWMmYbU,429
93
- jarvis/jarvis_utils/utils.py,sha256=ZwGMbHBQhPckq76W5v-Wbnoapy4vKX3QnvkPLesaHO0,9738
94
- jarvis_ai_assistant-0.1.180.dist-info/licenses/LICENSE,sha256=AGgVgQmTqFvaztRtCAXsAMryUymB18gZif7_l2e1XOg,1063
95
- jarvis_ai_assistant-0.1.180.dist-info/METADATA,sha256=hzV9sCqPJ3zunaI4KStiGReTpgBtM3xAKArTm-Cxacw,15291
96
- jarvis_ai_assistant-0.1.180.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
97
- jarvis_ai_assistant-0.1.180.dist-info/entry_points.txt,sha256=rjj61tZ7ahLi1R-JkJmX-IzIPPHD8mnwDZap1CnMe2s,973
98
- jarvis_ai_assistant-0.1.180.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
99
- jarvis_ai_assistant-0.1.180.dist-info/RECORD,,
91
+ jarvis/jarvis_utils/utils.py,sha256=atSK-2cUr7_tOIFsQzJnuQxebi7aFN4jtmaoXEaV4jM,10692
92
+ jarvis_ai_assistant-0.1.181.dist-info/licenses/LICENSE,sha256=AGgVgQmTqFvaztRtCAXsAMryUymB18gZif7_l2e1XOg,1063
93
+ jarvis_ai_assistant-0.1.181.dist-info/METADATA,sha256=0rpJwI0H5WiJsTHHQgmWeyeM7P4RZ7ei_L-Tnzt5ktI,15059
94
+ jarvis_ai_assistant-0.1.181.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
95
+ jarvis_ai_assistant-0.1.181.dist-info/entry_points.txt,sha256=Gy3DOP1PYLMK0GCj4rrP_9lkOyBQ39EK_lKGUSwn41E,869
96
+ jarvis_ai_assistant-0.1.181.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
97
+ jarvis_ai_assistant-0.1.181.dist-info/RECORD,,
@@ -1,8 +1,6 @@
1
1
  [console_scripts]
2
- jac = jarvis.jarvis_tools.ask_codebase:main
3
2
  jarvis = jarvis.jarvis_agent.jarvis:main
4
3
  jarvis-agent = jarvis.jarvis_agent.main:main
5
- jarvis-ask-codebase = jarvis.jarvis_tools.ask_codebase:main
6
4
  jarvis-code-agent = jarvis.jarvis_code_agent.code_agent:main
7
5
  jarvis-code-review = jarvis.jarvis_code_analysis.code_review:main
8
6
  jarvis-dev = jarvis.jarvis_dev.main:main
@@ -1,294 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- import os
3
- from typing import Any, Dict
4
-
5
- from jarvis.jarvis_agent import Agent
6
- from jarvis.jarvis_platform.registry import PlatformRegistry
7
- from jarvis.jarvis_utils.git_utils import find_git_root
8
- from jarvis.jarvis_utils.output import OutputType, PrettyOutput
9
- from jarvis.jarvis_utils.utils import init_env
10
-
11
-
12
- class AskCodebaseTool:
13
- """用于智能代码库查询和分析的工具
14
-
15
- 适用场景:
16
- - 查询特定功能所在的文件位置(支持所有文件类型)
17
- - 了解单个功能点的实现原理(支持所有文件类型)
18
- - 查找特定API或接口的用法(支持所有文件类型)
19
-
20
- 不适用场景:
21
- - 跨越多文件的大范围分析
22
- - 复杂系统架构的全面评估
23
- - 需要深入上下文理解的代码重构
24
- """
25
-
26
- name = "ask_codebase"
27
- description = "查询代码库中特定功能的位置和实现原理,适合定位功能所在文件和理解单点实现"
28
- labels = ['code', 'analysis', 'qa']
29
- parameters = {
30
- "type": "object",
31
- "properties": {
32
- "question": {
33
- "type": "string",
34
- "description": "关于代码库的问题,例如'登录功能在哪个文件实现?'或'如何实现了JWT验证?'"
35
- },
36
- "root_dir": {
37
- "type": "string",
38
- "description": "代码库根目录路径(可选)",
39
- "default": "."
40
- }
41
- },
42
- "required": ["question"]
43
- }
44
-
45
- def __init__(self, auto_complete: bool = True):
46
- self.auto_complete = auto_complete
47
-
48
- def execute(self, args: Dict[str, Any]) -> Dict[str, Any]:
49
- """Execute codebase analysis using an Agent with execute_script
50
-
51
- Args:
52
- args: Dictionary containing:
53
- - question: The question to answer, preferably about locating functionality
54
- or understanding implementation details of a specific feature
55
- - top_k: Optional number of files to analyze
56
- - root_dir: Optional root directory of the codebase
57
-
58
- Returns:
59
- Dict containing:
60
- - success: Boolean indicating success
61
- - stdout: Analysis result
62
- - stderr: Error message if any
63
-
64
- Note:
65
- This tool works best for focused questions about specific features or implementations.
66
- It is not designed for comprehensive multi-file analysis or complex architectural questions.
67
- """
68
- try:
69
- question = args["question"]
70
- root_dir = args.get("root_dir", ".")
71
-
72
- # Store current directory
73
- original_dir = os.getcwd()
74
-
75
- try:
76
- # Change to root_dir
77
- os.chdir(root_dir)
78
-
79
- # Get git root directory
80
- git_root = find_git_root() or os.getcwd()
81
-
82
- # Create system prompt for the Agent
83
- system_prompt = self._create_system_prompt(question, git_root)
84
-
85
- # Create summary prompt for the Agent
86
- summary_prompt = self._create_summary_prompt(question)
87
-
88
- # Create tools registry
89
- from jarvis.jarvis_tools.registry import ToolRegistry
90
- tool_registry = ToolRegistry()
91
- tool_registry.use_tools(["execute_script", "read_code", "methodology"])
92
-
93
- # Create and run Agent
94
- analyzer_agent = Agent(
95
- system_prompt=system_prompt,
96
- name=f"CodebaseAnalyzer",
97
- description=f"分析代码库中的功能实现和定位",
98
- summary_prompt=summary_prompt,
99
- platform=PlatformRegistry().get_thinking_platform(),
100
- output_handler=[tool_registry],
101
- execute_tool_confirm=False,
102
- auto_complete=self.auto_complete
103
- )
104
-
105
- # Run agent and get result
106
- task_input = f"回答关于代码库的问题: {question}"
107
- result = analyzer_agent.run(task_input)
108
-
109
- return {
110
- "success": True,
111
- "stdout": result,
112
- "stderr": ""
113
- }
114
- finally:
115
- # Always restore original directory
116
- os.chdir(original_dir)
117
- except Exception as e:
118
- error_msg = f"分析代码库失败: {str(e)}"
119
- PrettyOutput.print(error_msg, OutputType.WARNING)
120
- return {
121
- "success": False,
122
- "stdout": "",
123
- "stderr": error_msg
124
- }
125
-
126
- def _create_system_prompt(self, question: str, git_root: str) -> str:
127
- """创建Agent的system prompt
128
-
129
- Args:
130
- question: 用户问题
131
- git_root: Git仓库根目录
132
-
133
- Returns:
134
- 系统提示文本
135
- """
136
- return f"""# 代码库分析专家
137
-
138
- ## 任务描述
139
- 分析代码库,找出与用户问题最相关的信息,提供准确、具体的回答。
140
-
141
- ## 问题信息
142
- - 问题: {question}
143
- - 代码库根目录: {git_root}
144
-
145
- ## 工具使用优先级
146
- 1. **绝对优先使用 execute_script**:
147
- - 使用 fd 查找文件: `fd -t f -e py` 查找Python文件等
148
- - 使用 rg 搜索代码: `rg "pattern" --type py` 在Python文件中搜索等
149
- - 使用 loc 统计代码: `loc"` 统计代码量
150
-
151
- 2. **优先使用 read_code**:
152
- - 找到相关文件后优先使用read_code读取文件内容
153
- - 对大文件使用行范围参数读取指定区域
154
-
155
- ## 分析策略
156
- 1. 首先理解问题,确定需要查找的关键信息和代码组件
157
- 2. 使用fd命令查找可能相关的文件
158
- 3. 使用rg命令搜索关键词和代码模式
159
- 4. 使用read_code工具直接读取和分析相关文件内容
160
- 5. 根据文件内容提供具体、准确的回答
161
- 6. 确保分析的完整性,收集充分的信息后再得出结论,不要在只掌握部分信息就得出结论
162
- 7. 优先查阅README文件、文档目录和项目文档
163
- 8. 给出的任意结论都要有实际的代码支撑,不能假设和杜撰
164
-
165
- ## 分析步骤
166
- 1. **确定项目的编程语言**:
167
- - 使用loc命令: `loc` 统计各类文件数量
168
- - 查看构建文件和配置文件
169
- - 了解项目使用的框架和主要依赖
170
-
171
- 2. **探索代码库结构**:
172
- - 使用 `fd -t d` 了解目录结构
173
- - 使用 `fd README.md` 查找README文件
174
- - 使用 `fd -e md -g "doc*"` 查找文档文件
175
-
176
- 3. **定位相关文件**:
177
- - 使用fd和rg命令查找关键词
178
- - 示例: `rg -w "登录|login" --type py` 查找登录相关代码
179
- - 示例: `fd -e py -g "*auth*"` 查找认证相关文件
180
-
181
- 4. **深入分析代码**:
182
- - 使用read_code工具直接读取文件内容
183
- - 分析关键文件的实现细节
184
- - 使用rg识别功能的实现方式和关键逻辑
185
-
186
- 5. **回答问题**:
187
- - 提供基于直接分析代码的具体回答
188
- - 引用具体文件和代码片段作为依据
189
-
190
- ## 输出要求
191
- - 提供准确、具体的回答,避免模糊不清的描述
192
- - 引用具体文件路径和代码片段作为依据
193
- - 如果无法找到答案,明确说明并提供原因
194
- - 组织信息的逻辑清晰,便于理解
195
- - 对复杂概念提供简明解释
196
- - 确保全面收集相关信息后再形成结论
197
- - 明确区分已验证的信息和待验证的信息"""
198
-
199
- def _create_summary_prompt(self, question: str) -> str:
200
- """创建Agent的summary prompt
201
-
202
- Args:
203
- question: 用户问题
204
-
205
- Returns:
206
- 总结提示文本
207
- """
208
- return f"""# 代码库分析报告
209
-
210
- ## 报告要求
211
- 生成关于以下问题的清晰、准确的分析报告:
212
-
213
- **问题**: {question}
214
-
215
- 报告应包含:
216
-
217
- 1. **项目基本信息**:
218
- - 项目的主要编程语言和技术栈
219
- - 项目的主要框架和依赖
220
- - 项目的基本结构
221
-
222
- 2. **问题回答**:
223
- - 直接、具体地回答问题
224
- - 基于代码库中的实际代码
225
- - 避免模糊或推测性的回答
226
-
227
- 3. **核心发现**:
228
- - 相关文件和代码位置
229
- - 关键实现细节
230
- - 功能运作机制
231
-
232
- 4. **证据引用**:
233
- - 引用具体文件路径
234
- - 包含关键代码片段
235
- - 解释代码如何支持你的回答
236
-
237
- 5. **局限性**(如有):
238
- - 指出分析的任何局限
239
- - 说明任何无法确定的信息
240
-
241
- 6. **分析完整性**:
242
- - 确保在得出结论前已全面收集和分析相关信息
243
- - 避免基于部分信息形成不完整或偏颇的判断
244
- - 明确标识哪些是已确认的结论,哪些可能需要进一步验证
245
-
246
- 使用清晰的Markdown格式,重点突出对问题的回答和支持证据。"""
247
-
248
-
249
- def main():
250
- """
251
- 命令行入口点,允许将ask_codebase作为独立脚本运行
252
-
253
- 用法示例:
254
- ```
255
- python -m jarvis.jarvis_tools.ask_codebase "登录功能在哪个文件实现?" --root_dir /path/to/codebase
256
- ```
257
- 如果没有提供问题参数,则会进入交互式多行输入模式
258
- """
259
- import argparse
260
- import sys
261
-
262
- from jarvis.jarvis_utils.input import get_multiline_input
263
-
264
- init_env("欢迎使用 Jarvis-AskCodebase,您的智能代码库查询工具已准备就绪!")
265
-
266
- # 创建命令行参数解析器
267
- parser = argparse.ArgumentParser(description="智能代码库查询工具")
268
- parser.add_argument("question", nargs="?", help="关于代码库的问题")
269
- parser.add_argument("--root_dir", "-d", default=".", help="代码库根目录路径")
270
-
271
- # 解析命令行参数
272
- args = parser.parse_args()
273
-
274
- # 如果没有提供问题参数,使用多行输入
275
- if not args.question:
276
- args.question = get_multiline_input("请输入关于代码库的问题:")
277
-
278
- # 创建并执行工具
279
- tool = AskCodebaseTool(auto_complete=False)
280
- result = tool.execute({
281
- "question": args.question,
282
- "root_dir": args.root_dir
283
- })
284
-
285
- # 输出结果
286
- if result["success"]:
287
- PrettyOutput.print(result["stdout"], OutputType.SUCCESS)
288
- else:
289
- PrettyOutput.print(result["stderr"], OutputType.WARNING)
290
- sys.exit(1)
291
-
292
-
293
- if __name__ == "__main__":
294
- main()