jarvis-ai-assistant 0.1.148__tar.gz → 0.1.149__tar.gz

{jarvis_ai_assistant-0.1.148/src/jarvis_ai_assistant.egg-info → jarvis_ai_assistant-0.1.149}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: jarvis-ai-assistant
-Version: 0.1.148
+Version: 0.1.149
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire
@@ -40,32 +40,23 @@ Classifier: Programming Language :: Python :: 3.11
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: requests>=2.25.1
-Requires-Dist: pyyaml>=5.1
-Requires-Dist: colorama>=0.4.6
-Requires-Dist: prompt_toolkit>=3.0.0
-Requires-Dist: openai>=1.20.0
-Requires-Dist: numpy>=1.19.5
-Requires-Dist: faiss-cpu>=1.8.0
-Requires-Dist: sentence-transformers>=2.2.2
-Requires-Dist: bs4>=0.0.1
-Requires-Dist: PyMuPDF>=1.21.0
-Requires-Dist: python-docx>=0.8.11
-Requires-Dist: tiktoken>=0.3.0
-Requires-Dist: tqdm>=4.65.0
-Requires-Dist: docx>=0.2.4
-Requires-Dist: yaspin>=2.4.0
-Requires-Dist: rich>=13.3.1
-Requires-Dist: pygments>=2.15.0
-Requires-Dist: fuzzywuzzy>=0.18.0
-Requires-Dist: python-Levenshtein>=0.25.0
-Requires-Dist: jedi>=0.17.2
-Requires-Dist: psutil>=7.0.0
-Requires-Dist: fastapi>=0.115.4
-Requires-Dist: uvicorn>=0.33.0
-Requires-Dist: python-pptx>=1.0.0
-Requires-Dist: pandas>=2.0.0
-Requires-Dist: html2text>=2024.2.26
+Requires-Dist: requests==2.32.3
+Requires-Dist: colorama==0.4.6
+Requires-Dist: prompt_toolkit==3.0.50
+Requires-Dist: PyMuPDF==1.24.11
+Requires-Dist: yaspin==2.4.0
+Requires-Dist: pygments==2.19.1
+Requires-Dist: fuzzywuzzy==0.18.0
+Requires-Dist: jedi==0.19.2
+Requires-Dist: fastapi==0.115.12
+Requires-Dist: uvicorn==0.33.0
+Requires-Dist: pandas==2.0.3
+Requires-Dist: rich==14.0.0
+Requires-Dist: transformers==4.46.3
+Requires-Dist: python-docx==1.1.2
+Requires-Dist: python-pptx==1.0.2
+Requires-Dist: torch==2.4.1
+Requires-Dist: python-Levenshtein==0.25.1
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: black; extra == "dev"
@@ -82,6 +73,7 @@ Requires-Dist: mypy; extra == "dev"
 
 *您的智能开发和系统交互助手*
 
