iridet-bot 0.1.1a1__py3-none-any.whl

iribot/tools/write_file.py

@@ -0,0 +1,49 @@
+"""Write file tool"""
+from pathlib import Path
+from typing import Any, Dict
+from .base import BaseTool
+
+
+class WriteFileTool(BaseTool):
+    """Write content to files"""
+    
+    @property
+    def name(self) -> str:
+        return "write_file"
+    
+    @property
+    def description(self) -> str:
+        return "Write content to a file"
+    
+    @property
+    def parameters(self) -> Dict[str, Any]:
+        return {
+            "type": "object",
+            "properties": {
+                "file_path": {
+                    "type": "string",
+                    "description": "Path to the file to write"
+                },
+                "content": {
+                    "type": "string",
+                    "description": "Content to write to the file"
+                }
+            },
+            "required": ["file_path", "content"]
+        }
+    
+    def execute(self, file_path: str, content: str) -> Dict[str, Any]:
+        """Write content to file"""
+        try:
+            Path(file_path).parent.mkdir(parents=True, exist_ok=True)
+            with open(file_path, 'w', encoding='utf-8') as f:
+                f.write(content)
+            return {
+                "success": True,
+                "message": f"File written successfully: {file_path}"
+            }
+        except Exception as e:
+            return {
+                "success": False,
+                "error": str(e)
+            }

iridet_bot-0.1.1a1.dist-info/METADATA

