jarvis-ai-assistant 0.1.0__py3-none-any.whl

jarvis/tools/webpage.py

@@ -0,0 +1,90 @@
+from typing import Dict, Any
+import requests
+from bs4 import BeautifulSoup
+from ..utils import PrettyOutput, OutputType
+
+class WebpageTool:
+    name = "read_webpage"
+    description = "Read webpage content, supporting extraction of main text, title, and other information"
+    parameters = {
+        "type": "object",
+        "properties": {
+            "url": {
+                "type": "string",
+                "description": "URL of the webpage to read"
+            },
+            "extract_type": {
+                "type": "string",
+                "description": "Type of content to extract: 'text', 'title', or 'all'",
+                "enum": ["text", "title", "all"],
+                "default": "all"
+            }
+        },
+        "required": ["url"]
+    }
+
+    def execute(self, args: Dict) -> Dict[str, Any]:
+        """读取网页内容"""
+        try:
+            url = args["url"]
+            extract_type = args.get("extract_type", "all")
+            
+            # 设置请求头
+            headers = {
+                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
+            }
+            
+            # 发送请求
+            PrettyOutput.print(f"正在读取网页: {url}", OutputType.INFO)
+            response = requests.get(url, headers=headers, timeout=10)
+            response.raise_for_status()
+            
+            # 使用正确的编码
+            response.encoding = response.apparent_encoding
+            
+            # 解析HTML
+            soup = BeautifulSoup(response.text, 'html.parser')
+            
+            # 移除script和style标签
+            for script in soup(["script", "style"]):
+                script.decompose()
+            
+            result = {}
+            
+            # 提取标题
+            if extract_type in ["title", "all"]:
+                title = soup.title.string if soup.title else ""
+                result["title"] = title.strip() if title else "无标题"
+            
+            # 提取正文
+            if extract_type in ["text", "all"]:
+                text = soup.get_text(separator='\n', strip=True)
+                lines = [line.strip() for line in text.splitlines() if line.strip()]
+                result["text"] = "\n".join(lines)
+            
+            # 构建输出
+            output = []
+            if "title" in result:
+                output.append(f"标题: {result['title']}")
+                output.append("")
+            
+            if "text" in result:
+                output.append("正文内容:")
+                output.append(result["text"])
+            
+            return {
+                "success": True,
+                "stdout": "\n".join(output),
+                "stderr": ""
+            }
+
+        except requests.RequestException as e:
+            return {
+                "success": False,
+                "error": f"网页请求失败: {str(e)}"
+            }
+        except Exception as e:
+            return {
+                "success": False,
+                "error": f"解析网页失败: {str(e)}"
+            } 

jarvis/utils.py

