jarvis-ai-assistant 0.1.176__tar.gz → 0.1.177__tar.gz

{jarvis_ai_assistant-0.1.176/src/jarvis_ai_assistant.egg-info → jarvis_ai_assistant-0.1.177}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jarvis-ai-assistant
-Version: 0.1.176
+Version: 0.1.177
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire
@@ -201,7 +201,7 @@ OPENAI_API_BASE=https://api.openai.com/v1  # 可选，默认为官方API地址
 ### 环境变量配置
 | 变量名称 | 默认值 | 说明 |
 |----------|--------|------|
-| `JARVIS_MAX_TOKEN_COUNT` | 102400000 | 上下文窗口的最大token数量 |
+| `JARVIS_MAX_TOKEN_COUNT` | 960000 | 上下文窗口的最大token数量 |
 | `JARVIS_MAX_INPUT_TOKEN_COUNT` | 32000 | 输入的最大token数量 |
 | `JARVIS_AUTO_COMPLETE` | false | 是否启用自动完成功能（任务判定完成的时候会自动终止） |
 | `JARVIS_SHELL_NAME` | bash | 系统shell名称 |
@@ -211,9 +211,9 @@ OPENAI_API_BASE=https://api.openai.com/v1  # 可选，默认为官方API地址
 | `JARVIS_THINKING_MODEL` | JARVIS_MODEL | 推理任务使用的模型 |
 | `JARVIS_EXECUTE_TOOL_CONFIRM` | false | 执行工具前是否需要确认 |
 | `JARVIS_CONFIRM_BEFORE_APPLY_PATCH` | true | 应用补丁前是否需要确认 |
-| `JARVIS_MAX_TOOL_CALL_COUNT` | 20 | 最大连续工具调用次数 |
+| `JARVIS_MAX_TOOL_CALL_COUNT` | 20 | 最大连续工具调用次数，如果是0表示无限制 |
 | `JARVIS_AUTO_UPDATE` | true | 是否自动更新Jarvis（仅在以git仓库方式安装时有效） |
-| `JARVIS_MAX_BIG_CONTENT_SIZE` | 96000 | 最大大内容大小 |
+| `JARVIS_MAX_BIG_CONTENT_SIZE` | 1024000 | 最大大内容大小 |
 | `JARVIS_PRETTY_OUTPUT` | false | 是否启用PrettyOutput |
 
 所有配置编写到`~/.jarvis/env`文件中即可生效。

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/README.md

@@ -131,7 +131,7 @@ OPENAI_API_BASE=https://api.openai.com/v1  # 可选，默认为官方API地址
 ### 环境变量配置
 | 变量名称 | 默认值 | 说明 |
 |----------|--------|------|
-| `JARVIS_MAX_TOKEN_COUNT` | 102400000 | 上下文窗口的最大token数量 |
+| `JARVIS_MAX_TOKEN_COUNT` | 960000 | 上下文窗口的最大token数量 |
 | `JARVIS_MAX_INPUT_TOKEN_COUNT` | 32000 | 输入的最大token数量 |
 | `JARVIS_AUTO_COMPLETE` | false | 是否启用自动完成功能（任务判定完成的时候会自动终止） |
 | `JARVIS_SHELL_NAME` | bash | 系统shell名称 |
@@ -141,9 +141,9 @@ OPENAI_API_BASE=https://api.openai.com/v1  # 可选，默认为官方API地址
 | `JARVIS_THINKING_MODEL` | JARVIS_MODEL | 推理任务使用的模型 |
 | `JARVIS_EXECUTE_TOOL_CONFIRM` | false | 执行工具前是否需要确认 |
 | `JARVIS_CONFIRM_BEFORE_APPLY_PATCH` | true | 应用补丁前是否需要确认 |
-| `JARVIS_MAX_TOOL_CALL_COUNT` | 20 | 最大连续工具调用次数 |
+| `JARVIS_MAX_TOOL_CALL_COUNT` | 20 | 最大连续工具调用次数，如果是0表示无限制 |
 | `JARVIS_AUTO_UPDATE` | true | 是否自动更新Jarvis（仅在以git仓库方式安装时有效） |
-| `JARVIS_MAX_BIG_CONTENT_SIZE` | 96000 | 最大大内容大小 |
+| `JARVIS_MAX_BIG_CONTENT_SIZE` | 1024000 | 最大大内容大小 |
 | `JARVIS_PRETTY_OUTPUT` | false | 是否启用PrettyOutput |
 
 所有配置编写到`~/.jarvis/env`文件中即可生效。

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/pyproject.toml

