jarvis-ai-assistant 0.1.207__py3-none-any.whl → 0.1.209__py3-none-any.whl

jarvis/jarvis_tools/edit_file.py

@@ -19,8 +19,6 @@
 """
 from typing import Any, Dict
 
-from yaspin import yaspin
-
 from jarvis.jarvis_agent.edit_file_handler import EditFileHandler
 
 
@@ -62,12 +60,18 @@ class FileSearchReplaceTool:
                             "items": {
                                 "type": "object",
                                 "properties": {
-                                    "reason": {"type": "string", "description": "修改的原因"},
+                                    "reason": {
+                                        "type": "string",
+                                        "description": "修改的原因",
+                                    },
                                     "search": {
                                         "type": "string",
                                         "description": "需要查找的原始代码",
                                     },
-                                    "replace": {"type": "string", "description": "替换后的新代码"},
+                                    "replace": {
+                                        "type": "string",
+                                        "description": "替换后的新代码",
+                                    },
                                 },
                             },
                         },
@@ -79,7 +83,6 @@ class FileSearchReplaceTool:
         "required": ["files"],
     }
 
-
     def execute(self, args: Dict) -> Dict[str, Any]:
         """执行文件编辑操作，支持快速编辑和AI辅助编辑两种模式。
 
@@ -152,46 +155,46 @@ class FileSearchReplaceTool:
                     if file_exists and agent:
                         files = agent.get_user_data("files")
                         if not files or file_path not in files:
-                            file_results.append({
-                                "file": file_path,
-                                "success": False,
-                                "stdout": "",
-                                "stderr": f"请先读取文件 {file_path} 的内容后再编辑"
-                            })
+                            file_results.append(
+                                {
+                                    "file": file_path,
+                                    "success": False,
+                                    "stdout": "",
+                                    "stderr": f"请先读取文件 {file_path} 的内容后再编辑",
+                                }
+                            )
                             continue
 
-                    with yaspin(
-                        text=f"正在处理文件 {file_path}...", color="cyan"
-                    ) as spinner:
-                        # 首先尝试fast_edit模式
-                        success, temp_content = EditFileHandler._fast_edit(file_path, changes, spinner)
+                    print(f"⚙️ 正在处理文件 {file_path}...")
+                    # 首先尝试fast_edit模式
+                    success, temp_content = EditFileHandler._fast_edit(
+                        file_path, changes
+                    )
+                    if not success:
+                        # 如果fast_edit失败，尝试slow_edit模式
+                        success, temp_content = EditFileHandler._slow_edit(
+                            file_path, changes, agent
+                        )
                         if not success:
-                            # 如果fast_edit失败，尝试slow_edit模式
-                            success, temp_content = EditFileHandler._slow_edit(
-                                file_path,
-                                changes,
-                                spinner,
-                                agent
-                            )
-                            if not success:
-                                spinner.text = f"文件 {file_path} 处理失败"
-                                spinner.fail("❌")
-                                file_results.append({
+                            print(f"❌ 文件 {file_path} 处理失败")
+                            file_results.append(
+                                {
                                     "file": file_path,
                                     "success": False,
                                     "stdout": "",
-                                    "stderr": temp_content
-                                })
-                                continue
-                            else:
-                                spinner.text = f"文件 {file_path} 内容生成完成"
-                                spinner.ok("✅")
+                                    "stderr": temp_content,
+                                }
+                            )
+                            continue
                         else:
-                            spinner.text = f"文件 {file_path} 内容生成完成"
-                            spinner.ok("✅")
+                            print(f"✅ 文件 {file_path} 内容生成完成")
+                    else:
+                        print(f"✅ 文件 {file_path} 内容生成完成")
 
                     # 只有当所有替换操作都成功时，才写回文件
