jarvis-ai-assistant 0.1.0__tar.gz

jarvis_ai_assistant-0.1.0/MANIFEST.in

@@ -0,0 +1,4 @@
+include README.md
+include LICENSE
+include pyproject.toml
+recursive-include src/jarvis * 

jarvis_ai_assistant-0.1.0/PKG-INFO

@@ -0,0 +1,125 @@
+Metadata-Version: 2.1
+Name: jarvis-ai-assistant
+Version: 0.1.0
+Summary: Jarvis: An AI assistant that uses tools to interact with the system
+Home-page: https://github.com/skyfireitdiy/Jarvis
+Author: skyfire
+Author-email: Your Name <your.email@example.com>
+Project-URL: Homepage, https://github.com/skyfireitdiy/Jarvis
+Keywords: jarvis,ai,assistant,tools,automation
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: requests>=2.25.1
+Requires-Dist: beautifulsoup4>=4.9.3
+Requires-Dist: duckduckgo-search>=3.0.0
+Requires-Dist: pyyaml>=5.1
+Requires-Dist: ollama>=0.1.6
+Requires-Dist: sentence-transformers>=2.5.1
+Requires-Dist: chromadb>=0.4.24
+Provides-Extra: dev
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: black; extra == "dev"
+Requires-Dist: isort; extra == "dev"
+Requires-Dist: mypy; extra == "dev"
+
+<div align="center">
+
+# 🤖 Jarvis AI Assistant
+
+<p align="center">
+  <img src="docs/images/jarvis-logo.png" alt="Jarvis Logo" width="200"/>
+</p>
+
+[![PyPI version](https://badge.fury.io/py/jarvis-ai.svg)](https://badge.fury.io/py/jarvis-ai)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![Python 3.8+](https://img.shields.io/badge/python-3.8+-blue.svg)](https://www.python.org/downloads/)
+
+*Your intelligent assistant for development and system interaction*
+
+[Installation](#installation) •
+[Features](#features) •
+[Usage](#usage) •
+[Tools](#tools) •
+[Documentation](https://jarvis-ai.readthedocs.io/)
+
+</div>
+
+---
+
+## 🌟 Features
+
+🤖 **Multiple AI Models**
+- Ollama integration (llama3.2, qwen2.5:14b)
+- DuckDuckGo AI search capabilities
+
+🛠️ **Rich Tool Integration**
+- RAG (Retrieval-Augmented Generation)
+- File operations & Shell commands
+- Web search & content extraction
+- Python code execution with dependency management
+
+🔄 **Interactive Experience**
+- Natural language understanding
+- Context-aware responses
+- User-friendly interface
+
+## 🚀 Installation
+
+```bash
+pip install jarvis-ai-assistant
+```
+
+## 💡 Usage
+
+```bash
+# Quick Start
+jarvis
+
+# Using Specific Model
+jarvis --platform ollama --model qwen2.5:14b
+
+# Custom Ollama API
+jarvis --platform ollama --model llama3.2 --api-base http://localhost:11434
+```
+
+## 🧰 Tools
+
+| Tool | Description | Example |
+|------|-------------|---------|
+| 🔍 Search | Web search using DuckDuckGo | Search latest tech news |
+| 📚 RAG | Document querying with embeddings | Query your documentation |
+| 🐍 Python | Execute Python code | Run data analysis |
+| 🖥️ Shell | Execute system commands | Manage files and processes |
+| 📂 Files | Read/write operations | Handle configuration files |
+| 🌐 Web | Extract webpage content | Gather information |
+| 👤 User | Interactive input/confirmation | Get user preferences |
+
+
+## 🤝 Contributing
+
+Contributions are welcome! Please feel free to submit a Pull Request.
+
+1. Fork the repository
+2. Create your feature branch (`git checkout -b feature/AmazingFeature`)
+3. Commit your changes (`git commit -m 'Add some AmazingFeature'`)
+4. Push to the branch (`git push origin feature/AmazingFeature`)
+5. Open a Pull Request
+
+## 📄 License
+
+This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.
+
+---
+
+<div align="center">
+
+Made with ❤️ by [Your Name]
+
+</div>

jarvis_ai_assistant-0.1.0/README.md

@@ -0,0 +1,94 @@
+<div align="center">
+
+# 🤖 Jarvis AI Assistant
+
+<p align="center">
+  <img src="docs/images/jarvis-logo.png" alt="Jarvis Logo" width="200"/>
+</p>
+
+[![PyPI version](https://badge.fury.io/py/jarvis-ai.svg)](https://badge.fury.io/py/jarvis-ai)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![Python 3.8+](https://img.shields.io/badge/python-3.8+-blue.svg)](https://www.python.org/downloads/)
+
+*Your intelligent assistant for development and system interaction*
+
+[Installation](#installation) •
+[Features](#features) •
+[Usage](#usage) •
+[Tools](#tools) •
+[Documentation](https://jarvis-ai.readthedocs.io/)
+
+</div>
+
+---
+
+## 🌟 Features
+
+🤖 **Multiple AI Models**
+- Ollama integration (llama3.2, qwen2.5:14b)
+- DuckDuckGo AI search capabilities
+
+🛠️ **Rich Tool Integration**
+- RAG (Retrieval-Augmented Generation)
+- File operations & Shell commands
+- Web search & content extraction
+- Python code execution with dependency management
+
+🔄 **Interactive Experience**
+- Natural language understanding
+- Context-aware responses
+- User-friendly interface
+
+## 🚀 Installation
+
+```bash
+pip install jarvis-ai-assistant
+```
+
+## 💡 Usage
+
+```bash
+# Quick Start
+jarvis
+
+# Using Specific Model
+jarvis --platform ollama --model qwen2.5:14b
+
+# Custom Ollama API
+jarvis --platform ollama --model llama3.2 --api-base http://localhost:11434
+```
+
+## 🧰 Tools
+
+| Tool | Description | Example |
+|------|-------------|---------|
+| 🔍 Search | Web search using DuckDuckGo | Search latest tech news |
+| 📚 RAG | Document querying with embeddings | Query your documentation |
+| 🐍 Python | Execute Python code | Run data analysis |
+| 🖥️ Shell | Execute system commands | Manage files and processes |
+| 📂 Files | Read/write operations | Handle configuration files |
+| 🌐 Web | Extract webpage content | Gather information |
+| 👤 User | Interactive input/confirmation | Get user preferences |
+
+
+## 🤝 Contributing
+
+Contributions are welcome! Please feel free to submit a Pull Request.
+
+1. Fork the repository
+2. Create your feature branch (`git checkout -b feature/AmazingFeature`)
+3. Commit your changes (`git commit -m 'Add some AmazingFeature'`)
+4. Push to the branch (`git push origin feature/AmazingFeature`)
+5. Open a Pull Request
+
+## 📄 License
+
+This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.
+
+---
+
+<div align="center">
+
+Made with ❤️ by [Your Name]
+
+</div>

jarvis_ai_assistant-0.1.0/pyproject.toml

@@ -0,0 +1,45 @@
+[build-system]
+requires = ["setuptools>=45", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "jarvis-ai-assistant"
+version = "0.1.0"
+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" }]
+license = { file = "LICENSE" }
+classifiers = [
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.8",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+]
+keywords = ["jarvis", "ai", "assistant", "tools", "automation"]
+dependencies = [
+    "requests>=2.25.1",
+    "beautifulsoup4>=4.9.3",
+    "duckduckgo-search>=3.0.0",
+    "pyyaml>=5.1",
+    "ollama>=0.1.6",
+    "sentence-transformers>=2.5.1",
+    "chromadb>=0.4.24",
+]
+requires-python = ">=3.8"
+
+[project.optional-dependencies]
+dev = [
+    "pytest",
+    "black",
+    "isort",
+    "mypy",
+]
+
+[project.urls]
+Homepage = "https://github.com/skyfireitdiy/Jarvis"
+
+[project.scripts]
+jarvis = "jarvis.main:main"

jarvis_ai_assistant-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

jarvis_ai_assistant-0.1.0/setup.py

@@ -0,0 +1,41 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="jarvis-ai-assistant",
+    version="0.1.0",
+    author="skyfire",
+    author_email="skyfireitdiy@hotmail.com",
+    description="An AI assistant that uses various tools to interact with the system",
+    long_description=open("README.md").read(),
+    long_description_content_type="text/markdown",
+    url="https://github.com/skyfireitdiy/Jarvis",
+    packages=find_packages(where="src"),
+    package_dir={"": "src"},
+    include_package_data=True,
+    install_requires=[
+        "requests>=2.25.1",
+        "beautifulsoup4>=4.9.3",
+        "duckduckgo-search>=3.0.0",
+        "pyyaml>=5.1",
+        "ollama>=0.1.6",
+        "sentence-transformers>=2.5.1",
+        "chromadb>=0.4.24",
+    ],
+    entry_points={
+        "console_scripts": [
+            "jarvis=jarvis.main:main",
+        ],
+    },
+    python_requires=">=3.8",
+    classifiers=[
+        "Development Status :: 3 - Alpha",
+        "Intended Audience :: Developers",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
+        "Programming Language :: Python :: 3",
+        "Programming Language :: Python :: 3.8",
+        "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
+    ],
+) 

jarvis_ai_assistant-0.1.0/src/jarvis/.jarvis

@@ -0,0 +1 @@
+- 查看当前git仓库状态，如果有未提交的变更，自动根据当前仓库的文件修改内容总结生成一个git commit，然后执行git push推送到远端仓库，全流程不要询问用户

jarvis_ai_assistant-0.1.0/src/jarvis/__init__.py

@@ -0,0 +1,3 @@
+"""Jarvis AI Assistant"""
+
+__version__ = "0.1.0" 

jarvis_ai_assistant-0.1.0/src/jarvis/agent.py

@@ -0,0 +1,100 @@
+import json
+import subprocess
+from typing import Dict, Any, List, Optional
+from .tools import ToolRegistry
+from .utils import Spinner, PrettyOutput, OutputType, get_multiline_input
+from .models import BaseModel, OllamaModel
+import re
+import os
+from datetime import datetime
+
+class Agent:
+    def __init__(self, model: BaseModel, tool_registry: ToolRegistry):
+        self.model = model
+        self.tool_registry = tool_registry
+        # 编译正则表达式
+        self.tool_call_pattern = re.compile(r'<tool_call>\s*({[^}]+})\s*</tool_call>')
+        self.messages = [
+            {
+                "role": "system",
+                "content": """You are a rigorous AI assistant, all data must be obtained through tools, and no fabrication or speculation is allowed. """ + "\n" + self.tool_registry.tool_help_text()
+            }
+        ]
+        self.spinner = Spinner()
+
+    def _call_model(self, messages: List[Dict], use_tools: bool = True) -> Dict:
+        """调用模型获取响应"""
+        self.spinner.start()
+        try:
+            return self.model.chat(
+                messages=messages,
+                tools=self.tool_registry.get_all_tools() if use_tools else None
+            )
+        except Exception as e:
+            raise Exception(f"模型调用失败: {str(e)}")
+        finally:
+            self.spinner.stop()
+
+
+    def run(self, user_input: str) :
+        """处理用户输入并返回响应"""
+        # 检查是否是结束命令
+        self.clear_history()
+        self.messages.append({
+            "role": "user",
+            "content": user_input
+        })
+        while True:
+            try:
+                # 获取初始响应
+                response = self._call_model(self.messages)
+                current_response = response
+
+                # 将工具执行结果添加到对话
+                self.messages.append({
+                    "role": "assistant",
+                    "content": response["message"].get("content", ""),
+                    "tool_calls": current_response["message"]["tool_calls"]
+                })
+                
+                # 处理可能的多轮工具调用
+                if len(current_response["message"]["tool_calls"]) > 0:
+                    # 添加当前助手响应到输出（如果有内容）
+                    if current_response["message"].get("content"):
+                        PrettyOutput.print(current_response["message"]["content"], OutputType.SYSTEM)
+                        
+                    # 使用 ToolRegistry 的 handle_tool_calls 方法处理工具调用
+                    tool_result = self.tool_registry.handle_tool_calls(current_response["message"]["tool_calls"])
+                    PrettyOutput.print(tool_result, OutputType.RESULT)
+
+                    self.messages.append({
+                        "role": "tool",
+                        "content": tool_result
+                    })
+                    continue
+                
+                # 添加最终响应到对话历史和输出
+                final_content = current_response["message"].get("content", "")
+                
+                if final_content:
+                    PrettyOutput.print(final_content, OutputType.SYSTEM)
+                    
+                
+                # 如果没有工具调用且响应很短，可能需要继续对话
+                user_input = get_multiline_input("您可以继续输入，或输入空行结束当前任务")
+                if  not user_input:
+                    PrettyOutput.print("===============任务结束===============", OutputType.INFO)
+                    break
+                
+                self.messages.append({
+                    "role": "user",
+                    "content": user_input
+                })
+                
+            except Exception as e:
+                error_msg = f"处理响应时出错: {str(e)}"
+                PrettyOutput.print(error_msg, OutputType.ERROR)
+
+    def clear_history(self):
+        """清除对话历史，只保留系统提示"""
+        self.messages = [self.messages[0]] 

jarvis_ai_assistant-0.1.0/src/jarvis/main.py

@@ -0,0 +1,161 @@
+#!/usr/bin/env python3
+"""Command line interface for Jarvis."""
+
+import argparse
+import yaml
+import os
+import sys
+from pathlib import Path
+
+# 添加父目录到Python路径以支持导入
+sys.path.insert(0, str(Path(__file__).parent.parent))
+
+from jarvis.agent import Agent
+from jarvis.tools import ToolRegistry
+from jarvis.models import DDGSModel, OllamaModel
+from jarvis.utils import PrettyOutput, OutputType, get_multiline_input
+
+# 定义支持的平台和模型
+SUPPORTED_PLATFORMS = {
+    "ollama": {
+        "models": ["llama3.2", "qwen2.5:14b"],
+        "default": "qwen2.5:14b"
+    },
+    "ddgs": {
+        "models": ["gpt-4o-mini", "claude-3-haiku", "llama-3.1-70b", "mixtral-8x7b"],
+        "default": "gpt-4o-mini"
+    }
+}
+
+def load_tasks() -> list:
+    """Load tasks from .jarvis file if it exists."""
+    if not os.path.exists(".jarvis"):
+        return []
+    
+    try:
+        with open(".jarvis", "r", encoding="utf-8") as f:
+            tasks = yaml.safe_load(f)
+            
+        if not isinstance(tasks, list):
+            PrettyOutput.print("Warning: .jarvis file should contain a list of tasks", OutputType.ERROR)
+            return []
+            
+        return [str(task) for task in tasks if task]  # Convert all tasks to strings and filter out empty ones
+    except Exception as e:
+        PrettyOutput.print(f"Error loading .jarvis file: {str(e)}", OutputType.ERROR)
+        return []
+
+def select_task(tasks: list) -> str:
+    """Let user select a task from the list or skip."""
+    if not tasks:
+        return ""
+    
+    PrettyOutput.print("\nFound predefined tasks:", OutputType.INFO)
+    for i, task in enumerate(tasks, 1):
+        PrettyOutput.print(f"[{i}] {task}", OutputType.INFO)
+    PrettyOutput.print("[0] Skip predefined tasks", OutputType.INFO)
+    
+    while True:
+        try:
+            choice = input("\nSelect a task number (0 to skip): ").strip()
+            if not choice:
+                return ""
+            
+            choice = int(choice)
+            if choice == 0:
+                return ""
+            elif 1 <= choice <= len(tasks):
+                return tasks[choice - 1]
+            else:
+                PrettyOutput.print("Invalid choice. Please try again.", OutputType.ERROR)
+        except ValueError:
+            PrettyOutput.print("Please enter a valid number.", OutputType.ERROR)
+
+def main():
+    """Main entry point for Jarvis."""
+    parser = argparse.ArgumentParser(description="Jarvis AI Assistant")
+    
+    # 添加平台选择参数
+    parser.add_argument(
+        "--platform",
+        choices=list(SUPPORTED_PLATFORMS.keys()),
+        default="ollama",
+        help="选择运行平台 (默认: ollama)"
+    )
+    
+    # 添加模型选择参数
+    parser.add_argument(
+        "--model",
+        help="选择模型 (默认: 根据平台自动选择)"
+    )
+    
+    # 添加API基础URL参数
+    parser.add_argument(
+        "--api-base",
+        default="http://localhost:11434",
+        help="Ollama API基础URL (仅用于Ollama平台, 默认: http://localhost:11434)"
+    )
+    
+    args = parser.parse_args()
+    
+    # 验证并设置默认模型
+    if args.model:
+        if args.model not in SUPPORTED_PLATFORMS[args.platform]["models"]:
+            supported_models = ", ".join(SUPPORTED_PLATFORMS[args.platform]["models"])
+            PrettyOutput.print(
+                f"错误: 平台 {args.platform} 不支持模型 {args.model}\n"
+                f"支持的模型: {supported_models}",
+                OutputType.ERROR
+            )
+            return 1
+    else:
+        args.model = SUPPORTED_PLATFORMS[args.platform]["default"]
+
+    try:
+        # 根据平台创建相应的模型实例
+        if args.platform == "ollama":
+            model = OllamaModel(
+                model_name=args.model,
+                api_base=args.api_base
+            )
+            platform_name = f"Ollama ({args.model})"
+        else:  # ddgs
+            model = DDGSModel(model_name=args.model)
+            platform_name = f"DuckDuckGo Search ({args.model})"
+
+        tool_registry = ToolRegistry()
+        agent = Agent(model, tool_registry)
+
+        # 欢迎信息
+        PrettyOutput.print(f"Jarvis 已初始化 - {platform_name}", OutputType.SYSTEM)
+        
+        # 加载预定义任务
+        tasks = load_tasks()
+        if tasks:
+            selected_task = select_task(tasks)
+            if selected_task:
+                PrettyOutput.print(f"\n执行任务: {selected_task}", OutputType.INFO)
+                agent.run(selected_task)
+                return 0
+        
+        # 如果没有选择预定义任务，进入交互模式
+        while True:
+            try:
+                user_input = get_multiline_input("请输入您的任务(输入空行退出):")
+                if not user_input:
+                    break
+                agent.run(user_input)
+            except KeyboardInterrupt:
+                print("\n正在退出...")
+                break
+            except Exception as e:
+                PrettyOutput.print(f"错误: {str(e)}", OutputType.ERROR)
+
+    except Exception as e:
+        PrettyOutput.print(f"初始化错误: {str(e)}", OutputType.ERROR)
+        return 1
+
+    return 0
+
+if __name__ == "__main__":
+    exit(main())