@@ -8,7 +8,7 @@ default = true
 
 [project]
 name = "jarvis-ai-assistant"
-version = "0.1.176"
+version = "0.1.177"
 description = "Jarvis: An AI assistant that uses tools to interact with the system"
 readme = "README.md"
 authors = [{ name = "skyfire", email = "skyfireitdiy@hotmail.com" }]

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="jarvis-ai-assistant",
-    version="0.1.176",
+    version="0.1.177",
     author="skyfire",
     author_email="skyfireitdiy@hotmail.com",
     description="An AI assistant that uses various tools to interact with the system",

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/src/jarvis/__init__.py

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

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/src/jarvis/jarvis_agent/__init__.py

@@ -340,15 +340,17 @@ class Agent:
         complete_prompt = f"- 输出{ot('!!!COMPLETE!!!')}" if need_complete and self.auto_complete else ""
 
         addon_prompt = f"""
-请判断是否已经完成任务，如果已经完成：
-- 说明完成原因，不需要再有新的操作
-{complete_prompt}
-如果没有完成，请进行下一步操作：
-- 仅包含一个操作
-- 如果信息不明确，请请求用户补充
-- 如果执行过程中连续失败5次，请使用ask_user询问用户操作
-- 操作列表：
-{action_handlers}
+[系统提示开始]
+    请判断是否已经完成任务，如果已经完成：
+    - 说明完成原因，不需要再有新的操作
+    {complete_prompt}
+    如果没有完成，请进行下一步操作：
+    - 仅包含一个操作
+    - 如果信息不明确，请请求用户补充
+    - 如果执行过程中连续失败5次，请使用ask_user询问用户操作
+    - 操作列表：
+    {action_handlers}
+[系统提示结束]
 
 请继续。
 """
@@ -490,7 +492,7 @@ class Agent:
             return False, f"操作失败：检测到多个操作。一次只能执行一个操作。尝试执行的操作：{', '.join([handler.name() for handler in tool_list])}"
         if len(tool_list) == 0:
             return False, ""
-        if self.tool_call_count >= self.max_tool_call_count:
+        if self.max_tool_call_count > 0 and self.tool_call_count >= self.max_tool_call_count:
             if user_confirm(f"工具调用次数超过限制，是否继续执行？", True):
                 self.reset_tool_call_count()
             else:

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/src/jarvis/jarvis_agent/jarvis.py

@@ -3,7 +3,7 @@ import argparse
 import os
 import sys
 
-from typing import Dict  # 仅保留实际使用的类型导入
+from typing import Dict, List, Tuple  # 增加必要的类型导入
 
 from jarvis.jarvis_utils.config import get_data_dir
 from prompt_toolkit import prompt
@@ -39,10 +39,10 @@ def _load_tasks() -> Dict[str, str]:
                     for name, desc in user_tasks.items():
                         if desc:
                             tasks[str(name)] = str(desc)
-                spinner.text = "预定义任务加载完成"
+                spinner.text = f"预定义任务加载完成{pre_command_path}"
                 spinner.ok("✅")
             except (yaml.YAMLError, OSError):
-                spinner.text = "预定义任务加载失败"
+                spinner.text = f"预定义任务加载失败{pre_command_path}"
                 spinner.fail("❌")
 
     # Check .jarvis/pre-command in current directory
@@ -58,10 +58,10 @@ def _load_tasks() -> Dict[str, str]:
                     for name, desc in local_tasks.items():
                         if desc:
                             tasks[str(name)] = str(desc)
-                spinner.text = "预定义任务加载完成"
+                spinner.text = f"预定义任务加载完成{pre_command_path}"
                 spinner.ok("✅")
             except (yaml.YAMLError, OSError):
-                spinner.text = "预定义任务加载失败"
+                spinner.text = f"预定义任务加载失败{pre_command_path}"
                 spinner.fail("❌")
 
     return tasks
@@ -90,6 +90,7 @@ def _select_task(tasks: Dict[str, str]) -> str:
                 return ""
             if 1 <= choice <= len(task_names):
                 selected_task = tasks[task_names[choice - 1]]