@@ -0,0 +1,105 @@
+import sys
+import time
+import threading
+from typing import Optional
+from enum import Enum
+from datetime import datetime
+import colorama
+from colorama import Fore, Style
+
+# 初始化colorama
+colorama.init()
+
+class Spinner:
+    """加载动画类"""
+    def __init__(self, message: str = "思考中"):
+        self.spinner_chars = "⠋⠙⠹⠸⠼⠴⠦⠧⠇⠏"
+        self.message = message
+        self.running = False
+        self.spinner_thread = None
+
+    def _spin(self):
+        i = 0
+        while self.running:
+            sys.stdout.write(f"\r{Fore.BLUE}{self.spinner_chars[i]} {self.message}...{Style.RESET_ALL}")
+            sys.stdout.flush()
+            time.sleep(0.1)
+            i = (i + 1) % len(self.spinner_chars)
+        sys.stdout.write("\r" + " " * (len(self.message) + 10) + "\r")
+        sys.stdout.flush()
+
+    def start(self):
+        self.running = True
+        self.spinner_thread = threading.Thread(target=self._spin)
+        self.spinner_thread.start()
+
+    def stop(self):
+        self.running = False
+        if self.spinner_thread:
+            self.spinner_thread.join()
+
+class OutputType(Enum):
+    SYSTEM = "system"
+    CODE = "code"
+    RESULT = "result"
+    ERROR = "error"
+    INFO = "info"
+
+class PrettyOutput:
+    """美化输出类"""
+    @staticmethod
+    def format(text: str, output_type: OutputType, timestamp: bool = True) -> str:
+        # 颜色映射
+        colors = {
+            OutputType.SYSTEM: Fore.CYAN,
+            OutputType.CODE: Fore.GREEN,
+            OutputType.RESULT: Fore.BLUE,
+            OutputType.ERROR: Fore.RED,
+            OutputType.INFO: Fore.YELLOW
+        }
+
+        # 图标映射
+        icons = {
+            OutputType.SYSTEM: "🤖",
+            OutputType.CODE: "📝",
+            OutputType.RESULT: "✨",
+            OutputType.ERROR: "❌",
+            OutputType.INFO: "ℹ️"
+        }
+
+        color = colors.get(output_type, "")
+        icon = icons.get(output_type, "")
+        
+        # 添加时间戳
+        time_str = f"[{datetime.now().strftime('%H:%M:%S')}] " if timestamp else ""
+        
+        # 格式化输出
+        formatted_text = f"{color}{time_str}{icon} {text}{Style.RESET_ALL}"
+        
+        return formatted_text
+
+    @staticmethod
+    def print(text: str, output_type: OutputType, timestamp: bool = True):
+        print(PrettyOutput.format(text, output_type, timestamp)) 
+
+def get_multiline_input(tip: str) -> str:
+    """获取多行输入"""
+    PrettyOutput.print(tip + "\n", OutputType.INFO)
+    lines = []
+    
+    while True:
+        try:
+            line = input("... " if lines else ">>> ").strip()
+            # 检查是否结束输入
+            if not line:
+                if not lines:  # 如果是第一行就输入空行或finish
+                    return ""
+                break
+            
+            lines.append(line)
+            
+        except KeyboardInterrupt:
+            PrettyOutput.print("\n输入已取消", OutputType.ERROR)
+            return ""
+    
+    return "\n".join(lines).strip()

jarvis_ai_assistant-0.1.0.dist-info/METADATA

@@ -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.dist-info/RECORD

