jarvis-ai-assistant 0.1.58__py3-none-any.whl → 0.1.74__py3-none-any.whl

{jarvis_ai_assistant-0.1.58.dist-info → jarvis_ai_assistant-0.1.74.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: jarvis-ai-assistant
-Version: 0.1.58
+Version: 0.1.74
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire
@@ -44,6 +44,10 @@ Requires-Dist: colorama>=0.4.6
 Requires-Dist: prompt_toolkit>=3.0.0
 Requires-Dist: openai>=1.20.0
 Requires-Dist: playwright>=1.41.1
+Requires-Dist: numpy>=1.24.0
+Requires-Dist: faiss-cpu>=1.8.0
+Requires-Dist: sentence-transformers>=2.2.2
+Requires-Dist: bs4>=0.0.1
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: black; extra == "dev"
@@ -66,6 +70,8 @@ Dynamic: requires-python
 
 *Your intelligent assistant for development and system interaction*
 
+English | [简体中文](README_zh.md)
+
 [Features](#features) •
 [Usage](#usage) •
 [Configuration](#configuration) •
@@ -112,19 +118,26 @@ pip install jarvis-ai-assistant
 
 ## 🔧 Configuration
 
-Create a `.jarvis_env` file in your home directory with your API keys:
+Jarvis supports configuration through environment variables that can be set in the `~/.jarvis_env` file:
+
+| Environment Variable | Description | Default Value | Required |
+|---------|------|--------|------|
+| JARVIS_PLATFORM | AI platform to use, supports kimi/openai/ai8 etc | kimi | Yes |
+| JARVIS_MODEL | Model name to use | - | No |
+| JARVIS_CODEGEN_PLATFORM | AI platform for code generation | Same as JARVIS_PLATFORM | No |
+| JARVIS_CODEGEN_MODEL | Model name for code generation | Same as JARVIS_MODEL | No |
+| JARVIS_CHEAP_PLATFORM | AI platform for cheap operations | Same as JARVIS_PLATFORM | No |
+| JARVIS_CHEAP_MODEL | Model name for cheap operations | Same as JARVIS_MODEL | No |
+| JARVIS_EMBEDDING_MODEL | Embedding model for code analysis | BAAI/bge-large-zh-v1.5 | No |
+| OPENAI_API_KEY | API key for OpenAI platform | - | Required for OpenAI |
+| OPENAI_API_BASE | Base URL for OpenAI API | https://api.deepseek.com | No |
+| OPENAI_MODEL_NAME | Model name for OpenAI | deepseek-chat | No |
+| AI8_API_KEY | API key for AI8 platform | - | Required for AI8 |
+| AI8_MODEL | Model name for AI8 platform | deepseek-chat | No |
+| KIMI_API_KEY | API key for Kimi platform | - | Required for Kimi |
+| OYI_API_KEY | API key for OYI platform | - | Required for OYI |
+| OYI_MODEL | Model name for OYI platform | deepseek-chat | No |
 
-### For Kimi:
-```bash
-KIMI_API_KEY=your_kimi_api_key_here
-```
-
-### For OpenAI:
-```bash
-OPENAI_API_KEY=your_api_key_here
-OPENAI_API_BASE=your_api_base  # Optional, defaults to https://api.deepseek.com
-OPENAI_MODEL_NAME=your_model_name  # Optional, defaults to deepseek-chat
-```
 
 ## 🎯 Usage
 
@@ -133,15 +146,26 @@ OPENAI_MODEL_NAME=your_model_name  # Optional, defaults to deepseek-chat
 jarvis
 ```
 
+
 ### With Specific Model
 ```bash
 jarvis -p kimi  # Use Kimi platform
 jarvis -p openai  # Use OpenAI platform
 ```
 
-### Process Files
+### Code Modification
+```bash
+jarvis-coder --feature "Add new feature"  # Modify code to add new feature
+```
+
+### Codebase Search
 ```bash
-jarvis -f file1.py file2.py  # Process specific files
+jarvis-codebase --search "database connection"  # Search codebase
+```
+
+### Codebase Question
+```bash
+jarvis-codebase --ask "How to use the database?"  # Ask about codebase
 ```
 
 ### Keep Chat History
@@ -151,6 +175,7 @@ jarvis --keep-history  # Don't delete chat session after completion
 
 ## 🛠️ Tools
 
+
 ### Built-in Tools
 
 | Tool | Description |
@@ -160,27 +185,43 @@ jarvis --keep-history  # Don't delete chat session after completion
 | generate_tool | AI-powered tool generation and integration |
 | methodology | Experience accumulation and methodology management |
 | create_sub_agent | Create specialized sub-agents for specific tasks |
+| coder | Automatic code modification and generation tool |
+| codebase | Codebase management and search tool |
 
 ### Tool Locations
 - Built-in tools: `src/jarvis/tools/`
 - User tools: `~/.jarvis_tools/`
 
+
 ### Key Features
 
 #### 1. Self-Extending Capabilities
 - Tool generation through natural language description
 - Automatic code generation and integration
 - Dynamic capability expansion through sub-agents
+- Automatic code modification with version control
+- Codebase indexing and semantic search
 
 #### 2. Methodology Learning
 - Automatic experience accumulation from interactions
 - Pattern recognition and methodology extraction
 - Continuous refinement through usage
+- Code modification history tracking
+- Codebase analysis and documentation generation
 
 #### 3. Adaptive Problem Solving
 - Context-aware sub-agent creation
 - Dynamic tool composition
 - Learning from execution feedback
+- Codebase-aware problem solving
+- Multi-model collaboration for complex tasks
+
+#### 4. Code Intelligence
+- Automatic codebase indexing
+- Semantic code search
+- Code modification with git integration
+- Code analysis and documentation
+- Multi-model code generation
 
 ## 🎯 Extending Jarvis
 
@@ -241,55 +282,57 @@ Create a new Python file in `~/.jarvis_models/`:
 
 ```python
 from typing import Dict, List
-from jarvis.models.base import BaseModel
+from jarvis.models.base import BasePlatform
 from jarvis.utils import PrettyOutput, OutputType
 
-class CustomModel(BaseModel):
+class CustomPlatform(BasePlatform):
     """Custom model implementation"""
     
-    model_name = "custom"  # Model identifier
+    platform_name = "custom"  # Platform identifier
     
     def __init__(self):
         """Initialize model"""
-        # Add your initialization code here
+        # add initialization code
+        super().__init__()
         self.messages = []
         self.system_message = ""
-        
-    def set_system_message(self, message: str):
-        """Set system message"""
-        self.system_message = message
-        
+
+    def set_model_name(self, model_name: str):
+        """Set model name"""
+        self.model_name = model_name
+
     def chat(self, message: str) -> str:
-        """Execute chat with the model
+        """Chat with model
         
         Args:
-            message: User input message
+            message: user input message
             
         Returns:
-            str: Model response
+            str: model response
         """
         try:
-            # Implement chat logic here
-            PrettyOutput.print("发送请求...", OutputType.PROGRESS)
+            # implement chat logic
+            PrettyOutput.print("Sending request...", OutputType.PROGRESS)
             
-            # Add message to history
+            # add message to history
             self.messages.append({"role": "user", "content": message})
             
-            # Get response from your model
-            response = "Model response"
+            # get response from model
+            response = "model response"
             
-            # Add response to history
+            # add response to history
             self.messages.append({"role": "assistant", "content": response})
             
             return response
             
         except Exception as e:
-            PrettyOutput.print(f"对话失败: {str(e)}", OutputType.ERROR)
+            PrettyOutput.print(f"Chat failed: {str(e)}", OutputType.ERROR)
             raise Exception(f"Chat failed: {str(e)}")
-            
-    def name(self) -> str:
-        """Return model name"""
-        return self.model_name
+    
+    def upload_files(self, file_list: List[str]) -> List[Dict]:
+        """Upload files"""
+        # implement file upload logic
+        return []    
         
     def reset(self):
         """Reset model state"""
@@ -297,10 +340,22 @@ class CustomModel(BaseModel):
         if self.system_message:
             self.messages.append({"role": "system", "content": self.system_message})
             
+    def name(self) -> str:
+        """Return model name"""
+        return self.model_name
+            
     def delete_chat(self) -> bool:
         """Delete current chat session"""
         self.reset()
-        return True
+        return True  
+
+    def set_system_message(self, message: str):
+        """Set system message"""
+        self.system_message = message
+
+    def set_suppress_output(self, suppress: bool):
+        """Set whether to suppress output"""
+        self.suppress_output = suppress
 ```
 
 ### Development Guidelines

jarvis_ai_assistant-0.1.74.dist-info/RECORD

@@ -0,0 +1,33 @@
+jarvis/__init__.py,sha256=un8MKMurE4TQ58dJg7nKMX1AVpzjB6SOlLsT0fZsdWI,50
+jarvis/agent.py,sha256=jQAEJ3C8DEcmtCozXCieUcRKhhPSXJe8ylWuI4LSrOk,15032
+jarvis/main.py,sha256=7EcSlxa5JFFXBujzKDWdNtwX6axLhFFdJMc2GxTjfdk,6295
+jarvis/utils.py,sha256=vZV8sHj0ggZy4Rb8RxIujQhRWgeNEomhqVl4WXmpq7c,7498
+jarvis/jarvis_codebase/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jarvis/jarvis_codebase/main.py,sha256=G8ADdTgjZTxEjvP11IBbjHiNWCvuTkFiLkDKq_8UpI0,26947
+jarvis/jarvis_coder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jarvis/jarvis_coder/main.py,sha256=4YhQ20MD8ntdD5hjmjJbBHYEDPDZ2aPIgAviJCvtOXE,24458
+jarvis/models/__init__.py,sha256=mrOt67nselz_H1gX9wdAO4y2DY5WPXzABqJbr5Des8k,63
+jarvis/models/ai8.py,sha256=vgy-r_3HHxGMAalZrA65VWHC1PuwBTYgtprSgHkCbrk,12557
+jarvis/models/base.py,sha256=ShV1H8Unee4RMaiFO4idROQA0Hc6wu4dyeRPX5fcszk,1433
+jarvis/models/kimi.py,sha256=1iTB0Z_WOmCML3Ufsge6jmeKOYvccr7I5lS3JUXymU4,17611
+jarvis/models/openai.py,sha256=ayaBWAN5VexMcKVrjEPDNB-Q9wx0sCV9Z4BCrvwYJ9w,4315
+jarvis/models/oyi.py,sha256=X2c5SWDIuQDCCFBcEKbzIWEz3I34eOAi0d1XAFgxlpw,15001
+jarvis/models/registry.py,sha256=YpooKSpk5pSWfb5cBDz5wRfPK-abb9uuUZr4WBejqwI,7762
+jarvis/tools/__init__.py,sha256=7Rqyj5hBAv5cWDVr5T9ZTZASO7ssBHeQNm2_4ZARdkA,72
+jarvis/tools/base.py,sha256=EGRGbdfbLXDLwtyoWdvp9rlxNX7bzc20t0Vc2VkwIEY,652
+jarvis/tools/codebase_qa.py,sha256=LsowsgL7HBmdBwa7zXcYi_OkwOok4qbnzYWYsuZxHtU,2413
+jarvis/tools/coder.py,sha256=ZJfPInKms4Hj3-eQlBwamVsvZ-2nlZ-4jsqJ-tJc6mg,2040
+jarvis/tools/file_ops.py,sha256=h8g0eT9UvlJf4kt0DLXvdSsjcPj7x19lxWdDApeDfpg,3842
+jarvis/tools/generator.py,sha256=vVP3eN5cCDpRXf_fn0skETkPXAW1XZFWx9pt2_ahK48,5999
+jarvis/tools/methodology.py,sha256=UG6s5VYRcd9wrKX4cg6f7zJhet5AIcthFGMOAdevBiw,5175
+jarvis/tools/registry.py,sha256=MeTYNdZNRdhlgABviVxzbDPSgLpwDp2Nx2dGzedRu8U,7212
+jarvis/tools/search.py,sha256=1EqOVvLhg2Csh-i03-XeCrusbyfmH69FZ8khwZt8Tow,6131
+jarvis/tools/shell.py,sha256=UPKshPyOaUwTngresUw-ot1jHjQIb4wCY5nkJqa38lU,2520
+jarvis/tools/sub_agent.py,sha256=rEtAmSVY2ZjFOZEKr5m5wpACOQIiM9Zr_3dT92FhXYU,2621
+jarvis/tools/webpage.py,sha256=d3w3Jcjcu1ESciezTkz3n3Zf-rp_l91PrVoDEZnckOo,2391
+jarvis_ai_assistant-0.1.74.dist-info/LICENSE,sha256=AGgVgQmTqFvaztRtCAXsAMryUymB18gZif7_l2e1XOg,1063
+jarvis_ai_assistant-0.1.74.dist-info/METADATA,sha256=ajh1yMCWevAgppfPVYQxcKV4J143iOBz1SyBC9KGd9o,12399
+jarvis_ai_assistant-0.1.74.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+jarvis_ai_assistant-0.1.74.dist-info/entry_points.txt,sha256=QNUeqmUJd7nHufel2FO7cRttS1uKFfnbIyObv8eVyOY,140
+jarvis_ai_assistant-0.1.74.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
+jarvis_ai_assistant-0.1.74.dist-info/RECORD,,

{jarvis_ai_assistant-0.1.58.dist-info → jarvis_ai_assistant-0.1.74.dist-info}/entry_points.txt

@@ -1,3 +1,4 @@
 [console_scripts]
 jarvis = jarvis.main:main
+jarvis-codebase = jarvis.jarvis_codebase.main:main
 jarvis-coder = jarvis.jarvis_coder.main:main

jarvis/tools/bing_search.py

@@ -1,38 +0,0 @@
-from playwright.sync_api import sync_playwright
-from urllib.parse import quote
-
-def bing_search(query):
-    try:
-        with sync_playwright() as p:
-            browser = p.chromium.launch()
-            page = browser.new_page()
-            page.goto(
-                f"https://www.bing.com/search?form=QBRE&q={quote(query)}&cc=US"
-            )
-
-            page.wait_for_selector("#b_results", timeout=10000)
-            
-            summaries = page.evaluate("""() => {
-                const liElements = Array.from(
-                    document.querySelectorAll("#b_results > .b_algo")
-                );
-                return liElements.map((li) => {
-                    const abstractElement = li.querySelector(".b_caption > p");
-                    const linkElement = li.querySelector("a");
-                    const href = linkElement.getAttribute("href");
-                    const title = linkElement.textContent;
-                    const abstract = abstractElement ? abstractElement.textContent : "";
-                    return { href, title, abstract };
-                });
-            }""")
-            
-            browser.close()
-            print(summaries)
-            return summaries
-    except Exception as error:
-        print("An error occurred:", error)
-
-if __name__ == "__main__":
-    # results = bing_search("北京到西雅图的距离")
-    results = bing_search("北京到西雅图的距离")
-    print(results)

jarvis_ai_assistant-0.1.58.dist-info/RECORD

@@ -1,29 +0,0 @@
-jarvis/__init__.py,sha256=rc0PX9hG--lb8NGfVN3-VdS1SnQTkK_G_FvWr1ZO-LM,50
-jarvis/agent.py,sha256=s8VebNef8BbKM6D9IDWWN-XenEXEL9KoYnDLGKyZY58,12347
-jarvis/main.py,sha256=gXXtnrkkvGwEswJL6qiYjVrg3bpzye-GJeAe0Nf2B9o,6509
-jarvis/utils.py,sha256=JlkuC9RtspXH2VWDmj9nR0vnb8ie1gIsKc4vC7WRco8,7321
-jarvis/jarvis_coder/main.py,sha256=7Zc50dpPSM6hq4ASg1-yGZ83iS1NxuvQn70D9SPNbqY,30751
-jarvis/models/__init__.py,sha256=mrOt67nselz_H1gX9wdAO4y2DY5WPXzABqJbr5Des8k,63
-jarvis/models/ai8.py,sha256=aSiUazBrj4Fsit3tQX7jj8nM8MDBhHFnFrbQC6CvcGw,12745
-jarvis/models/base.py,sha256=39g0GcXrNu16RV3HzazIqleynCaQDhm0XzF0r0uMDro,1506
-jarvis/models/kimi.py,sha256=j4VtRip0WhTgI4PGL_YXL9Z-II-xHe6hvJW3vOY-ShA,17545
-jarvis/models/openai.py,sha256=FcZVcCK41jP4SB_q4kZ6bga2tYbL-s6zjpS4KU30-tw,4477
-jarvis/models/oyi.py,sha256=vVs1ru5K8WpofjwoYTxUQVvXL4Yc7ZH6nZhKYpEZcdk,15288
-jarvis/models/registry.py,sha256=iVBjN9ImEvGHcz8WR-z8pPMJQZI907o_nccVOFANhak,7951
-jarvis/tools/__init__.py,sha256=Kj1bKj34lwRDKMKHLOrLyQElf2lHbqA2tDgP359eaDo,71
-jarvis/tools/base.py,sha256=EGRGbdfbLXDLwtyoWdvp9rlxNX7bzc20t0Vc2VkwIEY,652
-jarvis/tools/bing_search.py,sha256=SvYeXM83eP9qREDcWguh0_r0m0npbcwXIyzEwwfreKU,1444
-jarvis/tools/file_ops.py,sha256=h8g0eT9UvlJf4kt0DLXvdSsjcPj7x19lxWdDApeDfpg,3842
-jarvis/tools/generator.py,sha256=vVP3eN5cCDpRXf_fn0skETkPXAW1XZFWx9pt2_ahK48,5999
-jarvis/tools/methodology.py,sha256=G3cOaHTMujGZBhDLhQEqyCV2NISizO3MXRuho1KfI6Y,5223
-jarvis/tools/registry.py,sha256=mlOAmUq3yzRz-7yvwrrCwbe5Lmw8eh1v8-_Fa5sezwI,7209
-jarvis/tools/search.py,sha256=o7EJqEtKKDqlENkXhxDuDA73djzeEH9j4BB0n61egso,4899
-jarvis/tools/shell.py,sha256=UPKshPyOaUwTngresUw-ot1jHjQIb4wCY5nkJqa38lU,2520
-jarvis/tools/sub_agent.py,sha256=rEtAmSVY2ZjFOZEKr5m5wpACOQIiM9Zr_3dT92FhXYU,2621
-jarvis/tools/webpage.py,sha256=d3w3Jcjcu1ESciezTkz3n3Zf-rp_l91PrVoDEZnckOo,2391
-jarvis_ai_assistant-0.1.58.dist-info/LICENSE,sha256=AGgVgQmTqFvaztRtCAXsAMryUymB18gZif7_l2e1XOg,1063
-jarvis_ai_assistant-0.1.58.dist-info/METADATA,sha256=upXDcw_I_71ZInBR5zWI90zGsm5YDBoFG64btCGews0,10049
-jarvis_ai_assistant-0.1.58.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-jarvis_ai_assistant-0.1.58.dist-info/entry_points.txt,sha256=ieRI4ilnGNx1R6LlzT2P510mJ27lhLesVZToezDjSd8,89
-jarvis_ai_assistant-0.1.58.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
-jarvis_ai_assistant-0.1.58.dist-info/RECORD,,