+                PrettyOutput.print(f"将要执行任务:\n {selected_task}", OutputType.INFO)
                 # 询问是否需要补充信息
                 need_additional = user_confirm("需要为此任务添加补充信息吗？", default=False)
                 if need_additional:
@@ -105,11 +106,10 @@ def _select_task(tasks: Dict[str, str]) -> str:
             PrettyOutput.print(f"选择任务失败: {str(val_err)}", OutputType.ERROR)
 
 
-
-
 def main() -> None:
-    """Jarvis main entry point"""
-    init_env()
+
+    
+    init_env("欢迎使用 Jarvis AI 助手，您的智能助理已准备就绪！")
     parser = argparse.ArgumentParser(description='Jarvis AI assistant')
     parser.add_argument('-p', '--platform', type=str, help='Platform to use')
     parser.add_argument('-m', '--model', type=str, help='Model to use')
@@ -138,7 +138,7 @@ def main() -> None:
 
         tasks = _load_tasks()
         if tasks and (selected_task := _select_task(tasks)):
-            PrettyOutput.print(f"执行任务: {selected_task}", OutputType.INFO)
+            PrettyOutput.print(f"开始执行任务: \n{selected_task}", OutputType.INFO)
             agent.run(selected_task)
             sys.exit(0)
 

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/src/jarvis/jarvis_agent/main.py

@@ -9,13 +9,13 @@ from jarvis.jarvis_utils.utils import init_env
 
 
 def load_config(config_path: str) -> dict:
-    """Load configuration from YAML file
+    """从YAML文件加载配置
 
-    Args:
-        config_path: Path to the YAML configuration file
+    参数:
+        config_path: YAML配置文件的路径
 
-    Returns:
-        dict: Configuration dictionary
+    返回:
+        dict: 配置字典
     """
     if not os.path.exists(config_path):
         PrettyOutput.print(f"配置文件 {config_path} 不存在，使用默认配置", OutputType.WARNING)
@@ -32,7 +32,7 @@ def load_config(config_path: str) -> dict:
 def main():
     """Main entry point for Jarvis agent"""
     # Initialize environment
-    init_env()
+    init_env("欢迎使用 Jarvis AI 助手，您的智能助理已准备就绪！")
 
     # Set up argument parser
     parser = argparse.ArgumentParser(description='Jarvis AI assistant')

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/src/jarvis/jarvis_code_agent/code_agent.py

@@ -31,7 +31,7 @@ from jarvis.jarvis_utils.git_utils import (
 from jarvis.jarvis_utils.input import get_multiline_input
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.utils import init_env, user_confirm
-
+from jarvis import __version__
 
 class CodeAgent:
     """Jarvis系统的代码修改代理。
@@ -372,10 +372,14 @@ class CodeAgent:
                     final_ret += f"# 应用补丁:\n```diff\n{diff}\n```"
 
                     # 修改后的提示逻辑
-                    addon_prompt = "如果对应语言有静态检查工具，请使用静态检查工具检查修改的代码，如果本次修改引入了警告和错误，请根据警告和错误信息修复代码\n"
-                    addon_prompt += "在引入警告和错误都被修复的前提下，如果用户的需求未完成，请继续修改代码，如果已经完成，请终止，不要实现任何超出用户需求外的内容\n"
-                    addon_prompt += "如果有任何信息不明确，调用工具获取信息\n"
-                    addon_prompt += "每次响应必须且只能包含一个操作\n"
+                    addon_prompt = """
+                    [系统提示开始]
+                    如果对应语言有静态检查工具，请使用静态检查工具检查修改的代码，如果本次修改引入了警告和错误，请根据警告和错误信息修复代码
+                    在引入警告和错误都被修复的前提下，如果用户的需求未完成，请继续修改代码，如果已经完成，请终止，不要实现任何超出用户需求外的内容
+                    如果有任何信息不明确，调用工具获取信息
+                    每次响应必须且只能包含一个操作
+                    [系统提示结束]
+                    """
 
                     agent.set_addon_prompt(addon_prompt)
 
@@ -403,7 +407,7 @@ class CodeAgent:
 
 def main() -> None:
     """Jarvis主入口点。"""
