jarvis-ai-assistant 0.1.181__tar.gz → 0.1.182__tar.gz

{jarvis_ai_assistant-0.1.181/src/jarvis_ai_assistant.egg-info → jarvis_ai_assistant-0.1.182}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jarvis-ai-assistant
-Version: 0.1.181
+Version: 0.1.182
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire

{jarvis_ai_assistant-0.1.181 → jarvis_ai_assistant-0.1.182}/pyproject.toml

@@ -8,7 +8,7 @@ default = true
 
 [project]
 name = "jarvis-ai-assistant"
-version = "0.1.181"
+version = "0.1.182"
 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.181 → jarvis_ai_assistant-0.1.182}/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="jarvis-ai-assistant",
-    version="0.1.181",
+    version="0.1.182",
     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.181 → jarvis_ai_assistant-0.1.182}/src/jarvis/__init__.py

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

{jarvis_ai_assistant-0.1.181 → jarvis_ai_assistant-0.1.182}/src/jarvis/jarvis_agent/__init__.py

@@ -2,7 +2,7 @@
 # 标准库导入
 import datetime
 import platform
-from typing import Any, Callable, List, Optional, Protocol, Tuple, Union
+from typing import Any, Callable, Dict, List, Optional, Protocol, Tuple, Union
 
 # 第三方库导入
 from yaspin import yaspin  # type: ignore
@@ -205,6 +205,8 @@ class Agent:
         if model_name is not None:
             self.model.set_model_name(model_name)
 
+        self.user_data: Dict[str, Any] = {}
+
         self.model.set_suppress_output(False)
 
         from jarvis.jarvis_tools.registry import ToolRegistry
@@ -319,6 +321,14 @@ class Agent:
         )
         self.first = True
 
+    def set_user_data(self, key: str, value: Any):
+        """设置用户数据"""
+        self.user_data[key] = value
+
+    def get_user_data(self, key: str) -> Optional[Any]:
+        """获取用户数据"""
+        return self.user_data.get(key, None)
+
     def set_use_tools(self, use_tools):
         """设置要使用的工具列表"""
         from jarvis.jarvis_tools.registry import ToolRegistry
@@ -372,7 +382,7 @@ class Agent:
         )
 
         addon_prompt = f"""
-[系统提示开始]
+<system_prompt>
     请判断是否已经完成任务，如果已经完成：
     - 直接输出完成原因，不需要再有新的操作，不要输出{ot("TOOL_CALL")}标签
     {complete_prompt}
@@ -381,7 +391,7 @@ class Agent:
     - 如果信息不明确，请请求用户补充
     - 如果执行过程中连续失败5次，请使用ask_user询问用户操作
     - 操作列表：{action_handlers}
-[系统提示结束]
+</system_prompt>
 
 请继续。
 """

jarvis_ai_assistant-0.1.182/src/jarvis/jarvis_agent/file_input_handler.py