+[视频介绍](#video-introduction) •
 [快速开始](#quick-start) •
 [配置说明](#configuration) •
 [工具说明](#tools) •
@@ -92,6 +84,11 @@ Requires-Dist: mypy; extra == "dev"
 
 ---
 
+## 📺 视频介绍<a id="video-introduction"></a>
+
+[![视频介绍](docs/images/intro.png)](https://player.bilibili.com/player.html?isOutside=true&aid=114306578382907&bvid=BV1x2dAYeEpM&cid=29314583629&p=1)
+
+
 ## 🚀 快速开始 <a id="quick-start"></a>
 ### 安装
 ```bash

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/README.md

@@ -8,6 +8,7 @@
 
 *您的智能开发和系统交互助手*
 
+[视频介绍](#video-introduction) •
 [快速开始](#quick-start) •
 [配置说明](#configuration) •
 [工具说明](#tools) •
@@ -18,6 +19,11 @@
 
 ---
 
+## 📺 视频介绍<a id="video-introduction"></a>
+
+[![视频介绍](docs/images/intro.png)](https://player.bilibili.com/player.html?isOutside=true&aid=114306578382907&bvid=BV1x2dAYeEpM&cid=29314583629&p=1)
+
+
 ## 🚀 快速开始 <a id="quick-start"></a>
 ### 安装
 ```bash

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "jarvis-ai-assistant"
-version = "0.1.148"
+version = "0.1.149"
 description = "Jarvis: An AI assistant that uses tools to interact with the system"
 readme = "README.md"
 authors = [{ name = "skyfire", email = "skyfireitdiy@hotmail.com" }]
@@ -22,32 +22,23 @@ classifiers = [
 ]
 keywords = ["jarvis", "ai", "assistant", "tools", "automation"]
 dependencies = [
-    "requests>=2.25.1",
-    "pyyaml>=5.1",
-    "colorama>=0.4.6",
-    "prompt_toolkit>=3.0.0",
-    "openai>=1.20.0",
-    "numpy>=1.19.5",
-    "faiss-cpu>=1.8.0",
-    "sentence-transformers>=2.2.2",
-    "bs4>=0.0.1",
-    "PyMuPDF>=1.21.0",
-    "python-docx>=0.8.11",
-    "tiktoken>=0.3.0",
-    "tqdm>=4.65.0",
-    "docx>=0.2.4",
-    "yaspin>=2.4.0",
-    "rich>=13.3.1",
-    "pygments>=2.15.0",
-    "fuzzywuzzy>=0.18.0",
-    "python-Levenshtein>=0.25.0",
-    "jedi>=0.17.2",
-    "psutil>=7.0.0",
-    "fastapi>=0.115.4",
-    "uvicorn>=0.33.0",
-    "python-pptx>=1.0.0",
-    "pandas>=2.0.0",
-    "html2text>=2024.2.26",
+    "requests==2.32.3",
+    "colorama==0.4.6",
+    "prompt_toolkit==3.0.50",
+    "PyMuPDF==1.24.11",
+    "yaspin==2.4.0",
+    "pygments==2.19.1",
+    "fuzzywuzzy==0.18.0",
+    "jedi==0.19.2",
+    "fastapi==0.115.12",
+    "uvicorn==0.33.0",
+    "pandas==2.0.3",
+    "rich==14.0.0",
+    "transformers==4.46.3",
+    "python-docx==1.1.2",
+    "python-pptx==1.0.2",
+    "torch==2.4.1",
+    "python-Levenshtein==0.25.1",
 ]
 requires-python = ">=3.8"
 

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="jarvis-ai-assistant",
-    version="0.1.148",
+    version="0.1.149",
     author="skyfire",
     author_email="skyfireitdiy@hotmail.com",
     description="An AI assistant that uses various tools to interact with the system",
@@ -13,32 +13,23 @@ setup(
     package_dir={"": "src"},
     include_package_data=True,
     install_requires=[
-        "requests>=2.25.1",
-        "pyyaml>=5.1",
-        "colorama>=0.4.6",
-        "prompt_toolkit>=3.0.0",
-        "openai>=1.20.0",
-        "numpy>=1.19.5",
-        "faiss-cpu>=1.8.0",
-        "sentence-transformers>=2.2.2",
-        "bs4>=0.0.1",
-        "PyMuPDF>=1.21.0",
-        "python-docx>=0.8.11",
-        "tiktoken>=0.3.0",
-        "tqdm>=4.65.0",
-        "docx>=0.2.4",
-        "yaspin>=2.4.0",
-        "rich>=13.3.1",
-        "pygments>=2.15.0",
-        "fuzzywuzzy>=0.18.0",
-        "python-Levenshtein>=0.25.0",
-        "jedi>=0.17.2",
-        "psutil>=7.0.0",
-        "fastapi>=0.115.4",
-        "uvicorn>=0.33.0",
-        "python-pptx>=1.0.0",
-        "pandas>=2.0.0",
-        "html2text>=2024.2.26",
+        "requests==2.32.3",
+        "colorama==0.4.6",
+        "prompt_toolkit==3.0.50",
+        "PyMuPDF==1.24.11",
+        "yaspin==2.4.0",
+        "pygments==2.19.1",
+        "fuzzywuzzy==0.18.0",
+        "jedi==0.19.2",
+        "fastapi==0.115.12",
+        "uvicorn==0.33.0",
+        "pandas==2.0.3",
+        "rich==14.0.0",
+        "transformers==4.46.3",
+        "python-docx==1.1.2",
+        "python-pptx==1.0.2",
+        "torch==2.4.1",
+        "python-Levenshtein==0.25.1",
     ],
     entry_points={
         "console_scripts": [

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/__init__.py

@@ -1,3 +1,3 @@
 """Jarvis AI Assistant"""
 
-__version__ = "0.1.148"
+__version__ = "0.1.149"

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/jarvis_agent/main.py

@@ -1,7 +1,6 @@
 import argparse
 import yaml
 import os
-from typing import Optional, List
 from jarvis.jarvis_agent import Agent
 from jarvis.jarvis_utils.input import get_multiline_input
 from jarvis.jarvis_utils.output import PrettyOutput, OutputType

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/jarvis_agent/patch.py

@@ -238,7 +238,8 @@ def revert_file(filepath: str):
         # 检查文件是否在版本控制中
         result = subprocess.run(
             ['git', 'ls-files', '--error-unmatch', filepath],
-            stderr=subprocess.PIPE
+            stderr=subprocess.PIPE,
+            text=False  # 禁用自动文本解码
         )
         if result.returncode == 0:
             subprocess.run(['git', 'checkout', 'HEAD',
@@ -248,33 +249,67 @@ def revert_file(filepath: str):
                 os.remove(filepath)
         subprocess.run(['git', 'clean', '-f', '--', filepath], check=True)
     except subprocess.CalledProcessError as e:
-        PrettyOutput.print(f"恢复文件失败: {str(e)}", OutputType.ERROR)
+        error_msg = e.stderr.decode('utf-8', errors='replace') if e.stderr else str(e)
+        PrettyOutput.print(f"恢复文件失败: {error_msg}", OutputType.ERROR)
 # 修改后的恢复函数
 
 
 def revert_change():
+    """恢复所有未提交的修改到HEAD状态"""
     import subprocess
-    subprocess.run(['git', 'reset', '--hard', 'HEAD'], check=True)
-    subprocess.run(['git', 'clean', '-fd'], check=True)
+    try:
+        # 检查是否为空仓库
+        head_check = subprocess.run(
+            ['git', 'rev-parse', '--verify', 'HEAD'],
+            stderr=subprocess.PIPE,
+            stdout=subprocess.PIPE
+        )
+        if head_check.returncode == 0:
+            subprocess.run(['git', 'reset', '--hard', 'HEAD'], check=True)
+        subprocess.run(['git', 'clean', '-fd'], check=True)
+    except subprocess.CalledProcessError as e:
+        return f"恢复更改失败: {str(e)}"
 # 修改后的获取差异函数
 
 
 def get_diff() -> str:
     """使用git获取暂存区差异"""
     import subprocess
+    
+    # 初始化状态
+    need_reset = False
+    
     try:
+        # 暂存所有修改
         subprocess.run(['git', 'add', '.'], check=True)
+        need_reset = True
+        
+        # 获取差异
         result = subprocess.run(
             ['git', 'diff', '--cached'],
             capture_output=True,
-            text=True,
+            text=False,
             check=True
         )
-        ret = result.stdout
-        subprocess.run(['git', "reset", "--mixed", "HEAD"], check=True)
+        
+        # 解码输出
+        try:
+            ret = result.stdout.decode('utf-8')
+        except UnicodeDecodeError:
+            ret = result.stdout.decode('utf-8', errors='replace')
+        
+        # 重置暂存区
+        subprocess.run(['git', "reset", "--mixed"], check=False)
         return ret
+        
     except subprocess.CalledProcessError as e:
+        if need_reset:
+            subprocess.run(['git', "reset", "--mixed"], check=False)
         return f"获取差异失败: {str(e)}"
+    except Exception as e:
+        if need_reset:
+            subprocess.run(['git', "reset", "--mixed"], check=False)
+        return f"发生意外错误: {str(e)}"
 
 
 def handle_commit_workflow() -> bool:

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/jarvis_code_analysis/code_review.py

@@ -4,7 +4,6 @@ import os
 import re
 import tempfile
 
-from httpx import get
 from yaspin import yaspin
 from jarvis.jarvis_platform.registry import PlatformRegistry
 from jarvis.jarvis_tools.read_code import ReadCodeTool

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/jarvis_git_utils/git_commiter.py

@@ -3,7 +3,6 @@ import shlex
 import subprocess
 from typing import Dict, Any, Optional
 import tempfile
-from click import Option
 import yaml
 from yaspin import yaspin
 from jarvis.jarvis_platform.registry import PlatformRegistry

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/jarvis_lsp/base.py

@@ -1,5 +1,5 @@
 from abc import ABC, abstractmethod
-from typing import List, Dict, Optional, Tuple, Any, Union
+from typing import List, Dict, Any, Union
 
 class BaseLSP(ABC):
     """Base class for Language Server Protocol integration.

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/jarvis_lsp/cpp.py

@@ -1,7 +1,7 @@
 import os
 import shutil
 import subprocess
-from typing import List, Dict, Optional, Tuple, Any
+from typing import List, Dict, Optional, Any
 import json
 from jarvis.jarvis_lsp.base import BaseLSP
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/jarvis_lsp/go.py

@@ -1,7 +1,7 @@
 import os
 import shutil
 import subprocess
-from typing import List, Dict, Optional, Tuple, Any
+from typing import List, Dict, Optional, Any
 import json
 from jarvis.jarvis_lsp.base import BaseLSP
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/jarvis_lsp/python.py

@@ -1,4 +1,4 @@
-from typing import List, Dict, Optional, Tuple, Any
+from typing import List, Dict, Any
 import jedi
 from jarvis.jarvis_lsp.base import BaseLSP
 

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/jarvis_lsp/rust.py

@@ -1,7 +1,7 @@
 import os
 import shutil
 import subprocess
-from typing import List, Dict, Optional, Tuple, Any
+from typing import List, Dict, Optional, Any
 import json
 from jarvis.jarvis_lsp.base import BaseLSP
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/jarvis_platform/base.py

@@ -1,6 +1,6 @@
 from abc import ABC, abstractmethod
 import re
-from typing import Dict, List, Tuple
+from typing import List, Tuple
 from jarvis.jarvis_utils.globals import clear_read_file_record
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.utils import ct, ot, get_context_token_count, while_success, while_true

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/jarvis_platform/kimi.py

@@ -7,7 +7,6 @@ import time
 from jarvis.jarvis_platform.base import BasePlatform
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.utils import while_success
-from yaspin import yaspin
 
 class KimiModel(BasePlatform):
     """Kimi model implementation"""

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/jarvis_tools/code_plan.py

@@ -15,7 +15,6 @@ from jarvis.jarvis_agent import Agent
 from jarvis.jarvis_platform.registry import PlatformRegistry
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.git_utils import find_git_root
-from jarvis.jarvis_utils.utils import init_env
 
 class CodePlanTool:
     """用于代码修改规划和需求分析的工具

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/jarvis_tools/file_analyzer.py

@@ -1,4 +1,4 @@
-from typing import Dict, Any, List
+from typing import Dict, Any
 import os
 
 from jarvis.jarvis_platform.registry import PlatformRegistry

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/jarvis_tools/read_code.py

@@ -1,7 +1,6 @@
 from typing import Dict, Any
 import os
 
-from pkg_resources import add_activation_listener
 from yaspin import yaspin
 
 from jarvis.jarvis_utils.globals import add_read_file_record

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/jarvis_tools/search_web.py

@@ -1,5 +1,4 @@
 import os
-import statistics
 from typing import Any, Dict
 from jarvis.jarvis_platform.registry import PlatformRegistry
 

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/jarvis_utils/embedding.py

@@ -1,12 +1,8 @@
 import os
-import numpy as np
-import torch
-from sentence_transformers import SentenceTransformer
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
-from typing import List, Any, Optional, Tuple
+from transformers import AutoTokenizer
+from typing import List
 import functools
 
-from yaspin.api import Yaspin
 from jarvis.jarvis_utils.output import PrettyOutput, OutputType
 
 # 全局缓存，避免重复加载模型

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/jarvis_utils/git_utils.py

@@ -74,14 +74,16 @@ def get_commits_between(start_hash: str, end_hash: str) -> List[Tuple[str, str]]
             ['git', 'log', f'{start_hash}..{end_hash}', '--pretty=format:%H|%s'],
             stdout=subprocess.PIPE,
             stderr=subprocess.PIPE,
-            text=True
+            text=False  # 禁用自动文本解码
         )
         if result.returncode != 0:
-            PrettyOutput.print(f"获取commit历史失败: {result.stderr}", OutputType.ERROR)
+            error_msg = result.stderr.decode('utf-8', errors='replace')
+            PrettyOutput.print(f"获取commit历史失败: {error_msg}", OutputType.ERROR)
             return []
 
+        output = result.stdout.decode('utf-8', errors='replace')
         commits = []
-        for line in result.stdout.splitlines():
+        for line in output.splitlines():
             if '|' in line:
                 commit_hash, message = line.split('|', 1)
                 commits.append((commit_hash, message))
@@ -94,18 +96,27 @@ def get_latest_commit_hash() -> str:
     """获取当前Git仓库的最新提交哈希值
 
     返回：
-        str: 提交哈希值，如果不在Git仓库或发生错误则返回空字符串
+        str: 提交哈希值，如果不在Git仓库、空仓库或发生错误则返回空字符串
     """
     try:
+        # 首先检查是否存在HEAD引用
+        head_check = subprocess.run(
+            ['git', 'rev-parse', '--verify', 'HEAD'],
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            text=False
+        )
+        if head_check.returncode != 0:
+            return ""  # 空仓库或无效HEAD
+
+        # 获取HEAD的完整哈希值
         result = subprocess.run(
             ['git', 'rev-parse', 'HEAD'],
             stdout=subprocess.PIPE,
             stderr=subprocess.PIPE,
-            text=True
+            text=False
         )
-        if result.returncode == 0:
-            return result.stdout.strip()
-        return ""
+        return result.stdout.decode('utf-8', errors='replace').strip() if result.returncode == 0 else ""
     except Exception:
         return ""
 def get_modified_line_ranges() -> Dict[str, Tuple[int, int]]:

{jarvis_ai_assistant-0.1.148 → jarvis_ai_assistant-0.1.149}/src/jarvis/jarvis_utils/output.py

@@ -11,7 +11,6 @@ from enum import Enum
 from datetime import datetime
 from typing import Optional
 from rich.panel import Panel
-from rich.box import HEAVY
 from rich.text import Text
 from rich.syntax import Syntax
 from rich.style import Style as RichStyle

jarvis_ai_assistant-0.1.149/src/jarvis/jarvis_utils/utils.py

@@ -0,0 +1,154 @@
+import os
+import time
+import hashlib
+from pathlib import Path
+from typing import List, Any, Callable
+from jarvis.jarvis_utils.config import get_max_input_token_count
+from jarvis.jarvis_utils.embedding import get_context_token_count
+from jarvis.jarvis_utils.input import get_single_line_input
+from jarvis.jarvis_utils.output import PrettyOutput, OutputType
+def init_env() -> None:
+    """初始化环境变量从~/.jarvis/env文件
+
+    功能：
+    1. 创建不存在的.jarvis目录
+    2. 加载环境变量到os.environ
+    3. 处理文件读取异常
+    """
+    jarvis_dir = Path.home() / ".jarvis"
+    env_file = jarvis_dir / "env"
+
+    # 检查~/.jarvis目录是否存在
+    if not jarvis_dir.exists():
+        jarvis_dir.mkdir(parents=True)
+    if env_file.exists():
+        try:
+            with open(env_file, "r", encoding="utf-8", errors="ignore") as f:
+                for line in f:
+                    line = line.strip()
+                    if line and not line.startswith(("#", ";")):
+                        try:
+                            key, value = line.split("=", 1)
+                            os.environ[key.strip()] = value.strip().strip("'").strip('"')
+                        except ValueError:
+                            continue
+        except Exception as e:
+            PrettyOutput.print(f"警告: 读取 {env_file} 失败: {e}", OutputType.WARNING)
+def while_success(func: Callable[[], Any], sleep_time: float = 0.1) -> Any:
+    """循环执行函数直到成功
+
+    参数：
+    func -- 要执行的函数
+    sleep_time -- 每次失败后的等待时间（秒）
+
+    返回：
+    函数执行结果
+    """
+    while True:
+        try:
+            return func()
+        except Exception as e:
+            PrettyOutput.print(f"执行失败: {str(e)}, 等待 {sleep_time}s...", OutputType.ERROR)
+            time.sleep(sleep_time)
+            continue
+def while_true(func: Callable[[], bool], sleep_time: float = 0.1) -> Any:
+    """循环执行函数直到返回True"""
+    while True:
+        ret = func()
+        if ret:
+            break
+        PrettyOutput.print(f"执行失败, 等待 {sleep_time}s...", OutputType.WARNING)
+        time.sleep(sleep_time)
+    return ret
+def get_file_md5(filepath: str)->str:
+    """计算文件内容的MD5哈希值
+
+    参数:
+        filepath: 要计算哈希的文件路径
+
+    返回:
+        str: 文件内容的MD5哈希值
+    """
+    return hashlib.md5(open(filepath, "rb").read(100*1024*1024)).hexdigest()
+def user_confirm(tip: str, default: bool = True) -> bool:
+    """提示用户确认是/否问题
+
+    参数:
+        tip: 显示给用户的消息
+        default: 用户直接回车时的默认响应
+
+    返回:
+        bool: 用户确认返回True，否则返回False
+    """
+    suffix = "[Y/n]" if default else "[y/N]"
+    ret = get_single_line_input(f"{tip} {suffix}: ")
+    return default if ret == "" else ret.lower() == "y"
+
+def get_file_line_count(filename: str) -> int:
+    """计算文件中的行数
+
+    参数:
+        filename: 要计算行数的文件路径
+
+    返回:
+        int: 文件中的行数，如果文件无法读取则返回0
+    """
+    try:
+        return len(open(filename, "r", encoding="utf-8", errors="ignore").readlines())
+    except Exception as e:
+        return 0
+
+
+def is_long_context(files: List[str]) -> bool:
+    """检查文件列表是否属于长上下文
+
+    判断标准：
+    当总token数超过最大上下文长度的80%时视为长上下文
+
+    参数：
+    files -- 要检查的文件路径列表
+
+    返回：
+    布尔值表示是否属于长上下文
+    """
+    max_input_token_count = get_max_input_token_count()
+    threshold = max_input_token_count * 0.8
+    total_tokens = 0
+
+    for file_path in files:
+        try:
+            with open(file_path, 'r', encoding='utf-8', errors="ignore") as f:
+                content = f.read()
+                total_tokens += get_context_token_count(content)
+
+                if total_tokens > threshold:
+                    return True
+        except Exception as e:
+            PrettyOutput.print(f"读取文件 {file_path} 失败: {e}", OutputType.WARNING)
+            continue
+
+    return total_tokens > threshold
+
+
+def ot(tag_name: str) -> str:
+    """生成HTML标签开始标记
+
+    参数：
+    tag_name -- HTML标签名称
+
+    返回：
+    格式化的开始标签字符串
+    """
+    return f"<{tag_name}>"
+
+def ct(tag_name: str) -> str:
+    """生成HTML标签结束标记
+
+    参数：
+    tag_name -- HTML标签名称
+
+    返回：
+    格式化的结束标签字符串
+    """
+    return f"</{tag_name}>"
+