-    init_env()
+    init_env("欢迎使用 Jarvis-CodeAgent，您的代码工程助手已准备就绪！")
 
     parser = argparse.ArgumentParser(description='Jarvis Code Agent')
     parser.add_argument('-p', '--platform', type=str,

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/src/jarvis/jarvis_code_analysis/code_review.py

@@ -661,7 +661,7 @@ def main():
     """CLI entry point"""
     import argparse
 
-    init_env()
+    init_env("欢迎使用 Jarvis-CodeReview，您的代码审查助手已准备就绪！")
 
     parser = argparse.ArgumentParser(description='Autonomous code review tool')
     subparsers = parser.add_subparsers(dest='type')

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/src/jarvis/jarvis_dev/main.py

@@ -1257,7 +1257,7 @@ def create_dev_team() -> MultiAgent:
 def main():
     """Main entry point for the development team simulation."""
 
-    init_env()
+    init_env("欢迎使用 Jarvis-Dev，您的开发团队已准备就绪！")
 
     # Create the development team
     dev_team = create_dev_team()

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/src/jarvis/jarvis_git_details/main.py

@@ -233,7 +233,7 @@ def main():
     """主函数，用于命令行接口"""
     import argparse
 
-    init_env()
+    init_env("欢迎使用 Jarvis-GitCommitAnalyzer，您的Git Commit分析助手已准备就绪！")
 
     parser = argparse.ArgumentParser(description='Git Commit Analyzer')
     group = parser.add_mutually_exclusive_group(required=True)

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/src/jarvis/jarvis_git_squash/main.py

@@ -46,7 +46,7 @@ class GitSquashTool:
             PrettyOutput.print(f"压缩提交失败: {str(e)}", OutputType.WARNING)
 
 def main():
-    init_env()
+    init_env("欢迎使用 Jarvis-GitSquash，您的Git压缩助手已准备就绪！")
     parser = argparse.ArgumentParser(description='Git squash tool')
     parser.add_argument('commit_hash', type=str, help='Base commit hash to squash from')
     parser.add_argument('--lang', type=str, default='Chinese', help='Language for commit messages')

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/src/jarvis/jarvis_git_utils/git_commiter.py

@@ -265,7 +265,7 @@ class GitCommitTool:
             }
 
 def main():
-    init_env()
+    init_env("欢迎使用 Jarvis-GitCommitTool，您的Git提交助手已准备就绪！")
     parser = argparse.ArgumentParser(description='Git commit tool')
     parser.add_argument('--lang', type=str, default='Chinese', help='Language for commit messages')
     parser.add_argument('--root-dir', type=str, default='.', help='Root directory of the Git repository')

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/src/jarvis/jarvis_multi_agent/main.py

@@ -10,7 +10,7 @@ def main():
     Returns:
         最终处理结果
     """
-    init_env()
+    init_env("欢迎使用 Jarvis-MultiAgent，您的多智能体系统已准备就绪！")
     import argparse
     parser = argparse.ArgumentParser(description="多智能体系统启动器")
     parser.add_argument("--config", "-c", required=True, help="YAML配置文件路径")

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/src/jarvis/jarvis_platform/base.py

@@ -58,7 +58,7 @@ class BasePlatform(ABC):
 
         if input_token_count > get_max_input_token_count():
             max_chunk_size = get_max_input_token_count() - 1024  # 留出一些余量
-            min_chunk_size = max_chunk_size // 2  # 最小块大小设为最大块大小的一半
+            min_chunk_size = get_max_input_token_count() - 2048
             inputs = split_text_into_chunks(message, max_chunk_size, min_chunk_size)
             with yaspin(text="正在提交长上下文...", color="cyan") as spinner:
                 prefix_prompt = f"""