@@ -0,0 +1,369 @@
+Metadata-Version: 2.4
+Name: iridet-bot
+Version: 0.1.1a1
+Summary: Full-stack AI agent with Python backend and Vue frontend
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: fastapi>=0.104.0
+Requires-Dist: uvicorn>=0.24.0
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: openai>=1.0.0
+Requires-Dist: pillow>=10.0.0
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: pydantic-settings>=2.0.0
+Requires-Dist: aiofiles>=23.0.0
+Requires-Dist: jinja2>=3.0.0
+
+# IriBot - Lightweight AI Agent Chat System
+
+A full-featured AI agent application with tool calling capabilities and real-time conversation experience. Built with Python FastAPI backend + Vue 3 frontend full-stack architecture.
+
+## 📦 PyPI 发布与 CLI
+
+- 安装：pip install iribot
+- 运行：iribot --host 127.0.0.1 --port 8000
+- 构建：使用 Makefile（make build，会自动构建前端并打包到后端静态资源）
+
+## ✨ Key Features
+
+### 🤖 AI Agent Conversation
+
+- Intelligent conversation powered by OpenAI API
+- Streaming response support for real-time AI replies
+- Image input support (vision capabilities)
+- Customizable system prompts
+
+### 🛠️ Tool Calling System
+
+The agent can autonomously call the following tools to complete tasks:
+
+- **File Operations**
+  - `read_file` - Read file contents
+  - `write_file` - Create or modify files
+  - `list_directory` - List directory contents
+
+- **Command Execution**
+  - `shell_start` - Start an interactive shell session
+  - `shell_run` - Execute commands in shell
+  - `shell_read` - Read shell output
+  - `shell_write` - Write input to shell
+  - `shell_stop` - Stop shell session
+
+### 💬 Session Management
+
+- Multi-session support, create multiple independent conversations
+- Persistent session history storage
+- Session list management (create, switch, delete)
+- Independent system prompts for each session
+
+### 🎨 Modern UI
+
+- Beautiful interface based on TDesign component library
+- Real-time tool call status display
+- Markdown message rendering support
+- Responsive design for different screen sizes
+
+## 🏗️ System Architecture
+
+```mermaid
+graph TB
+    subgraph Frontend["Frontend Layer"]
+        A[ChatSidebar<br/>Session List]
+        B[ChatContainer<br/>Chat View]
+        C[ToolCallMessage<br/>Tool Call View]
+        FE[Vue 3 + TDesign UI + Vite]
+    end
+
+    subgraph Backend["Backend Layer"]
+        D[main.py<br/>FastAPI Server]
+        E[agent.py<br/>AI Agent]
+        F[executor.py<br/>Tool Executor]
+        G[session_manager.py<br/>Session State Management]
+        H[tools/<br/>Tool Suite]
+        BE[FastAPI + OpenAI SDK]
+    end
+
+    subgraph External["External Services"]
+        I[OpenAI API / Compatible LLM Service<br/>GPT-4, GPT-3.5, or Custom Models]
+    end
+
+    Frontend -->|HTTP/SSE<br/>Server-Sent Events| Backend
+    Backend -->|OpenAI API| External
+
+    D --> E
+    E --> F
+    E --> G
+    F --> H
+
+    style Frontend fill:#e1f5ff
+    style Backend fill:#fff4e1
+    style External fill:#f0f0f0
+```
+
+### Data Flow
+
+```mermaid
+sequenceDiagram
+    participant User
+    participant Frontend
+    participant SessionManager
+    participant Agent
+    participant ToolExecutor
+    participant Tools
+    participant OpenAI
+
+    User->>Frontend: Input Message
+    Frontend->>SessionManager: POST /api/chat/stream
+    SessionManager->>SessionManager: Save user message
+    SessionManager->>Agent: Forward message
+    Agent->>OpenAI: Call OpenAI API
+
+    alt Text Response
+        OpenAI-->>Agent: Stream text content
+        Agent-->>Frontend: SSE stream
+        Frontend-->>User: Display in real-time
+    end
+
+    alt Tool Call Request
+        OpenAI-->>Agent: Return tool call request
+        Agent->>ToolExecutor: Execute tool
+        ToolExecutor->>Tools: Call specific tool
+
+        alt File Operations
+            Tools->>Tools: Read/Write file system
+        end
+
+        alt Shell Commands
+            Tools->>Tools: Execute shell commands
+        end
+
+        Tools-->>ToolExecutor: Return result
+        ToolExecutor-->>Agent: Tool execution result
+        Agent->>OpenAI: Send tool result
+        OpenAI-->>Agent: Continue generating response
+        Agent-->>Frontend: SSE stream
+        Frontend-->>User: Display response
+    end
+```
+
+## 🚀 Quick Start
+
+### Requirements
+
+- Python 3.8+
+- Node.js 16+
+- OpenAI API Key (or compatible LLM service)
+
+### Installation
+
+#### 1. Clone the Repository
+
+```bash
+git clone <repository-url>
+cd mybot
+```
+
+#### 2. Backend Setup
+
+```bash
+cd iribot
+
+# Create virtual environment (recommended)
+python -m venv venv
+source venv/bin/activate  # Windows: venv\Scripts\activate
+
+# Install dependencies
+pip install -r requirements.txt
+
+# Configure environment variables
+cp .env.example .env
+# Edit .env file and add your OpenAI API Key
+```
+
+`.env` configuration example:
+
+```ini
+OPENAI_API_KEY=sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
+OPENAI_MODEL=gpt-4-turbo-preview
+# OPENAI_BASE_URL=https://api.openai.com/v1  # Optional, use custom API endpoint
+DEBUG=false
+```
+
+#### 3. Frontend Setup
+
+```bash
+cd frontend
+
+# Install dependencies
+npm install
+```
+
+#### 4. Start Services
+
+##### Using Automated Scripts (Recommended)
+
+**Windows:**
+
+```bash
+# In project root directory
+./setup.bat
+```
+
+**Linux/macOS:**
+
+```bash
+# In project root directory
+chmod +x setup.sh
+./setup.sh
+```
+
+##### Manual Start
+
+**Backend:**
+
+```bash
+cd iribot
+uvicorn main:app --reload --port 8000
+```
+
+**Frontend:**
+
+```bash
+cd frontend
+npm run dev
+```
+
+## 🔧 Configuration
+
+### Backend Configuration
+
+Configure in `iribot/.env` file:
+
+| Config Item       | Description          | Default                |
+| ----------------- | -------------------- | ---------------------- |
+| `OPENAI_API_KEY`  | OpenAI API key       | Required               |
+| `OPENAI_MODEL`    | Model to use         | `gpt-4-vision-preview` |
+| `OPENAI_BASE_URL` | Custom API endpoint  | Empty (use official)   |
+| `DEBUG`           | Debug mode           | `false`                |
+| `BASH_PATH`       | Bash executable path | `bash`                 |
+
+### Frontend Configuration
+
+Frontend connects to backend via Vite proxy. Configuration file: `frontend/vite.config.js`
+
+```javascript
+export default {
+  server: {
+    proxy: {
+      "/api": {
+        target: "http://localhost:8000",
+        changeOrigin: true,
+      },
+    },
+  },
+};
+```
+
+## 🔌 API Endpoints
+
+### Session Management
+
+- `POST /api/sessions` - Create new session
+- `GET /api/sessions` - Get session list
+- `GET /api/sessions/{session_id}` - Get session details
+- `DELETE /api/sessions/{session_id}` - Delete session
+
+### Chat Interface
+
+- `POST /api/chat/stream` - Send message (SSE streaming response)
+
+### Tool Status
+
+- `GET /api/tools/status` - Get all tool statuses
+
+## 🛠️ Extension Development
+
+### Adding New Tools
+
+1. Create a new tool file in the `iribot/tools/` directory
+2. Inherit from `BaseTool` class:
+
+```python
+from tools.base import BaseTool
+
+class MyCustomTool(BaseTool):
+    @property
+    def name(self) -> str:
+        return "my_custom_tool"
+
+    @property
+    def description(self) -> str:
+        return "Tool description"
+
+    @property
+    def parameters(self) -> dict:
+        return {
+            "type": "object",
+            "properties": {
+                "param1": {
+                    "type": "string",
+                    "description": "Parameter description"
+                }
+            },
+            "required": ["param1"]
+        }
+
+    def execute(self, **kwargs) -> dict:
+        # Implement tool logic
+        return {
+            "success": True,
+            "result": "Execution result"
+        }
+```
+
+3. Register the tool in `executor.py`:
+
+```python
+def _register_default_tools(self):
+    # ... other tools
+    self.register_tool(MyCustomTool())
+```
+
+### Adding New Frontend Components
+
+Add tool call visualization components in the `frontend/src/components/tool-calls/` directory.
+
+## 📝 Tech Stack
+
+### Backend
+
+- **FastAPI** - Modern, fast web framework
+- **OpenAI SDK** - LLM interface calling
+- **Pydantic** - Data validation and settings management
+- **Uvicorn** - ASGI server
+
+### Frontend
+
+- **Vue 3** - Progressive JavaScript framework
+- **TDesign** - Enterprise-level UI component library
+- **Vite** - Next-generation frontend build tool
+- **Marked** - Markdown parser
+
+## 🤝 Contributing
+
+Issues and Pull Requests are welcome!
+
+## 📄 License
+
+MIT License
+
+## 🔗 Related Links
+
+- [OpenAI API Documentation](https://platform.openai.com/docs)
+- [FastAPI Documentation](https://fastapi.tiangolo.com/)
+- [Vue 3 Documentation](https://vuejs.org/)
+- [TDesign Documentation](https://tdesign.tencent.com/vue-next/overview)
+
+---
+
+**Note:** Using this project requires a valid OpenAI API Key or compatible LLM service endpoint.

iridet_bot-0.1.1a1.dist-info/RECORD

@@ -0,0 +1,24 @@
+iribot/.env.example,sha256=o8Njkm1VS5W_pv0fp-8act2VHrWm_6dQdMktOtm9fwM,96
+iribot/__init__.py,sha256=rlvjWIl_KyNPoj-2peEjdKwYVB6xOwdYoMClKCMuhas,72
+iribot/__main__.py,sha256=ls7HCIMJbxR8wU0izk-hR6LeXHfWJH6XfFUC-LNE6cE,102
+iribot/ag_ui_protocol.py,sha256=_P7J-zEYN9eHpU0ky-tl8iigZ_fMPaU-uxC9BxbJUXM,8099
+iribot/agent.py,sha256=no5lqHWG42balaIZaKRay2s3pxWMxH50CK8QzNJ1-C0,5203
+iribot/cli.py,sha256=fT_xcB5kfmzAJBPdpB-cP0vGzDRp0x-lJwrxF1klT6U,885
+iribot/config.py,sha256=8H_VmM6CTW_BxY6-46V_SAj_0c2mMhZ7lVu9sbQWnmo,1164
+iribot/executor.py,sha256=7mx2U4lTPf2tD0VnxjnTHu53k-WNG8pQaA0x1695JEY,2395
+iribot/main.py,sha256=qKSCgBEqXIxafFpEr5-7bPEYKLOIPta1TZ14XnYJiKM,11146
+iribot/models.py,sha256=4j2_vQ6ZZ9l-lhlFvVuwjuwz-cX0beLuX-9wFV9ZP2o,2360
+iribot/prompt_generator.py,sha256=Y-ctldfO-k7pCH5hqglfJ7i2hEY0iz-AJDKR1FB1Vto,3266
+iribot/session_manager.py,sha256=nWXIKTO7vv5D_7tKdU3YMEsdnZsHC3bVKg3MTHEd6Ig,7523
+iribot/templates/system_prompt.j2,sha256=mcqkwI4CTwsFIm7zqqI5a_axw0VRKZJNNZ9aYhDw99A,9631
+iribot/tools/__init__.py,sha256=e661yHzkCARwoeNO3NLl-SJLDfRAMChC37eECmi07x8,616
+iribot/tools/base.py,sha256=T2dbAI3BPDYRHHJ3wMCMWwpuI-kNa4b9KA0xxifNoBk,1791
+iribot/tools/execute_command.py,sha256=YhfFgCdhZAMkPthXoYW21iKlAH-0YABYzIBd1b-E2Gc,18530
+iribot/tools/list_directory.py,sha256=9MYO8hE7qG3qNap6fzLmmlJYfdjAKv1cYesNAYu3rYM,1299
+iribot/tools/read_file.py,sha256=hcuiEZpio92U4YOAwX1LxbewJQR52A1jodpUehywxwE,1100
+iribot/tools/write_file.py,sha256=6NAzz71iSvx-14fg4VqUU7l6t8_Dg8GFTjp_ZvwRlrU,1410
+iridet_bot-0.1.1a1.dist-info/METADATA,sha256=Vv6L9UCtu_p9apvcFiu_elGLig3vCGp4YlkBuyh873g,8724
+iridet_bot-0.1.1a1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+iridet_bot-0.1.1a1.dist-info/entry_points.txt,sha256=O4inBeR_cfncBFSCr5VcQUEwGMD7yKlcRQRwU-p6y0Y,43
+iridet_bot-0.1.1a1.dist-info/top_level.txt,sha256=byR3qpK9YFZen-PvnWbZNMIkq7pvuzmVlhLcTmePkYA,7
+iridet_bot-0.1.1a1.dist-info/RECORD,,

iridet_bot-0.1.1a1.dist-info/WHEEL

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

iridet_bot-0.1.1a1.dist-info/entry_points.txt

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

iridet_bot-0.1.1a1.dist-info/top_level.txt

@@ -0,0 +1 @@
+iribot