-                    if success and (temp_content != original_content or not file_exists):
+                    if success and (
+                        temp_content != original_content or not file_exists
+                    ):
                         # 确保目录存在
                         os.makedirs(
                             os.path.dirname(os.path.abspath(file_path)), exist_ok=True
@@ -208,24 +211,28 @@ class FileSearchReplaceTool:
                         PrettyOutput.print(stdout_message, OutputType.SUCCESS)
                         overall_success = True
 
-                        file_results.append({
-                            "file": file_path,
-                            "success": True,
-                            "stdout": stdout_message,
-                            "stderr": ""
-                        })
+                        file_results.append(
+                            {
+                                "file": file_path,
+                                "success": True,
+                                "stdout": stdout_message,
+                                "stderr": "",
+                            }
+                        )
 
                 except Exception as e:
                     stderr_message = f"处理文件 {file_path} 时出错: {str(e)}"
                     stderr_messages.append(stderr_message)
                     PrettyOutput.print(stderr_message, OutputType.WARNING)
                     file_success = False
-                    file_results.append({
-                        "file": file_path,
-                        "success": False,
-                        "stdout": "",
-                        "stderr": stderr_message
-                    })
+                    file_results.append(
+                        {
+                            "file": file_path,
+                            "success": False,
+                            "stdout": "",
+                            "stderr": stderr_message,
+                        }
+                    )
 
             except Exception as e:
                 error_msg = f"文件搜索替换操作失败: {str(e)}"
@@ -251,12 +258,14 @@ class FileSearchReplaceTool:
                     except:
                         stderr_messages.append(f"回滚文件失败: {file_path}")
 
-                file_results.append({
-                    "file": file_path,
-                    "success": False,
-                    "stdout": "",
-                    "stderr": error_msg
-                })
+                file_results.append(
+                    {
+                        "file": file_path,
+                        "success": False,
+                        "stdout": "",
+                        "stderr": error_msg,
+                    }
+                )
 
         # 整合所有错误信息到stderr
         all_stderr = []
@@ -267,5 +276,5 @@ class FileSearchReplaceTool:
         return {
             "success": overall_success,
             "stdout": "\n".join(stdout_messages) if overall_success else "",
-            "stderr": "\n".join(all_stderr) if not overall_success else ""
+            "stderr": "\n".join(all_stderr) if not overall_success else "",
         }

jarvis/jarvis_tools/file_analyzer.py

@@ -2,8 +2,6 @@
 import os
 from typing import Any, Dict
 
-from yaspin import yaspin  # type: ignore
-from yaspin.spinners import Spinners  # type: ignore
 
 from jarvis.jarvis_platform.registry import PlatformRegistry
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
@@ -47,9 +45,6 @@ class FileAnalyzerTool:
             file_paths = args["file_paths"]
             prompt = args["prompt"]
 
-            agent = args["agent"]
-            agent.reset_tool_call_count()
-
             # 验证文件路径
             valid_files = []
             for file_path in file_paths:
@@ -78,28 +73,24 @@ class FileAnalyzerTool:
             platform.set_system_prompt(system_message)
 
             # 上传文件
-            with yaspin(Spinners.dots, text="正在上传文件...") as spinner:
-                try:
-                    with spinner.hidden():
-                        upload_result = platform.upload_files(valid_files)
-                    if not upload_result:
-                        spinner.text = "文件上传失败"
-                        spinner.fail("❌")
-                        return {
-                            "success": False,
-                            "stdout": "",
-                            "stderr": "文件上传失败",
-                        }
-                    spinner.text = "文件上传成功"
-                    spinner.ok("✅")
-                except Exception as e:
-                    spinner.text = "文件上传失败"
-                    spinner.fail("❌")
+            print(f"📤 正在上传文件...")
+            try:
+                upload_result = platform.upload_files(valid_files)
+                if not upload_result:
+                    print(f"❌ 文件上传失败")
                     return {
                         "success": False,
                         "stdout": "",
-                        "stderr": f"文件上传失败: {str(e)}",
+                        "stderr": "文件上传失败",
                     }
+                print(f"✅ 文件上传成功")
+            except Exception as e:
+                print(f"❌ 文件上传失败: {str(e)}")
+                return {
+                    "success": False,
+                    "stdout": "",
+                    "stderr": f"文件上传失败: {str(e)}",
+                }
 
             platform.set_suppress_output(False)
 
@@ -111,11 +102,9 @@ class FileAnalyzerTool:
 请提供详细的分析结果和理由。"""
 
             # 发送请求并获取分析结果
-            with yaspin(Spinners.dots, text="正在分析文件...") as spinner:
-                with spinner.hidden():
-                    analysis_result = platform.chat_until_success(analysis_request)
-                spinner.text = "分析完成"
-                spinner.ok("✅")
+            print(f"🔍 正在分析文件...")
+            analysis_result = platform.chat_until_success(analysis_request)
+            print(f"✅ 分析完成")
 
             # 清理会话
             platform.delete_chat()

jarvis/jarvis_tools/read_code.py

@@ -2,8 +2,6 @@
 import os
 from typing import Any, Dict
 
-from yaspin import yaspin
-
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
 
@@ -46,89 +44,90 @@ class ReadCodeTool:
         """
         try:
             abs_path = os.path.abspath(filepath)
-            with yaspin(text=f"正在读取文件: {abs_path}...", color="cyan") as spinner:
-                # 文件存在性检查
-                if not os.path.exists(abs_path):
-                    return {
-                        "success": False,
-                        "stdout": "",
-                        "stderr": f"文件不存在: {abs_path}",
-                    }
-
-                # 文件大小限制检查（10MB）
-                if os.path.getsize(abs_path) > 10 * 1024 * 1024:
-                    return {
-                        "success": False,
-                        "stdout": "",
-                        "stderr": "文件过大 (>10MB)",
-                    }
-
-                # 读取文件内容
-                with open(abs_path, "r", encoding="utf-8", errors="ignore") as f:
-                    lines = f.readlines()
-
-                total_lines = len(lines)
-
-                # 处理空文件情况
-                if total_lines == 0:
-                    spinner.ok("✅")
-                    return {
-                        "success": True,
-                        "stdout": f"\n🔍 文件: {abs_path}\n📄 文件为空 (0行)\n",
-                        "stderr": "",
-                    }
-
-                # 处理特殊值-1表示文件末尾
-                if end_line == -1:
-                    end_line = total_lines
-                else:
-                    end_line = (
-                        max(1, min(end_line, total_lines))
-                        if end_line >= 0
-                        else total_lines + end_line + 1
-                    )
-
-                start_line = (
-                    max(1, min(start_line, total_lines))
-                    if start_line >= 0
-                    else total_lines + start_line + 1
-                )
+            print(f"📖 正在读取文件: {abs_path}...")
+            # 文件存在性检查
+            if not os.path.exists(abs_path):
+                return {
+                    "success": False,
+                    "stdout": "",
+                    "stderr": f"文件不存在: {abs_path}",
+                }
+
+            # 文件大小限制检查（10MB）
+            if os.path.getsize(abs_path) > 10 * 1024 * 1024:
+                return {
+                    "success": False,
+                    "stdout": "",
+                    "stderr": "文件过大 (>10MB)",
+                }
+
+            # 读取文件内容
+            with open(abs_path, "r", encoding="utf-8", errors="ignore") as f:
+                lines = f.readlines()
 
-                if start_line > end_line:
-                    spinner.fail("❌")
-                    return {
-                        "success": False,
-                        "stdout": "",
-                        "stderr": f"无效的行范围 [{start_line}-{end_line}] (总行数: {total_lines})",
-                    }
-
-                # 添加行号并构建输出内容
-                selected_lines = lines[start_line - 1 : end_line]
-                numbered_content = "".join(
-                    [
-                        f"{i:4d}:{line}"
-                        for i, line in enumerate(selected_lines, start=start_line)
-                    ]
+            total_lines = len(lines)
+
+            # 处理空文件情况
+            if total_lines == 0:
+                print(f"✅ 文件读取完成: {abs_path}")
+                return {
+                    "success": True,
+                    "stdout": f"\n🔍 文件: {abs_path}\n📄 文件为空 (0行)\n",
+                    "stderr": "",
+                }
+
+            # 处理特殊值-1表示文件末尾
+            if end_line == -1:
+                end_line = total_lines
+            else:
+                end_line = (
+                    max(1, min(end_line, total_lines))
+                    if end_line >= 0
+                    else total_lines + end_line + 1
                 )
 
-                # 构建输出格式
-                output = (
-                    f"\n🔍 文件: {abs_path}\n"
-                    f"📄 原始行号: {start_line}-{end_line} (共{total_lines}行) \n\n"
-                    f"{numbered_content}\n\n"
+            start_line = (
+                max(1, min(start_line, total_lines))
+                if start_line >= 0
+                else total_lines + start_line + 1
+            )
+
+            if start_line > end_line:
+                print(
+                    f"❌ 无效的行范围 [{start_line}-{end_line}] (总行数: {total_lines})"
                 )
-                spinner.text = f"文件读取完成: {abs_path}"
-                spinner.ok("✅")
-
-                if agent:
-                    files = agent.get_user_data("files")
-                    if files:
-                        files.append(abs_path)
-                    else:
-                        files = [abs_path]
-                    agent.set_user_data("files", files)
-
-                return {"success": True, "stdout": output, "stderr": ""}
+                return {
+                    "success": False,
+                    "stdout": "",
+                    "stderr": f"无效的行范围 [{start_line}-{end_line}] (总行数: {total_lines})",
+                }
+
+            # 添加行号并构建输出内容
+            selected_lines = lines[start_line - 1 : end_line]
+            numbered_content = "".join(
+                [
+                    f"{i:4d}:{line}"
+                    for i, line in enumerate(selected_lines, start=start_line)
+                ]
+            )
+
+            # 构建输出格式
+            output = (
+                f"\n🔍 文件: {abs_path}\n"
+                f"📄 原始行号: {start_line}-{end_line} (共{total_lines}行) \n\n"
+                f"{numbered_content}\n\n"
+            )
+            print(f"✅ 文件读取完成: {abs_path}")
+
+            if agent:
+                files = agent.get_user_data("files")
+                if files:
+                    files.append(abs_path)
+                else:
+                    files = [abs_path]
+                agent.set_user_data("files", files)
+
+            return {"success": True, "stdout": output, "stderr": ""}
 
         except Exception as e:
             PrettyOutput.print(str(e), OutputType.ERROR)

jarvis/jarvis_utils/builtin_replace_map.py

@@ -29,42 +29,7 @@ arguments:
 """,
         "description": "网页搜索",
     },
-    "Plan": {
-        "append": True,
-        "template": f"""
-请使用code_plan工具生成代码修改计划，必须严格遵守以下工具调用格式：
-
-{ot("TOOL_CALL")}
-want: 想要从执行结果中获取到的信息
-name: code_plan
-arguments:
-    requirement: "需要实现用户登录功能，包括用户名密码验证和JWT生成"
-{ct("TOOL_CALL")}
-
-请提供详细的需求描述和完整上下文信息：
-
-必须包含的上下文信息：
-1. 当前会话状态 - 当前正在处理的任务和进度
-2. 用户历史请求 - 与本任务相关的历史请求
-3. 系统状态 - 相关的系统配置和环境状态
-4. 已确定的相关文件 - 已经分析出的需要修改的文件列表
-
-示例：
-1. 需要实现用户登录功能，包括用户名密码验证和JWT生成
-2. 重构订单处理模块以提高性能
-3. 优化数据库查询性能，减少响应时间
-4. 添加支付网关集成功能
-5. 修改用户权限管理系统
-
-code_plan工具将：
-1. 分析项目结构确定相关文件
-2. 理解需求后制定详细修改步骤
-3. 按功能模块分组修改内容
-4. 评估修改影响范围
-5. 生成可执行的开发计划
-""",
-        "description": "生成代码修改计划",
-    },
+    
     "FindRelatedFiles": {
         "append": False,
         "template": f"""

jarvis/jarvis_utils/config.py

@@ -3,7 +3,7 @@ import os
 from functools import lru_cache
 from typing import Any, Dict, List
 
-import yaml
+import yaml  # type: ignore
 
 from jarvis.jarvis_utils.builtin_replace_map import BUILTIN_REPLACE_MAP
 
@@ -96,25 +96,21 @@ def get_max_input_token_count() -> int:
     return int(GLOBAL_CONFIG_DATA.get("JARVIS_MAX_INPUT_TOKEN_COUNT", "32000"))
 
 
-def is_auto_complete() -> bool:
-    """
-    检查是否启用了自动补全功能。
-
-    返回：
-        bool: 如果启用了自动补全则返回True，默认为False
-    """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_AUTO_COMPLETE", False) == True
-
-
 def get_shell_name() -> str:
     """
     获取系统shell名称。
 
     返回：
-        str: Shell名称（例如bash, zsh），默认为bash
+        str: Shell名称（例如bash, zsh, fish），默认为bash
+
+    获取顺序：
+    1. 先从GLOBAL_CONFIG_DATA中获取JARVIS_SHELL配置
+    2. 再从GLOBAL_CONFIG_DATA中获取SHELL配置
+    3. 最后从环境变量SHELL获取
+    4. 如果都未配置，则默认返回bash
     """
-    shell_path = GLOBAL_CONFIG_DATA.get("SHELL", "/bin/bash")
-    return os.path.basename(shell_path)
+    shell_path = GLOBAL_CONFIG_DATA.get("SHELL", os.getenv("SHELL", "/bin/bash"))
+    return os.path.basename(shell_path).lower()
 
 
 def get_normal_platform_name() -> str:
@@ -181,16 +177,6 @@ def is_confirm_before_apply_patch() -> bool:
     return GLOBAL_CONFIG_DATA.get("JARVIS_CONFIRM_BEFORE_APPLY_PATCH", False) == True
 
 
-def get_max_tool_call_count() -> int:
-    """
-    获取最大工具调用次数。
-
-    返回：
-        int: 最大连续工具调用次数，默认为20
-    """
-    return int(GLOBAL_CONFIG_DATA.get("JARVIS_MAX_TOOL_CALL_COUNT", "20"))
-
-
 def get_data_dir() -> str:
     """
     获取Jarvis数据存储目录路径。
@@ -199,20 +185,9 @@ def get_data_dir() -> str:
         str: 数据目录路径，优先从JARVIS_DATA_PATH环境变量获取，
              如果未设置或为空，则使用~/.jarvis作为默认值
     """
-    data_path = GLOBAL_CONFIG_DATA.get("JARVIS_DATA_PATH", "").strip()
-    if not data_path:
-        return os.path.expanduser("~/.jarvis")
-    return data_path
-
-
-def get_auto_update() -> bool:
-    """
-    获取是否自动更新git仓库。
-
-    返回：
-        bool: 如果需要自动更新则返回True，默认为True
-    """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_AUTO_UPDATE", True) == True
+    return os.path.expanduser(
+        GLOBAL_CONFIG_DATA.get("JARVIS_DATA_PATH", "~/.jarvis").strip()
+    )
 
 
 def get_max_big_content_size() -> int:
@@ -265,16 +240,6 @@ def is_print_prompt() -> bool:
     return GLOBAL_CONFIG_DATA.get("JARVIS_PRINT_PROMPT", False) == True
 
 
-def get_history_count() -> int:
-    """
-    获取是否启用历史记录功能。
-
-    返回：
-        bool: 如果启用历史记录则返回True，默认为False
-    """
-    return GLOBAL_CONFIG_DATA.get("JARVIS_USE_HISTORY_COUNT", 0)
-
-
 def get_mcp_config() -> List[Dict[str, Any]]:
     """
     获取MCP配置列表。

jarvis/jarvis_utils/embedding.py

@@ -1,17 +1,11 @@
 # -*- coding: utf-8 -*-
-import functools
-import os
-from typing import Any, List
+from typing import List
 
-from jarvis.jarvis_utils.config import get_data_dir
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 
-# 全局缓存，避免重复加载模型
-_global_tokenizers = {}
-
 
 def get_context_token_count(text: str) -> int:
-    """使用分词器获取文本的token数量。
+    """使用tiktoken获取文本的token数量。
 
     参数：
         text: 要计算token的输入文本
@@ -20,16 +14,10 @@ def get_context_token_count(text: str) -> int:
         int: 文本中的token数量
     """
     try:
-        from transformers import AutoTokenizer  # type: ignore
-        tokenizer : AutoTokenizer = load_tokenizer()
-        # 分批处理长文本，确保不超过模型最大长度
-        total_tokens = 0
-        chunk_size = 100  # 每次处理100个字符，避免超过模型最大长度（考虑到中文字符可能被编码成多个token）
-        for i in range(0, len(text), chunk_size):
-            chunk = text[i : i + chunk_size]
-            tokens = tokenizer.encode(chunk)  # type: ignore
-            total_tokens += len(tokens)
-        return total_tokens
+        import tiktoken
+
+        encoding = tiktoken.get_encoding("cl100k_base")
+        return len(encoding.encode(text))
     except Exception as e:
         PrettyOutput.print(f"计算token失败: {str(e)}", OutputType.WARNING)
         return len(text) // 4  # 每个token大约4个字符的粗略估计
@@ -84,36 +72,3 @@ def split_text_into_chunks(
         PrettyOutput.print(f"文本分割失败: {str(e)}", OutputType.WARNING)
         # 发生错误时回退到简单的字符分割
         return [text[i : i + max_length] for i in range(0, len(text), max_length)]
-
-
-@functools.lru_cache(maxsize=1)
-def load_tokenizer() -> Any:
-    """
-    加载用于文本处理的分词器，使用缓存避免重复加载。
-
-    返回：
-        AutoTokenizer: 加载的分词器
-    """
-
-    from transformers import AutoTokenizer  # type: ignore
-
-    model_name = "gpt2"
-    cache_dir = os.path.join(get_data_dir(), "huggingface", "hub")
-
-    # 检查全局缓存
-    if model_name in _global_tokenizers:
-        return _global_tokenizers[model_name]
-
-    try:
-        tokenizer = AutoTokenizer.from_pretrained(
-            model_name, cache_dir=cache_dir, local_files_only=True
-        )
-    except Exception:
-        tokenizer = AutoTokenizer.from_pretrained(
-            model_name, cache_dir=cache_dir, local_files_only=False
-        )
-
-    # 保存到全局缓存
-    _global_tokenizers[model_name] = tokenizer
-
-    return tokenizer  # type: ignore