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

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,16 +96,6 @@ 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名称。
@@ -119,10 +109,6 @@ def get_shell_name() -> str:
     3. 最后从环境变量SHELL获取
     4. 如果都未配置，则默认返回bash
     """
-    shell_name = GLOBAL_CONFIG_DATA.get("JARVIS_SHELL")
-    if shell_name:
-        return shell_name.lower()
-    
     shell_path = GLOBAL_CONFIG_DATA.get("SHELL", os.getenv("SHELL", "/bin/bash"))
     return os.path.basename(shell_path).lower()
 
@@ -191,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数据存储目录路径。
@@ -209,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:
@@ -275,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

jarvis/jarvis_utils/git_utils.py

@@ -12,15 +12,15 @@ Git工具模块
 import os
 import re
 import subprocess
+import sys
 from typing import Any, Dict, List, Set, Tuple
 
-from jarvis.jarvis_utils.config import (get_auto_update,
-                                        is_confirm_before_apply_patch)
+from jarvis.jarvis_utils.config import is_confirm_before_apply_patch
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.utils import user_confirm
 
 
-def find_git_root(start_dir: str = ".") -> str:
+def find_git_root_and_cd(start_dir: str = ".") -> str:
     """
     切换到给定路径的Git根目录，如果不是Git仓库则初始化。
 
@@ -213,7 +213,9 @@ def handle_commit_workflow() -> bool:
     Returns:
         bool: 提交是否成功
     """
-    if is_confirm_before_apply_patch() and not user_confirm("是否要提交代码？", default=True):
+    if is_confirm_before_apply_patch() and not user_confirm(
+        "是否要提交代码？", default=True
+    ):
         revert_change()
         return False
 
@@ -337,13 +339,11 @@ def check_and_update_git_repo(repo_path: str) -> bool:
         bool: 是否执行了更新
     """
     curr_dir = os.path.abspath(os.getcwd())
-    git_root = find_git_root(repo_path)
+    git_root = find_git_root_and_cd(repo_path)
     if git_root is None:
         return False
 
     try:
-        if not get_auto_update():
-            return False
         # 检查是否有未提交的修改
         if has_uncommitted_changes():
             return False
@@ -394,7 +394,49 @@ def check_and_update_git_repo(repo_path: str) -> bool:
                 f"Jarvis已更新到tag {remote_tag_result.stdout.strip()}",
                 OutputType.SUCCESS,
             )
-            return True
+            
+            # 执行pip安装更新代码
+            try:
+                PrettyOutput.print("正在安装更新后的代码...", OutputType.INFO)
+                
+                # 检查是否在虚拟环境中
+                in_venv = hasattr(sys, 'real_prefix') or (
+                    hasattr(sys, 'base_prefix') and sys.base_prefix != sys.prefix
+                )
+                
+                # 尝试普通安装
+                install_cmd = [sys.executable, "-m", "pip", "install", "-e", "."]
+                result = subprocess.run(
+                    install_cmd,
+                    cwd=git_root,
+                    capture_output=True,
+                    text=True
+                )
+                
+                if result.returncode == 0:
+                    PrettyOutput.print("代码更新安装成功", OutputType.SUCCESS)
+                    return True
+                
+                # 处理权限错误
+                error_msg = result.stderr.strip()
+                if not in_venv and ("Permission denied" in error_msg or "not writeable" in error_msg):
+                    if user_confirm("检测到权限问题，是否尝试用户级安装(--user)？", True):
+                        user_result = subprocess.run(
+                            install_cmd + ["--user"],
+                            cwd=git_root,
+                            capture_output=True,
+                            text=True
+                        )
+                        if user_result.returncode == 0:
+                            PrettyOutput.print("用户级代码安装成功", OutputType.SUCCESS)
+                            return True
+                        error_msg = user_result.stderr.strip()
+                
+                PrettyOutput.print(f"代码安装失败: {error_msg}", OutputType.ERROR)
+                return False
+            except Exception as e:
+                PrettyOutput.print(f"安装过程中发生意外错误: {str(e)}", OutputType.ERROR)
+                return False
         return False
     except Exception as e:
         PrettyOutput.print(f"Git仓库更新检查失败: {e}", OutputType.WARNING)
@@ -424,7 +466,9 @@ def get_diff_file_list() -> List[str]:
         subprocess.run(["git", "reset"], check=True)
 
         if result.returncode != 0:
-            PrettyOutput.print(f"获取差异文件列表失败: {result.stderr}", OutputType.ERROR)
+            PrettyOutput.print(
+                f"获取差异文件列表失败: {result.stderr}", OutputType.ERROR
+            )
             return []
 
         return [f for f in result.stdout.splitlines() if f]
@@ -559,7 +603,9 @@ def confirm_add_new_files() -> None:
             need_confirm = True
 
         if binary_files:
-            output_lines.append(f"检测到{len(binary_files)}个二进制文件(选择N将重新检测)")
+            output_lines.append(
+                f"检测到{len(binary_files)}个二进制文件(选择N将重新检测)"
+            )
             output_lines.append("二进制文件列表:")
             output_lines.extend(f"  - {file}" for file in binary_files)
             need_confirm = True
@@ -580,7 +626,10 @@ def confirm_add_new_files() -> None:
         if not _check_conditions(new_files, added_lines, binary_files):
             break
 
-        if not user_confirm("是否要添加这些变更（如果不需要请修改.gitignore文件以忽略不需要的文件）？", False):
+        if not user_confirm(
+            "是否要添加这些变更（如果不需要请修改.gitignore文件以忽略不需要的文件）？",
+            False,
+        ):
             continue
 
         break

jarvis/jarvis_utils/http.py

@@ -0,0 +1,104 @@
+# -*- coding: utf-8 -*-
+
+import requests
+from typing import Any, Dict, Optional, Union
+
+
+def get_session() -> requests.Session:
+    """
+    获取一个永不超时的 requests.Session 对象
+
+    返回:
+        requests.Session 对象
+    """
+    session = requests.Session()
+
+    # 设置默认请求头以优化连接
+    session.headers.update(
+        {"Connection": "keep-alive"}
+    )
+
+    return session
+
+
+# 增强版本的 HTTP 请求方法（带重试机制，解决连接中断问题）
+def post(
+    url: str,
+    data: Optional[Union[Dict[str, Any], str, bytes]] = None,
+    json: Optional[Dict[str, Any]] = None,
+    **kwargs,
+) -> requests.Response:
+    """
+    发送增强版永不超时的 POST 请求，包含重试机制
+
+    参数:
+        url: 请求的 URL
+        data: (可选) 请求体数据 (表单数据或原始数据)
+        json: (可选) JSON 数据，会自动设置 Content-Type
+        **kwargs: 其他传递给 requests.post 的参数
+
+    返回:
+        requests.Response 对象
+
+    注意:
+        此方法使用增强的永不超时设置，包含自动重试机制，适用于解决"Response ended prematurely"等连接问题
+    """
+    session = get_session()
+    return session.post(url=url, data=data, json=json, **kwargs)
+
+
+def get(url: str, **kwargs) -> requests.Response:
+    """
+    发送增强版永不超时的 GET 请求，包含重试机制
+
+    参数:
+        url: 请求的 URL
+        **kwargs: 其他传递给 requests.get 的参数
+
+    返回:
+        requests.Response 对象
+
+    注意:
+        此方法使用增强的永不超时设置，包含自动重试机制，适用于解决"Response ended prematurely"等连接问题
+    """
+    session = get_session()
+    return session.get(url=url, **kwargs)
+
+
+def put(
+    url: str, data: Optional[Union[Dict[str, Any], str, bytes]] = None, **kwargs
+) -> requests.Response:
+    """
+    发送增强版永不超时的 PUT 请求，包含重试机制
+
+    参数:
+        url: 请求的 URL
+        data: (可选) 请求体数据 (表单数据或原始数据)
+        **kwargs: 其他传递给 requests.put 的参数
+
+    返回:
+        requests.Response 对象
+
+    注意:
+        此方法使用增强的永不超时设置，包含自动重试机制，适用于解决"Response ended prematurely"等连接问题
+    """
+    session = get_session()
+    return session.put(url=url, data=data, **kwargs)
+
+
+def delete(url: str, **kwargs) -> requests.Response:
+    """
+    发送增强版永不超时的 DELETE 请求，包含重试机制
+
+    参数:
+        url: 请求的 URL
+        **kwargs: 其他传递给 requests.delete 的参数
+
+    返回:
+        requests.Response 对象
+
+    注意:
+        此方法使用增强的永不超时设置，包含自动重试机制，适用于解决"Response ended prematurely"等连接问题
+    """
+    session = get_session()
+    return session.delete(url=url, **kwargs)