@@ -0,0 +1,112 @@
+# -*- coding: utf-8 -*-
+
+
+import os
+import re
+from typing import Any, Tuple
+
+from yaspin import yaspin
+
+from jarvis.jarvis_tools.file_operation import FileOperationTool
+from jarvis.jarvis_utils.output import OutputType, PrettyOutput
+from jarvis.jarvis_utils.utils import is_context_overflow
+
+
+def file_input_handler(user_input: str, agent: Any) -> Tuple[str, bool]:
+    """Process user input containing file references and read file contents.
+    
+    Args:
+        user_input: Input string that may contain file references in format:
+            - 'file_path' (whole file)
+            - 'file_path:start_line,end_line' (line range)
+            - 'file_path:start_line:end_line' (alternative range format)
+        agent: Agent object for further processing (currently unused)
+        
+    Returns:
+        Tuple[str, bool]: 
+            - Processed prompt string with file contents prepended
+            - Boolean indicating if context overflow occurred
+    """
+    prompt = user_input
+    files = []
+
+    file_refs = re.findall(r"'([^']+)'", user_input)
+    for ref in file_refs:
+        # Handle file:start,end or file:start:end format
+        if ':' in ref:
+            file_path, line_range = ref.split(':', 1)
+            # Initialize with default values
+            start_line = 1  # 1-based
+            end_line = -1
+
+            # Process line range if specified
+            if ',' in line_range or ':' in line_range:
+                try:
+                    raw_start, raw_end = map(int, re.split(r'[,:]', line_range))
+
+                    # Handle special values and Python-style negative indices
+                    try:
+                        with open(file_path, 'r', encoding='utf-8', errors="ignore") as f:
+                            total_lines = len(f.readlines())
+                    except FileNotFoundError:
+                        PrettyOutput.print(f"文件不存在: {file_path}", OutputType.WARNING)
+                        continue
+                    # Process start line (0 means whole file, negative means from end)
+                    if raw_start == 0:  # 0表示整个文件
+                        start_line = 1
+                        end_line = total_lines
+                    else:
+                        start_line = raw_start if raw_start > 0 else total_lines + raw_start + 1
+
+                    # Process end line
+                    if raw_end == 0:  # 0表示整个文件（如果start也是0）
+                        end_line = total_lines
+                    else:
+                        end_line = raw_end if raw_end > 0 else total_lines + raw_end + 1
+
+                    # Auto-correct ranges
+                    start_line = max(1, min(start_line, total_lines))
+                    end_line = max(start_line, min(end_line, total_lines))
+
+                    # Final validation
+                    if start_line < 1 or end_line > total_lines or start_line > end_line:
+                        raise ValueError
+
+                except:
+                    continue
+
+            # Add file if it exists
+            if os.path.isfile(file_path):
+                files.append({
+                    "path": file_path,
+                    "start_line": start_line,
+                    "end_line": end_line
+                })
+        else:
+            # Handle simple file path
+            if os.path.isfile(ref):
+                files.append({
+                    "path": ref,
+                    "start_line": 1,  # 1-based
+                    "end_line": -1
+                })
+
+    # Read and process files if any were found
+    if files:
+        with yaspin(text="正在读取文件...", color="cyan") as spinner:
+            old_prompt = prompt
+            result = FileOperationTool().execute({"operation":"read","files": files})
+            if result["success"]:
+                spinner.text = "文件读取完成"
+                spinner.ok("✅")
+                # Prepend file contents to prompt and check for overflow
+                prompt = f"""{prompt} 
+
+<file_context>
+{result["stdout"]}
+</file_context>"""
+                if is_context_overflow(prompt):
+                    return old_prompt, False
+
+    return prompt, False
+

{jarvis_ai_assistant-0.1.181 → jarvis_ai_assistant-0.1.182}/src/jarvis/jarvis_code_agent/code_agent.py

@@ -15,6 +15,7 @@ from yaspin import yaspin  # type: ignore
 from jarvis import __version__
 from jarvis.jarvis_agent import Agent
 from jarvis.jarvis_agent.builtin_input_handler import builtin_input_handler
+from jarvis.jarvis_agent.file_input_handler import file_input_handler
 from jarvis.jarvis_agent.shell_input_handler import shell_input_handler
 # 忽略yaspin的类型检查
 from jarvis.jarvis_code_agent.lint import get_lint_tools
@@ -108,6 +109,7 @@ class CodeAgent:
             output_handler=[tool_registry],
             platform=platform_instance,
             input_handler=[
+                file_input_handler,
                 shell_input_handler,
                 builtin_input_handler
             ],

{jarvis_ai_assistant-0.1.181 → jarvis_ai_assistant-0.1.182}/src/jarvis/jarvis_tools/edit_file.py

@@ -138,6 +138,7 @@ class FileSearchReplaceTool:
         
         file_path = args["file"]
         changes = args["changes"]
+        agent = args.get("agent", None)
         
         # 创建已处理文件变量，用于失败时回滚
         original_content = None