@@ -66,10 +66,13 @@ class BasePlatform(ABC):
                 """
                 while_true(lambda: while_success(lambda: self.chat(prefix_prompt), 5), 5)
                 submit_count = 0
+                length = 0
                 for input in inputs:
                     submit_count += 1
-                    spinner.text = f"正在提交第{submit_count}部分（共{len(inputs)}部分）"
-                    while_true(lambda: while_success(lambda: self.chat(f"<part_content>{input}</part_content>请返回已收到"), 5), 5)
+                    length += len(input)
+                    spinner.text = f"正在提交第{submit_count}部分（共{len(inputs)}部分({length}/{len(message)})）"
+                    list(while_true(lambda: while_success(lambda: self.chat(f"<part_content>{input}</part_content>请返回已收到"), 5), 5))
+                    spinner.write(f"提交第{submit_count}部分完成，当前进度：{length}/{len(message)}")
                 spinner.text = "提交完成"
                 spinner.ok("✅")
             response = while_true(lambda: while_success(lambda: self._chat("内容已经全部提供完毕，请继续"), 5), 5)

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/src/jarvis/jarvis_platform/yuanbao.py

@@ -294,7 +294,7 @@ class YuanbaoPlatform(BasePlatform):
             with open(file_path, 'rb') as file:
                 file_content = file.read()
 
-            spinner.write(f"ℹ️ 上传文件大小: {len(file_content)}")
+            spinner.write(f"ℹ️  上传文件大小: {len(file_content)}")
                 
             # Prepare headers for PUT request
             host = f"{upload_info['bucketName']}.{upload_info.get('accelerateDomain', 'cos.accelerate.myqcloud.com')}"

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/src/jarvis/jarvis_platform_manager/main.py

@@ -606,7 +606,7 @@ def main():
     """Main function"""
     import argparse
 
-    init_env()
+    init_env("欢迎使用 Jarvis-PlatformManager，您的平台管理助手已准备就绪！")
 
     parser = argparse.ArgumentParser(description='Jarvis AI 平台')
     subparsers = parser.add_subparsers(dest='command', help='可用子命令')

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/src/jarvis/jarvis_smart_shell/main.py

@@ -5,6 +5,8 @@ import os
 import sys
 from typing import Optional
 
+from sympy import false
+
 from jarvis.jarvis_platform.registry import PlatformRegistry
 from jarvis.jarvis_utils.config import get_shell_name
 from jarvis.jarvis_utils.input import get_multiline_input
@@ -113,8 +115,8 @@ def process_request(request: str) -> Optional[str]:
         return None
 
 def main() -> int:
-    # 创建参数解析器
-    init_env()
+    # 创建参数解析器s
+    init_env("")
     parser = argparse.ArgumentParser(
         description="将自然语言要求转换为shell命令",
         formatter_class=argparse.RawDescriptionHelpFormatter,

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/src/jarvis/jarvis_tools/ask_codebase.py

@@ -259,7 +259,7 @@ def main():
     import sys
     from jarvis.jarvis_utils.input import get_multiline_input
 
-    init_env()
+    init_env("欢迎使用 Jarvis-AskCodebase，您的智能代码库查询工具已准备就绪！")
 
     # 创建命令行参数解析器
     parser = argparse.ArgumentParser(description="智能代码库查询工具")

{jarvis_ai_assistant-0.1.176 → jarvis_ai_assistant-0.1.177}/src/jarvis/jarvis_tools/cli/main.py

@@ -29,7 +29,7 @@ def main() -> int:
     import argparse
     import json
 
-    init_env()
+    init_env("欢迎使用 Jarvis-Tools，您的工具系统已准备就绪！")
 
     parser = argparse.ArgumentParser(description="Jarvis 工具系统命令行界面")
     subparsers = parser.add_subparsers(dest="command", help="命令")
@@ -47,6 +47,10 @@ def main() -> int:
         "--args-file", type=str, help="从文件加载工具参数 (JSON格式)"
     )
 
+    # 统计子命令
+    stat_parser = subparsers.add_parser("stat", help="显示工具调用统计信息")
+    stat_parser.add_argument("--json", action="store_true", help="以JSON格式输出")
+
     args = parser.parse_args()
 
     # 初始化工具注册表
@@ -72,6 +76,29 @@ def main() -> int:
                     print(f"   参数:")
                     print(tool["parameters"])  # 显示详细参数信息
 
+    elif args.command == "stat":
+        from tabulate import tabulate
+        stats = registry._get_tool_stats()
+        tools = registry.get_all_tools()
+        
+        # 构建统计表格数据
+        table_data = []
+        for tool in tools:
+            name = tool["name"]
+            count = stats.get(name, 0)
+            table_data.append([name, count])
+        
+        # 按调用次数降序排序
+        table_data.sort(key=lambda x: x[1], reverse=True)
+        
+        if args.json:
+            print(json.dumps(dict(table_data), indent=2))
+        else:
+            PrettyOutput.section("工具调用统计", OutputType.SYSTEM)
+            print(tabulate(table_data, headers=["工具名称", "调用次数"], tablefmt="grid"))
+            
+        return 0
+
     elif args.command == "call":
         tool_name = args.tool_name
         tool_obj = registry.get_tool(tool_name)