@@ -0,0 +1,35 @@
+jarvis/.jarvis,sha256=S4ZMmLqlVLHAPmnwHFQ3vYt0gnDFp-KQRH4okBh8Hpw,209
+jarvis/__init__.py,sha256=01cN4ZP2Yp7XKuyO8I-3HXUXEK_bGvuxNDlGslRbbFs,49
+jarvis/agent.py,sha256=hEatrIothG6C_RuOsygZz4ez1aF1EycnUPwY2Krwiwk,4092
+jarvis/main.py,sha256=wAeeGoRq8fFiTWUCQ31MYVeU9o6SZ48ORUow0jlMJJE,5259
+jarvis/models.py,sha256=ZQAyc39e_UsNmHkME6ATp6053safQ7Gog-oWwETsrMM,4415
+jarvis/utils.py,sha256=-fOPAgiKVPC6DevHM7A7IJvvIzvvFlGM0pHBKyBXDcA,3044
+jarvis/__pycache__/agent.cpython-313.pyc,sha256=I1JYUKC-zedURwuxQQjd9saj_yQKUhpdbb1KNWlY6Ss,4940
+jarvis/__pycache__/models.cpython-313.pyc,sha256=fot2VOV0lgODg4b_WuD-kkq-g7uZGLuZlYXiljlW13U,6350
+jarvis/__pycache__/tools.cpython-313.pyc,sha256=lAD4LrnnWzNZQmHXGfZ_2l7oskOpr2_2OC-gdFhxQY8,33933
+jarvis/__pycache__/utils.cpython-313.pyc,sha256=o8Fubq8Dcrvv3ATzQ9BnZoZjQmU3_FNPrfTt0ANdKdI,5804
+jarvis/tools/__init__.py,sha256=FNF5X32UzKvlT08-hPz7-7vcIyqqMngQYj36LXyNbV0,553
+jarvis/tools/base.py,sha256=OoLmfcg6P5deCzskGnP_6-CdbHV1ThoecpYbf0ip-oY,5074
+jarvis/tools/file_ops.py,sha256=05Vc4u-NGMjLD15WI52eL_nt_RyYt-Z_cXJnnBepf-c,3781
+jarvis/tools/python_script.py,sha256=_eK8LqNs-Mz50zdcgwbjdd8-qAeOl6kJ_qRDvWawGMw,5006
+jarvis/tools/rag.py,sha256=eBrn1fdqy-nd2nR0b4oH1EQpthTYVen-sYDhC5Ypl38,5647
+jarvis/tools/search.py,sha256=dyJmeP_s2tWv5KQejOl-TuVF12B6SXViiXEyTemD1Wo,1412
+jarvis/tools/shell.py,sha256=uGqwUGmqFg1cJkyTg7zyPk3jRwEr-woKTmqcTKdbbg8,1979
+jarvis/tools/user_confirmation.py,sha256=p0JverifHJfnALeIhtKtaVBBVEGkgSpUzT-PSycG4NY,1850
+jarvis/tools/user_interaction.py,sha256=xOaoYsCKTa9-K3mg5tsu1zegn1HKcO40k2aJ3VygZqM,3073
+jarvis/tools/webpage.py,sha256=UTEomu5j7jbOw8c5az2jsjv5E7LeokWKj1QahvZO7xc,3077
+jarvis/tools/__pycache__/__init__.cpython-313.pyc,sha256=wktmOMuGMwQjQ--eRzooUNsL5YjY0bS_CbcSFUWze54,674
+jarvis/tools/__pycache__/base.cpython-313.pyc,sha256=9gpGOdHaHuk5m9lFrZ_hVUZ1RQSOOwvKOy9MIJFaqzk,7327
+jarvis/tools/__pycache__/file_ops.cpython-313.pyc,sha256=bawv11xhWSj5wCtW0QeJLI-InhUBUXaSkogq6rZzvTQ,3636
+jarvis/tools/__pycache__/python_script.cpython-313.pyc,sha256=8JpryqTovEiTvBlWAK1KjZmPvHUuPc9GT9rTXBEQoJc,6693
+jarvis/tools/__pycache__/rag.cpython-313.pyc,sha256=JH6-PSZRMKAvTZqCwlRXJGClxYXNMs-vetU0q7hBLz0,6064
+jarvis/tools/__pycache__/search.cpython-313.pyc,sha256=VX9zztOwIsPCkYwev0A0XJGyu4Tr0PQcQkbbqx8vfB0,1870
+jarvis/tools/__pycache__/shell.cpython-313.pyc,sha256=Xtqt-LzMRfsCXeCE7QxFY8TR8XEeQG_jic73YLkhXQw,2302
+jarvis/tools/__pycache__/user_confirmation.cpython-313.pyc,sha256=wK3Ev10lHSUSRvoYmi7A0GzxYkzU-C4Wfhs5qW_HBqs,2271
+jarvis/tools/__pycache__/user_interaction.cpython-313.pyc,sha256=RuVZ-pmiPBDywY3efgXSfohMAciC1avMGPmBK5qlnew,3305
+jarvis/tools/__pycache__/webpage.cpython-313.pyc,sha256=VcpkaV8IyOOtebedXjn8ybjr8AglU-3-Cs80yzE4FYo,3628
+jarvis_ai_assistant-0.1.0.dist-info/METADATA,sha256=EbX9DRefJ8ugI5t5qQH1HeHfT4c21YyMYez0OI_Rye8,3675
+jarvis_ai_assistant-0.1.0.dist-info/WHEEL,sha256=A3WOREP4zgxI0fKrHUG8DC8013e3dK3n7a6HDbcEIwE,91
+jarvis_ai_assistant-0.1.0.dist-info/entry_points.txt,sha256=iKu7OMfew9dtfGhW71gIMTg4wvafuPqKb4wyQOnMAGU,44
+jarvis_ai_assistant-0.1.0.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
+jarvis_ai_assistant-0.1.0.dist-info/RECORD,,

jarvis_ai_assistant-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (75.7.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

jarvis_ai_assistant-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+jarvis = jarvis.main:main

jarvis_ai_assistant-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+jarvis