@@ -153,6 +154,17 @@ class FileSearchReplaceTool:
                     with open(file_path, 'r', encoding='utf-8') as f:
                         content = f.read()
                         original_content = content
+
+                if file_exists and agent:
+                    files = agent.get_user_data("files")
+                    if not files or files.get(file_path, None) is None:
+                        return {
+                            "success": False,
+                            "stdout": "",
+                            "stderr": f"请先读取文件 {file_path} 的内容后再编辑"
+                        }
+
+
                 with yaspin(text=f"正在处理文件 {file_path}...", color="cyan") as spinner:
                     success, temp_content = fast_edit(file_path, changes, spinner)
                     if not success:

{jarvis_ai_assistant-0.1.181 → jarvis_ai_assistant-0.1.182}/src/jarvis/jarvis_tools/file_operation.py

@@ -50,7 +50,7 @@ class FileOperationTool:
         return None  # 如果没有合适的处理器，返回None
     
     def _handle_single_file(self, operation: str, filepath: str, content: str = "",
-                          start_line: int = 1, end_line: int = -1) -> Dict[str, Any]:
+                          start_line: int = 1, end_line: int = -1, agent: Any = None) -> Dict[str, Any]:
         """Handle operations for a single file"""
         try:
             abs_path = os.path.abspath(filepath)
@@ -128,6 +128,15 @@ class FileOperationTool:
                     
                     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,
@@ -173,7 +182,7 @@ class FileOperationTool:
         """
         try:
             operation = args["operation"].strip()
-
+            agent = args.get("agent", None)
             if "files" not in args or not isinstance(args["files"], list):
                 return {
                     "success": False,
@@ -194,7 +203,8 @@ class FileOperationTool:
                     file_info["path"].strip(),
                     content,
                     file_info.get("start_line", 1),
-                    file_info.get("end_line", -1)
+                    file_info.get("end_line", -1),
+                    agent
                 )
 
                 if result["success"]:

{jarvis_ai_assistant-0.1.181 → jarvis_ai_assistant-0.1.182}/src/jarvis/jarvis_tools/read_code.py

@@ -31,7 +31,7 @@ class ReadCodeTool:
         "required": ["files"]
     }
 
-    def _handle_single_file(self, filepath: str, start_line: int = 1, end_line: int = -1) -> Dict[str, Any]:
+    def _handle_single_file(self, filepath: str, start_line: int = 1, end_line: int = -1, agent: Any = None) -> Dict[str, Any]:
         """处理单个文件的读取操作
 
         Args:
@@ -99,6 +99,14 @@ class ReadCodeTool:
                 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,
@@ -123,6 +131,7 @@ class ReadCodeTool:
             Dict[str, Any]: 包含成功状态、输出内容和错误信息的字典
         """
         try:
+            agent = args.get("agent", None)
             if "files" not in args or not isinstance(args["files"], list):
                 return {
                     "success": False,
@@ -140,7 +149,8 @@ class ReadCodeTool:
                 result = self._handle_single_file(
                     file_info["path"].strip(),
                     file_info.get("start_line", 1),
-                    file_info.get("end_line", -1)
+                    file_info.get("end_line", -1),
+                    agent
                 )
 
                 if result["success"]:

{jarvis_ai_assistant-0.1.181 → jarvis_ai_assistant-0.1.182/src/jarvis_ai_assistant.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jarvis-ai-assistant
-Version: 0.1.181
+Version: 0.1.182
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire

{jarvis_ai_assistant-0.1.181 → jarvis_ai_assistant-0.1.182}/src/jarvis_ai_assistant.egg-info/SOURCES.txt

@@ -6,6 +6,7 @@ setup.py
 src/jarvis/__init__.py
 src/jarvis/jarvis_agent/__init__.py
 src/jarvis/jarvis_agent/builtin_input_handler.py
+src/jarvis/jarvis_agent/file_input_handler.py
 src/jarvis/jarvis_agent/jarvis.py
 src/jarvis/jarvis_agent/main.py
 src/jarvis/jarvis_agent/output_handler.py