jarvis-ai-assistant 0.1.119__tar.gz → 0.1.120__tar.gz

{jarvis_ai_assistant-0.1.119/src/jarvis_ai_assistant.egg-info → jarvis_ai_assistant-0.1.120}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: jarvis-ai-assistant
-Version: 0.1.119
+Version: 0.1.120
 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.119 → jarvis_ai_assistant-0.1.120}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "jarvis-ai-assistant"
-version = "0.1.119"
+version = "0.1.120"
 description = "Jarvis: An AI assistant that uses tools to interact with the system"
 readme = "README.md"
 authors = [{ name = "Your Name", email = "your.email@example.com" }]

{jarvis_ai_assistant-0.1.119 → jarvis_ai_assistant-0.1.120}/setup.py

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

@@ -1,3 +1,3 @@
 """Jarvis AI Assistant"""
 
-__version__ = "0.1.119" 
+__version__ = "0.1.120" 

{jarvis_ai_assistant-0.1.119 → jarvis_ai_assistant-0.1.120}/src/jarvis/jarvis_code_agent/patch.py

@@ -21,23 +21,20 @@ class PatchOutputHandler(OutputHandler):
     
     def prompt(self) -> str:
         return """
-# 📝 Code Modification Format
-Use specific blocks for different operations:
-
 # 🔄 REPLACE: Modify existing code
 <REPLACE>
 File: path/to/file
 Lines: [start,end] or [start,end)
------
-new_content
+[new content]
+...
 </REPLACE>
 
 # ➕ INSERT: Add new code
 <INSERT>
 File: path/to/file
 Line: position
------
-new_content
+[new content]
+...
 </INSERT>
 
 # 🗑️ DELETE: Remove existing code
@@ -49,8 +46,8 @@ Lines: [start,end] or [start,end)
 # 🆕 NEW_FILE: Create new file
 <NEW_FILE>
 File: path/to/file
------
-new_content
+[new content]
+...
 </NEW_FILE>
 
 # ➡️ MOVE_FILE: Relocate a file
@@ -77,7 +74,6 @@ File: path/to/file
    - Omit for NEW_FILE/REMOVE_FILE
 
 3. Content
-   - Use "-----" separator
    - Maintain original indentation
    - Follow existing code style
 
@@ -86,7 +82,6 @@ File: path/to/file
 <REPLACE>
 File: src/utils.py
 Lines: [9,13]
------
 def updated_function():
     # Replaces lines 9-13 inclusive
     return "new_implementation"
@@ -96,7 +91,6 @@ def updated_function():
 <REPLACE>
 File: src/calculator.py
 Lines: [5,8)
------
 def new_calculation():
     # Replaces lines 5-7 (excludes line 8)
     return 42
@@ -106,7 +100,6 @@ def new_calculation():
 <INSERT>
 File: src/main.py
 Line: 19
------
     # Inserted before line 19
     new_feature()
 </INSERT>
@@ -114,7 +107,6 @@ Line: 19
 ## NEW_FILE Example
 <NEW_FILE>
 File: src/new_module.py
------
 # New file creation
 def feature_entry():
     pass
@@ -204,9 +196,20 @@ def _parse_patch(patch_str: str) -> Dict[str, List[Dict[str, Any]]]:
             else:
                 continue
         
-        # Get content (after separator)
-        separator_index = next((i for i, line in enumerate(lines) if line.strip() == "-----"), -1)
-        content = '\n'.join(lines[separator_index + 1:]) if separator_index != -1 else ''
+        # Get content (after metadata)
+        if patch_type in ['REPLACE', 'DELETE']:
+            content_start = 2  # File + Lines
+        elif patch_type == 'INSERT':
+            content_start = 2   # File + Line
+        elif patch_type == 'NEW_FILE':
+            content_start = 1   # File
+        elif patch_type == 'MOVE_FILE':
+            content_start = 2   # File + NewPath
+        elif patch_type == 'REMOVE_FILE':
+            content_start = 1   # File
+        
+        content_lines = lines[content_start:]
+        content = '\n'.join(content_lines).strip()
 
         if filepath not in result:
             result[filepath] = []
@@ -364,6 +367,9 @@ def handle_code_operation(filepath: str, patch: Dict[str, Any]):
     end_line = patch.get('end_line', 0)
     new_content = patch.get('content', '').splitlines(keepends=True)
 
+    if not new_content:
+        new_content = ['']
+
     PrettyOutput.print(f"patch_type: {patch_type}\nstart_line: {start_line}\nend_line: {end_line}\nnew_content:\n{new_content}", OutputType.INFO)
 
     if new_content and new_content[-1] and new_content[-1][-1] != '\n':

{jarvis_ai_assistant-0.1.119 → jarvis_ai_assistant-0.1.120}/src/jarvis/jarvis_platform/base.py

@@ -45,9 +45,22 @@ class BasePlatform(ABC):
             duration = end_time - start_time
             char_count = len(response)
             
+            # Calculate token count and tokens per second
+            try:
+                from jarvis.jarvis_utils import get_context_token_count
+                token_count = get_context_token_count(response)
+                tokens_per_second = token_count / duration if duration > 0 else 0
+            except Exception as e:
+                PrettyOutput.print(f"Tokenization failed: {str(e)}", OutputType.WARNING)
+                token_count = 0
+                tokens_per_second = 0
+
             # Print statistics
-            PrettyOutput.print(f"对话完成 - 耗时: {duration:.2f}秒, 输出字符数: {char_count}", OutputType.INFO)
-            
+            PrettyOutput.print(
+                f"对话完成 - 耗时: {duration:.2f}秒, 输出字符数: {char_count}, 输出Token数量: {token_count}, 每秒Token数量: {tokens_per_second:.2f}",
+                OutputType.INFO,
+            )
+
             # Keep original think tag handling
             response = re.sub(r'<<think>>.*?</</think>>', '', response, flags=re.DOTALL)
             return response

{jarvis_ai_assistant-0.1.119 → jarvis_ai_assistant-0.1.120/src/jarvis_ai_assistant.egg-info}/PKG-INFO

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