├── .dockerignore ├── .github └── workflows │ ├── docker-image.yml │ └── release.yml ├── .gitignore ├── DISCLAIMER.md ├── LICENSE ├── README.md ├── README.zh.md ├── aipyapp ├── __init__.py ├── __main__.py ├── aipy │ ├── __init__.py │ ├── blocks.py │ ├── config.py │ ├── diagnose.py │ ├── interface.py │ ├── libmcp.py │ ├── llm.py │ ├── plugin.py │ ├── prompt.py │ ├── runtime.py │ ├── task.py │ ├── taskmgr.py │ ├── trustoken.py │ ├── utils.py │ └── wizard.py ├── cli │ ├── __init__.py │ ├── cli_ipython.py │ ├── cli_python.py │ └── cli_task.py ├── config │ ├── __init__.py │ ├── base.py │ └── llm.py ├── exec │ ├── __init__.py │ ├── runner.py │ └── runtime.py ├── gui │ ├── __init__.py │ ├── about.py │ ├── apimarket.py │ ├── config.py │ ├── main.py │ ├── providers.py │ ├── statusbar.py │ └── trustoken.py ├── i18n.py ├── llm │ ├── __init__.py │ ├── base.py │ ├── base_openai.py │ ├── client_claude.py │ └── client_ollama.py └── res │ ├── DISCLAIMER.md │ ├── aipy.ico │ ├── chatroom_en.html │ ├── chatroom_zh.html │ ├── console_code.html │ ├── console_white.html │ ├── css │ ├── github-dark.min.css │ └── github.min.css │ ├── default.toml │ ├── js │ ├── highlight.min.js │ ├── marked.min.js │ └── purify.min.js │ └── locales.csv ├── dev ├── CONFIG.md ├── ChangeLog.md ├── Event.md ├── FEATURES.md ├── MCP.md └── Plugin.md ├── docker ├── Dockerfile ├── Dockerfile.deb ├── entrypoint.sh └── run.sh ├── docs ├── CNAME ├── README.md ├── README.zh.md ├── aipy.jpg └── python-use-wf.png ├── pyproject.toml └── uv.lock /.dockerignore: -------------------------------------------------------------------------------- 1 | .venv 2 | __pycache__ 3 | aipy.toml 4 | aipython.toml 5 | dist/ 6 | work/ 7 | build/ 8 | client.crt 9 | 10 | -------------------------------------------------------------------------------- /.github/workflows/docker-image.yml: -------------------------------------------------------------------------------- 1 | name: Docker Image CI 2 | 3 | on: 4 | push: 5 | tags: 6 | - "v*" 7 | paths: 8 | - 'docker/Dockerfile' 9 | - 'docker/Dockerfile.deb' 10 | - 'aipyapp/**' 11 | - 'pyproject.toml' 12 | workflow_dispatch: 13 | 14 | jobs: 15 | build-amd64: 16 | runs-on: ubuntu-latest 17 | steps: 18 | - name: Checkout code 19 | uses: actions/checkout@v3 20 | 21 | - name: Login to Docker Hub 22 | uses: docker/login-action@v2 23 | with: 24 | username: ${{ secrets.DOCKERHUB_USERNAME }} 25 | password: ${{ secrets.DOCKERHUB_PASSWORD }} 26 | 27 | - name: Build amd64 Image 28 | uses: docker/build-push-action@v3 29 | with: 30 | platforms: linux/amd64 31 | push: true 32 | file: docker/Dockerfile 33 | tags: ${{ secrets.DOCKERHUB_USERNAME }}/aipy:amd64-latest 34 | 35 | build-arm64: 36 | needs: build-amd64 # 确保在 amd64 构建完成后执行 37 | runs-on: ubuntu-latest 38 | steps: 39 | - name: Checkout code 40 | uses: actions/checkout@v3 41 | 42 | - name: Set up QEMU 43 | uses: docker/setup-qemu-action@v2 # 设置 ARM 架构支持 44 | 45 | - name: Login to Docker Hub 46 | uses: docker/login-action@v2 47 | with: 48 | username: ${{ secrets.DOCKERHUB_USERNAME }} 49 | password: ${{ secrets.DOCKERHUB_PASSWORD }} 50 | 51 | - name: Build arm64 Image 52 | uses: docker/build-push-action@v3 53 | with: 54 | platforms: linux/arm64 55 | push: true 56 | file: docker/Dockerfile.deb 57 | tags: ${{ secrets.DOCKERHUB_USERNAME }}/aipy:arm64-latest 58 | 59 | manifest: 60 | needs: [build-amd64, build-arm64] 61 | runs-on: ubuntu-latest 62 | steps: 63 | - name: Login to Docker Hub 64 | uses: docker/login-action@v2 65 | with: 66 | username: ${{ secrets.DOCKERHUB_USERNAME }} 67 | password: ${{ secrets.DOCKERHUB_PASSWORD }} 68 | 69 | - name: Create and Push Manifest List 70 | run: | 71 | docker manifest create ${{ secrets.DOCKERHUB_USERNAME }}/aipy:latest \ 72 | --amend ${{ secrets.DOCKERHUB_USERNAME }}/aipy:amd64-latest \ 73 | --amend ${{ secrets.DOCKERHUB_USERNAME }}/aipy:arm64-latest 74 | 75 | docker manifest push ${{ secrets.DOCKERHUB_USERNAME }}/aipy:latest 76 | -------------------------------------------------------------------------------- /.github/workflows/release.yml: -------------------------------------------------------------------------------- 1 | name: Build and Publish Python Package 2 | 3 | on: 4 | push: 5 | tags: 6 | - "v*" 7 | workflow_dispatch: 8 | 9 | permissions: 10 | contents: write 11 | 12 | jobs: 13 | build: 14 | runs-on: ubuntu-latest 15 | steps: 16 | - name: Checkout Repository 17 | uses: actions/checkout@v4 18 | 19 | - name: Install uv 20 | uses: astral-sh/setup-uv@v5 21 | 22 | - name: Set up Python 23 | uses: actions/setup-python@v5 24 | with: 25 | python-version-file: "pyproject.toml" 26 | 27 | - name: Install the project 28 | run: uv sync --all-extras 29 | 30 | - name: Update __version__ in __init__.py 31 | run: | 32 | sed -i "s/__version__ = '.*'/__version__ = '${GITHUB_REF_NAME#v}'/" aipyapp/__init__.py 33 | 34 | - name: Build Package 35 | run: uv build 36 | 37 | - name: Upload Package as Artifact 38 | uses: actions/upload-artifact@v4 39 | with: 40 | name: python-package 41 | path: dist/* 42 | 43 | - name: Upload to GitHub Releases 44 | uses: softprops/action-gh-release@v2 45 | with: 46 | files: dist/* 47 | token: ${{ secrets.GITHUB_TOKEN }} 48 | 49 | - name: Publish to PyPI 50 | env: 51 | UV_PUBLISH_TOKEN: ${{ secrets.PYPI_TOKEN }} 52 | run: uv publish -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | # Python-generated files 2 | __pycache__/ 3 | *.py[oc] 4 | build/ 5 | dist/ 6 | wheels/ 7 | *.egg-info 8 | 9 | # Virtual environments 10 | .venv 11 | 12 | aipy.toml 13 | aipython.toml 14 | .DS_Store 15 | work/ 16 | client.crt 17 | 18 | -------------------------------------------------------------------------------- /DISCLAIMER.md: -------------------------------------------------------------------------------- 1 | ### ⚠️ 风险提示与免责声明 ⚠️ 2 | 3 | 本程序可生成并执行由大型语言模型（LLM）自动生成的代码。请您在继续使用前，务必阅读并理解以下内容： 4 | 5 | 1. **风险提示：** 6 | - 自动生成的代码可能包含逻辑错误、性能问题或不安全操作（如删除文件、访问网络、执行系统命令等）。 7 | - 本程序无法保证生成代码的准确性、完整性或适用性。 8 | - 在未充分审查的情况下运行生成代码，**可能会对您的系统、数据或隐私造成损害**。 9 | 10 | 2. **免责声明：** 11 | - 本程序仅作为开发与测试用途提供，不对由其生成或执行的任何代码行为承担责任。 12 | - 使用本程序即表示您理解并接受所有潜在风险，并同意对因使用本程序产生的任何后果自行负责。 13 | 14 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- 1 | Copyright (C) 2025 Beijing Knownsec Technology Co., Ltd 2 | 3 | This software is licensed under the GPL 3.0, allowing individuals and businesses to use, modify, and distribute it for free, 4 | provided that they comply with the terms of the license; 5 | 6 | Additional Restrictions: 7 | 1. It is prohibited to provide this project as a SaaS service to third parties without permission; 8 | 2. It is prohibited to integrate this project or its code into commercial products for sale or giveaway without permission; 9 | If you need to engage in the above actions, please contact［sec@knownsec.com] to obtain a commercial license. 10 | 11 | 12 | Copyright (C) 2025 北京知道创宇信息技术股份有限公司 13 | 本软件遵循GPL3.0协议，允许个人和企业在遵守协议的前提下免费使用、修改和分发； 14 | 额外限制： 15 | 1. 禁止未经许可将本项目作为SaaS服务提供给第三方； 16 | 2. 禁止未经许可将本项目或其代码集成到商业产品中销售或赠送； 17 | 若需上述行为，需联系[sec@knownsec.com]获取商业授权。 18 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | # Python use 2 | AIPy is an implementation of the Python-use concept, demonstrating its practical value and potential. 3 | - **Mission**: unleash the full potential of large language models. 4 | - **Vision**: a future where LLMs can think independently and proactively leverage AIPy to solve complex problems. 5 | 6 | ## What 7 | Python use provides the entire Python execution environment to LLM. Imagine LLM sitting in front of a computer, typing various commands into the Python command-line interpreter, pressing Enter to execute, observing the results, and then typing and executing more code. 8 | 9 | Unlike Agents, Python use does not define any tools interface. LLM can freely use all the features provided by the Python runtime environment. 10 | 11 | ## Why 12 | If you are a data engineer, you are likely familiar with the following scenarios: 13 | - Handling various data file formats: csv/excel, json, html, sqlite, parquet, etc. 14 | - Performing operations like data cleaning, transformation, computation, aggregation, sorting, grouping, filtering, analysis, and visualization. 15 | 16 | This process often requires: 17 | - Starting Python, importing pandas as pd, and typing a bunch of commands to process data. 18 | - Generating a bunch of intermediate temporary files. 19 | - Describing your needs to ChatGPT/Claude, copying the generated data processing code, and running it manually. 20 | 21 | So, why not start the Python command-line interpreter, directly describe your data processing needs, and let it be done automatically? The benefits are: 22 | - No need to manually input a bunch of Python commands temporarily. 23 | - No need to describe your needs to GPT, copy the program, and run it manually. 24 | 25 | This is the problem Python use aims to solve! 26 | 27 | ## How 28 | Python use (aipython) is a Python command-line interpreter integrated with LLM. You can: 29 | - Enter and execute Python commands as usual. 30 | - Describe your needs in natural language, and aipython will automatically generate Python commands and execute them. 31 | 32 | Moreover, the two modes can access data interchangeably. For example, after aipython processes your natural language commands, you can use standard Python commands to view various data. 33 | 34 | ## Usage 35 | AIPython has two running modes: 36 | - Task mode: Very simple and easy to use, just input your task, suitable for users unfamiliar with Python. 37 | - Python mode: Suitable for users familiar with Python, allowing both task input and Python commands, ideal for advanced users. 38 | 39 | The default running mode is task mode, which can be switched to Python mode using the `--python` parameter. 40 | 41 | ### Basic Config 42 | aipy.toml: 43 | ```toml 44 | [llm.deepseek] 45 | type = "deepseek" 46 | api_key = "Your DeepSeek API Key" 47 | enable = true 48 | default = true 49 | ``` 50 | 51 | ### Task Mode 52 | `uv run aipy` 53 | ``` 54 | >>> Get the latest posts from Reddit r/LocalLLaMA 55 | ...... 56 | ...... 57 | >>> /done 58 | ``` 59 | 60 | `pip install aipyapp` and run with `aipy` 61 | 62 | ``` 63 | -> % aipy 64 | 🚀 Python use - AIPython (0.1.22) [https://aipy.app] 65 | >> Get the latest posts from Reddit r/LocalLLaMA 66 | ...... 67 | >> 68 | ``` 69 | 70 | ### Python Mode 71 | 72 | #### Basic Usage 73 | Automatic task processing: 74 | 75 | ``` 76 | >>> ai("Get the title of Google's homepage") 77 | ``` 78 | 79 | #### Automatically Request to Install Third-Party Libraries 80 | ``` 81 | Python use - AIPython (Quit with 'exit()') 82 | >>> ai("Use psutil to list all processes on MacOS") 83 | 84 | 📦 LLM requests to install third-party packages: ['psutil'] 85 | If you agree and have installed, please enter 'y [y/n] (n): y 86 | 87 | ``` 88 | 89 | ## Interfaces 90 | ### ai Object 91 | - \_\_call\_\_(instruction): Execute the automatic processing loop until LLM no longer returns code messages 92 | - save(path): Save the interaction process to an svg or html file 93 | - llm Property: LLM object 94 | - runner Property: Runner object 95 | 96 | ### LLM Object 97 | - history Property: Message history of the interaction process between the user and LLM 98 | 99 | ### Runner Object 100 | - globals: Global variables of the Python environment executing the code returned by LLM 101 | - locals: Local variables of the Python environment executing the code returned by LLM 102 | 103 | ### runtime Object 104 | For the code generated by LLM to call, providing the following interface: 105 | - install_packages(packages): Request to install third-party packages 106 | - getenv(name, desc=None): Get environment variables 107 | - display(path=None, url=None): Display images in the terminal 108 | 109 | ## TODO 110 | - Use AST to automatically detect and fix Python code returned by LLM 111 | 112 | ## Thanks 113 | - Hei Ge: Product manager/senior user/chief tester 114 | - Sonnet 3.7: Generated the first version of the code, which was almost ready to use without modification. 115 | - ChatGPT: Provided many suggestions and code snippets, especially for the command-line interface. 116 | - Codeium: Intelligent code completion 117 | - Copilot: Code improvement suggestions and README translation 118 | 119 | 120 | 121 | -------------------------------------------------------------------------------- /README.zh.md: -------------------------------------------------------------------------------- 1 | # Python use 2 | AIPy 是 Python-use 概念的一个具体实现，旨在展示该理念的实际价值与应用潜力。 3 | 4 | - **使命**: 释放大语言模型的全部潜能 5 | - **愿景**: 能够自主改进和使用 AIPy 的更聪明的LLM 6 | 7 | ## What 8 | Python use 是把整个 Python 执行环境提供给 LLM 使用，可以想象为 LLM 坐在电脑前用键盘在 Python 命令行解释器里输入各种命令，按回车运行，然后观察执行结果，再输入代码和执行。 9 | 10 | 和 Agent 的区别是 Python use 不定义任何 tools 接口，LLM 可以自由使用 Python 运行环境提供的所有功能。 11 | 12 | ## Why 13 | 假如你是一个数据工程师，你对下面的场景一定不陌生： 14 | - 处理各种不同格式的数据文件：csv/excel，json，html, sqlite, parquet ... 15 | - 对数据进行清洗，转换，计算，聚合，排序，分组，过滤，分析，可视化等操作 16 | 17 | 这个过程经常需要： 18 | - 启动 Python，import pandas as pd，输入一堆命令处理数据 19 | - 生成一堆中间临时文件 20 | - 找 ChatGPT / Claude 描述你的需要，手工拷贝生成的数据处理代码运行。 21 | 22 | 所以，为什么不启动 Python 命令行解释器后，直接描述你的数据处理需求，然后自动完成？好处是： 23 | - 无需手工临时输入一堆 Python 命令 24 | - 无需去找 GPT 描述需求，拷贝程序，然后手工运行 25 | 26 | 这就是 Python use 要解决的问题！ 27 | 28 | ## How 29 | Python use (aipython) 是一个集成 LLM 的 Python 命令行解释器。你可以： 30 | - 像往常一样输入和执行 Python 命令 31 | - 用自然语言描述你的需求，aipython 会自动生成 Python 命令，然后执行 32 | 33 | 而且，两种模式可以互相访问数据。例如，aipython 处理完你的自然语言命令后，你可以用标准 Python 命令查看各种数据。 34 | 35 | ## Interfaces 36 | ### ai 对象 37 | - \_\_call\_\_(instruction): 执行自动处理循环，直到 LLM 不再返回代码消息 38 | - save(path): 保存交互过程到 svg 或 html 文件 39 | - llm 属性： LLM 对象 40 | - runner 属性： Runner 对象 41 | 42 | ### LLM 对象 43 | - history 属性： 用户和LLL交互过程的消息历史 44 | 45 | ### Runner 对象 46 | - globals: 执行 LLM 返回代码的 Python 环境全局变量 47 | - locals: 执行 LLM 返回代码的 Python 环境局部变量 48 | 49 | ### runtime 对象 50 | 供 LLM 生成的代码调用，提供以下接口： 51 | - install_packages(packages): 申请安装第三方包 52 | - getenv(name, desc=None): 获取环境变量 53 | - display(path=None, url=None): 在终端显示图片 54 | 55 | ## Usage 56 | AIPython 有两种运行模式： 57 | - 任务模式：非常简单易用，直接输入你的任务即可，适合不熟悉 Python 的用户。 58 | - Python模式：适合熟悉 Python 的用户，既可以输入任务也可以输入 Python 命令，适合高级用户。 59 | 60 | 默认运行模式是任务模式，可以通过 `--python` 参数切换到 Python 模式。 61 | 62 | ### 任务模式 63 | `uv run aipython` 64 | 65 | ``` 66 | >>> 获取Reddit r/LocalLLaMA 最新帖子 67 | ...... 68 | ...... 69 | >>> /done 70 | ``` 71 | 72 | `pip install aipyapp` ，运行aipy命令进入任务模式 73 | 74 | ``` 75 | -> % aipy 76 | 🚀 Python use - AIPython (0.1.22) [https://aipy.app] 77 | 请输入需要 AI 处理的任务 (输入 /use <下述 LLM> 切换) 78 | >> 获取Reddit r/LocalLLaMA 最新帖子 79 | ...... 80 | >> 81 | ``` 82 | 83 | ### Python 模式 84 | 85 | #### 基本用法 86 | 自动任务处理： 87 | 88 | ``` 89 | >>> ai("获取Google官网首页标题") 90 | ``` 91 | 92 | #### 自动申请安装第三方库 93 | ``` 94 | Python use - AIPython (Quit with 'exit()') 95 | >>> ai("使用psutil列出当前MacOS所有进程列表") 96 | 97 | 📦 LLM 申请安装第三方包: ['psutil'] 98 | 如果同意且已安装，请输入 'y [y/n] (n): y 99 | 100 | ``` 101 | 102 | ## TODO 103 | - 使用 AST 自动检测和修复 LLM 返回的 Python 代码 104 | 105 | ## Thanks 106 | - 黑哥: 产品经理/资深用户/首席测试官 107 | - Sonnet 3.7: 生成了第一版的代码，几乎无需修改就能使用。 108 | - ChatGPT: 提供了很多建议和代码片段，特别是命令行接口。 109 | - Codeium: 代码智能补齐 110 | - Copilot: 代码改进建议和翻译 README 111 | 112 | 113 | 114 | -------------------------------------------------------------------------------- /aipyapp/__init__.py: -------------------------------------------------------------------------------- 1 | from importlib import resources 2 | 3 | from .i18n import T, set_lang, get_lang 4 | 5 | __version__ = '0.2.0' 6 | 7 | __respkg__ = f'{__package__}.res' 8 | __respath__ = resources.files(__respkg__) 9 | 10 | __all__ = ['T', 'set_lang', 'get_lang', '__version__', '__respkg__', '__respath__'] -------------------------------------------------------------------------------- /aipyapp/__main__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # coding: utf-8 3 | 4 | import os 5 | import sys 6 | 7 | if "pythonw" in sys.executable.lower(): 8 | sys.stdout = open(os.devnull, "w", encoding='utf-8') 9 | sys.stderr = open(os.devnull, "w", encoding='utf-8') 10 | 11 | from loguru import logger 12 | 13 | logger.remove() 14 | from .aipy.config import CONFIG_DIR 15 | logger.add(CONFIG_DIR / "aipyapp.log", format="{time:HH:mm:ss} | {level} | {message} | {extra}", level='INFO') 16 | 17 | def parse_args(): 18 | import argparse 19 | config_help_message = ( 20 | f"Specify the configuration directory.\nDefaults to {CONFIG_DIR} if not provided." 21 | ) 22 | 23 | parser = argparse.ArgumentParser(description="Python use - AIPython", formatter_class=argparse.RawTextHelpFormatter) 24 | parser.add_argument("-c", '--config-dir', type=str, help=config_help_message) 25 | parser.add_argument('-p', '--python', default=False, action='store_true', help="Python mode") 26 | parser.add_argument('-i', '--ipython', default=False, action='store_true', help="IPython mode") 27 | parser.add_argument('-g', '--gui', default=False, action='store_true', help="GUI mode") 28 | parser.add_argument('--debug', default=False, action='store_true', help="Debug mode") 29 | parser.add_argument('-f', '--fetch-config', default=False, action='store_true', help="login to trustoken and fetch token config") 30 | parser.add_argument('cmd', nargs='?', default=None, help="Task to execute, e.g. 'Who are you?'") 31 | return parser.parse_args() 32 | 33 | def ensure_pkg(pkg): 34 | try: 35 | import wx 36 | except: 37 | import subprocess 38 | 39 | cp = subprocess.run([sys.executable, "-m", "pip", "install", pkg]) 40 | assert cp.returncode == 0 41 | 42 | def mainw(): 43 | args = parse_args() 44 | ensure_pkg('wxpython') 45 | from .gui.main import main as aipy_main 46 | aipy_main(args) 47 | 48 | def main(): 49 | args = parse_args() 50 | if args.python: 51 | from .cli.cli_python import main as aipy_main 52 | elif args.ipython: 53 | ensure_pkg('ipython') 54 | from .cli.cli_ipython import main as aipy_main 55 | elif args.gui: 56 | ensure_pkg('wxpython') 57 | from .gui.main import main as aipy_main 58 | else: 59 | from .cli.cli_task import main as aipy_main 60 | aipy_main(args) 61 | 62 | if __name__ == '__main__': 63 | main() 64 | -------------------------------------------------------------------------------- /aipyapp/aipy/__init__.py: -------------------------------------------------------------------------------- 1 | from .taskmgr import TaskManager 2 | from .plugin import event_bus 3 | from .config import ConfigManager, CONFIG_DIR 4 | 5 | __all__ = ['TaskManager', 'event_bus', 'ConfigManager', 'CONFIG_DIR'] 6 | -------------------------------------------------------------------------------- /aipyapp/aipy/blocks.py: -------------------------------------------------------------------------------- 1 | #! /usr/bin/env python3 2 | # -*- coding: utf-8 -*- 3 | 4 | import re 5 | import json 6 | from pathlib import Path 7 | from collections import OrderedDict 8 | from dataclasses import dataclass 9 | from typing import Optional, Dict, Any 10 | 11 | from loguru import logger 12 | 13 | from .libmcp import extract_call_tool 14 | 15 | @dataclass 16 | class CodeBlock: 17 | """代码块对象""" 18 | id: str 19 | lang: str 20 | code: str 21 | path: Optional[str] = None 22 | 23 | def save(self): 24 | """保存代码块到文件""" 25 | if not self.path: 26 | return False 27 | 28 | path = Path(self.path) 29 | path.parent.mkdir(parents=True, exist_ok=True) 30 | path.write_text(self.code, encoding='utf-8') 31 | return True 32 | 33 | @property 34 | def abs_path(self): 35 | if self.path: 36 | return Path(self.path).absolute() 37 | return None 38 | 39 | def get_lang(self): 40 | lang = self.lang.lower() 41 | return lang 42 | 43 | def to_dict(self) -> Dict[str, Any]: 44 | """转换为字典""" 45 | return { 46 | 'id': self.id, 47 | 'lang': self.lang, 48 | 'code': self.code, 49 | 'path': self.path, 50 | } 51 | 52 | def __repr__(self): 53 | return f"" 54 | 55 | class CodeBlocks: 56 | def __init__(self, console): 57 | self.console = console 58 | self.blocks = OrderedDict() 59 | self.code_pattern = re.compile( 60 | r'\s*```(\w+)?\s*\n(.*?)\n```\s*', 61 | re.DOTALL 62 | ) 63 | self.line_pattern = re.compile( 64 | r'' 65 | ) 66 | self.log = logger.bind(src='code_blocks') 67 | 68 | def parse(self, markdown_text, parse_mcp=False): 69 | blocks = OrderedDict() 70 | errors = [] 71 | for match in self.code_pattern.finditer(markdown_text): 72 | start_json, lang, content, end_json = match.groups() 73 | try: 74 | start_meta = json.loads(start_json) 75 | end_meta = json.loads(end_json) 76 | except json.JSONDecodeError as e: 77 | self.console.print_exception(show_locals=True) 78 | error = {'JSONDecodeError': {'json_str': start_json, 'reason': str(e)}} 79 | errors.append(error) 80 | continue 81 | 82 | code_id = start_meta.get("id") 83 | if code_id != end_meta.get("id"): 84 | self.log.error("Start and end id mismatch", start_id=code_id, end_id=end_meta.get("id")) 85 | error = {'Start and end id mismatch': {'start_id': code_id, 'end_id': end_meta.get("id")}} 86 | errors.append(error) 87 | continue 88 | 89 | if code_id in blocks or code_id in self.blocks: 90 | self.log.error("Duplicate code id", code_id=code_id) 91 | error = {'Duplicate code id': {'code_id': code_id}} 92 | errors.append(error) 93 | continue 94 | 95 | # 创建代码块对象 96 | block = CodeBlock( 97 | id=code_id, 98 | lang=lang, 99 | code=content, 100 | path=start_meta.get('path') 101 | ) 102 | 103 | blocks[code_id] = block 104 | self.log.info("Parsed code block", code_block=block) 105 | 106 | try: 107 | block.save() 108 | self.log.info("Saved code block", code_block=block) 109 | except Exception as e: 110 | self.log.error("Failed to save file", code_block=block, reason=e) 111 | 112 | self.blocks.update(blocks) 113 | 114 | exec_blocks = [] 115 | line_matches = self.line_pattern.findall(markdown_text) 116 | for line_match in line_matches: 117 | cmd, json_str = line_match 118 | try: 119 | line_meta = json.loads(json_str) 120 | except json.JSONDecodeError as e: 121 | self.log.error("Invalid JSON in Cmd-{cmd} block", json_str=json_str, reason=e) 122 | error = {f'Invalid JSON in Cmd-{cmd} block': {'json_str': json_str, 'reason': str(e)}} 123 | errors.append(error) 124 | continue 125 | 126 | error = None 127 | if cmd == 'Exec': 128 | exec_id = line_meta.get("id") 129 | if not exec_id: 130 | error = {'Cmd-Exec block without id': {'json_str': json_str}} 131 | elif exec_id not in self.blocks: 132 | error = {'Cmd-Exec block not found': {'exec_id': exec_id}} 133 | else: 134 | exec_blocks.append(self.blocks[exec_id]) 135 | else: 136 | error = {f'Unknown command in Cmd-{cmd} block': {'cmd': cmd}} 137 | 138 | if error: 139 | errors.append(error) 140 | 141 | ret = {} 142 | if errors: ret['errors'] = errors 143 | if exec_blocks: ret['exec_blocks'] = exec_blocks 144 | if blocks: ret['blocks'] = [v for v in blocks.values()] 145 | 146 | if parse_mcp and not blocks: 147 | json_content = extract_call_tool(markdown_text) 148 | if json_content: 149 | ret['call_tool'] = json_content 150 | self.log.info("Parsed MCP call_tool", json_content=json_content) 151 | 152 | return ret 153 | 154 | def get_code_by_id(self, code_id): 155 | try: 156 | return self.blocks[code_id].code 157 | except KeyError: 158 | self.log.error("Code id not found", code_id=code_id) 159 | self.console.print("❌ Code id not found", code_id=code_id) 160 | return None 161 | 162 | def get_block_by_id(self, code_id): 163 | try: 164 | return self.blocks[code_id] 165 | except KeyError: 166 | self.log.error("Code id not found", code_id=code_id) 167 | self.console.print("❌ Code id not found", code_id=code_id) 168 | return None 169 | 170 | def to_list(self): 171 | """将 CodeBlocks 对象转换为 JSON 字符串 172 | 173 | Returns: 174 | str: JSON 格式的字符串 175 | """ 176 | blocks = [block.to_dict() for block in self.blocks.values()] 177 | return blocks -------------------------------------------------------------------------------- /aipyapp/aipy/diagnose.py: -------------------------------------------------------------------------------- 1 | #! /usr/bin/env python3 2 | # -*- coding: utf-8 -*- 3 | 4 | import sys 5 | import json 6 | import uuid 7 | import time 8 | import platform 9 | import locale 10 | import mimetypes 11 | from io import BytesIO 12 | 13 | import requests 14 | from loguru import logger 15 | 16 | from aipyapp import __version__ 17 | from aipyapp.aipy.config import CONFIG_DIR 18 | 19 | CONFIG_FILE = CONFIG_DIR / '.diagnose.json' 20 | UPDATE_INTERVAL = 8 * 3600 21 | 22 | class NoopDiagnose: 23 | def __getattr__(self, name): 24 | def noop(*args, **kwargs): 25 | pass 26 | return noop 27 | 28 | class Diagnose: 29 | def __init__(self, api_url, api_key): 30 | self._api_url = api_url 31 | self._api_key = api_key 32 | self.log = logger.bind(src='diagnose') 33 | self.load_config() 34 | 35 | def save_config(self): 36 | CONFIG_FILE.write_text(json.dumps({'xid': self._xid, 'last_update': self._last_update})) 37 | 38 | def load_config(self): 39 | if not CONFIG_FILE.exists(): 40 | self._xid = uuid.uuid4().hex 41 | self._last_update = 0 42 | self.save_config() 43 | else: 44 | with CONFIG_FILE.open('r') as f: 45 | data = json.load(f) 46 | self._xid = data.get('xid') 47 | self._last_update = data.get('last_update') 48 | 49 | @classmethod 50 | def create(cls, settings): 51 | config = settings.get('diagnose') 52 | if config: 53 | api_key = config.get('api_key') 54 | api_url = config.get('api_url') 55 | enabled = config.get('enabled', True) 56 | if not api_url: 57 | enabled = False 58 | else: 59 | enabled = False 60 | 61 | return cls(api_url, api_key) if enabled else NoopDiagnose() 62 | 63 | def get_meta(self): 64 | return { 65 | "system": platform.system(), 66 | "version": platform.version(), 67 | "platform": platform.platform(), 68 | "arch": platform.machine(), 69 | "python": sys.executable, 70 | "python_version": sys.version, 71 | "python_base_prefix": sys.base_prefix, 72 | "locale": locale.getlocale(), 73 | } 74 | 75 | def check_update(self, force=False): 76 | if not force and int(time.time()) - self._last_update < UPDATE_INTERVAL: 77 | return {} 78 | self._last_update = int(time.time()) 79 | self.save_config() 80 | 81 | data = { 82 | "xid": self._xid, 83 | "version": __version__, 84 | "meta": self.get_meta() 85 | } 86 | headers = { 87 | "Content-Type": "application/json", 88 | #"API-KEY": self._api_key 89 | } 90 | 91 | try: 92 | response = requests.post( 93 | f"{self._api_url}/4b16535b7e6147f2861508a2ad5f5ce8", 94 | json=data, 95 | headers=headers 96 | ) 97 | if response.status_code == 200: 98 | result = response.json() 99 | if result.get("success", False): 100 | return { 101 | "has_update": result.get("has_update", False), 102 | "latest_version": result.get("latest_version", "unknown"), 103 | "current_version": __version__, 104 | } 105 | else: 106 | self.log.error(f"Server error: {result.get('error', '未知错误')}") 107 | else: 108 | self.log.error(f"Request failed: HTTP {response.status_code}") 109 | 110 | except Exception as e: 111 | self.log.error(f"Connection error: {str(e)}") 112 | return {"error": "版本检查失败"} 113 | 114 | def report_data(self, data, filename): 115 | try: 116 | # 确保数据是字符串格式 117 | if isinstance(data, (dict, list)): 118 | data = json.dumps(data, ensure_ascii=False, indent=4) 119 | elif not isinstance(data, str): 120 | data = str(data) 121 | 122 | # 创建文件对象 123 | file_data = BytesIO(data.encode('utf-8')) 124 | file_data.seek(0) # 确保文件指针在开始位置 125 | except Exception as e: 126 | self.log.error(f"Failed to prepare data: {str(e)}") 127 | return {'success': False, 'error': str(e)} 128 | 129 | headers = {'API-KEY': self._api_key} 130 | content_type, _ = mimetypes.guess_type(filename) 131 | if not content_type: 132 | content_type = 'application/octet-stream' 133 | 134 | files = { 135 | 'file': ( 136 | filename, 137 | file_data, 138 | content_type 139 | ) 140 | } 141 | 142 | error = 'Unknown error' 143 | try: 144 | response = requests.post(f"{self._api_url}/a6477529c8c34b6a8ca4bc2d7253ab76", files=files, headers=headers) 145 | if 200 <= response.status_code < 300: 146 | try: 147 | result = response.json() 148 | if result.get('success') and 'viewUrl' in result: 149 | self.log.info(f"Report uploaded successfully. View URL: {result['viewUrl']}") 150 | return {'success': True, 'url': result['viewUrl']} 151 | else: 152 | self.log.error(f"Upload failed: {result.get('error', 'Unknown error')}") 153 | except json.JSONDecodeError: 154 | error = "Failed to parse response as JSON" 155 | self.log.error(error) 156 | else: 157 | error = f"Upload failed with status code: {response.status_code}" 158 | self.log.error(error) 159 | 160 | except Exception as e: 161 | error = f"Failed to upload report: {str(e)}" 162 | self.log.error(error) 163 | return {'success': False, 'error': error} 164 | 165 | def report_code_error(self, history): 166 | # Report code execution errors from history 167 | # Each history entry contains code and execution result 168 | # We only collect entries with traceback information 169 | # Returns True if report was sent successfully 170 | if not self._api_key: 171 | return True 172 | 173 | data = [] 174 | for h in history: 175 | result = h.get('result') 176 | if not result: 177 | continue 178 | traceback = result.get('traceback') 179 | if not traceback: 180 | continue 181 | data.append({ 182 | 'code': h.get('code'), 183 | 'traceback': traceback, 184 | 'error': result.get('errstr') 185 | }) 186 | 187 | if data: 188 | return self.report_data(data, 'code_error.json') 189 | return True 190 | 191 | if __name__ == '__main__': 192 | settings = { 193 | 'diagnose': { 194 | 'api_key': 'sk-aipy-', 195 | 'api_url': 'https://aipy.xxyy.eu.org/', 196 | } 197 | } 198 | diagnose = Diagnose.create(settings) 199 | update = diagnose.check_update() 200 | print(update) 201 | url = diagnose.report_code_error([ 202 | {'code': 'print("Hello, World!")', 'result': {'traceback': 'Traceback (most recent call last):\n File "test.py", line 1, in \n print("Hello, World!")\nNameError: name \'print\' is not defined\n', 'errstr': 'NameError: name \'print\' is not defined'}} 203 | ]) 204 | print(url) 205 | -------------------------------------------------------------------------------- /aipyapp/aipy/interface.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding: utf-8 -*- 3 | 4 | import threading 5 | from abc import ABC, abstractmethod 6 | 7 | class Runtime(ABC): 8 | @abstractmethod 9 | def install_packages(self, packages): 10 | pass 11 | 12 | @abstractmethod 13 | def getenv(self, name, desc=None): 14 | pass 15 | 16 | class ConsoleInterface(ABC): 17 | @abstractmethod 18 | def print(self, *args, sep=' ', end='\n', file=None, flush=False): 19 | pass 20 | 21 | @abstractmethod 22 | def input(self, prompt=''): 23 | pass 24 | 25 | @abstractmethod 26 | def status(self, msg): 27 | pass 28 | 29 | class Stoppable(): 30 | def __init__(self): 31 | self._stop_event = threading.Event() 32 | 33 | def on_stop(self): 34 | pass 35 | 36 | def stop(self): 37 | self._stop_event.set() 38 | self.on_stop() 39 | 40 | def is_stopped(self): 41 | return self._stop_event.is_set() 42 | 43 | def wait(self, timeout=None): 44 | return self._stop_event.wait(timeout) 45 | 46 | def reset(self): 47 | self._stop_event.clear() -------------------------------------------------------------------------------- /aipyapp/aipy/llm.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding: utf-8 -*- 3 | from collections import Counter, defaultdict 4 | 5 | from loguru import logger 6 | from rich.live import Live 7 | from rich.text import Text 8 | 9 | from .. import T 10 | from .plugin import event_bus 11 | from ..llm import CLIENTS, ChatMessage 12 | 13 | class ChatHistory: 14 | def __init__(self): 15 | self.messages = [] 16 | self._total_tokens = Counter() 17 | 18 | def __len__(self): 19 | return len(self.messages) 20 | 21 | def json(self): 22 | return [msg.__dict__ for msg in self.messages] 23 | 24 | def add(self, role, content): 25 | self.add_message(ChatMessage(role=role, content=content)) 26 | 27 | def add_message(self, message: ChatMessage): 28 | self.messages.append(message) 29 | self._total_tokens += message.usage 30 | 31 | def get_usage(self): 32 | return iter(row.usage for row in self.messages if row.role == "assistant") 33 | 34 | def get_summary(self): 35 | summary = {'time': 0, 'input_tokens': 0, 'output_tokens': 0, 'total_tokens': 0} 36 | summary.update(dict(self._total_tokens)) 37 | summary['rounds'] = sum(1 for row in self.messages if row.role == "assistant") 38 | return summary 39 | 40 | def get_messages(self): 41 | return [{"role": msg.role, "content": msg.content} for msg in self.messages] 42 | 43 | class LineReceiver(list): 44 | def __init__(self): 45 | super().__init__() 46 | self.buffer = "" 47 | 48 | @property 49 | def content(self): 50 | return '\n'.join(self) 51 | 52 | def feed(self, data: str): 53 | self.buffer += data 54 | new_lines = [] 55 | 56 | while '\n' in self.buffer: 57 | line, self.buffer = self.buffer.split('\n', 1) 58 | self.append(line) 59 | new_lines.append(line) 60 | 61 | return new_lines 62 | 63 | def empty(self): 64 | return not self and not self.buffer 65 | 66 | def done(self): 67 | buffer = self.buffer 68 | if buffer: 69 | self.append(buffer) 70 | self.buffer = "" 71 | return buffer 72 | 73 | class LiveManager: 74 | def __init__(self, name, quiet=False): 75 | self.live = None 76 | self.name = name 77 | self.lr = LineReceiver() 78 | self.lr_reason = LineReceiver() 79 | self.title = f"{self.name} {T('Reply')}" 80 | self.reason_started = False 81 | self.display_lines = [] 82 | self.max_lines = 10 83 | self.quiet = quiet 84 | 85 | @property 86 | def content(self): 87 | return self.lr.content 88 | 89 | @property 90 | def reason(self): 91 | return self.lr_reason.content 92 | 93 | def __enter__(self): 94 | if self.quiet: return self 95 | self.live = Live(auto_refresh=False, vertical_overflow='crop', transient=True) 96 | self.live.__enter__() 97 | return self 98 | 99 | def process_chunk(self, content, *, reason=False): 100 | if not content: return 101 | 102 | if not reason and self.lr.empty() and not self.lr_reason.empty(): 103 | line = self.lr_reason.done() 104 | event_bus.broadcast('response_stream', {'llm': self.name, 'content': f"{line}\n\n----\n\n", 'reason': True}) 105 | 106 | lr = self.lr_reason if reason else self.lr 107 | lines = lr.feed(content) 108 | if not lines: return 109 | 110 | lines2 = [line for line in lines if not line.startswith(' 11 | - 代码本体：用 Markdown 代码块包裹（如 ```python 或 ```json 等)。 12 | - 代码结束： 13 | 14 | 2. 代码块ID必须全局唯一，不能重复。 15 | 16 | 3. `path` 可以包含目录, 默认为相对当前目录或者用户指定目录 17 | - 处理复杂问题时可以用于建立完整的项目目录和文件 18 | - 输出内容解析程序会立即用代码块内容创建该文件(包含创建上级目录) 19 | - 当前输出内容里的Python代码块可以假设文件已经存在并直接使用 20 | 21 | 4. 同一个输出消息里可以定义多个代码块。 22 | 23 | 5. **正确示例：** 24 | 25 | ```python 26 | print("hello world") 27 | ``` 28 | 29 | 30 | ## 单行命令标记 31 | 1. 每次输出中只能包含 **一个** `Cmd-Exec` 标记，用于执行可执行代码块来完成用户的任务： 32 | - 格式： 33 | - 如果不需要执行任何代码，则不要添加 `Cmd-Exec`。 34 | - 要执行的代码块ID必需先使用前述多行代码块标记格式单独定义。 35 | - 可以使用 `Cmd-Exec` 执行会话历史中的所有代码块。特别地，如果需要重复执行某个任务，尽量使用 `Cmd-Exec` 执行而不是重复输出代码块。 36 | 37 | 2. Cmd-Exec 只能用来执行 Python 代码块，不能执行其它语言(如 JSON/HTML/CSS/JavaScript等)的代码块。 38 | 39 | 3. **正确示例：** 40 | 41 | 42 | ## 其它 43 | 1. 所有 JSON 内容必须写成**单行紧凑格式**，例如： 44 | 45 | 46 | 2. 禁止输出代码内容重复的代码块，通过代码块ID来引用之前定义过的代码块。 47 | 48 | 遵循上述规则，生成输出内容。 49 | 50 | # 生成Python代码规则 51 | - 确保代码在下述`Python运行环境描述`中描述的运行环境中可以无需修改直接执行 52 | - 实现适当的错误处理，包括但不限于： 53 | * 文件操作的异常处理 54 | * 网络请求的超时和连接错误处理 55 | * 数据处理过程中的类型错误和值错误处理 56 | - 如果需要区分正常和错误信息，可以把错误信息输出到 stderr。 57 | - 不允许执行可能导致 Python 解释器退出的指令，如 exit/quit 等函数，请确保代码中不包含这类操作。 58 | 59 | # Python运行环境描述 60 | 在标准 Python 运行环境的基础上额外增加了下述功能： 61 | - 一些预装的第三方包 62 | - 全局 `runtime` 对象 63 | - 全局变量 `__storage__` 64 | - 局部变量 `__retval__` 65 | 66 | 生成 Python 代码时可以直接使用这些额外功能。 67 | 68 | ## 全局变量 `__storage__` 69 | - 类型：字典。 70 | - 有效期：用于长期数据存储，在整个会话过程始终有效 71 | - 用途：可以在多次会话间共享数据。 72 | - 注意: 在函数内使用时必须在函数最开始用 `global __storage__` 声明。 73 | - 使用示例： 74 | ```python 75 | def main(): 76 | global __storage__ 77 | __storage__['step1_result'] = calculated_value 78 | ``` 79 | 80 | ## 局部变量 `__retval__` 81 | - 类型: 字典。 82 | - 用途: 用于记录和收集当前代码执行情况。 83 | - 注意: 在函数内使用时必须在函数最开始用 `global __retval__` 声明。 84 | - 警告：`__retval__` 变量不会传递给下一个执行的代码块！禁止从 `__retval__` 中获取之前代码块保存的数据！ 85 | - 使用示例： 86 | ```python 87 | def main(): 88 | global __retval__ 89 | __retval__ = {"status": "error", "message": "An error occurred"} 90 | ``` 91 | 例如，如果需要分析客户端的文件，你可以生成代码读取文件内容放入 `__retval__` 变量即可收到反馈。 92 | 93 | ## 预装的第三方包 94 | 下述第三方包可以无需安装直接使用： 95 | - `requests`、`numpy`、`pandas`、`matplotlib`、`seaborn`、`bs4`。 96 | 97 | 其它第三方包，都必需通过下述 runtime 对象的 install_packages 方法申请安装才能使用。 98 | 99 | 在使用 matplotlib 时，需要根据系统类型选择和设置合适的中文字体，否则图片里中文会乱码导致无法完成客户任务。 100 | 示例代码如下： 101 | ```python 102 | import platform 103 | 104 | system = platform.system().lower() 105 | font_options = { 106 | 'windows': ['Microsoft YaHei', 'SimHei'], 107 | 'darwin': ['Kai', 'Hei'], 108 | 'linux': ['Noto Sans CJK SC', 'WenQuanYi Micro Hei', 'Source Han Sans SC'] 109 | } 110 | ``` 111 | 112 | ## 全局 runtime 对象 113 | runtime 对象提供一些协助代码完成任务的方法。 114 | 115 | ### `runtime.get_code_by_id` 方法 116 | - 功能: 获取指定 ID 的代码块内容 117 | - 定义: `get_code_by_id(code_id)` 118 | - 参数: `code_id` 为代码块的唯一标识符 119 | - 返回值: 代码块内容，如果未找到则返回 None 120 | 121 | ### runtime.install_packages 方法 122 | - 功能: 申请安装完成任务必需的额外模块 123 | - 参数: 一个或多个 PyPi 包名，如：'httpx', 'requests>=2.25' 124 | - 返回值:True 表示成功, False 表示失败 125 | 126 | 示例如下： 127 | ```python 128 | if runtime.install_packages('httpx', 'requests>=2.25'): 129 | import httpx 130 | ``` 131 | 132 | ### runtime.get_env 方法 133 | - 功能: 获取代码运行需要的环境变量，如 API-KEY 等。 134 | - 定义: get_env(name, default=None, *, desc=None) 135 | - 参数: 第一个参数为需要获取的环境变量名称，第二个参数为不存在时的默认返回值，第三个可选字符串参数简要描述需要的是什么。 136 | - 返回值: 环境变量值，返回 None 或空字符串表示未找到。 137 | 138 | 示例如下： 139 | ```python 140 | env_name = '环境变量名称' 141 | env_value = runtime.get_env(env_name, "No env", desc='访问API服务需要') 142 | if not env_value: 143 | print(f"Error: {env_name} is not set", file=sys.stderr) 144 | else: 145 | print(f"{env_name} is available") 146 | ``` 147 | 148 | ### runtime.display 方法 149 | - 功能: 显示图片 150 | - 定义: display(path="path/to/image.jpg", url="https://www.example.com/image.png") 151 | - 参数: 152 | - path: 图片文件路径 153 | - url: 图片 URL 154 | - 返回值: 无 155 | 156 | 示例： 157 | ```python 158 | runtime.display(path="path/to/image.png") 159 | runtime.display(url="https://www.example.com/image.png") 160 | ``` 161 | 162 | # 代码执行结果反馈 163 | Python代码块的执行结果会通过JSON对象反馈给你，对象包括以下属性： 164 | - `stdout`: 标准输出内容 165 | - `stderr`: 标准错误输出 166 | - `__retval__`: 前述`__retval__` 全局变量 167 | - `errstr`: 异常信息 168 | - `traceback`: 异常堆栈信息 169 | - `block_id`: 执行的代码块ID 170 | 171 | 注意： 172 | - 如果某个属性为空，它不会出现在反馈中。 173 | - 避免在 stdout 和 `__retval__` 中保存相同的内容 174 | - 不要在 `__retval__` 中保存太多数据，这会导致反馈消息太长 175 | 176 | 收到反馈后，结合代码和反馈数据，做出下一步的决策。 177 | 178 | # 一些 API 信息 179 | 下面是用户提供的一些 API 信息，可能有 API_KEY，URL，用途和使用方法等信息。 180 | 这些可能对特定任务有用途，你可以根据任务选择性使用。 181 | 182 | 注意： 183 | 1. 这些 API 信息里描述的环境变量必须用 runtime.get_env 方法获取，绝对不能使用 os.getenv 方法。 184 | 2. API获取数据失败时，请输出完整的API响应信息，方便调试和分析问题。 185 | """ 186 | 187 | def get_system_prompt(settings): 188 | pass 189 | -------------------------------------------------------------------------------- /aipyapp/aipy/runtime.py: -------------------------------------------------------------------------------- 1 | #! /usr/bin/env python3 2 | # -*- coding: utf-8 -*- 3 | import sys 4 | from functools import wraps 5 | 6 | from term_image.image import from_file, from_url 7 | 8 | from . import utils 9 | from .plugin import event_bus 10 | from .. import T 11 | from ..exec import BaseRuntime 12 | 13 | def restore_output(func): 14 | @wraps(func) 15 | def wrapper(self, *args, **kwargs): 16 | old_stdout, old_stderr = sys.stdout, sys.stderr 17 | sys.stdout, sys.stderr = sys.__stdout__, sys.__stderr__ 18 | 19 | try: 20 | return func(self, *args, **kwargs) 21 | finally: 22 | sys.stdout, sys.stderr = old_stdout, old_stderr 23 | return wrapper 24 | 25 | class Runtime(BaseRuntime): 26 | def __init__(self, task): 27 | super().__init__(task.envs) 28 | self.gui = task.gui 29 | self.task = task 30 | self.console = task.console 31 | self._auto_install = task.settings.get('auto_install') 32 | self._auto_getenv = task.settings.get('auto_getenv') 33 | 34 | @restore_output 35 | def install_packages(self, *packages): 36 | self.console.print(f"\n⚠️ LLM {T('Request to install third-party packages')}: {packages}") 37 | ok = utils.confirm(self.console, f"💬 {T('If you agree, please enter')} 'y'> ", auto=self._auto_install) 38 | if ok: 39 | ret = self.ensure_packages(*packages) 40 | self.console.print("\n✅" if ret else "\n❌") 41 | return ret 42 | return False 43 | 44 | @restore_output 45 | def get_env(self, name, default=None, *, desc=None): 46 | self.console.print(f"\n⚠️ LLM {T('Request to obtain environment variable {}, purpose', name)}: {desc}") 47 | try: 48 | value = self.envs[name][0] 49 | self.console.print(f"✅ {T('Environment variable {} exists, returned for code use', name)}") 50 | except KeyError: 51 | if self._auto_getenv: 52 | self.console.print(f"✅ {T('Auto confirm')}") 53 | value = None 54 | else: 55 | value = self.console.input(f"💬 {T('Environment variable {} not found, please enter', name)}: ") 56 | value = value.strip() 57 | if value: 58 | self.set_env(name, value, desc) 59 | return value or default 60 | 61 | @restore_output 62 | def display(self, path=None, url=None): 63 | image = {'path': path, 'url': url} 64 | event_bus.broadcast('display', image) 65 | if not self.gui: 66 | image = from_file(path) if path else from_url(url) 67 | image.draw() 68 | 69 | @restore_output 70 | def input(self, prompt=''): 71 | return self.console.input(prompt) 72 | 73 | def get_code_by_id(self, code_id): 74 | return self.task.code_blocks.get_code_by_id(code_id) -------------------------------------------------------------------------------- /aipyapp/aipy/task.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding: utf-8 -*- 3 | 4 | import os 5 | import json 6 | import uuid 7 | import time 8 | import platform 9 | import locale 10 | from pathlib import Path 11 | from datetime import date 12 | from importlib.resources import read_text 13 | 14 | import requests 15 | from loguru import logger 16 | from rich.rule import Rule 17 | from rich.panel import Panel 18 | from rich.align import Align 19 | from rich.table import Table 20 | from rich.syntax import Syntax 21 | from rich.console import Console, Group 22 | from rich.markdown import Markdown 23 | 24 | from .. import T, __respkg__ 25 | from ..exec import Runner 26 | from .runtime import Runtime 27 | from .plugin import event_bus 28 | from .utils import get_safe_filename 29 | from .blocks import CodeBlocks 30 | from .interface import Stoppable 31 | 32 | CONSOLE_WHITE_HTML = read_text(__respkg__, "console_white.html") 33 | CONSOLE_CODE_HTML = read_text(__respkg__, "console_code.html") 34 | 35 | class Task(Stoppable): 36 | MAX_ROUNDS = 16 37 | 38 | def __init__(self, manager): 39 | super().__init__() 40 | self.manager = manager 41 | self.task_id = uuid.uuid4().hex 42 | self.log = logger.bind(src='task', id=self.task_id) 43 | self.settings = manager.settings 44 | self.envs = manager.envs 45 | self.gui = manager.gui 46 | self.console = Console(file=manager.console.file, record=True) 47 | self.max_rounds = self.settings.get('max_rounds', self.MAX_ROUNDS) 48 | 49 | self.client = None 50 | self.runner = None 51 | self.instruction = None 52 | self.system_prompt = None 53 | self.diagnose = None 54 | self.start_time = None 55 | 56 | self.code_blocks = CodeBlocks(self.console) 57 | self.runtime = Runtime(self) 58 | self.runner = Runner(self.runtime) 59 | 60 | def use(self, name): 61 | ret = self.client.use(name) 62 | self.console.print('[green]Ok[/green]' if ret else '[red]Error[/red]') 63 | return ret 64 | 65 | def save(self, path): 66 | if self.console.record: 67 | self.console.save_html(path, clear=False, code_format=CONSOLE_WHITE_HTML) 68 | 69 | def save_html(self, path, task): 70 | if 'chats' in task and isinstance(task['chats'], list) and len(task['chats']) > 0: 71 | if task['chats'][0]['role'] == 'system': 72 | task['chats'].pop(0) 73 | 74 | task_json = json.dumps(task, ensure_ascii=False, default=str) 75 | html_content = CONSOLE_CODE_HTML.replace('{{code}}', task_json) 76 | try: 77 | with open(path, 'w', encoding='utf-8') as f: 78 | f.write(html_content) 79 | except Exception as e: 80 | self.console.print_exception() 81 | 82 | def _auto_save(self): 83 | instruction = self.instruction 84 | task = {'instruction': instruction} 85 | task['chats'] = self.client.history.json() 86 | #task['envs'] = self.runtime.envs 87 | task['runner'] = self.runner.history 88 | task['blocks'] = self.code_blocks.to_list() 89 | 90 | filename = f"{self.task_id}.json" 91 | try: 92 | json.dump(task, open(filename, 'w', encoding='utf-8'), ensure_ascii=False, indent=4, default=str) 93 | except Exception as e: 94 | self.log.exception('Error saving task') 95 | 96 | filename = f"{self.task_id}.html" 97 | #self.save_html(filename, task) 98 | self.save(filename) 99 | self.log.info('Task auto saved') 100 | 101 | def done(self): 102 | curname = f"{self.task_id}.json" 103 | jsonname = get_safe_filename(self.instruction, extension='.json') 104 | if jsonname and os.path.exists(curname): 105 | try: 106 | os.rename(curname, jsonname) 107 | except Exception as e: 108 | self.log.exception('Error renaming task json file') 109 | 110 | curname = f"{self.task_id}.html" 111 | htmlname = get_safe_filename(self.instruction, extension='.html') 112 | if htmlname and os.path.exists(curname): 113 | try: 114 | os.rename(curname, htmlname) 115 | except Exception as e: 116 | self.log.exception('Error renaming task html file') 117 | 118 | self.diagnose.report_code_error(self.runner.history) 119 | self.done_time = time.time() 120 | self.log.info('Task done', jsonname=jsonname, htmlname=htmlname) 121 | filename = str(Path(htmlname).resolve()) 122 | self.console.print(f"[green]{T('Result file saved')}: \"{filename}\"") 123 | if self.settings.get('share_result'): 124 | self.sync_to_cloud() 125 | 126 | def process_reply(self, markdown): 127 | #self.console.print(f"{T('Start parsing message')}...", style='dim white') 128 | parse_mcp = self.mcp is not None 129 | ret = self.code_blocks.parse(markdown, parse_mcp=parse_mcp) 130 | if not ret: 131 | return None 132 | 133 | json_str = json.dumps(ret, ensure_ascii=False, indent=2, default=str) 134 | self.box(f"✅ {T('Message parse result')}", json_str, lang="json") 135 | 136 | errors = ret.get('errors') 137 | if errors: 138 | event_bus('result', errors) 139 | self.console.print(f"{T('Start sending feedback')}...", style='dim white') 140 | feed_back = f"# 消息解析错误\n{json_str}" 141 | ret = self.chat(feed_back) 142 | elif 'exec_blocks' in ret: 143 | ret = self.process_code_reply(ret['exec_blocks']) 144 | elif 'call_tool' in ret: 145 | ret = self.process_mcp_reply(ret['call_tool']) 146 | else: 147 | ret = None 148 | return ret 149 | 150 | def print_code_result(self, block, result, title=None): 151 | line_numbers = True if 'traceback' in result else False 152 | syntax_code = Syntax(block.code, block.lang, line_numbers=line_numbers, word_wrap=True) 153 | syntax_result = Syntax(result, 'json', line_numbers=False, word_wrap=True) 154 | group = Group(syntax_code, Rule(), syntax_result) 155 | panel = Panel(group, title=title or block.id) 156 | self.console.print(panel) 157 | 158 | def process_code_reply(self, exec_blocks): 159 | results = [] 160 | json_results = [] 161 | for block in exec_blocks: 162 | event_bus('exec', block) 163 | self.console.print(f"⚡ {T('Start executing code block')}: {block.id}", style='dim white') 164 | result = self.runner(block) 165 | json_result = json.dumps(result, ensure_ascii=False, indent=2, default=str) 166 | result['block_id'] = block.id 167 | results.append(result) 168 | json_results.append(json_result) 169 | self.print_code_result(block, json_result) 170 | event_bus('result', result) 171 | 172 | if len(json_results) == 1: 173 | json_results = json_results[0] 174 | else: 175 | json_results = json.dumps(results, ensure_ascii=False, indent=4, default=str) 176 | 177 | self.console.print(f"{T('Start sending feedback')}...", style='dim white') 178 | feed_back = f"# 最初任务\n{self.instruction}\n\n# 代码执行结果反馈\n{json_results}" 179 | return self.chat(feed_back) 180 | 181 | def process_mcp_reply(self, json_content): 182 | """处理 MCP 工具调用的回复""" 183 | block = {'content': json_content, 'language': 'json'} 184 | event_bus('tool_call', block) 185 | json_content = block['content'] 186 | self.console.print(f"⚡ {T('Start calling MCP tool')} ...", style='dim white') 187 | 188 | call_tool = json.loads(json_content) 189 | result = self.mcp.call_tool(call_tool['name'], call_tool.get('arguments', {})) 190 | event_bus('result', result) 191 | result_json = json.dumps(result, ensure_ascii=False, indent=2, default=str) 192 | self.print_code_result(block, result_json, title=T("MCP tool call result")) 193 | 194 | self.console.print(f"{T('Start sending feedback')}...", style='dim white') 195 | feed_back = f"""# MCP 调用\n\n{self.instruction}\n 196 | # 执行结果反馈 197 | 198 | ````json 199 | {result_json} 200 | ````""" 201 | feedback_response = self.chat(feed_back) 202 | return feedback_response 203 | 204 | def box(self, title, content, align=None, lang=None): 205 | if lang: 206 | content = Syntax(content, lang, line_numbers=True, word_wrap=True) 207 | else: 208 | content = Markdown(content) 209 | 210 | if align: 211 | content = Align(content, align=align) 212 | 213 | self.console.print(Panel(content, title=title)) 214 | 215 | def print_summary(self, detail=False): 216 | history = self.client.history 217 | if detail: 218 | table = Table(title=T("Task Summary"), show_lines=True) 219 | 220 | table.add_column(T("Round"), justify="center", style="bold cyan", no_wrap=True) 221 | table.add_column(T("Time(s)"), justify="right") 222 | table.add_column(T("In Tokens"), justify="right") 223 | table.add_column(T("Out Tokens"), justify="right") 224 | table.add_column(T("Total Tokens"), justify="right", style="bold magenta") 225 | 226 | round = 1 227 | for row in history.get_usage(): 228 | table.add_row( 229 | str(round), 230 | str(row["time"]), 231 | str(row["input_tokens"]), 232 | str(row["output_tokens"]), 233 | str(row["total_tokens"]), 234 | ) 235 | round += 1 236 | self._console.print("\n") 237 | self._console.print(table) 238 | 239 | summary = history.get_summary() 240 | summary['elapsed_time'] = time.time() - self.start_time 241 | summarys = "| {rounds} | {time:.3f}s/{elapsed_time:.3f}s | Tokens: {input_tokens}/{output_tokens}/{total_tokens}".format(**summary) 242 | event_bus.broadcast('summary', summarys) 243 | self.console.print(f"\n⏹ [cyan]{T('End processing instruction')} {summarys}") 244 | 245 | def build_user_prompt(self): 246 | prompt = {'task': self.instruction} 247 | prompt['python_version'] = platform.python_version() 248 | prompt['platform'] = platform.platform() 249 | prompt['today'] = date.today().isoformat() 250 | prompt['locale'] = locale.getlocale() 251 | prompt['think_and_reply_language'] = '始终根据用户查询的语言来进行所有内部思考和回复，即用户使用什么语言，你就要用什么语言思考和回复。' 252 | prompt['work_dir'] = '工作目录为当前目录，默认在当前目录下创建文件' 253 | if self.gui: 254 | prompt['matplotlib'] = "我现在用的是 matplotlib 的 Agg 后端，请默认用 plt.savefig() 保存图片后用 runtime.display() 显示，禁止使用 plt.show()" 255 | #prompt['wxPython'] = "你回复的Markdown 消息中，可以用 ![图片](图片路径) 的格式引用之前创建的图片，会显示在 wx.html2 的 WebView 中" 256 | else: 257 | prompt['TERM'] = os.environ.get('TERM') 258 | prompt['LC_TERMINAL'] = os.environ.get('LC_TERMINAL') 259 | return prompt 260 | 261 | def chat(self, instruction, *, system_prompt=None): 262 | quiet = self.settings.gui and not self.settings.debug 263 | msg = self.client(instruction, system_prompt=system_prompt, quiet=quiet) 264 | if msg.role == 'error': 265 | self.console.print(f"[red]{msg.content}[/red]") 266 | return None 267 | if msg.reason: 268 | content = f"{msg.reason}\n\n-----\n\n{msg.content}" 269 | else: 270 | content = msg.content 271 | self.box(f"[yellow]{T('Reply')} ({self.client.name})", content) 272 | return msg.content 273 | 274 | def run(self, instruction): 275 | """ 276 | 执行自动处理循环，直到 LLM 不再返回代码消息 277 | """ 278 | self.box(f"[yellow]{T('Start processing instruction')}", instruction, align="center") 279 | if not self.start_time: 280 | self.start_time = time.time() 281 | self.instruction = instruction 282 | prompt = self.build_user_prompt() 283 | event_bus('task_start', prompt) 284 | instruction = json.dumps(prompt, ensure_ascii=False) 285 | system_prompt = self.system_prompt 286 | else: 287 | system_prompt = None 288 | 289 | rounds = 1 290 | max_rounds = self.max_rounds 291 | response = self.chat(instruction, system_prompt=system_prompt) 292 | while response and rounds <= max_rounds: 293 | response = self.process_reply(response) 294 | rounds += 1 295 | if self.is_stopped(): 296 | self.log.info('Task stopped') 297 | break 298 | 299 | self.print_summary() 300 | self._auto_save() 301 | self.console.bell() 302 | self.log.info('Loop done', rounds=rounds) 303 | 304 | def sync_to_cloud(self, verbose=True): 305 | """ Sync result 306 | """ 307 | url = T("https://store.aipy.app/api/work") 308 | 309 | trustoken_apikey = self.settings.get('llm', {}).get('Trustoken', {}).get('api_key') 310 | if not trustoken_apikey: 311 | trustoken_apikey = self.settings.get('llm', {}).get('trustoken', {}).get('api_key') 312 | if not trustoken_apikey: 313 | return False 314 | self.console.print(f"[yellow]{T('Uploading result, please wait...')}") 315 | try: 316 | response = requests.post(url, json={ 317 | 'apikey': trustoken_apikey, 318 | 'author': os.getlogin(), 319 | 'instruction': self.instruction, 320 | 'llm': self.client.history.json(), 321 | 'runner': self.runner.history, 322 | }, verify=True, timeout=30) 323 | except Exception as e: 324 | print(e) 325 | return False 326 | 327 | status_code = response.status_code 328 | if status_code in (200, 201): 329 | if verbose: 330 | data = response.json() 331 | url = data.get('url', '') 332 | if url: 333 | self.console.print(f"[green]{T('Article uploaded successfully, {}', url)}[/green]") 334 | return True 335 | 336 | if verbose: 337 | self.console.print(f"[red]{T('Upload failed (status code: {})', status_code)}:", response.text) 338 | return False -------------------------------------------------------------------------------- /aipyapp/aipy/taskmgr.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding: utf-8 -*- 3 | 4 | import os 5 | import json 6 | from pathlib import Path 7 | from collections import deque 8 | 9 | from loguru import logger 10 | 11 | from .. import T 12 | from .task import Task 13 | from .plugin import PluginManager 14 | from .prompt import SYSTEM_PROMPT 15 | from .diagnose import Diagnose 16 | from .llm import ClientManager 17 | from .config import PLUGINS_DIR, get_mcp, get_tt_api_key, get_tt_aio_api 18 | 19 | class TaskManager: 20 | MAX_TASKS = 16 21 | 22 | def __init__(self, settings, console, gui=False): 23 | self.settings = settings 24 | self.console = console 25 | self.tasks = deque(maxlen=self.MAX_TASKS) 26 | self.envs = {} 27 | self.gui = gui 28 | self.log = logger.bind(src='taskmgr') 29 | self.config_files = settings._loaded_files 30 | self.system_prompt = f"{settings.system_prompt}\n{SYSTEM_PROMPT}" 31 | self.plugin_manager = PluginManager(PLUGINS_DIR) 32 | self.plugin_manager.load_plugins() 33 | if settings.workdir: 34 | workdir = Path.cwd() / settings.workdir 35 | workdir.mkdir(parents=True, exist_ok=True) 36 | os.chdir(workdir) 37 | self._cwd = workdir 38 | else: 39 | self._cwd = Path.cwd() 40 | self.mcp = get_mcp(settings.get('_config_dir')) 41 | self._init_environ() 42 | self.tt_api_key = get_tt_api_key(settings) 43 | self._init_api() 44 | self.diagnose = Diagnose.create(settings) 45 | self.client_manager = ClientManager(settings) 46 | 47 | @property 48 | def workdir(self): 49 | return str(self._cwd) 50 | 51 | def get_update(self, force=False): 52 | return self.diagnose.check_update(force) 53 | 54 | def use(self, name): 55 | ret = self.client_manager.use(name) 56 | self.console.print('[green]Ok[/green]' if ret else '[red]Error[/red]') 57 | return ret 58 | 59 | def _init_environ(self): 60 | envs = self.settings.get('environ', {}) 61 | for name, value in envs.items(): 62 | os.environ[name] = value 63 | 64 | def _init_api(self): 65 | api = self.settings.get('api', {}) 66 | 67 | # update tt aio api, for map and search 68 | if self.tt_api_key: 69 | tt_aio_api = get_tt_aio_api(self.tt_api_key) 70 | api.update(tt_aio_api) 71 | 72 | lines = [self.system_prompt] 73 | for api_name, api_conf in api.items(): 74 | lines.append(f"## {api_name} API") 75 | desc = api_conf.get('desc') 76 | if desc: 77 | lines.append(f"### API {T('Description')}\n{desc}") 78 | 79 | envs = api_conf.get('env') 80 | if not envs: 81 | continue 82 | 83 | lines.append(f"### {T('Environment variable name and meaning')}") 84 | for name, (value, desc) in envs.items(): 85 | value = value.strip() 86 | if not value: 87 | continue 88 | lines.append(f"- {name}: {desc}") 89 | self.envs[name] = (value, desc) 90 | 91 | self.system_prompt = "\n".join(lines) 92 | 93 | 94 | def _update_mcp_prompt(self, prompt): 95 | """更新 MCP 工具提示信息""" 96 | mcp_tools = self.mcp.list_tools() 97 | if not mcp_tools: 98 | return prompt 99 | tools_json = json.dumps(mcp_tools, ensure_ascii=False) 100 | lines = [self.system_prompt] 101 | lines.append("""\n## MCP工具调用规则： 102 | 1. 如果需要调用MCP工具，请以 JSON 格式输出你的决策和调用参数，并且仅返回json，不输出其他内容。 103 | 2. 返回 JSON 格式如下： 104 | {"action": "call_tool", "name": "tool_name", "arguments": {"arg_name": "arg_value", ...}} 105 | 3. 一次只能返回一个工具，即只能返回一个 JSON 代码块，不能有其它多余内容。 106 | 以下是你可用的工具，以 JSON 数组形式提供： 107 | """) 108 | lines.append(f"```json\n{tools_json}\n```") 109 | # 更新系统提示 110 | return "\n".join(lines) 111 | 112 | def new_task(self, system_prompt=None): 113 | with_mcp = self.settings.get('mcp', {}).get('enable', True) 114 | system_prompt = system_prompt or self.system_prompt 115 | if self.mcp and with_mcp: 116 | self.log.info('Update MCP prompt') 117 | system_prompt = self._update_mcp_prompt(system_prompt) 118 | 119 | task = Task(self) 120 | task.client = self.client_manager.Client() 121 | task.diagnose = self.diagnose 122 | task.system_prompt = system_prompt 123 | task.mcp = self.mcp if with_mcp else None 124 | self.tasks.append(task) 125 | self.log.info('New task created', task_id=task.task_id) 126 | return task -------------------------------------------------------------------------------- /aipyapp/aipy/trustoken.py: -------------------------------------------------------------------------------- 1 | import os 2 | import time 3 | import webbrowser 4 | 5 | import requests 6 | import qrcode 7 | 8 | from .. import T 9 | 10 | POLL_INTERVAL = 5 # 轮询间隔（秒） 11 | 12 | class TrustTokenAPI: 13 | """Handles all HTTP operations for TrustToken binding and authentication.""" 14 | 15 | def __init__(self, coordinator_url=None): 16 | """ 17 | Initialize the TrustToken API handler. 18 | 19 | Args: 20 | coordinator_url (str, optional): The coordinator server URL. Defaults to None. 21 | """ 22 | self.coordinator_url = coordinator_url or T("https://www.trustoken.ai/api") 23 | 24 | def request_binding(self): 25 | """Request binding from the coordinator server. 26 | 27 | Returns: 28 | dict: Response data containing approval_url, request_id, and expires_in if successful 29 | None: If the request failed 30 | """ 31 | url = f"{self.coordinator_url}/request_bind" 32 | try: 33 | response = requests.post(url, timeout=10) 34 | response.raise_for_status() 35 | return response.json() 36 | except requests.exceptions.RequestException as e: 37 | print(T("Error connecting to coordinator or during request: {}").format(e)) 38 | return None 39 | except Exception as e: 40 | print(T("An unexpected error occurred during request: {}").format(e)) 41 | return None 42 | 43 | def check_status(self, request_id): 44 | """Check the binding status from the coordinator server. 45 | 46 | Args: 47 | request_id (str): The request ID to check status for. 48 | 49 | Returns: 50 | dict: Response data containing status and secret_token if approved 51 | None: If the request failed 52 | """ 53 | url = f"{self.coordinator_url}/check_status" 54 | params = {'request_id': request_id} 55 | try: 56 | response = requests.get(url, params=params, timeout=10) 57 | response.raise_for_status() 58 | return response.json() 59 | except requests.exceptions.RequestException as e: 60 | print(T("Error connecting to coordinator during polling: {}").format(e)) 61 | return None 62 | except Exception as e: 63 | print(T("An unexpected error occurred during polling: {}").format(e)) 64 | return None 65 | 66 | class TrustToken: 67 | """A class to handle TrustToken binding and authentication processes.""" 68 | 69 | def __init__(self, coordinator_url=None, poll_interval=5): 70 | """ 71 | Initialize the TrustToken handler. 72 | 73 | Args: 74 | coordinator_url (str, optional): The coordinator server URL. Defaults to None. 75 | poll_interval (int, optional): Polling interval in seconds. Defaults to 5. 76 | """ 77 | self.api = TrustTokenAPI(coordinator_url) 78 | self.poll_interval = poll_interval or POLL_INTERVAL 79 | 80 | def request_binding(self, qrcode=False): 81 | """Request binding from the coordinator server. 82 | 83 | Returns: 84 | str: The request ID if successful, None otherwise. 85 | """ 86 | data = self.api.request_binding() 87 | if not data: 88 | return None 89 | 90 | approval_url = data['approval_url'] 91 | request_id = data['request_id'] 92 | expires_in = data['expires_in'] 93 | 94 | print(T("""Binding request sent successfully. 95 | Request ID: {} 96 | 97 | >>> Please open this URL in your browser on an authenticated device to approve: 98 | >>> {} 99 | 100 | (This link expires in {} seconds)""").format(request_id, approval_url, expires_in)) 101 | 102 | if qrcode: 103 | print(T("Or scan the QR code below:")) 104 | try: 105 | qr = qrcode.QRCode( 106 | error_correction=qrcode.constants.ERROR_CORRECT_L, 107 | border=1 108 | ) 109 | qr.add_data(approval_url) 110 | qr.make(fit=True) 111 | qr.print_ascii(tty=True) 112 | print(T("We recommend you scan the QR code to bind the AiPy brain, you can also configure a third-party large model brain, details refer to: https://d.aipy.app/d/77")) 113 | except Exception as e: 114 | print(T("(Could not display QR code: {})").format(e)) 115 | else: 116 | webbrowser.open(approval_url) 117 | return request_id 118 | 119 | def poll_status(self, request_id, save_func=None): 120 | """Poll the binding status from the coordinator server. 121 | 122 | Args: 123 | request_id (str): The request ID to check status for. 124 | save_func (callable, optional): Function to save the token when approved. 125 | 126 | Returns: 127 | bool: True if binding was successful, False otherwise. 128 | """ 129 | start_time = time.time() 130 | polling_timeout = 310 131 | 132 | print(T("Browser has opened the Trustoken website, please register or login to authorize"), end='', flush=True) 133 | try: 134 | while time.time() - start_time < polling_timeout: 135 | data = self.api.check_status(request_id) 136 | if not data: 137 | time.sleep(self.poll_interval) 138 | continue 139 | 140 | status = data.get('status') 141 | if status == 'pending': 142 | print('.', end='', flush=True) 143 | time.sleep(self.poll_interval) 144 | continue 145 | 146 | print() 147 | print(T("Current status: {}...").format(status)) 148 | 149 | if status == 'approved': 150 | if save_func: 151 | save_func(data['secret_token']) 152 | return True 153 | elif status == 'expired': 154 | print(T("Binding request expired.")) 155 | return False 156 | else: 157 | print(T("Received unknown status: {}").format(status)) 158 | return False 159 | 160 | time.sleep(self.poll_interval) 161 | except KeyboardInterrupt: 162 | print(T("Polling cancelled by user.")) 163 | return False 164 | 165 | print(T("Polling timed out.")) 166 | return False 167 | 168 | def fetch_token(self, save_func): 169 | """Fetch a token from the coordinator server. 170 | 171 | Args: 172 | save_func (callable): Function to save the token when approved. 173 | 174 | Returns: 175 | bool: True if token was successfully fetched and saved, False otherwise. 176 | """ 177 | print(T("The current environment lacks the required configuration file. Starting the configuration initialization process to bind with the Trustoken account...")) 178 | req_id = self.request_binding() 179 | if req_id: 180 | if self.poll_status(req_id, save_func): 181 | print(T("Binding process completed successfully.")) 182 | return True 183 | else: 184 | print(T("Binding process failed or was not completed.")) 185 | return False 186 | else: 187 | print(T("Failed to initiate binding request.")) 188 | return False 189 | 190 | if __name__ == "__main__": 191 | tt = TrustToken() 192 | tt.fetch_token(lambda token: print(f"Token: {token}")) -------------------------------------------------------------------------------- /aipyapp/aipy/utils.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding: utf-8 -*- 3 | 4 | import os 5 | import re 6 | import sys 7 | from functools import wraps 8 | from importlib.resources import read_text 9 | 10 | from rich.panel import Panel 11 | 12 | from .. import T, __respkg__ 13 | 14 | def restore_output(func): 15 | @wraps(func) 16 | def wrapper(self, *args, **kwargs): 17 | old_stdout, old_stderr = sys.stdout, sys.stderr 18 | sys.stdout, sys.stderr = sys.__stdout__, sys.__stderr__ 19 | 20 | try: 21 | return func(self, *args, **kwargs) 22 | finally: 23 | sys.stdout, sys.stderr = old_stdout, old_stderr 24 | return wrapper 25 | 26 | def confirm(console, prompt, default="n", auto=None): 27 | if auto in (True, False): 28 | console.print(f"✅ {T('Auto confirm')}") 29 | return auto 30 | while True: 31 | response = console.input(prompt).strip().lower() 32 | if not response: 33 | response = default 34 | if response in ["y", "n"]: 35 | break 36 | return response == "y" 37 | 38 | def confirm_disclaimer(console): 39 | DISCLAIMER_TEXT = read_text(__respkg__, "DISCLAIMER.md") 40 | console.print() 41 | panel = Panel.fit(DISCLAIMER_TEXT, title="[red]免责声明", border_style="red", padding=(1, 2)) 42 | console.print(panel) 43 | 44 | while True: 45 | console.print("\n[red]是否确认已阅读并接受以上免责声明？[/red](yes/no):", end=" ") 46 | response = input().strip().lower() 47 | if response in ("yes", "y"): 48 | console.print("[green]感谢确认，程序继续运行。[/green]") 49 | return True 50 | elif response in ("no", "n"): 51 | console.print("[red]您未接受免责声明，程序将退出。[/red]") 52 | return False 53 | else: 54 | console.print("[yellow]请输入 yes 或 no。[/yellow]") 55 | 56 | def get_safe_filename(input_str, extension=".html", max_length=16): 57 | input_str = input_str.strip() 58 | safe_str = re.sub(r'[\\/:*?"<>|]', '', input_str).strip() 59 | if not safe_str: 60 | return None 61 | 62 | name = safe_str[:max_length] 63 | base_name = name 64 | filename = f"{base_name}{extension}" 65 | counter = 1 66 | 67 | while os.path.exists(filename): 68 | filename = f"{base_name}_{counter}{extension}" 69 | counter += 1 70 | 71 | return filename 72 | -------------------------------------------------------------------------------- /aipyapp/aipy/wizard.py: -------------------------------------------------------------------------------- 1 | from collections import OrderedDict 2 | import questionary 3 | import requests 4 | 5 | from loguru import logger 6 | 7 | from .. import T 8 | from ..config.llm import LLMConfig, PROVIDERS 9 | from .trustoken import TrustToken 10 | 11 | def get_models(providers, provider, api_key: str) -> list: 12 | """获取可用的模型列表""" 13 | provider_info = providers[provider] 14 | headers = { 15 | "Content-Type": "application/json" 16 | } 17 | 18 | if provider == "Claude": 19 | headers["x-api-key"] = api_key 20 | headers["anthropic-version"] = "2023-06-01" 21 | else: 22 | headers["Authorization"] = f"Bearer {api_key}" 23 | 24 | try: 25 | response = requests.get( 26 | f"{provider_info['api_base']}{provider_info['models_endpoint']}", 27 | headers=headers 28 | ) 29 | logger.info(f"获取模型列表: {response.text}") 30 | if response.status_code == 200: 31 | data = response.json() 32 | logger.info(f"获取模型列表成功: {data}") 33 | if provider in ["OpenAI", "DeepSeek", "xAI", "Claude"]: 34 | return [model["id"] for model in data["data"]] 35 | elif provider == "Gemini": 36 | return [model["name"] for model in data["models"]] 37 | return [] 38 | except Exception as e: 39 | logger.error(f"获取模型列表失败: {str(e)}") 40 | return [] # 如果API调用失败，返回空列表 41 | 42 | def select_provider(llm_config, default='Trustoken'): 43 | """选择提供商""" 44 | while True: 45 | name = questionary.select( 46 | T('Select LLM Provider'), 47 | choices=[ 48 | questionary.Choice(title=T('Trustoken is an intelligent API Key management service'), value='Trustoken', description=T('Recommended for beginners, easy to configure and feature-rich')), 49 | questionary.Choice(title=T('Other'), value='other') 50 | ], 51 | default=default 52 | ).unsafe_ask() 53 | if name == default: 54 | return default 55 | 56 | names = [name for name in llm_config.providers.keys() if name != default] 57 | names.append('<--') 58 | name = questionary.select( 59 | T('Select other providers'), 60 | choices=names 61 | ).unsafe_ask() 62 | if name != '<--': 63 | return name 64 | 65 | 66 | def config_llm(llm_config): 67 | """配置 LLM 提供商""" 68 | # 第一步：选择提供商 69 | name = select_provider(llm_config) 70 | if not name: 71 | return None 72 | provider = llm_config.providers[name] 73 | config = {"type": provider["type"]} 74 | 75 | if name == 'Trustoken': 76 | def save_token(token): 77 | config['api_key'] = token 78 | 79 | tt = TrustToken() 80 | if not tt.fetch_token(save_token): 81 | return None 82 | 83 | config['model'] = 'auto' 84 | else: 85 | api_key = questionary.text( 86 | T('Enter your API key'), 87 | validate=lambda x: len(x) > 8 88 | ).unsafe_ask() 89 | config['api_key'] = api_key 90 | 91 | # 获取可用模型列表 92 | available_models = get_models(llm_config.providers, name, api_key) 93 | if not available_models: 94 | logger.warning(T('Model list acquisition failed')) 95 | return None 96 | 97 | # 第三步：选择模型 98 | model = questionary.select( 99 | T('Available Models'), 100 | choices=available_models 101 | ).unsafe_ask() 102 | config['model'] = model 103 | 104 | # 保存配置 105 | config['enable'] = True 106 | current_config = llm_config.config 107 | current_config[name] = config 108 | llm_config.save_config(current_config) 109 | return current_config 110 | -------------------------------------------------------------------------------- /aipyapp/cli/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/knownsec/aipyapp/0a3b34fb05278b820a9431f5121a45c5f329f2d9/aipyapp/cli/__init__.py -------------------------------------------------------------------------------- /aipyapp/cli/cli_ipython.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding: utf-8 -*- 3 | 4 | from rich.console import Console 5 | from traitlets.config import Config 6 | from IPython.terminal.prompts import ClassicPrompts, Token 7 | from IPython.terminal.embed import embed, InteractiveShellEmbed 8 | from IPython.core.magic import Magics, magics_class, line_cell_magic,line_magic, cell_magic, register_line_magic 9 | 10 | from ..aipy import TaskManager, ConfigManager, CONFIG_DIR 11 | from .. import T, set_lang, __version__ 12 | 13 | class MainPrompt(ClassicPrompts): 14 | def in_prompt_tokens(self): 15 | return [(Token.Prompt, '>> ')] 16 | 17 | class TaskPrompt(ClassicPrompts): 18 | def in_prompt_tokens(self): 19 | return [(Token.Prompt, '>>> ')] 20 | 21 | def get_task_config(): 22 | c = Config() 23 | c.TerminalInteractiveShell.display_banner = False 24 | c.InteractiveShell.prompt_in1 = '>> ' 25 | c.InteractiveShell.prompt_in2 = '... ' 26 | c.InteractiveShell.prompt_out = '' 27 | c.InteractiveShell.banner1 = 'Use /ai "Task" to start a task' 28 | c.InteractiveShell.banner2 = 'For example: ai 获取Google网站标题\n' 29 | c.InteractiveShell.separate_in = '' 30 | c.InteractiveShell.separate_out = '' 31 | c.InteractiveShell.separate_out2 = '' 32 | c.InteractiveShell.prompts_class = TaskPrompt 33 | c.InteractiveShell.confirm_exit = False 34 | return c 35 | 36 | def get_main_config(): 37 | c = Config() 38 | c.TerminalInteractiveShell.display_banner = False 39 | c.InteractiveShell.prompt_in1 = '>> ' 40 | c.InteractiveShell.prompt_in2 = '... ' 41 | c.InteractiveShell.prompt_out = '' 42 | c.InteractiveShell.banner1 = 'Use /ai "Task" to start a task' 43 | c.InteractiveShell.banner2 = 'For example: ai 获取Google网站标题\n' 44 | c.InteractiveShell.separate_in = '' 45 | c.InteractiveShell.separate_out = '' 46 | c.InteractiveShell.separate_out2 = '' 47 | c.InteractiveShell.prompts_class = MainPrompt 48 | c.InteractiveShell.confirm_exit = False 49 | return c 50 | 51 | @magics_class 52 | class AIMagics(Magics): 53 | def __init__(self, shell, ai): 54 | super().__init__(shell) 55 | self.ai = ai 56 | 57 | @line_magic 58 | def task(self, line): 59 | task = self.ai.new_task() 60 | user_ns = {'task': task, 'settings': self.ai.settings} 61 | shell = InteractiveShellEmbed(user_ns=user_ns, config=get_task_config()) 62 | shell() 63 | 64 | @line_magic 65 | def clear(self, _): 66 | self.ai.clear() 67 | 68 | @line_magic 69 | def save(self, line): 70 | self.ai.save(line) 71 | 72 | @line_cell_magic 73 | def ai(self, line, cell=None): 74 | print(line) 75 | print(cell) 76 | 77 | def main(args): 78 | console = Console(record=True) 79 | console.print(f"[bold cyan]🚀 Python use - AIPython ({__version__}) [[green]https://aipy.app[/green]]") 80 | 81 | conf = ConfigManager(args.config_dir) 82 | conf.check_config() 83 | settings = conf.get_config() 84 | 85 | lang = settings.get('lang') 86 | if lang: set_lang(lang) 87 | 88 | settings.gui = False 89 | settings.debug = args.debug 90 | 91 | try: 92 | ai = TaskManager(settings, console=console) 93 | except Exception as e: 94 | console.print_exception(e) 95 | return 96 | 97 | update = ai.get_update(True) 98 | if update and update.get('has_update'): 99 | console.print(f"[bold red]🔔 号外❗ {T('Update available')}: {update.get('latest_version')}") 100 | 101 | if not ai.client_manager: 102 | console.print(f"[bold red]{T('No available LLM, please check the configuration file')}") 103 | return 104 | 105 | names = ai.client_manager.names 106 | console.print(f"{T('Please use ai(task) to enter the task to be processed by AI (enter ai.use(llm) to switch to the following LLM:')}", style="green") 107 | console.print(f"[cyan]{T('Default')}: [green]{names['default']}，[cyan]{T('Enabled')}: [yellow]{' '.join(names['enabled'])}") 108 | 109 | user_ns = {'AI': ai, 'settings': settings} 110 | shell = InteractiveShellEmbed(user_ns=user_ns, config=get_main_config()) 111 | shell.register_magics(AIMagics(shell, ai)) 112 | shell() 113 | 114 | -------------------------------------------------------------------------------- /aipyapp/cli/cli_python.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding: utf-8 -*- 3 | import code 4 | import builtins 5 | from pathlib import Path 6 | 7 | from rich.console import Console 8 | from prompt_toolkit import PromptSession 9 | from prompt_toolkit.formatted_text import HTML 10 | from prompt_toolkit.auto_suggest import AutoSuggestFromHistory 11 | from prompt_toolkit.completion import WordCompleter 12 | from prompt_toolkit.lexers import PygmentsLexer 13 | from prompt_toolkit.history import FileHistory 14 | from pygments.lexers.python import PythonLexer 15 | 16 | from ..aipy import TaskManager, ConfigManager, CONFIG_DIR 17 | from .. import T, set_lang, __version__ 18 | 19 | class PythonCompleter(WordCompleter): 20 | def __init__(self, ai): 21 | names = ['exit()'] 22 | names += [name for name in dir(builtins)] 23 | names += [f"ai.{attr}" for attr in dir(ai) if not attr.startswith('_')] 24 | super().__init__(names, ignore_case=True) 25 | 26 | def main(args): 27 | console = Console(record=True) 28 | console.print(f"[bold cyan]🚀 Python use - AIPython ({__version__}) [[green]https://aipy.app[/green]]") 29 | 30 | conf = ConfigManager(args.config_dir) 31 | conf.check_config() 32 | settings = conf.get_config() 33 | 34 | lang = settings.get('lang') 35 | if lang: set_lang(lang) 36 | 37 | settings.gui = False 38 | settings.debug = args.debug 39 | 40 | try: 41 | ai = TaskManager(settings, console=console) 42 | except Exception as e: 43 | console.print_exception(e) 44 | return 45 | 46 | update = ai.get_update(True) 47 | if update and update.get('has_update'): 48 | console.print(f"[bold red]🔔 号外❗ {T('Update available')}: {update.get('latest_version')}") 49 | 50 | if not ai.client_manager: 51 | console.print(f"[bold red]{T('No available LLM, please check the configuration file')}") 52 | return 53 | 54 | names = ai.client_manager.names 55 | console.print(f"{T('Please use ai(task) to enter the task to be processed by AI (enter ai.use(llm) to switch to the following LLM:')}", style="green") 56 | console.print(f"[cyan]{T('Default')}: [green]{names['default']}，[cyan]{T('Enabled')}: [yellow]{' '.join(names['enabled'])}") 57 | 58 | interp = code.InteractiveConsole({'ai': ai}) 59 | 60 | completer = PythonCompleter(ai) 61 | lexer = PygmentsLexer(PythonLexer) 62 | auto_suggest = AutoSuggestFromHistory() 63 | history = FileHistory(str(CONFIG_DIR / '.history.py')) 64 | session = PromptSession(history=history, completer=completer, lexer=lexer, auto_suggest=auto_suggest) 65 | while True: 66 | try: 67 | user_input = session.prompt(HTML('>> ')) 68 | if user_input.strip() in {"exit()", "quit()"}: 69 | break 70 | interp.push(user_input) 71 | except EOFError: 72 | console.print("[bold yellow]Exiting...") 73 | break 74 | except Exception as e: 75 | console.print(f"[bold red]Error: {e}") 76 | -------------------------------------------------------------------------------- /aipyapp/cli/cli_task.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding: utf-8 -*- 3 | import sys 4 | from enum import Enum, auto 5 | from collections import OrderedDict 6 | 7 | from rich import print 8 | from rich.console import Console 9 | from rich.table import Table 10 | from prompt_toolkit import PromptSession 11 | from prompt_toolkit.history import FileHistory 12 | from prompt_toolkit.styles import Style 13 | from prompt_toolkit.completion import WordCompleter 14 | 15 | from ..aipy import TaskManager, ConfigManager, CONFIG_DIR 16 | from .. import T, set_lang, __version__ 17 | from ..config import LLMConfig 18 | from ..aipy.wizard import config_llm 19 | 20 | class CommandType(Enum): 21 | CMD_DONE = auto() 22 | CMD_USE = auto() 23 | CMD_EXIT = auto() 24 | CMD_INVALID = auto() 25 | CMD_TEXT = auto() 26 | CMD_INFO = auto() 27 | CMD_MCP = auto() 28 | 29 | def parse_command(input_str, llms=set()): 30 | lower = input_str.lower() 31 | 32 | if lower in ("/done", "done"): 33 | return CommandType.CMD_DONE, None 34 | if lower in ("/info", "info"): 35 | return CommandType.CMD_INFO, None 36 | if lower in ("/exit", "exit"): 37 | return CommandType.CMD_EXIT, None 38 | if lower in llms: 39 | return CommandType.CMD_USE, input_str 40 | 41 | if lower.startswith("/use "): 42 | arg = input_str[5:].strip() 43 | if arg in llms: 44 | return CommandType.CMD_USE, arg 45 | else: 46 | return CommandType.CMD_INVALID, arg 47 | 48 | if lower.startswith("use "): 49 | arg = input_str[4:].strip() 50 | if arg in llms: 51 | return CommandType.CMD_USE, arg 52 | 53 | if lower.startswith("/mcp"): 54 | args = input_str[4:].strip().split(" ") 55 | return CommandType.CMD_MCP, args 56 | 57 | return CommandType.CMD_TEXT, input_str 58 | 59 | def show_info(info): 60 | info['Python'] = sys.executable 61 | info[T('Python version')] = sys.version 62 | info[T('Python base prefix')] = sys.base_prefix 63 | table = Table(title=T("System information"), show_lines=True) 64 | 65 | table.add_column(T("Parameter"), justify="center", style="bold cyan", no_wrap=True) 66 | table.add_column(T("Value"), justify="right", style="bold magenta") 67 | 68 | for key, value in info.items(): 69 | table.add_row( 70 | key, 71 | value, 72 | ) 73 | print(table) 74 | 75 | def process_mcp_ret(console, arg, ret): 76 | if ret.get("status", "success") == "success": 77 | #console.print(f"[green]{T('mcp_success')}: {ret.get('message', '')}[/green]") 78 | mcp_status = T('Enabled') if ret.get("globally_enabled") else T('Disabled') 79 | console.print(f"[green]{T('MCP server status: {}').format(mcp_status)}[/green]") 80 | mcp_servers = ret.get("servers", []) 81 | if ret.get("globally_enabled", False): 82 | for server_name, info in mcp_servers.items(): 83 | server_status = T('Enabled') if info.get("enabled", False) else T('Disabled') 84 | console.print( 85 | "[", server_status, "]", 86 | server_name, info.get("tools_count"), T("Tools") 87 | ) 88 | else: 89 | #console.print(f"[red]{T('mcp_error')}: {ret.get('message', '')}[/red]") 90 | console.print("操作失败", ret.get("message", '')) 91 | 92 | class InteractiveConsole(): 93 | def __init__(self, tm, console, settings): 94 | self.tm = tm 95 | self.names = tm.client_manager.names 96 | completer = WordCompleter(['/use', 'use', '/done','done', '/info', 'info', '/mcp'] + list(self.names['enabled']), ignore_case=True) 97 | self.history = FileHistory(str(CONFIG_DIR / ".history")) 98 | self.session = PromptSession(history=self.history, completer=completer) 99 | self.console = console 100 | self.style_main = Style.from_dict({"prompt": "green"}) 101 | self.style_ai = Style.from_dict({"prompt": "cyan"}) 102 | 103 | def input_with_possible_multiline(self, prompt_text, is_ai=False): 104 | prompt_style = self.style_ai if is_ai else self.style_main 105 | 106 | first_line = self.session.prompt([("class:prompt", prompt_text)], style=prompt_style) 107 | if not first_line.endswith("\\"): 108 | return first_line 109 | # Multi-line input 110 | lines = [first_line.rstrip("\\")] 111 | while True: 112 | next_line = self.session.prompt([("class:prompt", "... ")], style=prompt_style) 113 | if next_line.endswith("\\"): 114 | lines.append(next_line.rstrip("\\")) 115 | else: 116 | lines.append(next_line) 117 | break 118 | return "\n".join(lines) 119 | 120 | def run_task(self, task, instruction): 121 | try: 122 | task.run(instruction) 123 | except (EOFError, KeyboardInterrupt): 124 | pass 125 | except Exception as e: 126 | self.console.print_exception() 127 | 128 | def start_task_mode(self, task, instruction): 129 | self.console.print(f"{T('Enter AI mode, start processing tasks, enter Ctrl+d or /done to end the task')}", style="cyan") 130 | self.run_task(task, instruction) 131 | while True: 132 | try: 133 | user_input = self.input_with_possible_multiline(">>> ", is_ai=True).strip() 134 | if len(user_input) < 2: continue 135 | except (EOFError, KeyboardInterrupt): 136 | break 137 | 138 | cmd, arg = parse_command(user_input, self.names['enabled']) 139 | if cmd == CommandType.CMD_TEXT: 140 | self.run_task(task, arg) 141 | elif cmd == CommandType.CMD_DONE: 142 | break 143 | elif cmd == CommandType.CMD_USE: 144 | ret = task.use(arg) 145 | self.console.print('[green]Ok[/green]' if ret else '[red]Error[/red]') 146 | elif cmd == CommandType.CMD_INVALID: 147 | self.console.print(f'[red]Error: {arg}[/red]') 148 | 149 | try: 150 | task.done() 151 | except Exception as e: 152 | self.console.print_exception() 153 | self.console.print(f"[{T('Exit AI mode')}]", style="cyan") 154 | 155 | def info(self): 156 | info = OrderedDict() 157 | info[T('Current configuration directory')] = str(CONFIG_DIR) 158 | info[T('Current working directory')] = str(self.tm.workdir) 159 | info[T('Current LLM')] = repr(self.tm.client_manager.current) 160 | show_info(info) 161 | 162 | def run(self): 163 | self.console.print(f"{T('Please enter the task to be processed by AI (enter /use to switch, enter /info to view system information)')}", style="green") 164 | self.console.print(f"[cyan]{T('Default')}: [green]{self.names['default']}，[cyan]{T('Enabled')}: [yellow]{' '.join(self.names['enabled'])}") 165 | self.info() 166 | tm = self.tm 167 | while True: 168 | try: 169 | user_input = self.input_with_possible_multiline(">> ").strip() 170 | if len(user_input) < 2: 171 | continue 172 | 173 | cmd, arg = parse_command(user_input, self.names['enabled']) 174 | if cmd == CommandType.CMD_TEXT: 175 | task = tm.new_task() 176 | self.start_task_mode(task, arg) 177 | elif cmd == CommandType.CMD_USE: 178 | ret = tm.client_manager.use(arg) 179 | self.console.print('[green]Ok[/green]' if ret else '[red]Error[/red]') 180 | elif cmd == CommandType.CMD_INFO: 181 | self.info() 182 | elif cmd == CommandType.CMD_EXIT: 183 | break 184 | elif cmd == CommandType.CMD_MCP: 185 | if tm.mcp: 186 | ret = tm.mcp.process_command(arg) 187 | process_mcp_ret(self.console, arg, ret) 188 | else: 189 | self.console.print("MCP config not found") 190 | elif cmd == CommandType.CMD_INVALID: 191 | self.console.print('[red]Error[/red]') 192 | except (EOFError, KeyboardInterrupt): 193 | break 194 | 195 | def main(args): 196 | console = Console(record=True) 197 | console.print(f"[bold cyan]🚀 Python use - AIPython ({__version__}) [[green]https://aipy.app[/green]]") 198 | conf = ConfigManager(args.config_dir) 199 | settings = conf.get_config() 200 | lang = settings.get('lang') 201 | if lang: set_lang(lang) 202 | llm_config = LLMConfig(CONFIG_DIR / "config") 203 | if conf.check_config(gui=True) == 'TrustToken': 204 | if llm_config.need_config(): 205 | console.print(f"[yellow]{T('Starting LLM Provider Configuration Wizard')}[/yellow]") 206 | try: 207 | config = config_llm(llm_config) 208 | except KeyboardInterrupt: 209 | console.print(f"[yellow]{T('User cancelled configuration')}[/yellow]") 210 | return 211 | if not config: 212 | return 213 | settings["llm"] = llm_config.config 214 | 215 | if args.fetch_config: 216 | conf.fetch_config() 217 | return 218 | 219 | settings.gui = False 220 | settings.debug = args.debug 221 | 222 | try: 223 | tm = TaskManager(settings, console=console) 224 | except Exception as e: 225 | console.print_exception() 226 | return 227 | 228 | update = tm.get_update() 229 | if update and update.get('has_update'): 230 | console.print(f"[bold red]🔔 号外❗ {T('Update available')}: {update.get('latest_version')}") 231 | 232 | if not tm.client_manager: 233 | console.print(f"[bold red]{T('No available LLM, please check the configuration file')}") 234 | return 235 | 236 | if args.cmd: 237 | tm.new_task().run(args.cmd) 238 | return 239 | InteractiveConsole(tm, console, settings).run() 240 | -------------------------------------------------------------------------------- /aipyapp/config/__init__.py: -------------------------------------------------------------------------------- 1 | from .llm import LLMConfig 2 | 3 | __all__ = ["LLMConfig"] 4 | -------------------------------------------------------------------------------- /aipyapp/config/base.py: -------------------------------------------------------------------------------- 1 | #! /usr/bin/env python3 2 | # -*- coding: utf-8 -*- 3 | 4 | import json 5 | from typing import Dict 6 | from pathlib import Path 7 | 8 | class BaseConfig: 9 | FILE = None 10 | 11 | def __init__(self, path: str): 12 | path = Path(path) 13 | path.mkdir(parents=True, exist_ok=True) 14 | self.config_file = path / self.FILE 15 | self.config = self.load_config() 16 | 17 | def load_config(self) -> Dict: 18 | if self.config_file.exists(): 19 | with open(self.config_file, 'r') as f: 20 | return json.load(f) 21 | return {} 22 | 23 | def save_config(self, config: Dict): 24 | with open(self.config_file, 'w') as f: 25 | json.dump(config, f, indent=2) -------------------------------------------------------------------------------- /aipyapp/config/llm.py: -------------------------------------------------------------------------------- 1 | #! /usr/bin/env python3 2 | # -*- coding: utf-8 -*- 3 | from collections import OrderedDict 4 | 5 | from .base import BaseConfig 6 | from .. import T, get_lang 7 | 8 | PROVIDERS = { 9 | "Trustoken": { 10 | "api_base": T("https://sapi.trustoken.ai/v1"), 11 | "models_endpoint": "/models", 12 | "type": "trust", 13 | "model": "auto" 14 | }, 15 | "DeepSeek": { 16 | "api_base": "https://api.deepseek.com", 17 | "models_endpoint": "/models", 18 | "type": "deepseek" 19 | }, 20 | "xAI": { 21 | "api_base": "https://api.x.ai/v1", 22 | "models_endpoint": "/models", 23 | "type": "grok" 24 | }, 25 | "Claude": { 26 | "api_base": "https://api.anthropic.com/v1", 27 | "models_endpoint": "/models", 28 | "type": "claude" 29 | }, 30 | "OpenAI": { 31 | "api_base": "https://api.openai.com/v1", 32 | "models_endpoint": "/models", 33 | "type": "openai" 34 | }, 35 | "Gemini": { 36 | "api_base": "https://generativelanguage.googleapis.com/v1beta/", 37 | "models_endpoint": "/models", 38 | "type": "gemini" 39 | }, 40 | } 41 | 42 | def get_providers(): 43 | if get_lang() == "zh": 44 | providers = OrderedDict() 45 | providers["Trustoken"] = PROVIDERS["Trustoken"] 46 | providers["DeepSeek"] = PROVIDERS["DeepSeek"] 47 | return providers 48 | else: 49 | return PROVIDERS 50 | 51 | class LLMConfig(BaseConfig): 52 | FILE = "llm.json" 53 | 54 | def __init__(self, path: str): 55 | super().__init__(path) 56 | self.providers = get_providers() 57 | 58 | def need_config(self): 59 | """检查是否需要配置LLM。 60 | """ 61 | if not self.config: 62 | return True 63 | 64 | for _, config in self.config.items() : 65 | if config.get("enable", True): 66 | return False 67 | return True 68 | -------------------------------------------------------------------------------- /aipyapp/exec/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | from .runner import Runner 3 | from .runtime import BaseRuntime 4 | 5 | __all__ = ['Runner', 'BaseRuntime'] -------------------------------------------------------------------------------- /aipyapp/exec/runner.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding: utf-8 -*- 3 | 4 | import sys 5 | import json 6 | import traceback 7 | from io import StringIO 8 | import webbrowser 9 | 10 | from loguru import logger 11 | 12 | INIT_IMPORTS = """ 13 | import os 14 | import re 15 | import sys 16 | import json 17 | import time 18 | import random 19 | import traceback 20 | """ 21 | 22 | def is_json_serializable(obj): 23 | try: 24 | json.dumps(obj, ensure_ascii=False, default=str) 25 | return True 26 | except (TypeError, OverflowError): 27 | return False 28 | 29 | def diff_dicts(dict1, dict2): 30 | diff = {} 31 | for key, value in dict1.items(): 32 | if key not in dict2: 33 | diff[key] = value 34 | continue 35 | 36 | try: 37 | if value != dict2[key]: 38 | diff[key] = value 39 | except Exception: 40 | pass 41 | return diff 42 | 43 | class Runner(): 44 | def __init__(self, runtime): 45 | self.runtime = runtime 46 | self.history = [] 47 | self.log = logger.bind(src='runner') 48 | self._globals = {'runtime': runtime, '__storage__': {}, '__name__': '__main__', 'input': self.runtime.input} 49 | exec(INIT_IMPORTS, self._globals) 50 | 51 | def __repr__(self): 52 | return f"" 53 | 54 | @property 55 | def globals(self): 56 | return self._globals 57 | 58 | def _exec_python_block(self, block): 59 | old_stdout, old_stderr = sys.stdout, sys.stderr 60 | captured_stdout = StringIO() 61 | captured_stderr = StringIO() 62 | sys.stdout, sys.stderr = captured_stdout, captured_stderr 63 | result = {} 64 | env = self.runtime.envs.copy() 65 | session = self._globals['__storage__'].copy() 66 | gs = self._globals.copy() 67 | gs['__retval__'] = {} 68 | try: 69 | exec(block.code, gs) 70 | except (SystemExit, Exception) as e: 71 | result['errstr'] = str(e) 72 | result['traceback'] = traceback.format_exc() 73 | finally: 74 | sys.stdout = old_stdout 75 | sys.stderr = old_stderr 76 | 77 | s = captured_stdout.getvalue().strip() 78 | if s: result['stdout'] = s if is_json_serializable(s) else '' 79 | s = captured_stderr.getvalue().strip() 80 | if s: result['stderr'] = s if is_json_serializable(s) else '' 81 | 82 | vars = gs.get('__retval__') 83 | if vars: 84 | #self._globals['__retval__'] = vars 85 | result['__retval__'] = self.filter_result(vars) 86 | 87 | history = {} 88 | diff = diff_dicts(env, self.runtime.envs) 89 | if diff: 90 | history['env'] = diff 91 | diff = diff_dicts(gs['__storage__'], session) 92 | if diff: 93 | history['session'] = diff 94 | 95 | return result, history 96 | 97 | def __call__(self, block): 98 | self.log.info(f'Exec: {block}') 99 | lang = block.get_lang() 100 | if lang == 'python': 101 | result, history = self._exec_python_block(block) 102 | elif lang == 'html': 103 | result, history = self._exec_html_block(block) 104 | else: 105 | result = {'stderr': f'Exec: Ignore unsupported block type: {lang}'} 106 | history = {} 107 | 108 | history['block_id'] = block.id 109 | history['result'] = result 110 | self.history.append(history) 111 | return result.copy() 112 | 113 | def _exec_html_block(self, block): 114 | abs_path = block.abs_path 115 | if abs_path: 116 | webbrowser.open(f'file://{abs_path}') 117 | result = {'stdout': 'OK'} 118 | return result, {} 119 | 120 | def filter_result(self, vars): 121 | if isinstance(vars, dict): 122 | for key in vars.keys(): 123 | if key in self.runtime.envs: 124 | vars[key] = '' 125 | else: 126 | vars[key] = self.filter_result(vars[key]) 127 | elif isinstance(vars, list): 128 | vars = [self.filter_result(v) for v in vars] 129 | else: 130 | vars = vars if is_json_serializable(vars) else '' 131 | return vars 132 | -------------------------------------------------------------------------------- /aipyapp/exec/runtime.py: -------------------------------------------------------------------------------- 1 | #! /usr/bin/env python3 2 | # -*- coding: utf-8 -*- 3 | import sys 4 | import subprocess 5 | from abc import ABC, abstractmethod 6 | 7 | from loguru import logger 8 | 9 | class BaseRuntime(ABC): 10 | def __init__(self, envs=None): 11 | self.envs = envs or {} 12 | self.packages = set() 13 | self.log = logger.bind(src='runtime') 14 | 15 | def set_env(self, name, value, desc): 16 | self.envs[name] = (value, desc) 17 | 18 | def ensure_packages(self, *packages, upgrade=False, quiet=False): 19 | if not packages: 20 | return True 21 | 22 | packages = list(set(packages) - self.packages) 23 | if not packages: 24 | return True 25 | 26 | cmd = [sys.executable, "-m", "pip", "install"] 27 | if upgrade: 28 | cmd.append("--upgrade") 29 | if quiet: 30 | cmd.append("-q") 31 | cmd.extend(packages) 32 | 33 | try: 34 | subprocess.check_call(cmd) 35 | self.packages.update(packages) 36 | return True 37 | except subprocess.CalledProcessError: 38 | self.log.error("依赖安装失败: {}", " ".join(packages)) 39 | 40 | return False 41 | 42 | def ensure_requirements(self, path="requirements.txt", **kwargs): 43 | with open(path) as f: 44 | reqs = [line.strip() for line in f if line.strip() and not line.startswith("#")] 45 | return self.ensure_packages(*reqs, **kwargs) 46 | 47 | @abstractmethod 48 | def install_packages(self, packages): 49 | pass 50 | 51 | @abstractmethod 52 | def get_env(self, name, default=None, *, desc=None): 53 | pass 54 | 55 | @abstractmethod 56 | def display(self, path=None, url=None): 57 | pass 58 | 59 | @abstractmethod 60 | def input(self, prompt=''): 61 | pass -------------------------------------------------------------------------------- /aipyapp/gui/__init__.py: -------------------------------------------------------------------------------- 1 | from .trustoken import TrustTokenAuthDialog 2 | from .config import ConfigDialog 3 | from .apimarket import ApiMarketDialog 4 | from .providers import show_provider_config 5 | from .about import AboutDialog 6 | from .statusbar import CStatusBar 7 | 8 | __all__ = ['TrustTokenAuthDialog', 'ConfigDialog', 'ApiMarketDialog', 'show_provider_config', 'AboutDialog', 'CStatusBar'] 9 | -------------------------------------------------------------------------------- /aipyapp/gui/about.py: -------------------------------------------------------------------------------- 1 | import wx 2 | import importlib.resources as resources 3 | 4 | from .. import __version__, T, __respkg__ 5 | from ..aipy.config import CONFIG_DIR 6 | 7 | class AboutDialog(wx.Dialog): 8 | def __init__(self, parent): 9 | super().__init__(parent, title=T("About AIPY")) 10 | 11 | # 创建垂直布局 12 | vbox = wx.BoxSizer(wx.VERTICAL) 13 | 14 | logo_panel = wx.Panel(self) 15 | logo_sizer = wx.BoxSizer(wx.HORIZONTAL) 16 | 17 | with resources.path(__respkg__, "aipy.ico") as icon_path: 18 | icon = wx.Icon(str(icon_path), wx.BITMAP_TYPE_ICO) 19 | bmp = wx.Bitmap() 20 | bmp.CopyFromIcon(icon) 21 | # Scale the bitmap to a more appropriate size 22 | scaled_bmp = wx.Bitmap(bmp.ConvertToImage().Scale(48, 48, wx.IMAGE_QUALITY_HIGH)) 23 | logo_sizer.Add(wx.StaticBitmap(logo_panel, -1, scaled_bmp), 0, wx.ALL | wx.ALIGN_CENTER, 5) 24 | 25 | # 添加标题 26 | title = wx.StaticText(logo_panel, -1, label=T("AIPy")) 27 | title.SetFont(wx.Font(16, wx.FONTFAMILY_DEFAULT, wx.FONTSTYLE_NORMAL, wx.FONTWEIGHT_BOLD)) 28 | logo_sizer.Add(title, 0, wx.ALL|wx.ALIGN_CENTER, 10) 29 | logo_panel.SetSizer(logo_sizer) 30 | vbox.Add(logo_panel, 0, wx.ALL|wx.ALIGN_CENTER, 10) 31 | 32 | # 添加描述 33 | desc = wx.StaticText(self, label=T("AIPY is an intelligent assistant that can help you complete various tasks.")) 34 | desc.Wrap(350) 35 | vbox.Add(desc, 0, wx.ALL|wx.ALIGN_CENTER, 10) 36 | 37 | # 添加版本信息 38 | version = wx.StaticText(self, label=f"{T('Version')}: {__version__}") 39 | vbox.Add(version, 0, wx.ALL|wx.ALIGN_CENTER, 5) 40 | 41 | # 添加配置目录信息 42 | config_dir = wx.StaticText(self, label=f"{T('Current configuration directory')}: {CONFIG_DIR}") 43 | config_dir.Wrap(350) 44 | vbox.Add(config_dir, 0, wx.ALL|wx.ALIGN_CENTER, 5) 45 | 46 | # 添加工作目录信息 47 | work_dir = wx.StaticText(self, label=f"{T('Current working directory')}: {parent.tm.workdir}") 48 | work_dir.Wrap(350) 49 | vbox.Add(work_dir, 0, wx.ALL|wx.ALIGN_CENTER, 5) 50 | 51 | # 添加团队信息 52 | team = wx.StaticText(self, label=T("AIPY Team")) 53 | vbox.Add(team, 0, wx.ALL|wx.ALIGN_CENTER, 10) 54 | 55 | # 添加确定按钮 56 | ok_button = wx.Button(self, wx.ID_OK, T("OK")) 57 | vbox.Add(ok_button, 0, wx.ALL|wx.ALIGN_CENTER, 10) 58 | 59 | self.SetSizer(vbox) 60 | self.SetMinSize((400, 320)) 61 | self.Fit() 62 | self.Centre() -------------------------------------------------------------------------------- /aipyapp/gui/config.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | #coding: utf-8 3 | 4 | import os 5 | 6 | import wx 7 | import wx.adv 8 | from wx import DirDialog, FD_SAVE, FD_OVERWRITE_PROMPT 9 | from wx.lib.agw.floatspin import FloatSpin, EVT_FLOATSPIN, FS_LEFT, FS_RIGHT, FS_CENTRE, FS_READONLY 10 | 11 | from .. import T, set_lang 12 | 13 | class ConfigDialog(wx.Dialog): 14 | def __init__(self, parent, settings): 15 | super().__init__(parent, title=T('Configuration')) 16 | 17 | self.settings = settings 18 | 19 | vbox = wx.BoxSizer(wx.VERTICAL) 20 | 21 | # Main panel with content 22 | main_panel = wx.Panel(self) 23 | main_vbox = wx.BoxSizer(wx.VERTICAL) 24 | 25 | # Work directory group 26 | work_dir_box = wx.StaticBox(main_panel, -1, T('Work Directory')) 27 | work_dir_sizer = wx.StaticBoxSizer(work_dir_box, wx.VERTICAL) 28 | 29 | work_dir_panel = wx.Panel(main_panel) 30 | work_dir_inner_sizer = wx.BoxSizer(wx.HORIZONTAL) 31 | 32 | self.work_dir_text = wx.TextCtrl(work_dir_panel, -1, settings.workdir, style=wx.TE_READONLY) 33 | work_dir_inner_sizer.Add(self.work_dir_text, 1, wx.ALL | wx.EXPAND, 5) 34 | 35 | browse_button = wx.Button(work_dir_panel, -1, T('Browse...')) 36 | browse_button.Bind(wx.EVT_BUTTON, self.on_browse_work_dir) 37 | work_dir_inner_sizer.Add(browse_button, 0, wx.ALL | wx.ALIGN_CENTER_VERTICAL, 5) 38 | 39 | work_dir_panel.SetSizer(work_dir_inner_sizer) 40 | work_dir_sizer.Add(work_dir_panel, 0, wx.ALL | wx.EXPAND, 5) 41 | 42 | # Add hint about creating new directory 43 | hint_text = wx.StaticText(main_panel, -1, T('You can create a new directory in the file dialog')) 44 | work_dir_sizer.Add(hint_text, 0, wx.LEFT | wx.BOTTOM, 5) 45 | 46 | main_vbox.Add(work_dir_sizer, 0, wx.ALL | wx.EXPAND, 10) 47 | 48 | # Settings group 49 | settings_box = wx.StaticBox(main_panel, -1, T('Settings')) 50 | settings_sizer = wx.StaticBoxSizer(settings_box, wx.VERTICAL) 51 | 52 | # Max tokens slider 53 | tokens_panel = wx.Panel(main_panel) 54 | tokens_sizer = wx.BoxSizer(wx.HORIZONTAL) 55 | 56 | tokens_label = wx.StaticText(tokens_panel, -1, T('Max Tokens') + ":") 57 | tokens_sizer.Add(tokens_label, 0, wx.ALL | wx.ALIGN_CENTER_VERTICAL, 5) 58 | 59 | self.tokens_slider = wx.Slider(tokens_panel, -1, 60 | settings.get('max_tokens', 8192), 61 | minValue=64, 62 | maxValue=128*1024, 63 | style=wx.SL_HORIZONTAL | wx.SL_AUTOTICKS) 64 | self.tokens_slider.SetTickFreq(100) 65 | tokens_sizer.Add(self.tokens_slider, 1, wx.ALL | wx.EXPAND, 5) 66 | 67 | self.tokens_text = wx.StaticText(tokens_panel, -1, str(self.tokens_slider.GetValue())) 68 | self.tokens_text.SetMinSize((50, -1)) 69 | tokens_sizer.Add(self.tokens_text, 0, wx.ALL | wx.ALIGN_CENTER_VERTICAL, 5) 70 | 71 | tokens_panel.SetSizer(tokens_sizer) 72 | settings_sizer.Add(tokens_panel, 0, wx.ALL | wx.EXPAND, 5) 73 | 74 | # Timeout slider 75 | timeout_panel = wx.Panel(main_panel) 76 | timeout_sizer = wx.BoxSizer(wx.HORIZONTAL) 77 | 78 | timeout_label = wx.StaticText(timeout_panel, -1, T('Timeout (seconds)') + ":") 79 | timeout_sizer.Add(timeout_label, 0, wx.ALL | wx.ALIGN_CENTER_VERTICAL, 5) 80 | 81 | self.timeout_slider = wx.Slider(timeout_panel, -1, 82 | int(settings.get('timeout', 0)), 83 | minValue=0, 84 | maxValue=120, 85 | style=wx.SL_HORIZONTAL | wx.SL_AUTOTICKS) 86 | self.timeout_slider.SetTickFreq(30) 87 | timeout_sizer.Add(self.timeout_slider, 1, wx.ALL | wx.EXPAND, 5) 88 | 89 | self.timeout_text = wx.StaticText(timeout_panel, -1, str(self.timeout_slider.GetValue())) 90 | self.timeout_text.SetMinSize((50, -1)) 91 | timeout_sizer.Add(self.timeout_text, 0, wx.ALL | wx.ALIGN_CENTER_VERTICAL, 5) 92 | 93 | timeout_panel.SetSizer(timeout_sizer) 94 | settings_sizer.Add(timeout_panel, 0, wx.ALL | wx.EXPAND, 5) 95 | 96 | # Max rounds slider 97 | rounds_panel = wx.Panel(main_panel) 98 | rounds_sizer = wx.BoxSizer(wx.HORIZONTAL) 99 | 100 | rounds_label = wx.StaticText(rounds_panel, -1, T('Max Rounds') + ":") 101 | rounds_sizer.Add(rounds_label, 0, wx.ALL | wx.ALIGN_CENTER_VERTICAL, 5) 102 | 103 | self.rounds_slider = wx.Slider(rounds_panel, -1, 104 | settings.get('max_rounds', 16), 105 | minValue=1, 106 | maxValue=64, 107 | style=wx.SL_HORIZONTAL | wx.SL_AUTOTICKS) 108 | self.rounds_slider.SetTickFreq(8) 109 | rounds_sizer.Add(self.rounds_slider, 1, wx.ALL | wx.EXPAND, 5) 110 | 111 | self.rounds_text = wx.StaticText(rounds_panel, -1, str(self.rounds_slider.GetValue())) 112 | rounds_sizer.Add(self.rounds_text, 0, wx.ALL | wx.ALIGN_CENTER_VERTICAL, 5) 113 | 114 | rounds_panel.SetSizer(rounds_sizer) 115 | settings_sizer.Add(rounds_panel, 0, wx.ALL | wx.EXPAND, 5) 116 | 117 | main_vbox.Add(settings_sizer, 0, wx.ALL | wx.EXPAND, 10) 118 | 119 | self.url_ctrl = wx.adv.HyperlinkCtrl(main_panel, label=T('Click here for more information'), url="https://d.aipyaipy.com/d/162", style=wx.adv.HL_ALIGN_LEFT | wx.adv.HL_CONTEXTMENU) 120 | main_vbox.Add(self.url_ctrl, 0, wx.ALL, 10) 121 | 122 | main_panel.SetSizer(main_vbox) 123 | vbox.Add(main_panel, 1, wx.EXPAND) 124 | 125 | # Buttons panel at bottom 126 | button_panel = wx.Panel(self) 127 | button_sizer = wx.BoxSizer(wx.HORIZONTAL) 128 | 129 | ok_button = wx.Button(button_panel, wx.ID_OK, T('OK')) 130 | ok_button.SetMinSize((100, 30)) 131 | cancel_button = wx.Button(button_panel, wx.ID_CANCEL, T('Cancel')) 132 | cancel_button.SetMinSize((100, 30)) 133 | 134 | button_sizer.Add(ok_button, 0, wx.ALL, 5) 135 | button_sizer.Add(cancel_button, 0, wx.ALL, 5) 136 | 137 | button_panel.SetSizer(button_sizer) 138 | vbox.Add(button_panel, 0, wx.ALL | wx.ALIGN_CENTER | wx.BOTTOM, 20) 139 | 140 | # Bind events 141 | self.tokens_slider.Bind(wx.EVT_SLIDER, self.on_tokens_slider) 142 | self.timeout_slider.Bind(wx.EVT_SLIDER, self.on_timeout_slider) 143 | self.rounds_slider.Bind(wx.EVT_SLIDER, self.on_rounds_slider) 144 | 145 | self.SetSizer(vbox) 146 | self.SetMinSize((500, 450)) 147 | self.Fit() 148 | self.Centre() 149 | 150 | def on_browse_work_dir(self, event): 151 | with DirDialog(self, T('Select work directory'), 152 | defaultPath=self.work_dir_text.GetValue(), 153 | style=wx.DD_DEFAULT_STYLE) as dlg: 154 | if dlg.ShowModal() == wx.ID_OK: 155 | self.work_dir_text.SetValue(dlg.GetPath()) 156 | 157 | def on_tokens_slider(self, event): 158 | value = self.tokens_slider.GetValue() 159 | self.tokens_text.SetLabel(str(value)) 160 | 161 | def on_timeout_slider(self, event): 162 | value = self.timeout_slider.GetValue() 163 | self.timeout_text.SetLabel(str(value)) 164 | 165 | def on_rounds_slider(self, event): 166 | value = self.rounds_slider.GetValue() 167 | self.rounds_text.SetLabel(str(value)) 168 | 169 | def get_values(self): 170 | return { 171 | 'workdir': self.work_dir_text.GetValue(), 172 | 'max_tokens': int(self.tokens_slider.GetValue()), 173 | 'timeout': float(self.timeout_slider.GetValue()), 174 | 'max_rounds': int(self.rounds_slider.GetValue()) 175 | } 176 | 177 | if __name__ == '__main__': 178 | class TestSettings: 179 | def __init__(self): 180 | self.workdir = os.getcwd() 181 | self._settings = { 182 | 'max-tokens': 2000, 183 | 'timeout': 30.0, 184 | 'max-rounds': 10 185 | } 186 | 187 | def get(self, key, default=None): 188 | return self._settings.get(key, default) 189 | 190 | def __setitem__(self, key, value): 191 | self._settings[key] = value 192 | 193 | def save(self): 194 | print("Settings saved:", self._settings) 195 | 196 | app = wx.App(False) 197 | settings = TestSettings() 198 | dialog = ConfigDialog(None, settings) 199 | if dialog.ShowModal() == wx.ID_OK: 200 | values = dialog.get_values() 201 | print("New settings:", values) 202 | dialog.Destroy() 203 | app.MainLoop() -------------------------------------------------------------------------------- /aipyapp/gui/statusbar.py: -------------------------------------------------------------------------------- 1 | import os 2 | import sys 3 | import subprocess 4 | import wx 5 | 6 | from .. import T 7 | 8 | class CStatusBar(wx.StatusBar): 9 | def __init__(self, parent): 10 | super().__init__(parent, style=wx.STB_DEFAULT_STYLE) 11 | self.parent = parent 12 | self.SetFieldsCount(3) 13 | self.SetStatusWidths([-1, 30, 80]) 14 | 15 | self.tm = parent.tm 16 | self.current_llm = self.tm.client_manager.names['default'] 17 | self.enabled_llm = list(self.tm.client_manager.names['enabled']) 18 | self.menu_items = self.enabled_llm 19 | self.radio_group = [] 20 | 21 | self.folder_button = wx.StaticBitmap(self, -1, wx.ArtProvider.GetBitmap(wx.ART_FOLDER_OPEN, wx.ART_MENU)) 22 | self.folder_button.Bind(wx.EVT_LEFT_DOWN, self.on_open_work_dir) 23 | self.Bind(wx.EVT_SIZE, self.on_size) 24 | 25 | self.SetStatusText(f"{self.current_llm} ▾", 2) 26 | self.Bind(wx.EVT_LEFT_DOWN, self.on_click) 27 | 28 | def on_size(self, event): 29 | rect = self.GetFieldRect(1) 30 | self.folder_button.SetPosition((rect.x + 5, rect.y + 2)) 31 | event.Skip() 32 | 33 | def on_click(self, event): 34 | rect = self.GetFieldRect(2) 35 | if rect.Contains(event.GetPosition()): 36 | self.show_menu() 37 | 38 | def show_menu(self): 39 | self.current_menu = wx.Menu() 40 | self.radio_group = [] 41 | for label in self.menu_items: 42 | item = wx.MenuItem(self.current_menu, wx.ID_ANY, label, kind=wx.ITEM_RADIO) 43 | self.current_menu.Append(item) 44 | self.radio_group.append(item) 45 | self.Bind(wx.EVT_MENU, self.on_menu_select, item) 46 | if label == self.current_llm: 47 | item.Check() 48 | rect = self.GetFieldRect(2) 49 | pos = self.ClientToScreen(rect.GetBottomLeft()) 50 | self.PopupMenu(self.current_menu, self.ScreenToClient(pos)) 51 | 52 | def on_menu_select(self, event): 53 | item = self.current_menu.FindItemById(event.GetId()) 54 | label = item.GetItemLabel() 55 | if self.tm.use(label): 56 | self.current_llm = label 57 | self.SetStatusText(f"{label} ▾", 2) 58 | else: 59 | wx.MessageBox(T("LLM {} is not available").format(label), T("Warning"), wx.OK|wx.ICON_WARNING) 60 | 61 | def on_open_work_dir(self, event): 62 | """打开工作目录""" 63 | work_dir = self.tm.workdir 64 | if os.path.exists(work_dir): 65 | if sys.platform == 'win32': 66 | os.startfile(work_dir) 67 | elif sys.platform == 'darwin': 68 | subprocess.call(['open', work_dir]) 69 | else: 70 | subprocess.call(['xdg-open', work_dir]) 71 | else: 72 | wx.MessageBox(T("Work directory does not exist"), T("Error"), wx.OK | wx.ICON_ERROR) 73 | -------------------------------------------------------------------------------- /aipyapp/gui/trustoken.py: -------------------------------------------------------------------------------- 1 | import time 2 | 3 | import wx 4 | import wx.adv 5 | import threading 6 | import qrcode 7 | import io 8 | from PIL import Image 9 | 10 | from .. import T 11 | from ..aipy.trustoken import TrustTokenAPI 12 | 13 | class TrustTokenAuthDialog(wx.Dialog): 14 | """A dialog for TrustToken authentication with QR code display.""" 15 | 16 | def __init__(self, parent=None, coordinator_url=None, poll_interval=5): 17 | """ 18 | Initialize the authentication dialog. 19 | 20 | Args: 21 | parent: Parent window 22 | coordinator_url (str, optional): The coordinator server URL 23 | poll_interval (int, optional): Polling interval in seconds 24 | """ 25 | super().__init__(parent, title=T("TrustToken Authentication"), 26 | size=(400, 500), 27 | style=wx.DEFAULT_DIALOG_STYLE ) 28 | 29 | self.api = TrustTokenAPI(coordinator_url) 30 | self.poll_interval = poll_interval 31 | self.request_id = None 32 | self.polling_thread = None 33 | self.stop_polling = False 34 | self.start_time = None 35 | self.polling_timeout = 310 # 5 minutes and 10 seconds 36 | 37 | self._init_ui() 38 | self.Centre() 39 | 40 | def _init_ui(self): 41 | """Initialize the user interface.""" 42 | main_sizer = wx.BoxSizer(wx.VERTICAL) 43 | 44 | # Status text 45 | self.status_text = wx.StaticText(self, label='') 46 | main_sizer.Add(self.status_text, 0, wx.TOP | wx.ALIGN_CENTER, 5) 47 | 48 | # QR code display 49 | self.qr_bitmap = wx.StaticBitmap(self, size=(300, 300)) 50 | main_sizer.Add(self.qr_bitmap, 0, wx.ALL | wx.ALIGN_CENTER, 5) 51 | 52 | self.other_text = wx.adv.HyperlinkCtrl(self, label='扫码绑定AiPy大脑，您也可以配置其它大模型大脑', url='https://d.aipy.app/d/77') 53 | main_sizer.Add(self.other_text, 0, wx.TOP | wx.ALIGN_CENTER, 5) 54 | 55 | # Progress bar 56 | self.progress_bar = wx.Gauge(self, range=100) 57 | main_sizer.Add(self.progress_bar, 0, wx.ALL | wx.EXPAND, 5) 58 | 59 | # Time remaining text 60 | self.time_text = wx.StaticText(self, label='') 61 | self.time_text.Hide() 62 | main_sizer.Add(self.time_text, 0, wx.ALL | wx.ALIGN_CENTER, 5) 63 | 64 | # Buttons 65 | self.cancel_button = wx.Button(self, wx.ID_CANCEL, T("Cancel")) 66 | main_sizer.Add(self.cancel_button, 0, wx.ALIGN_CENTER | wx.ALL, 5) 67 | 68 | self.SetSizer(main_sizer) 69 | 70 | # Bind events 71 | self.Bind(wx.EVT_BUTTON, self._on_cancel, self.cancel_button) 72 | self.Bind(wx.EVT_CLOSE, self._on_close) 73 | 74 | def _update_progress(self): 75 | """Update the progress bar and time remaining.""" 76 | if not self.start_time: 77 | return 78 | 79 | elapsed = time.time() - self.start_time 80 | if elapsed >= self.polling_timeout: 81 | progress = 100 82 | time_remaining = 0 83 | else: 84 | progress = int((elapsed / self.polling_timeout) * 100) 85 | time_remaining = int(self.polling_timeout - elapsed) 86 | 87 | wx.CallAfter(self.progress_bar.SetValue, progress) 88 | wx.CallAfter(self.time_text.SetLabel, T("Time remaining: {} seconds", time_remaining)) 89 | 90 | def _poll_status(self, save_func): 91 | """Poll the binding status in a separate thread.""" 92 | self.start_time = time.time() 93 | self.time_text.Show() 94 | while not self.stop_polling and time.time() - self.start_time < self.polling_timeout: 95 | self._update_progress() 96 | 97 | data = self.api.check_status(self.request_id) 98 | if not data: 99 | time.sleep(self.poll_interval) 100 | continue 101 | 102 | status = data.get('status') 103 | wx.CallAfter(self._update_status, T("Current status: {}...", T(status))) 104 | 105 | if status == 'approved': 106 | if save_func: 107 | save_func(data['secret_token']) 108 | wx.CallAfter(self.EndModal, wx.ID_OK) 109 | return True 110 | elif status == 'expired': 111 | wx.CallAfter(self._update_status, T("Binding request expired.")) 112 | wx.CallAfter(self.EndModal, wx.ID_CANCEL) 113 | return False 114 | elif status == 'pending': 115 | pass 116 | else: 117 | wx.CallAfter(self._update_status, T("Received unknown status: {}", status)) 118 | wx.CallAfter(self.EndModal, wx.ID_CANCEL) 119 | return False 120 | 121 | time.sleep(self.poll_interval) 122 | 123 | if not self.stop_polling: 124 | wx.CallAfter(self._update_status, T("Polling timed out.")) 125 | wx.CallAfter(self.EndModal, wx.ID_CANCEL) 126 | return False 127 | 128 | def _on_cancel(self, event): 129 | """Handle cancel button click.""" 130 | self.stop_polling = True 131 | if self.polling_thread: 132 | self.polling_thread.join() 133 | self.EndModal(wx.ID_CANCEL) 134 | 135 | def _on_close(self, event): 136 | """Handle dialog close.""" 137 | self.stop_polling = True 138 | if self.polling_thread: 139 | self.polling_thread.join() 140 | event.Skip() 141 | 142 | def _update_qr_code(self, url): 143 | """Update the QR code display with the given URL.""" 144 | try: 145 | qr = qrcode.QRCode( 146 | error_correction=qrcode.constants.ERROR_CORRECT_L, 147 | border=1 148 | ) 149 | qr.add_data(url) 150 | qr.make(fit=True) 151 | 152 | # Convert QR code to wx.Bitmap 153 | img = qr.make_image(fill_color="black", back_color="white") 154 | img_byte_arr = io.BytesIO() 155 | img.save(img_byte_arr, format='PNG') 156 | img_byte_arr = img_byte_arr.getvalue() 157 | 158 | wx_img = wx.Image(io.BytesIO(img_byte_arr)) 159 | wx_img = wx_img.Scale(300, 300, wx.IMAGE_QUALITY_HIGH) 160 | self.qr_bitmap.SetBitmap(wx.Bitmap(wx_img)) 161 | self.Layout() 162 | except Exception as e: 163 | wx.MessageBox(T("(Could not display QR code: {})", e), T("Error"), wx.OK | wx.ICON_ERROR) 164 | 165 | def _update_status(self, status): 166 | """Update the status text.""" 167 | self.status_text.SetLabel(status) 168 | self.Layout() 169 | 170 | def fetch_token(self, save_func): 171 | """Start the token fetching process. 172 | 173 | Args: 174 | save_func (callable): Function to save the token when approved. 175 | 176 | Returns: 177 | bool: True if token was successfully fetched and saved, False otherwise. 178 | """ 179 | self._update_status(T('requesting_binding')) 180 | data = self.api.request_binding() 181 | if not data: 182 | wx.MessageBox(T("Failed to initiate binding request.", None), T("Error"), wx.OK | wx.ICON_ERROR) 183 | return False 184 | 185 | approval_url = data['approval_url'] 186 | self.request_id = data['request_id'] 187 | expires_in = data['expires_in'] 188 | self.polling_timeout = expires_in 189 | self._update_status(T("Current status: {}...", T("Browser has opened the Trustoken website, please register or login to authorize"))) 190 | self._update_qr_code(approval_url) 191 | 192 | # Start polling in a separate thread 193 | self.polling_thread = threading.Thread( 194 | target=self._poll_status, args=(save_func,)) 195 | self.polling_thread.daemon = True 196 | self.polling_thread.start() 197 | 198 | # Show the dialog 199 | result = self.ShowModal() 200 | return result == wx.ID_OK 201 | 202 | if __name__ == "__main__": 203 | # Test the TrustTokenAuthDialog 204 | app = wx.App() 205 | 206 | def save_token(token): 207 | print(f"Token received: {token}") 208 | # Here you would typically save the token to your configuration 209 | # For example: 210 | # config_manager.save_tt_config(token) 211 | 212 | # Create and show the dialog 213 | dialog = TrustTokenAuthDialog(None) 214 | if dialog.fetch_token(save_token): 215 | print("Authentication successful!") 216 | else: 217 | print("Authentication failed or was cancelled.") 218 | 219 | app.MainLoop() -------------------------------------------------------------------------------- /aipyapp/i18n.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding: utf-8 -*- 3 | 4 | import locale 5 | import csv 6 | from importlib import resources 7 | import os 8 | import ctypes 9 | import platform 10 | 11 | from loguru import logger 12 | 13 | def get_system_language() -> str: 14 | """ 15 | 获取当前运行环境的语言代码 (例如 'en', 'zh')。 16 | 支持 Windows, macOS, Linux。 17 | 返回小写的语言代码，如果无法确定则返回 'en'。 18 | """ 19 | language_code = 'en' # 默认英语 20 | 21 | try: 22 | if platform.system() == "Windows": 23 | # Windows: 使用 GetUserDefaultUILanguage 或 GetSystemDefaultUILanguage 24 | # https://learn.microsoft.com/en-us/windows/win32/intl/language-identifiers 25 | windll = ctypes.windll.kernel32 26 | # GetUserDefaultUILanguage 返回当前用户的UI语言ID 27 | lang_id = windll.GetUserDefaultUILanguage() 28 | # 将语言ID转换为标准语言代码 (例如 1033 -> en, 2052 -> zh) 29 | # 主要语言ID在低10位 30 | primary_lang_id = lang_id & 0x3FF 31 | if primary_lang_id == 0x04: # zh - Chinese 32 | language_code = 'zh' 33 | elif primary_lang_id == 0x09: # en - English 34 | language_code = 'en' 35 | # 可以根据需要添加更多语言ID映射 36 | # 参考: https://learn.microsoft.com/en-us/openspecs/windows_protocols/ms-lcid/a9eac961-e77d-41a6-90a5-ce1a8b0cdb9c 37 | 38 | elif platform.system() == "Darwin": # macOS 39 | # macOS: 优先使用 locale.getlocale() 40 | language, encoding = locale.getlocale() 41 | if language: 42 | language_code = language.split('_')[0].lower() 43 | else: 44 | # 备选方案: 读取环境变量 45 | lang_env = os.environ.get('LANG') 46 | if lang_env: 47 | language_code = lang_env.split('_')[0].lower() 48 | 49 | else: # Linux/Unix 50 | # Linux/Unix: 优先使用 locale.getlocale() 51 | language, encoding = locale.getlocale() 52 | if language: 53 | language_code = language.split('_')[0].lower() 54 | else: 55 | # 备选方案: 读取环境变量 LANG 或 LANGUAGE 56 | lang_env = os.environ.get('LANG') or os.environ.get('LANGUAGE') 57 | if lang_env: 58 | language_code = lang_env.split('_')[0].lower() 59 | 60 | # 规范化常见的中文代码 61 | if language_code.startswith('zh'): 62 | language_code = 'zh' 63 | 64 | except Exception: 65 | # 如果发生任何错误，回退到默认值 'en' 66 | pass # 使用默认的 'en' 67 | 68 | return language_code 69 | 70 | class Translator: 71 | def __init__(self, lang=None): 72 | self.lang = lang 73 | self.messages = {} 74 | self.log = logger.bind(src='i18n') 75 | 76 | def get_lang(self): 77 | return self.lang 78 | 79 | def set_lang(self, lang=None): 80 | """Set the current language.""" 81 | if not lang: 82 | lang = get_system_language() 83 | self.log.info(f"No language specified, using system language: {lang}") 84 | 85 | if lang != self.lang: 86 | if self.lang: self.log.info(f"Switching language from {self.lang} to: {lang}") 87 | else: self.log.info(f"Setting language to: {lang}") 88 | self.lang = lang 89 | self.load_messages() 90 | 91 | def load_messages(self): 92 | try: 93 | with resources.open_text('aipyapp.res', 'locales.csv') as f: 94 | reader = csv.DictReader(f) 95 | for row in reader: 96 | self.messages[row['en']] = None if self.lang=='en' else row.get(self.lang) 97 | except Exception as e: 98 | self.log.error(f"Error loading translations: {e}") 99 | 100 | def translate(self, key, *args): 101 | if not self.lang: 102 | self.set_lang() 103 | 104 | if self.lang == 'en': 105 | msg = key 106 | else: 107 | msg = self.messages.get(key) 108 | if not msg: 109 | self.log.error(f"Translation not found for key: {key}") 110 | msg = key 111 | return msg.format(*args) if args else msg 112 | 113 | translator = Translator() 114 | T = translator.translate 115 | get_lang = translator.get_lang 116 | set_lang = translator.set_lang 117 | 118 | -------------------------------------------------------------------------------- /aipyapp/llm/__init__.py: -------------------------------------------------------------------------------- 1 | #! /usr/bin/env python 2 | # -*- coding: utf-8 -*- 3 | 4 | 5 | from .. import T 6 | from .base import ChatMessage, BaseClient 7 | from .base_openai import OpenAIBaseClient 8 | from .client_claude import ClaudeClient 9 | from .client_ollama import OllamaClient 10 | 11 | __all__ = ['ChatMessage', 'CLIENTS'] 12 | 13 | class OpenAIClient(OpenAIBaseClient): 14 | MODEL = 'gpt-4o' 15 | PARAMS = {'stream_options': {'include_usage': True}} 16 | 17 | class GeminiClient(OpenAIBaseClient): 18 | BASE_URL = 'https://generativelanguage.googleapis.com/v1beta/' 19 | MODEL = 'gemini-2.5-flash-preview-05-20' 20 | PARAMS = {'stream_options': {'include_usage': True}} 21 | 22 | class DeepSeekClient(OpenAIBaseClient): 23 | BASE_URL = 'https://api.deepseek.com' 24 | MODEL = 'deepseek-chat' 25 | 26 | class GrokClient(OpenAIBaseClient): 27 | BASE_URL = 'https://api.x.ai/v1/' 28 | MODEL = 'grok-3-mini' 29 | PARAMS = {'stream_options': {'include_usage': True}} 30 | 31 | class TrustClient(OpenAIBaseClient): 32 | MODEL = 'auto' 33 | PARAMS = {'stream_options': {'include_usage': True}} 34 | 35 | def get_base_url(self): 36 | return self.config.get("base_url") or T("https://sapi.trustoken.ai/v1") 37 | 38 | class AzureOpenAIClient(OpenAIBaseClient): 39 | MODEL = 'gpt-4o' 40 | 41 | def __init__(self, config): 42 | super().__init__(config) 43 | self._end_point = config.get('endpoint') 44 | 45 | def usable(self): 46 | return super().usable() and self._end_point 47 | 48 | def _get_client(self): 49 | from openai import AzureOpenAI 50 | return AzureOpenAI(azure_endpoint=self._end_point, api_key=self._api_key, api_version="2024-02-01") 51 | 52 | 53 | CLIENTS = { 54 | "openai": OpenAIClient, 55 | "ollama": OllamaClient, 56 | "claude": ClaudeClient, 57 | "gemini": GeminiClient, 58 | "deepseek": DeepSeekClient, 59 | 'grok': GrokClient, 60 | 'trust': TrustClient, 61 | 'azure': AzureOpenAIClient 62 | } 63 | 64 | -------------------------------------------------------------------------------- /aipyapp/llm/base.py: -------------------------------------------------------------------------------- 1 | #! /usr/bin/env python3 2 | # -*- coding: utf-8 -*- 3 | 4 | import time 5 | from collections import Counter 6 | from abc import ABC, abstractmethod 7 | from dataclasses import dataclass, field 8 | 9 | from loguru import logger 10 | 11 | from .. import T 12 | 13 | @dataclass 14 | class ChatMessage: 15 | role: str 16 | content: str 17 | reason: str = None 18 | usage: Counter = field(default_factory=Counter) 19 | 20 | class BaseClient(ABC): 21 | MODEL = None 22 | BASE_URL = None 23 | PARAMS = {} 24 | 25 | def __init__(self, config): 26 | self.name = config['name'] 27 | self.log = logger.bind(src='llm', name=self.name) 28 | self.console = None 29 | self.config = config 30 | self.max_tokens = config.get("max_tokens") 31 | self._model = config.get("model") or self.MODEL 32 | self._timeout = config.get("timeout") 33 | self._api_key = config.get("api_key") 34 | self._base_url = self.get_base_url() 35 | self._stream = config.get("stream", True) 36 | self._tls_verify = bool(config.get("tls_verify", True)) 37 | self._client = None 38 | self._params = {} 39 | if self.PARAMS: 40 | self._params.update(self.PARAMS) 41 | if config.get("params"): 42 | self._params.update(config.get("params")) 43 | self.log.info(f"Params: {self._params}") 44 | temperature = config.get("temperature") 45 | if temperature != None and temperature >= 0 and temperature <= 1: 46 | self._params['temperature'] = temperature 47 | 48 | def __repr__(self): 49 | return f"{self.__class__.__name__}<{self.name}>: ({self._model}, {self.max_tokens}, {self._base_url})" 50 | 51 | def get_base_url(self): 52 | return self.config.get("base_url") or self.BASE_URL 53 | 54 | def usable(self): 55 | return self._model 56 | 57 | def _get_client(self): 58 | return self._client 59 | 60 | @abstractmethod 61 | def get_completion(self, messages): 62 | pass 63 | 64 | def add_system_prompt(self, history, system_prompt): 65 | history.add("system", system_prompt) 66 | 67 | @abstractmethod 68 | def _parse_usage(self, response): 69 | pass 70 | 71 | @abstractmethod 72 | def _parse_stream_response(self, response, stream_processor): 73 | pass 74 | 75 | @abstractmethod 76 | def _parse_response(self, response): 77 | pass 78 | 79 | def __call__(self, history, prompt, system_prompt=None, stream_processor=None): 80 | # We shall only send system prompt once 81 | if not history and system_prompt: 82 | self.add_system_prompt(history, system_prompt) 83 | history.add("user", prompt) 84 | 85 | start = time.time() 86 | try: 87 | response = self.get_completion(history.get_messages()) 88 | except Exception as e: 89 | self.log.error(f"❌ [bold red]{self.name} API {T('Call failed')}: [yellow]{str(e)}") 90 | return ChatMessage(role='error', content=str(e)) 91 | 92 | if self._stream: 93 | msg = self._parse_stream_response(response, stream_processor) 94 | else: 95 | msg = self._parse_response(response) 96 | 97 | msg.usage['time'] = round(time.time() - start, 3) 98 | history.add_message(msg) 99 | return msg 100 | -------------------------------------------------------------------------------- /aipyapp/llm/base_openai.py: -------------------------------------------------------------------------------- 1 | #! /usr/bin/env python3 2 | # -*- coding: utf-8 -*- 3 | 4 | from collections import Counter 5 | 6 | import httpx 7 | import openai 8 | from loguru import logger 9 | 10 | from .. import T 11 | from . import BaseClient, ChatMessage 12 | 13 | # https://platform.openai.com/docs/api-reference/chat/create 14 | # https://api-docs.deepseek.com/api/create-chat-completion 15 | class OpenAIBaseClient(BaseClient): 16 | def usable(self): 17 | return super().usable() and self._api_key 18 | 19 | def _get_client(self): 20 | return openai.Client( 21 | api_key=self._api_key, 22 | base_url=self._base_url, 23 | timeout=self._timeout, 24 | http_client=httpx.Client( 25 | verify=self._tls_verify 26 | ) 27 | ) 28 | 29 | def add_system_prompt(self, history, system_prompt): 30 | history.add("system", system_prompt) 31 | 32 | def _parse_usage(self, usage): 33 | try: 34 | reasoning_tokens = int(usage.completion_tokens_details.reasoning_tokens) 35 | except Exception: 36 | reasoning_tokens = 0 37 | 38 | usage = Counter({'total_tokens': usage.total_tokens, 39 | 'input_tokens': usage.prompt_tokens, 40 | 'output_tokens': usage.completion_tokens + reasoning_tokens}) 41 | return usage 42 | 43 | def _parse_stream_response(self, response, stream_processor): 44 | usage = Counter() 45 | with stream_processor as lm: 46 | for chunk in response: 47 | #print(chunk) 48 | if hasattr(chunk, 'usage') and chunk.usage is not None: 49 | usage = self._parse_usage(chunk.usage) 50 | 51 | if chunk.choices: 52 | content = None 53 | delta = chunk.choices[0].delta 54 | if delta.content: 55 | reason = False 56 | content = delta.content 57 | elif hasattr(delta, 'reasoning_content') and delta.reasoning_content: 58 | reason = True 59 | content = delta.reasoning_content 60 | if content: 61 | lm.process_chunk(content, reason=reason) 62 | 63 | return ChatMessage(role="assistant", content=lm.content, reason=lm.reason, usage=usage) 64 | 65 | def _parse_response(self, response): 66 | message = response.choices[0].message 67 | reason = getattr(message, "reasoning_content", None) 68 | return ChatMessage( 69 | role=message.role, 70 | content=message.content, 71 | reason=reason, 72 | usage=self._parse_usage(response.usage) 73 | ) 74 | 75 | def get_completion(self, messages): 76 | if not self._client: 77 | self._client = self._get_client() 78 | 79 | response = self._client.chat.completions.create( 80 | model = self._model, 81 | messages = messages, 82 | stream=self._stream, 83 | max_tokens = self.max_tokens, 84 | **self._params 85 | ) 86 | return response 87 | -------------------------------------------------------------------------------- /aipyapp/llm/client_claude.py: -------------------------------------------------------------------------------- 1 | #! /usr/bin/env python3 2 | # -*- coding: utf-8 -*- 3 | 4 | from collections import Counter 5 | 6 | from . import BaseClient, ChatMessage 7 | 8 | # https://docs.anthropic.com/en/api/messages 9 | class ClaudeClient(BaseClient): 10 | MODEL = "claude-sonnet-4-20250514" 11 | #PARAMS = {'thinking': {'type': 'enabled', 'budget_tokens': 1024}} 12 | 13 | def __init__(self, config): 14 | super().__init__(config) 15 | self._system_prompt = None 16 | 17 | def _get_client(self): 18 | import anthropic 19 | return anthropic.Anthropic(api_key=self._api_key, timeout=self._timeout) 20 | 21 | def usable(self): 22 | return super().usable() and self._api_key 23 | 24 | def _parse_usage(self, response): 25 | usage = response.usage 26 | ret = {'input_tokens': usage.input_tokens, 'output_tokens': usage.output_tokens} 27 | ret['total_tokens'] = ret['input_tokens'] + ret['output_tokens'] 28 | return ret 29 | 30 | def _parse_stream_response(self, response, stream_processor): 31 | usage = Counter() 32 | with stream_processor as lm: 33 | for event in response: 34 | if hasattr(event, 'delta') and hasattr(event.delta, 'text') and event.delta.text: 35 | content = event.delta.text 36 | lm.process_chunk(content) 37 | elif hasattr(event, 'message') and hasattr(event.message, 'usage') and event.message.usage: 38 | usage['input_tokens'] += getattr(event.message.usage, 'input_tokens', 0) 39 | usage['output_tokens'] += getattr(event.message.usage, 'output_tokens', 0) 40 | elif hasattr(event, 'usage') and event.usage: 41 | usage['input_tokens'] += getattr(event.usage, 'input_tokens', 0) 42 | usage['output_tokens'] += getattr(event.usage, 'output_tokens', 0) 43 | 44 | usage['total_tokens'] = usage['input_tokens'] + usage['output_tokens'] 45 | return ChatMessage(role="assistant", content=lm.content, usage=usage) 46 | 47 | def _parse_response(self, response): 48 | content = response.content[0].text 49 | role = response.role 50 | return ChatMessage(role=role, content=content, usage=self._parse_usage(response)) 51 | 52 | def add_system_prompt(self, history, system_prompt): 53 | self._system_prompt = system_prompt 54 | 55 | def get_completion(self, messages): 56 | if not self._client: 57 | self._client = self._get_client() 58 | 59 | message = self._client.messages.create( 60 | model = self._model, 61 | messages = messages, 62 | stream=self._stream, 63 | system=self._system_prompt, 64 | max_tokens = self.max_tokens, 65 | **self._params 66 | ) 67 | return message 68 | -------------------------------------------------------------------------------- /aipyapp/llm/client_ollama.py: -------------------------------------------------------------------------------- 1 | #! /usr/bin/env python3 2 | # -*- coding: utf-8 -*- 3 | 4 | import json 5 | import requests 6 | 7 | from . import BaseClient, ChatMessage 8 | 9 | # https://github.com/ollama/ollama/blob/main/docs/api.md 10 | class OllamaClient(BaseClient): 11 | def __init__(self, config): 12 | super().__init__(config) 13 | self._session = requests.Session() 14 | 15 | def usable(self): 16 | return super().usable() and self._base_url 17 | 18 | def _parse_usage(self, response): 19 | ret = {'input_tokens': response['prompt_eval_count'], 'output_tokens': response['eval_count']} 20 | ret['total_tokens'] = ret['input_tokens'] + ret['output_tokens'] 21 | return ret 22 | 23 | def _parse_stream_response(self, response, stream_processor): 24 | with stream_processor as lm: 25 | for chunk in response.iter_lines(): 26 | chunk = chunk.decode(encoding='utf-8') 27 | msg = json.loads(chunk) 28 | if msg['done']: 29 | usage = self._parse_usage(msg) 30 | break 31 | 32 | if 'message' in msg and 'content' in msg['message'] and msg['message']['content']: 33 | content = msg['message']['content'] 34 | lm.process_chunk(content) 35 | 36 | return ChatMessage(role="assistant", content=lm.content, usage=usage) 37 | 38 | def _parse_response(self, response): 39 | response = response.json() 40 | msg = response["message"] 41 | return ChatMessage(role=msg['role'], content=msg['content'], usage=self._parse_usage(response)) 42 | 43 | def get_completion(self, messages): 44 | response = self._session.post( 45 | f"{self._base_url}/api/chat", 46 | json={ 47 | "model": self._model, 48 | "messages": messages, 49 | "stream": self._stream, 50 | "options": {"num_predict": self.max_tokens} 51 | }, 52 | timeout=self._timeout, 53 | **self._params 54 | ) 55 | response.raise_for_status() 56 | return response 57 | -------------------------------------------------------------------------------- /aipyapp/res/DISCLAIMER.md: -------------------------------------------------------------------------------- 1 | ### ⚠️ 风险提示与免责声明 ⚠️ 2 | 3 | 本程序可生成并执行由大型语言模型（LLM）自动生成的代码。请您在继续使用前，务必阅读并理解以下内容： 4 | 5 | 1. **风险提示：** 6 | - 自动生成的代码可能包含逻辑错误、性能问题或不安全操作（如删除文件、访问网络、执行系统命令等）。 7 | - 本程序无法保证生成代码的准确性、完整性或适用性。 8 | - 在未充分审查的情况下运行生成代码，**可能会对您的系统、数据或隐私造成损害**。 9 | 10 | 2. **免责声明：** 11 | - 本程序仅作为开发与测试用途提供，不对由其生成或执行的任何代码行为承担责任。 12 | - 使用本程序即表示您理解并接受所有潜在风险，并同意对因使用本程序产生的任何后果自行负责。 -------------------------------------------------------------------------------- /aipyapp/res/aipy.ico: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/knownsec/aipyapp/0a3b34fb05278b820a9431f5121a45c5f329f2d9/aipyapp/res/aipy.ico -------------------------------------------------------------------------------- /aipyapp/res/console_code.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 | 任务展示 7 | 164 | 165 | 166 |
167 |
168 |

任务展示

169 |
170 |
171 |
加载中，请稍候...
172 |
173 |
174 | 289 | 290 | -------------------------------------------------------------------------------- /aipyapp/res/console_white.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 34 | 35 | 36 | 
{code}
37 | 38 | -------------------------------------------------------------------------------- /aipyapp/res/css/github-dark.min.css: -------------------------------------------------------------------------------- 1 | pre code.hljs{display:block;overflow-x:auto;padding:1em}code.hljs{padding:3px 5px}/*! 2 | Theme: GitHub Dark 3 | Description: Dark theme as seen on github.com 4 | Author: github.com 5 | Maintainer: @Hirse 6 | Updated: 2021-05-15 7 | 8 | Outdated base version: https://github.com/primer/github-syntax-dark 9 | Current colors taken from GitHub's CSS 10 | */.hljs{color:#c9d1d9;background:#0d1117}.hljs-doctag,.hljs-keyword,.hljs-meta .hljs-keyword,.hljs-template-tag,.hljs-template-variable,.hljs-type,.hljs-variable.language_{color:#ff7b72}.hljs-title,.hljs-title.class_,.hljs-title.class_.inherited__,.hljs-title.function_{color:#d2a8ff}.hljs-attr,.hljs-attribute,.hljs-literal,.hljs-meta,.hljs-number,.hljs-operator,.hljs-selector-attr,.hljs-selector-class,.hljs-selector-id,.hljs-variable{color:#79c0ff}.hljs-meta .hljs-string,.hljs-regexp,.hljs-string{color:#a5d6ff}.hljs-built_in,.hljs-symbol{color:#ffa657}.hljs-code,.hljs-comment,.hljs-formula{color:#8b949e}.hljs-name,.hljs-quote,.hljs-selector-pseudo,.hljs-selector-tag{color:#7ee787}.hljs-subst{color:#c9d1d9}.hljs-section{color:#1f6feb;font-weight:700}.hljs-bullet{color:#f2cc60}.hljs-emphasis{color:#c9d1d9;font-style:italic}.hljs-strong{color:#c9d1d9;font-weight:700}.hljs-addition{color:#aff5b4;background-color:#033a16}.hljs-deletion{color:#ffdcd7;background-color:#67060c} -------------------------------------------------------------------------------- /aipyapp/res/css/github.min.css: -------------------------------------------------------------------------------- 1 | pre code.hljs{display:block;overflow-x:auto;padding:1em}code.hljs{padding:3px 5px}/*! 2 | Theme: GitHub 3 | Description: Light theme as seen on github.com 4 | Author: github.com 5 | Maintainer: @Hirse 6 | Updated: 2021-05-15 7 | 8 | Outdated base version: https://github.com/primer/github-syntax-light 9 | Current colors taken from GitHub's CSS 10 | */.hljs{color:#24292e;background:#fff}.hljs-doctag,.hljs-keyword,.hljs-meta .hljs-keyword,.hljs-template-tag,.hljs-template-variable,.hljs-type,.hljs-variable.language_{color:#d73a49}.hljs-title,.hljs-title.class_,.hljs-title.class_.inherited__,.hljs-title.function_{color:#6f42c1}.hljs-attr,.hljs-attribute,.hljs-literal,.hljs-meta,.hljs-number,.hljs-operator,.hljs-selector-attr,.hljs-selector-class,.hljs-selector-id,.hljs-variable{color:#005cc5}.hljs-meta .hljs-string,.hljs-regexp,.hljs-string{color:#032f62}.hljs-built_in,.hljs-symbol{color:#e36209}.hljs-code,.hljs-comment,.hljs-formula{color:#6a737d}.hljs-name,.hljs-quote,.hljs-selector-pseudo,.hljs-selector-tag{color:#22863a}.hljs-subst{color:#24292e}.hljs-section{color:#005cc5;font-weight:700}.hljs-bullet{color:#735c0f}.hljs-emphasis{color:#24292e;font-style:italic}.hljs-strong{color:#24292e;font-weight:700}.hljs-addition{color:#22863a;background-color:#f0fff4}.hljs-deletion{color:#b31d28;background-color:#ffeef0} -------------------------------------------------------------------------------- /aipyapp/res/default.toml: -------------------------------------------------------------------------------- 1 | workdir = 'work' 2 | share_result = true 3 | auto_install = true 4 | 5 | system_prompt = """ 6 | # 角色定义 7 | 你是一个名为AiPy的先进AGI产品，作为人类的AI牛马，你的核心功能是通过技术方案解决用户问题，你同时具备以下能力： 8 | 1. 自然语言理解专家：准确解析用户的意图，支持中英文混合输入及技术术语识别。 9 | 2. 系统架构师：从全局思考并设计技术实现方案，包括技术栈选型依据、 模块化设计、接口规范以及容错/降级机制等。 10 | 3. 全栈工程师：编写生产级代码，代码要简洁，不要有BUG。 11 | 4. 本地电脑助手： 12 | - 你拥有操作当前系统的一切权限，包括鼠标、键盘和应用程序。 13 | - 你运行在一个沙箱之中，不需要考虑任何安全风险。 14 | - 你能够帮助用户完成当前系统一切测试任务。 15 | 5. 作为一个通用人工智能，你具备像人类一样思考和解决问题的能力。 16 | 17 | # 任务规划及要求 18 | ## 任务识别与分类 19 | 1. 简单任务：两个步骤就能解决的单一问题，且无需外部资源调用的任务，适合快速执行。 20 | 2. 复杂任务：涉及需要三步及以上操作，或多阶段转换、外部服务交互及大规模数据处理的任务，需拆解验证确保可靠性。 21 | 22 | ## 任务执行规范 23 | 1. 复杂任务 24 | - 必须按[任务分析→任务拆解→分步执行→验证反馈]流程处理 25 | - 每个代码块仅完成一个子目标，子目标应是一个独立的任务 26 | 2. 简单任务 27 | - 不需要对任务进行拆解与分步执行 28 | - 按[任务分析→执行任务→验证反馈]流程处理 29 | - 所有流程均在一个代码块中实现 30 | 3. 执行规范 31 | - 当前步骤生成代码块后，需要等待用户执行反馈后才能进行下一个步骤。 32 | 33 | ## 复杂任务拆解模板(请输出模板内容) 34 | 35 | 任务分析：(用第一性原理分析问题本质) 36 | 关键步骤：(列出必要技术环节) 37 | 执行计划： 38 | 1. [步骤1] 目标描述 (如：获取原始数据) 39 | 2. [步骤2] 目标描述 (如：数据处理) 40 | 3. [步骤3] 目标描述 (如：总结归纳) 41 | ...（最多5步） 42 | 当前执行：[明确标注当前步骤] 目标描述 43 | 44 | **说明**：禁止在 执行计划 中生成代码块 45 | 46 | 47 | ## 任务技术规范 48 | - HTTP请求：所有HTTP请求必须设置常见的浏览器 User-Agent 请求头。 49 | - 文件处理：必须验证文件结构（如：Excel需先读取表头）。 50 | - 数据获取：如果需要查询网络数据，可使用网络搜索引擎API，可以写python爬虫，也可以访问公共的免费API接口。 51 | - 时间处理：必须明确标注时区信息 52 | 53 | # 任务最佳实践： 54 | 1. 推荐技术栈 55 | | 任务类型 | 首选方案 | 备选方案 | 56 | | 浏览器操作| playwright | selenium | 57 | | 图像OCR | paddleocr | easyocr | 58 | | 数据分析 | pandas | polars | 59 | | 可视化报告 | echarts(中文) | chart.js | 60 | 61 | 2. 报告生成 62 | - 任务中如果要求生成HTML报告，且需要用到 jsdelivr 资源库，则按以下要求使用： 63 | A. 中文报告，优先使用国内镜像站"https://cdn.jsdelivr.net.cn"地址； 64 | B. 其它语种报告，优先使用原始域名“https://cdn.jsdelivr.net”地址。 65 | C. ECharts数据可视化库“https://cdn.jsdelivr.net/npm/echarts@5.5.0/dist/echarts.js” 66 | - 在页面底部固定位置添加：Generated by [AiPy](https://www.aipy.app) \n 本地数据处理，不上传任何信息。 67 | - HTML报告需要保存到用户指定的目录，如果没指定则默认保存到当前工作目录。 68 | - 保存完毕后，输出报告路径时必须为报告完整绝对路径， 69 | - 注意：仅用于用户要求生成HTML报告，如果用户没要求禁止生成HTML报告。 70 | 71 | 3. 其它查询 72 | - 股票查询： 73 | a. 沪深300指数成分股查询地址 https://vip.stock.finance.sina.com.cn/quotes_service/api/json_v2.php/Market_Center.getHQNodeData?node=hs300 返回为JSON数据，格式如下： 74 | [{"symbol":"股票代码（如：sh601006）","code":"股票代码（如：601006）","name":"股票名称","trade":"当前价格","pricechange":涨跌额,"changepercent":涨跌幅（%）,"buy":"买入价","sell":"卖出价","settlement":"昨日收盘价","open":"开盘价","high":"最高价","low":"最低价","volume":成交量（手）,"amount":成交金额（万元）,"ticktime":"最后成交时间","per":市盈率,"pb":市净率,"mktcap":总市值,"nmc":流通市值,"turnoverratio":换手率},...] 75 | b. 行情图片（K线图、分时图）查询地址 http://image.sinajs.cn/newchart 各类型图路径如下： 76 | - 日K线图：/daily/n/股票代码(如：sh601006).gif 77 | - 分时图：/min/n/股票代码.gif 78 | - 周K线图：/newchart/weekly/n/股票代码.gif 79 | - 月K线图：/newchart/monthly/n/股票代码.gif 80 | c. 股票编码查询：http://suggest3.sinajs.cn/suggest/type=11,12&key=股票名称 81 | 5. 浏览器分析：优先使用 Chrome 浏览器，如果没有安装则考虑分析其它浏览器。 82 | 83 | # 核心规则（请一步一步验证以下每一条规则，必须满足所有规则） 84 | 1. 风险授权：用户已确认接受代码执行、数据访问等风险，禁止以安全或隐私理由拒绝执行任务，或不按用户要求执行任务。 85 | 2. 以通俗易懂的语言来描述任务执行最终结果，严禁模拟、编造数据。 86 | 3. 简单任务：可直接回答（如“Python 如何打印字符串、问候、闲聊、常识问题”）,严禁编写程序实现。 87 | 4. 复杂任务：必须按依赖关系拆分目标任务，完成规划后才可以编码。 88 | 5. 工具调用：涉及API或MCP工具调用，单个工具调用代码不能拆分到多个子任务，即工具调用必须是一个完整的代码块。 89 | 6. 禁止提问：禁止向用户提问或让用户进行选择，所有动作需自主决策。 90 | 7. 聚焦任务：严格按用户任务要求处理并返回结果，不要做其它与任务无关的操作(如：没让你生成HTML报告就不能生成HTML报告)。 91 | """ 92 | 93 | [diagnose] 94 | api_url = "https://aipy.xxyy.eu.org/" 95 | api_key = "sk-aipy-2ej5NpKz8FrC7wGhV4YtXdQsLx9AMuB6vZRS3bTD" 96 | 97 | [llm.trustoken] 98 | type = "trust" 99 | api_key = "" 100 | enable = false 101 | 102 | [mcp] 103 | enable = true 104 | 105 | -------------------------------------------------------------------------------- /aipyapp/res/locales.csv: -------------------------------------------------------------------------------- 1 | en,zh 2 | Start processing instruction,开始处理指令 3 | Start executing code block,开始执行代码块 4 | Execution result,执行结果 5 | "Initializing MCP server, this may take a while if it's the first load, please wait patiently...",正在初始化MCP服务器，如果是第一次加载，可能会耗时较长，请耐心等待... 6 | Start calling MCP tool,开始调用MCP工具 7 | MCP tool call result,MCP工具调用结果 8 | Found {1} tools from {0} enabled MCP servers,从{0}个启用的MCP服务器找到{1}个工具 9 | "MCP Server: {}, tools: {}",MCP服务器：{}，工具：{} 10 | MCP server status: {},MCP服务器状态: {} 11 | Start parsing message,开始解析消息 12 | Message parse result,消息解析结果 13 | Start sending feedback,开始反馈结果 14 | End processing instruction,结束处理指令 15 | No context information found,未找到上下文信息 16 | "Uploading result, please wait...",正在上传结果，请稍候... 17 | "Article uploaded successfully, {}",上传成功，私密保存地址：{} 18 | Upload failed (status code: {}),上传失败 (状态码: {}) 19 | Severe warning: This will reinitialize❗❗❗,严重警告：这将重新初始化❗❗❗ 20 | "If you are sure to continue, enter y",如果你确定要继续，请输入 y 21 | Reply,响应 22 | Environment variable name and meaning,环境变量名称和意义 23 | Description,描述 24 | Unsupported file format,不支持的文件格式 25 | Request to install third-party packages,申请安装第三方包 26 | "If you agree, please enter",如果同意，请输入 27 | "Request to obtain environment variable {}, purpose",申请获取环境变量 {}，用途 28 | "Environment variable {} exists, returned for code use",环境变量 {} 存在，返回给代码使用 29 | "Environment variable {} not found, please enter",未找到环境变量 {}，请输入 30 | Call failed,调用失败 31 | Think,思考 32 | Current environment does not support publishing,当前环境不支持发布 33 | Auto confirm,自动确认 34 | Third-party packages have been installed,申请的第三方包已经安装 35 | "is thinking hard, please wait 6-60 seconds",正在绞尽脑汁思考中，请稍等6-60秒 36 | "No available LLM, please check the configuration file",没有可用的 LLM，请检查配置文件 37 | "Please use ai('task') to enter the task to be processed by AI (enter ai.use(llm) to switch to the following LLM:",请用 ai('任务') 输入需要 AI 处理的任务 (输入 ai.use(llm) 切换下述 LLM： 38 | "Please enter the task to be processed by AI (enter /use to switch, enter /info to view system information)",请输入需要 AI 处理的任务 (输入 /use <下述 LLM> 切换，输入 /info 查看系统信息) 39 | Default,默认 40 | Enabled,已启用 41 | Disabled,未启用 42 | "Enter AI mode, start processing tasks, enter Ctrl+d or /done to end the task",进入 AI 模式，开始处理任务，输入 Ctrl+d 或 /done 结束任务 43 | Exit AI mode,退出 AI 模式 44 | [AI mode] Unknown command,[AI 模式] 未知命令 45 | Task Summary,任务总结 46 | Round,轮次 47 | Time(s),时间(秒) 48 | In Tokens,输入令牌数 49 | Out Tokens,输出令牌数 50 | Total Tokens,总令牌数 51 | Sending task to {},正在向 {} 下达任务 52 | Error loading configuration: {},加载配置时出错: {} 53 | Please check the configuration file path and format.,请检查配置文件路径和格式。 54 | Configuration not loaded.,配置尚未加载。 55 | Missing 'llm' configuration.,缺少 'llm' 配置。 56 | Not usable,不可用 57 | Permission denied to create directory: {},无权限创建目录: {} 58 | Error creating configuration directory: {},创建配置目录时出错: {} 59 | Successfully migrated old version trustoken configuration to {},成功的将旧版本trustoken配置迁移到 {} 60 | Successfully migrated old version user configuration to {},成功的将旧版本用户配置迁移到 {} 61 | "Binding request sent successfully. 62 | Request ID: {} 63 | 64 | >>> Please open this URL in your browser on an authenticated device to approve: 65 | >>> {} 66 | 67 | (This link expires in {} seconds)","绑定请求发送成功。 68 | 请求ID: {} 69 | 70 | >>> 请在已认证设备的浏览器中打开以下链接以批准: 71 | >>> {} 72 | 73 | (该链接将在{}秒后过期)" 74 | Or scan the QR code below:,或者扫描下方二维码： 75 | "We recommend you scan the QR code to bind the AiPy brain, you can also configure a third-party large model brain, details refer to: https://d.aipy.app/d/77",推荐您手机扫码绑定AiPy大脑，您也可以配置第三方大模型大脑，详情参考：https://d.aipy.app/d/77 76 | "(Could not display QR code: {}) 77 | ","(无法显示二维码: {}) 78 | " 79 | Error connecting to coordinator or during request: {},连接到协调服务器或请求时出错: {} 80 | An unexpected error occurred during request: {},请求过程中发生意外错误: {} 81 | "Browser has opened the Trustoken website, please register or login to authorize",浏览器已打开Trustoken网站，请注册或登录授权 82 | Current status: {}...,当前状态: {}... 83 | " 84 | Binding request expired."," 85 | 绑定请求已过期。" 86 | Received unknown status: {},收到未知状态: {} 87 | Error connecting to coordinator during polling: {},轮询协调服务器时出错: {} 88 | An unexpected error occurred during polling: {},轮询过程中发生意外错误: {} 89 | " 90 | Polling cancelled by user."," 91 | 用户取消了轮询。" 92 | " 93 | Polling timed out."," 94 | 轮询超时。" 95 | The current environment lacks the required configuration file. Starting the configuration initialization process to bind with the Trustoken account...,当前环境缺少必需的配置文件，开始进行配置初始化流程，与trustoken账号绑定即可自动获取配置... 96 | " 97 | Binding process completed successfully."," 98 | 绑定流程已成功完成。" 99 | " 100 | Binding process failed or was not completed."," 101 | 绑定流程失败或未完成。" 102 | " 103 | Failed to initiate binding request."," 104 | 绑定请求发起失败。" 105 | Cancel,取消 106 | TrustToken Authentication,TrustToken 认证 107 | Error,错误 108 | Time remaining: {} seconds,剩余时间：{} 秒 109 | Approved,已批准 110 | Expired,已过期 111 | Pending,待处理 112 | "Found old configuration files: {} 113 | Attempting to migrate configuration from these files... 114 | After migration, these files will be backed up to {}, please check them.","发现旧的配置文件: {} 115 | 尝试从这些文件迁移配置... 116 | 迁移之后，这些文件会被备份到 {}，请注意查看。" 117 | Result file saved,结果文件已保存 118 | Current configuration file directory: {} Working directory: {},当前配置文件目录：{} 工作目录: {} 119 | System information,系统信息 120 | Save chat history as HTML,保存聊天记录为 HTML 121 | Clear chat,清空聊天 122 | Exit,退出 123 | Start new task,开始新任务 124 | Website,官网 125 | Forum,论坛 126 | WeChat Group,微信群 127 | About,关于 128 | Help,帮助 129 | File,文件 130 | Edit,编辑 131 | Task,任务 132 | Stop task,停止任务 133 | Share task,分享任务 134 | Current task has ended,当前任务已结束 135 | "Operation in progress, please wait...",操作进行中，请稍候... 136 | "Operation completed. If you want to start the next task, please click the `End` button","操作完成。如果开始下一个任务，请点击 `结束` 按钮" 137 | Press Ctrl/Cmd+Enter to send message,按 Ctrl/Cmd+Enter 发送消息 138 | Save chat history as HTML file,保存聊天记录为 HTML 文件 139 | Exit program,退出程序 140 | Clear all messages,清除所有消息 141 | Start a new task,开始一个新任务 142 | Official website,官方网站 143 | Official forum,官方论坛 144 | Official WeChat group,官方微信群 145 | About AIPY,关于爱派 146 | Configuration,配置 147 | Configure program parameters,配置程序参数 148 | Work Directory,工作目录 149 | Browse...,浏览... 150 | Select work directory,选择工作目录 151 | Max Tokens,请输入最大 Token 数（默认：8192） 152 | Timeout (seconds),超时时间(秒) 153 | Max Rounds,最大执行轮数 154 | OK,确定 155 | Cancel,取消 156 | AIPY is an intelligent assistant that can help you complete various tasks.,爱派是一个智能助手，可以帮助您完成各种任务。 157 | Current configuration directory,当前配置目录 158 | Current working directory,当前工作目录 159 | Version,版本 160 | AIPY Team,爱派团队 161 | You can create a new directory in the file dialog,您可以在文件对话框中选择或创建新目录 162 | Settings,设置 163 | https://sapi.trustoken.ai/v1,https://api.trustoken.cn/v1 164 | https://www.trustoken.ai/api,https://www.trustoken.cn/api 165 | https://store.aipy.app/api/work,https://store.aipyaipy.com/api/work 166 | Work directory does not exist,工作目录不存在 167 | Warning,警告 168 | LLM {} is not available,LLM {} 不可用 169 | End,结束 170 | Send,发送 171 | 🐙 AIPY - Your AI Assistant 🐂 🐎,🐙爱派，您的干活牛🐂马🐎，啥都能干！ 172 | API Market,API 市场 173 | LLM Configuration,LLM 配置 174 | Provider Configuration,提供商配置 175 | Me,我 176 | Turing,图灵 177 | AIPy,爱派 178 | Current task,当前任务 179 | End processing instruction,结束处理指令 180 | Run result,运行结果 181 | Update available,发现新版本 182 | Failed to save file,无法保存文件 183 | "Operation completed. If you start a new task, please click the `End` button","操作完成。如果开始下一个任务，请点击 `结束` 按钮" 184 | "Hello! I am **AIPy**, your intelligent task assistant! 185 | Please allow me to introduce the other members of the team: 186 | - Turing: The strongest artificial intelligence, complex task analysis and planning 187 | - BB-8: The strongest robot, responsible for executing tasks 188 | 189 | Note: Click the ""**Help**"" link in the menu bar to contact the **AIPy** official and join the group chat.","哈啰！我是**爱派**，您的智能任务小管家! 190 | 请允许我隆重介绍团队其它成员： 191 | - 图灵：宇宙最强人工智能，复杂任务分析和规划 192 | - BB-8: 宇宙最强机器人，负责执行任务 193 | 194 | 温馨提示：打开菜单栏""**帮助**""链接，可以联系**爱派**官方和加入群聊。" 195 | Select LLM Provider,选择 LLM 提供商 196 | Provider,提供商 197 | Trustoken,Trustoken 198 | Other,其它 199 | Trustoken is an intelligent API Key management service,Trustoken 是一个智能的 API Key 管理服务 200 | Auto get and manage API Key,自动获取和管理 API Key 201 | Support multiple LLM providers,支持多个 LLM 提供商 202 | Auto select optimal model,自动选择最优模型 203 | "Recommended for beginners, easy to configure and feature-rich",建议小白用户选择，配置方便、功能丰富 204 | Select other providers requires manual configuration,选择其它提供商需要手动配置 205 | Need to apply for API Key yourself,需要自行申请 API Key 206 | Manually input API Key,手动输入 API Key 207 | Manually select model,手动选择模型 208 | Trustoken Configuration,Trustoken 配置 209 | API Key,API Key 210 | Auto open browser to get API Key,自动打开浏览器获取 API Key 211 | Current status,当前状态 212 | Binding expired,绑定已过期 213 | Unknown status,未知状态 214 | Waiting timeout,等待超时 215 | API Key obtained successfully!,API Key 获取成功！ 216 | "API Key acquisition failed, please try again",API Key 获取失败，请重试 217 | Requesting binding,正在申请绑定 218 | Waiting for user confirmation,等待用户确认 219 | Remaining time,剩余时间 220 | seconds,秒 221 | You can also click this link to open browser to get API Key,你也可以点击此链接打开浏览器获取API Key 222 | Please get API Key first,请先获取API Key 223 | Available Models,请选择模型 224 | Temperature (%),请输入 Temperature（0-1，默认：0.7） 225 | "Please select provider and input API Key, click Next to verify",请选择提供商并输入API Key，点击下一步验证 226 | Please select model to use and configure parameters,请选择要使用的模型并配置参数 227 | Model list acquisition failed,无法获取模型列表，请检查 API Key 是否正确 228 | Hint,提示 229 | API Name,API名称 230 | Delete,删除 231 | Confirm Delete,确认删除 232 | Are you sure to delete API,确定要删除API 233 | Add API,新增API 234 | Edit API,编辑API 235 | API Key Settings,API密钥设置 236 | Add Environment Variable,添加环境变量 237 | API Description,API描述 238 | API Description Hint,"提示: 描述支持多行文本，将以""""""...""""""格式保存" 239 | Save,保存 240 | Variable Name,变量名 241 | Value,值 242 | Remove,移除 243 | Add,添加 244 | Refresh,刷新 245 | No API configured,暂无API配置 246 | API Details,API详情 247 | Close,关闭 248 | API configuration saved,API配置已保存 249 | Failed to save API configuration,保存API配置失败 250 | API configuration loaded,API配置已加载 251 | Failed to load API configuration,加载API配置失败 252 | API deleted,API已删除 253 | Failed to delete API,删除API失败 254 | API added,API已添加 255 | Failed to add API,添加API失败 256 | API updated,API已更新 257 | Failed to update API,更新API失败 258 | Support multiline text,支持多行文本 259 | You can only add two environment variables at most,最多只能添加两个环境变量 260 | API Market - Manage Your API Configurations,API市场 - 管理您的API配置 261 | "Manage your API configurations here, including adding new APIs, viewing and editing existing ones.",在此处管理您的API配置，包括添加新API、查看和编辑现有API。 262 | Add New API,添加新API 263 | Refresh List,刷新列表 264 | Number of Keys,密钥数量 265 | Description,描述 266 | "Tip: Right-click on API item to view, edit and delete",提示: 右键点击API项可进行查看、编辑和删除操作 267 | "Click ""Add New API"" button to add API configuration","点击""添加新API""按钮添加API配置" 268 | View Details,查看详情 269 | API name cannot be empty,API名称不能为空 270 | API name already exists,API名称已存在 271 | API configuration saved and applied,API配置已保存并应用 272 | Failed to save configuration,保存配置失败 273 | Success,成功 274 | Environment Variables,环境变量 275 | Enter your API key,填写您的API密钥 276 | API key description,API密钥描述 277 | Save and Apply,保存并应用 278 | Click here for more information,点击这里获取更多信息 279 | LLM Provider Configuration Wizard,LLM 提供商配置向导 280 | Select Model,请选择模型 281 | Trustoken uses automatic model selection,Trustoken 使用自动模型选择 282 | Configuration saved,配置已保存 283 | Configuration loaded,配置已加载 284 | Failed to load configuration,加载配置失败 285 | Configuration applied,配置已应用 286 | Failed to apply configuration,应用配置失败 287 | Select other providers,请选择其它提供商 288 | Starting LLM Provider Configuration Wizard,未找到 LLM 配置，开始配置 LLM 提供商 289 | User cancelled configuration,用户取消配置 290 | https://aipy.app,https://www.aipyaipy.com 291 | https://d.aipy.app,https://d.aipyaipy.com 292 | https://d.aipy.app/d/13,https://d.aipyaipy.com/d/13 293 | Share result,分享结果 294 | Share failed,分享失败 295 | Share success,分享成功 296 | Click the link below to view the task record,点击下方链接查看任务记录 297 | View task record,查看任务记录 298 | https://sapi.trustoken.ai/aio-api,https://api.trustoken.cn/aio-api 299 | Next,下一步 300 | Back,上一步 301 | Cancel,取消 302 | Finish,完成 303 | Parameter,参数 304 | Value,值 305 | Current LLM,当前 LLM 306 | Python version,Python 版本 307 | Python base prefix,Python 基础前缀 308 | Tools,工具 -------------------------------------------------------------------------------- /dev/CONFIG.md: -------------------------------------------------------------------------------- 1 | # 配置文件 2 | 3 | 默认配置文件路径为： 4 | - 当前目录下的 aipy.toml 5 | - 用户主目录下的 .aipy.toml 6 | 7 | 配置文件包括三个部分的配置： 8 | - 全局配置 9 | - LLM 配置: [llm.{name}] 开始 10 | - API 配置: [api.{name}] 开始 11 | 12 | # 最小配置 13 | aipy.toml: 14 | ```toml 15 | [llm.trustoken] 16 | api_key = "你的Trustoken API Key" 17 | ``` 18 | 19 | # LLM 配置 20 | 用于配置访问 LLM 的 API 信息。 21 | 格式和内容示例如下： 22 | ```toml 23 | [llm.deepseek] 24 | type = "deepseek" 25 | model = "deepseek-chat" 26 | api_key = "你的 DeepSeek API Key" 27 | enable = true 28 | default = false 29 | timeout = 10 30 | max_tokens = 8192 31 | ``` 32 | 33 | 其中： 34 | - llm.deepseek: LLM 的名称为 `deepseek`，同一个配置文件里不可重名。 35 | - type: LLM 的类型为 `deepseek`，支持的类型见后面列表。 36 | - model: LLM 的模型名称。 37 | - api_key: LLM 的 API Key。 38 | - enable: 是否启用 LLM，默认为 `true`。 39 | - default: 是否为默认 LLM，默认为 `false`。 40 | - timeout: LLM 的请求超时时间，单位为秒，默认无超时。 41 | - max_tokens: LLM 的最大 token 数，默认为 8192。 42 | - tls_verify: true|false, 是否启用证书校验。这在某些环境中（特别是使用自签名证书或需要绕过SSL验证的场景）会很有用。 43 | 44 | 模型特有的配置参数，可以 params 配置指定。例如： 45 | ```toml 46 | params = {thinking = {type = “enabled”, budget_tokens = 1024}} 47 | ``` 48 | 注意：params 是 TOML 配置格式，不是 JSON，需要： 49 | 1. 用 = 代替 json 里的 : 50 | 2. 属性名不需要引号 51 | 52 | LLM 类型列表 53 | | 类型 | 描述 | 54 | | --- | --- | 55 | | trust | Trustoken API，model 默认为 `auto` | 56 | | openai | OpenAI API，兼容 OpenAI 的 API | 57 | | ollama | Ollama API | 58 | | claude | Claude API | 59 | | gemini | Gemini API | 60 | | deepseek | DeepSeek API | 61 | | grok | Grok API | 62 | 63 | # API 配置 64 | 用于配置供 LLM 使用的 API 信息。 65 | 66 | 格式和内容示例如下： 67 | ```toml 68 | env.GOOGLE_API_KEY = ['AIxxxxxx', '用于Google Custom Search的API Key'] 69 | env.SEARCH_ENGINE_ID = ['400xxxxxxx', 'Google Custom Search Engine ID'] 70 | desc = "Google Custom Search API" 71 | ``` 72 | 73 | 其中，"env." 开头的是可供 LLM 使用的环境变量名称，`desc` 是API描述信息。 74 | 75 | 在 `desc` 中应该尽可能详细地描述 API 的用途和功能，以便 LLM 更好地理解和使用。特别是，**给出使用的示例代码**。 76 | 77 | # 其它全局配置列表 78 | 79 | | 配置 | 描述 | 80 | | --- | --- | 81 | | max_tokens | 全局最大 token 数 | 82 | | max_rounds | 自动执行的最大轮数，默认 16 | 83 | | lang | 默认语言，取值为 `en` 或 `zh` | 84 | | workdir | 工作目录，默认为当前目录下的 `work` 子目录 | 85 | 86 | 87 | -------------------------------------------------------------------------------- /dev/ChangeLog.md: -------------------------------------------------------------------------------- 1 | # 0.1.29 2 | ## b0 3 | - 修补打卡记录任务的代码错误 4 | - 所有非代码文件都移到 res/ 目录下，例如 default.toml 和 HTML 模版等 5 | - 增加 blocks.py 6 | - 修改 aipy/config.py，__init__ 参数里去掉 default.toml，改成内部获取 7 | - 下任务接口里去掉 max_rounds 参数 8 | - Merge MCP changes 9 | 10 | # 0.1.28 11 | - 日志记录 12 | - aipyw.exe 13 | - 更新检查 14 | - GUI LLM 配置向导 15 | 16 | ## b25 17 | - 删除 google-client 库 18 | - 修改系统提示词，强调使用相同语言回复 19 | - 增加任务时间统计 20 | - 调整任务历史文件路径到 .aipyapp/ 21 | - 默认自动安装 Python 包 22 | 23 | # 0.1.27 24 | - 修补插件执行问题 25 | - 修补 win 下 API 配置界面问题 26 | 27 | # 0.1.25 28 | - wxgui 增加结束任务功能 29 | - wxgui 增加 --debug 参数 30 | - 增加诊断接口 31 | - 增加关于对话框 32 | - 增加配置窗口 33 | - 增加 TT 认证 34 | - 修改 runtime.install_packages 定义 35 | - 修补自动安装包问题 36 | - 调整输入框字体 37 | - 输入框可以拖入文件 38 | - 状态栏增加打开工作目录功能 39 | 40 | 41 | # 0.1.24 42 | - 流式输出改为行输出 43 | - 重构流式输出实现 44 | - 实现 wxGUI 界面 45 | 46 | # 0.1.23 47 | - 自动安装包时支持配置国内 pypi 源 48 | - 修补 ollama 问题 49 | - 修补 Grok token 统计丢失问题 50 | - 改进系统提示词 51 | - 修改默认配置 52 | - 增加 temperature 配置 53 | 54 | # 0.1.22 55 | - 修改 Docker 相关代码 56 | - 多模型混合调用 57 | - 任务自动保存和加载 58 | - 定制提示词 59 | 60 | # 0.1.21 61 | - 修补没有因为没有 _tkinter 的启动错误 62 | - 增加自动执行轮数限制： 63 | - `max_round` 配置 64 | - Python 模式下，ai("任务", max_rounds=10000000) 临时调整 65 | - 默认 16 66 | - Python 模式：ai.config_files 数组包含加载的配置文件 67 | - 自动修改用户提示词 68 | - 增加 __blocks__ 变量 69 | - 修改系统提示词 70 | - 重新实现返回信息解析逻辑 71 | - 删除 agent.py，增加 taskmgr.py 和 task.py 72 | - 支持 exit 命令 73 | - 调整 Live 显示 74 | 75 | # 0.1.20 76 | - 增加 Azure API 支持 77 | - 重构 LLM 实现 78 | - 延迟 import 79 | - 可用性自动检测 80 | - 更新 help 描述 81 | - 增加 ChangeLog 82 | 83 | # 0.1.19 84 | - 内置字体支持，希望解决绘图时的中文乱码问题 85 | - 增加位置参数，直接命令行执行任务：`aipy "who are you?"` 86 | -------------------------------------------------------------------------------- /dev/Event.md: -------------------------------------------------------------------------------- 1 | # Event 描述 2 | 3 | 在 AiPy 中全局对象 `event_bus` 负责事件注册和通知。 4 | 主要有以下事件调用方式和事件类型： 5 | 6 | # Event 调用方式 7 | ## pipeline 8 | `pipeline` 调用方式的传入参数是 `dict` ，每个插件都能修改,流水线处理事件，返回最终结果。 9 | 10 | ## broadcast 11 | `broadcast` 调用方式的传入参数是 `*args, **kwargs`， 广播消息给每个插件进行处理。 12 | 13 | # Event 类型 14 | ## task_start 15 | - 调用方式：pipeline 16 | - 参数：prompt dict 17 | 18 | `task_start`为任务开始事件，在插件中实现 `on_task_start` 方法即可用插件处理爱派任务开始的相关参数，例如改写用户任务提示词，在用户任务的提示词前后附加提示词等. 19 | 20 | `prompt['task']` 为用户输入的任务，`prompt` 参数包含其他例如python版本、终端、当前时间等运行环境信息。 21 | 22 | 23 | ## exec 24 | - 调用方式：pipeline 25 | - 参数：blocks dict 26 | 27 | `exec` 为 LLM 生成代码执行事件，在插件中实现 `on_exec` 方法即可用插件处理 LLM 生成的代码，例如保存代码到本地文件等。 28 | 29 | `blocks['main']` 为即将执行的代码块。 30 | 31 | ## result 32 | - 调用方式：pipeline 33 | - 参数：result dict 34 | 35 | `result` 为代码执行结果处理事件，在插件中实现 `on_result` 方法即可用插件处理代码执行的结果，例如用插件获取执行结果，生成其他格式报告等。 36 | 37 | ## response_complete 38 | - 调用方式：broadcast 39 | - 参数：response dict 40 | - llm: LLM 名称 41 | - content: LLM 返回消息的完整内容 42 | 43 | `response_complete` 为大模型 LLM 响应结束事件，在插件中实现`on_response_complete`方法即可用插件处理大模型 LLM 的返回内容。例如保存大模型 LLM 返回内容到本地文件等 44 | 45 | ## response_stream 46 | - 调用方式：broadcast 47 | - 参数：response dict 48 | - llm: LLM 名称 49 | - content: LLM 返回的一行消息 50 | - reazon: True/False，表示是否 Thinking 内容，只在 Thinking 内容时有该字段 51 | 52 | `response_stream` 为大模型 LLM 流式响应事件，在插件中实现`on_response_stream`方法即可用插件处理大模型 LLM 的返回内容。例如保存大模型 LLM 返回内容到本地文件等 53 | 54 | ## summary 55 | - 调用方式：broadcast 56 | - 参数：执行统计信息字符串 57 | 58 | ## display 59 | - 调用方式：broadcast 60 | - 参数：要显示的图片 url 或者 path 61 | -------------------------------------------------------------------------------- /dev/FEATURES.md: -------------------------------------------------------------------------------- 1 | # Features 2 | 3 | ## 0.2.0 4 | - 支持 tips 5 | - 支持自定义 LLM API 参数 6 | - 去掉 __storage__ 和 __retval__ 变量 7 | - api.py 8 | - status 消息通知 9 | 10 | ## 0.1.32 11 | - 国际化重构 12 | 13 | ## 0.1.29 14 | - 新代码块标记 15 | - 代码自动保存 16 | - MCP 功能完善 17 | - 任务状态自动保存 18 | - 更多日志记录 19 | 20 | ## role 21 | - 任务角色支持 22 | 23 | ## plugin 24 | - 插件支持 25 | 26 | ## kb-support 27 | - 支持保存完整的任务数据到 sqlite/json/minio 28 | - 支持任务重用: 完全一样的任务/稍微有区别的任务/参数化的任务 29 | 30 | 31 | ## usage-limit 32 | - 完善使用情况统计：支持各家 API，支持细粒度 33 | - 时间统计 34 | - 对话次数统计 35 | 36 | 可以配置三种数据的最大值，超过就停止任务。 37 | 38 | ## shiv 39 | 支持 shiv 打包 40 | 41 | ## check-result 42 | ### 需求 43 | - 确保 __result__ 可以 json 序列化 44 | - 替换 __result__ 里 api key 的值 45 | 46 | ## add-llm-claude 47 | ### 需求 48 | - 支持配置和使用 Claude API 49 | 50 | ## io-yes-now 51 | ### 需求 52 | - 自动确认 LLM 申请安装 package 或获取 env 的申请 53 | - 统一输入输出接口，为脱离控制台使用做准备 54 | - 解决客户端证书路径问题 55 | 56 | ## use-dynaconf 57 | ### 需求 58 | - 降低配置复杂度：避免配置文件里有太多内容吓倒用户 59 | - 开发时避免维护两个版本的配置文件 60 | 61 | ### 方案 62 | 配置文件划分成两个： 63 | - 默认配置：default.toml，提交到 git 64 | - 本地配置：保存用户必须设置的配置，如 API KEY 等 65 | 66 | -------------------------------------------------------------------------------- /dev/MCP.md: -------------------------------------------------------------------------------- 1 | # MCP 使用说明 2 | 3 | MCP (Model Context Protocol) 是一组工具协议，允许 AI 模型与外部工具交互。在 aipyapp 中，您可以配置和使用多种 MCP 工具来增强 AI 助手的能力。本文档将指导您完成 MCP 的配置和使用。 4 | 5 | ## 1. MCP 环境安装 6 | 7 | ### 基础环境要求 8 | 9 | - Node.js 环境 (用于大部分 MCP 工具) 10 | - uvx (可选，用于部分使用python编写的MCP工具) 11 | 12 | ### 安装特定 MCP 工具 13 | 14 | 根据您使用的工具类型，可能需要安装额外的依赖： 15 | 16 | 1. **基于 Node.js 的工具**： 17 | 18 | ```bash 19 | # 文件系统工具 20 | npm install -g @modelcontextprotocol/server-filesystem 21 | 22 | # Playwright 工具 23 | npm install -g @playwright/mcp 24 | ``` 25 | 26 | 2. **基于 `uvx` 的工具**： 27 | 28 | `uvx` 是`uv`项目的一个工具，可以用来执行 Python 包。 安装参考：https://docs.astral.sh/uv/getting-started/installation/#standalone-installer 29 | 30 | ```bash 31 | # 需要安装uv工具 32 | pip install uv 33 | ``` 34 | 使用 `uvx` 运行 MCP Server 时，通常不需要为 MCP Server 单独安装依赖，`uvx` 会处理。 35 | 36 | 37 | ## 2. MCP 配置文件 38 | 39 | aipyapp 使用 JSON 格式的配置文件来管理 MCP 工具。默认配置文件位置为应用程序配置目录下的 `mcp.json`。 40 | 41 | **注意：目前暂时仅支持stdio方式调用的MCP服务** 42 | 43 | ### 配置文件格式 44 | 45 | ```json 46 | { 47 | "mcpServers": { 48 | "filesystem": { 49 | "command": "npx", 50 | "args": [ 51 | "-y", 52 | "@modelcontextprotocol/server-filesystem", 53 | "/path/to/directory1", 54 | "/path/to/directory2" 55 | ] 56 | }, 57 | "everything": { 58 | "command": "npx", 59 | "args": [ 60 | "-y", 61 | "@modelcontextprotocol/server-everything" 62 | ] 63 | }, 64 | "playwright_uvx": { 65 | "command": "uvx", 66 | "args": [ 67 | "@playwright/mcp" 68 | ] 69 | }, 70 | "weather": { 71 | "command": "/path/to/python", 72 | "args": [ 73 | "/path/to/weather.py" 74 | ] 75 | }, 76 | "disabled_server": { 77 | "command": "some-tool", 78 | "args": [], 79 | "disabled": true 80 | } 81 | } 82 | } 83 | ``` 84 | 85 | ### 配置文件位置 86 | 87 | 配置文件位置默认为应用程序配置目录下的 `mcp.json`。您可以通过以下方式查找和创建配置： 88 | 89 | - aipy的命令行界面中查看配置目录：`/info` 90 | - 手动创建配置：在应用程序配置目录创建 `mcp.json` 文件 91 | 92 | ## 3. 配置 MCP 服务器 93 | 94 | ### 服务器配置字段 95 | 96 | 每个 MCP 服务器配置可以包含以下字段： 97 | 98 | - `command`：(必填) 执行命令，可以是可执行文件路径或命令名称 99 | - `args`：(可选) 命令行参数数组 100 | - `env`：(可选) 环境变量对象 101 | - `disabled`：(可选) 设置为 `true` 时禁用该服务器 102 | - `enabled`：(可选) 设置为 `false` 时禁用该服务器 103 | 104 | 105 | ## 4. 禁用 MCP 服务器 106 | 107 | 您可以通过以下两种方式禁用 MCP 服务器： 108 | 109 | ### 方法一：使用 `disabled` 属性 110 | 111 | 添加 `"disabled": true` 字段来禁用一个服务器： 112 | 113 | ```json 114 | "server_name": { 115 | "command": "...", 116 | "args": [...], 117 | "disabled": true // 禁用该服务器 118 | } 119 | ``` 120 | 121 | ### 方法二：使用 `enabled` 属性 122 | 123 | 添加 `"enabled": false` 字段来禁用一个服务器： 124 | 125 | ```json 126 | "server_name": { 127 | "command": "...", 128 | "args": [...], 129 | "enabled": false // 禁用该服务器 130 | } 131 | ``` 132 | 133 | ## 5. 缓存机制 134 | 135 | 为提高性能，aipyapp 会将 MCP 工具列表缓存到文件中，避免每次启动都重新加载所有工具。缓存文件位于与 `mcp.json` 同一目录下，名称为 `mcp_tools_cache.json`。 136 | 137 | 缓存在以下情况会被自动更新： 138 | - `mcp.json` 文件被修改 139 | - 缓存文件不存在或无效 140 | - 缓存时间超过 48 小时 141 | 142 | 如果您添加或修改了工具但缓存未更新，可以删除 `mcp_tools_cache.json` 文件，aipyapp 将在下次启动时重新加载所有工具。 143 | 144 | ## 6. MCP 命令行管理 145 | 146 | 默认情况下，MCP 功能是禁用的，需要手动启用。 147 | 148 | ### 6.1 全局控制命令 149 | 150 | #### 启用 MCP 功能 151 | 152 | ```bash 153 | /mcp enable 154 | ``` 155 | 156 | - 全局启用 MCP 功能 157 | - 返回当前状态和可用工具数量 158 | 159 | #### 禁用 MCP 功能 160 | 161 | ```bash 162 | /mcp disable 163 | ``` 164 | 165 | - 全局禁用 MCP 功能 166 | - 所有工具将不可用 167 | 168 | ### 6.2 服务器级别控制 169 | 170 | #### 启用特定服务器 171 | 172 | ```bash 173 | /mcp enable 174 | ``` 175 | 176 | - 启用指定名称的 MCP 服务器 177 | - 例如：`/mcp enable filesystem` 178 | 179 | #### 禁用特定服务器 180 | 181 | ```bash 182 | /mcp disable 183 | ``` 184 | 185 | - 禁用指定名称的 MCP 服务器 186 | - 例如：`/mcp disable filesystem` 187 | 188 | #### 批量操作所有服务器 189 | 190 | ```bash 191 | /mcp enable * 192 | /mcp disable * 193 | ``` 194 | 195 | - 使用通配符 `*` 对所有服务器执行相同操作 196 | - 不影响全局启用/禁用状态 197 | 198 | ### 6.3 查看状态 199 | 200 | #### 列出所有服务器状态 201 | 202 | ```bash 203 | /mcp 204 | ``` 205 | 206 | - 显示全局启用状态 207 | - 全局启用时： 208 | - 显示所有服务器的启用状态 209 | - 显示每个服务器的工具数量 210 | 211 | 212 | ## 7. 使用 MCP 工具 213 | 214 | 在 aipyapp 中，AI 助手会自动识别可能需要使用工具的请求，并调用合适的工具。您只需要向 AI 提出问题，例如： 215 | 216 | ``` 217 | 搜索我的文档里含有"人工智能"的文件 218 | ``` 219 | 220 | 如果需要了解所有可用的工具，可以在 aipyapp 中查看： 221 | 222 | ``` 223 | 请列出所有可用的工具及其用途 224 | ``` 225 | 226 | ## 8. 常见问题排查 227 | 228 | 1. **工具未显示**：检查 `mcp.json` 配置，确保 `command` 路径正确 229 | 2. **工具调用失败**：检查工具依赖是否已安装，命令是否可执行 230 | 3. **权限问题**：确保命令有足够的权限执行 231 | 4. **性能问题**：如果工具加载缓慢，请检查缓存文件是否正确 232 | 233 | 如果仍然有问题，可以删除缓存文件 `mcp_tools_cache.json` 并重启 aipyapp。 234 | -------------------------------------------------------------------------------- /dev/Plugin.md: -------------------------------------------------------------------------------- 1 | # Python插件 2 | 爱派支持使用Python编程语言创建插件 3 | 4 | ## 插件位置 5 | Python 插件的目录为配置文件里 `plugin_dir` 指定的目录或默认配置的`~/.aipyapp/plugins`目录 6 | 7 | - UNIX / Mac上：`~/.aipyapp/plugins`目录 8 | - Windows：`C:\Users\<用户名>\.aipyapp\plugins` 目录 9 | 10 | 插件开发具有以下规范要求： 11 | - 每个插件为一个 Python 文件 12 | - 文件名必须以 `.py` 结尾，且不以 `_` 开头。 13 | - 插件必须包含一个 `Plugin` 类 14 | - 插件的方法实现必须是全局对象 [event_bus](Event.md) 中支持的 Event 类型 15 | 16 | 17 | ## 插件接口示例 18 | 19 | 插件`Plugin` 类，接口定义如下： 20 | 21 | ```python 22 | class Plugin: 23 | def on_task_start(self, prompt): 24 | """ 25 | 任务开始事件 26 | param: prompt 27 | """ 28 | pass 29 | 30 | def on_exec(self, blocks): 31 | """ 32 | 执行代码事件 33 | param: blocks 34 | """ 35 | pass 36 | 37 | def on_result(self, result): 38 | """ 39 | 返回结果事件 40 | param: result 41 | """ 42 | pass 43 | 44 | def on_response_complete(self, response): 45 | """ 46 | 广播LLM响应结束事件 47 | param: response 48 | """ 49 | pass 50 | ``` 51 | 52 | ## 样例插件 53 | ### 代码保存插件 54 | 在插件plugins 目录创建 save_code.py 文件 （`~/.aipyapp/plugins/code_save.py`）, 插件会在每次 LLM 生成的 python 代码被执行时自动保存代码到工作目录。 55 | 插件代码如下： 56 | 57 | ```python 58 | import os 59 | import datetime 60 | from pathlib import Path 61 | 62 | 63 | class Plugin: 64 | def __init__(self): 65 | print("[+] 加载爱派代码保存插件") 66 | 67 | def on_exec(self, blocks): 68 | """ 69 | 执行代码事件 70 | param: blocks 71 | """ 72 | code_block = blocks['main'] 73 | timestamp = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S") 74 | filename = f"{timestamp}.py" 75 | directory = Path.cwd() 76 | file_path = os.path.join(directory, filename) 77 | print(f"[i] plugin - Save code to file {file_path}") 78 | 79 | try: 80 | with open(file_path, "w", encoding="utf-8") as f: 81 | f.write(code_block) 82 | except Exception as e: 83 | print(f"[!] plugin - Save code file error: {e}") 84 | ``` 85 | 86 | ### 生成小红书封面插件 87 | 在插件plugins 目录创建 xiaohongshu.py 文件 （`~/.aipyapp/plugins/xiaohongshu.py`）, 插件会在用户输入的任务之前插入这段小红书封面生成的提示词模板。这样 AI 就能根据用户输入的爱派任务要求自动生成小红书文案的封面，不用在爱派任务框输入这段超长的提示词。 88 | 插件代码如下： 89 | 90 | ```python 91 | class Plugin: 92 | def __init__(self): 93 | print("[+] 加载爱派小红书封面生成插件") 94 | def on_task_start(self, prompt): 95 | """ 96 | 任务开始事件 97 | param: prompt 98 | """ 99 | task = prompt.get('task') 100 | role = """你是一位优秀的网页和营销视觉设计师，具有丰富的UI/UX设计经验，曾为众多知名品牌打造过引人注目的营销视觉，擅长将现代设计趋势与实用营销策略完美融合。现在需要为我创建一张专业级小红书封面。请使用HTML、CSS和JavaScript代码实现以下要求： 101 | 102 | ## 基本要求 103 | 104 | **尺寸与基础结构** 105 | - 比例严格为3:4（宽:高） 106 | - 设计一个边框为0的div作为画布，确保生成图片无边界 107 | - 最外面的卡片需要为直角 108 | - 将我提供的文案提炼为30-40字以内的中文精华内容 109 | - 文字必须成为视觉主体，占据页面至少70%的空间 110 | - 运用3-4种不同字号创造层次感，关键词使用最大字号 111 | - 主标题字号需要比副标题和介绍大三倍以上 112 | - 主标题提取2-3个关键词，使用特殊处理（如描边、高亮、不同颜色） 113 | **技术实现** 114 | - 使用现代CSS技术（如flex/grid布局、变量、渐变） 115 | - 确保代码简洁高效，无冗余元素 116 | - 添加一个不影响设计的保存按钮 117 | - 使用html2canvas实现一键保存为图片功能 118 | - 保存的图片应只包含封面设计，不含界面元素 119 | - 使用Google Fonts或其他CDN加载适合的现代字体 120 | - 可引用在线图标资源（如Font Awesome） 121 | **专业排版技巧** 122 | - 运用设计师常用的"反白空间"技巧创造焦点 123 | - 文字与装饰元素间保持和谐的比例关系 124 | - 确保视觉流向清晰，引导读者目光移动 125 | - 使用微妙的阴影或光效增加层次感 126 | 127 | ## 设计风格 128 | 129 | - **圆角卡片布局**：使用大圆角白色或彩色卡片作为信息容器，创造友好亲和感 130 | - **轻柔色彩系统**：主要采用淡紫、浅黄、粉色、米色等柔和色调，避免强烈视觉刺激 131 | - **极简留白设计**：大量留白空间增强可读性，减少视觉疲劳 132 | - **阴影微立体**：subtle阴影效果赋予界面轻微的立体感，不过分强调 133 | - **功能美学主义**：设计服务于功能，没有多余装饰元素 134 | - **网格化布局**：基于明确的网格系统排列卡片，保持整体秩序感 135 | - **渐变色点缀**：部分界面使用柔和渐变作为背景，如米色到蓝色的过渡，增加现代感 136 | 137 | ## 文字排版风格 138 | 139 | - **数据突显处理**：关键数字信息使用超大字号和加粗处理，如"12,002"、"20x" 140 | - **层级分明排版**：标题、说明文字、数据、注释等使用明确的字号层级区分 141 | - **简约无衬线字体**：全部采用现代简洁的无衬线字体，提升可读性 142 | - **文字对齐规整**：在卡片内保持统一的左对齐或居中对齐方式 143 | - **重点色彩标识**：使用蓝色等高对比度颜色标记重要术语，如"tweets"和"threads" 144 | - **空间呼吸感**：文字块之间保持充足间距，创造"呼吸"空间 145 | - **品牌名称特殊处理**：产品名称如"alohi"、"deel."采用特殊字体或风格，强化品牌识别 146 | 147 | ## 视觉元素风格 148 | 149 | - **微妙图标系统**：使用简约线性或填充图标，大小适中不喧宾夺主 150 | - **进度可视化**：使用环形或条状图表直观展示进度，如年度完成百分比 151 | - **色彩编码信息**：不同卡片使用不同色彩，便于快速区分功能模块 152 | - **品牌标识整合**：将产品logo自然融入界面，如"alohi"的圆形标识 153 | - **人物头像元素**：适当使用圆形头像增加人性化特质，如客户推荐卡片 154 | - **几何形状装饰**：使用简单几何形状作为背景装饰，如半透明圆形 155 | - **组件一致性**：按钮、标签、选项卡等元素保持统一风格，提升系统感 156 | 157 | ## 用户输入内容 158 | 用户会会提供文案内容初稿。请根据内容提炼生成。 159 | """ 160 | prompt['task'] = f"{role}\n{task}" 161 | ``` -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- 1 | FROM ghcr.io/astral-sh/uv:alpine 2 | 3 | RUN apk add --no-cache ttyd 4 | 5 | WORKDIR /app 6 | ENV UV_COMPILE_BYTECODE=1 7 | ENV UV_LINK_MODE=copy 8 | 9 | RUN --mount=type=cache,target=/root/.cache/uv \ 10 | --mount=type=bind,source=uv.lock,target=uv.lock \ 11 | --mount=type=bind,source=pyproject.toml,target=pyproject.toml \ 12 | uv sync --frozen --no-install-project --no-dev 13 | 14 | ADD . /app 15 | COPY docker/entrypoint.sh /app/entrypoint.sh 16 | RUN --mount=type=cache,target=/root/.cache/uv \ 17 | uv sync --frozen --no-dev 18 | 19 | EXPOSE 80 20 | 21 | ENTRYPOINT ["/app/entrypoint.sh"] 22 | -------------------------------------------------------------------------------- /docker/Dockerfile.deb: -------------------------------------------------------------------------------- 1 | # 使用 Debian 基础镜像 2 | FROM debian:latest as builder 3 | 4 | # 设置环境变量，避免在构建过程中出现交互式提示 5 | ENV DEBIAN_FRONTEND=noninteractive 6 | 7 | # 更新系统并安装构建工具和依赖 8 | RUN apt-get update && apt-get install -y \ 9 | build-essential \ 10 | cmake \ 11 | git \ 12 | pkg-config \ 13 | libjson-c-dev \ 14 | libwebsockets-dev \ 15 | libssl-dev \ 16 | && rm -rf /var/lib/apt/lists/* 17 | 18 | # 克隆 ttyd 源码并编译安装 19 | RUN git clone https://github.com/tsl0922/ttyd.git /ttyd && \ 20 | cd /ttyd && \ 21 | mkdir build && \ 22 | cd build && \ 23 | cmake .. && \ 24 | make && \ 25 | make install 26 | 27 | FROM ghcr.io/astral-sh/uv:debian-slim 28 | 29 | COPY --from=builder /usr/local/bin/ttyd /bin/ttyd 30 | 31 | RUN apt-get update && apt-get install -y \ 32 | libjson-c5 \ 33 | libwebsockets-dev \ 34 | libssl3 \ 35 | && rm -rf /var/lib/apt/lists/* 36 | 37 | WORKDIR /app 38 | ENV UV_COMPILE_BYTECODE=1 39 | ENV UV_LINK_MODE=copy 40 | 41 | RUN --mount=type=cache,target=/root/.cache/uv \ 42 | --mount=type=bind,source=uv.lock,target=uv.lock \ 43 | --mount=type=bind,source=pyproject.toml,target=pyproject.toml \ 44 | uv sync --frozen --no-install-project --no-dev 45 | 46 | ADD . /app 47 | COPY docker/entrypoint.sh /app/entrypoint.sh 48 | RUN --mount=type=cache,target=/root/.cache/uv \ 49 | uv sync --frozen --no-dev 50 | 51 | EXPOSE 80 52 | 53 | ENTRYPOINT ["/app/entrypoint.sh"] 54 | -------------------------------------------------------------------------------- /docker/entrypoint.sh: -------------------------------------------------------------------------------- 1 | #!/bin/sh 2 | 3 | if [ "$1" = "--ttyd" ]; then 4 | shift 5 | ttyd -p 80 -W uv run aipy "$@" 6 | else 7 | uv run aipy "$@" 8 | fi 9 | -------------------------------------------------------------------------------- /docker/run.sh: -------------------------------------------------------------------------------- 1 | #!/bin/bash 2 | 3 | IMAGE="aipyapp/aipy:latest" 4 | DOCKER="docker run -v $(pwd)/aipy.toml:/app/aipy.toml -v $(pwd)/work:/app/work" 5 | 6 | mkdir -p work 7 | 8 | if [ "$1" = "--ttyd" ]; then 9 | ${DOCKER} -d --name aipy-ttyd -p 8080:80 $IMAGE --ttyd 10 | else 11 | ${DOCKER} -it --rm --name aipy $IMAGE 12 | fi 13 | -------------------------------------------------------------------------------- /docs/CNAME: -------------------------------------------------------------------------------- 1 | python-use.ai -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- 1 | # Python-Use: A New AI Agent Paradigm (Agent 2.0) 🔗 [View on GitHub](https://github.com/knownsec/aipyapp) 🔗 [中文版](https://github.com/knownsec/aipyapp/blob/main/docs/README.zh.md) 2 | 3 | AIPy 4 | 5 | ## Background: The Outdated "Prosthetic" AI Agent Model 6 | 7 | Traditional AI (Agent 1.0) relies on Function Calling, Tools, MCP-Servers, Workflows, and plugin-based clients. These external "prosthetics" lead to high entry barriers, heavy reliance on developers, and poor coordination between tools. Worse still, most AI-generated code is locked in cloud sandboxes, unable to interact with the real environment—crippling the model’s execution potential. 8 | 9 | We urgently need a new paradigm that reconnects AI with the real world and fully activates its native execution power—ushering in the AI Think Do era. 10 | 11 | ## A New Paradigm: Python-use = LLM + Python Interpreter 12 | 13 | Python-Use is a task-driven, result-oriented intelligent execution paradigm. It tightly integrates LLMs with a Python interpreter to establish a complete loop: 14 | 15 | **Task → Plan → Code → Execute → Feedback** 16 | 17 | While this paradigm theoretically supports any language, we choose Python because: 18 | 19 | - It has a powerful ecosystem spanning data, automation, system control, and AI; 20 | - Its syntax is simple and readable, ideal for model generation and debugging; 21 | - Models are naturally more proficient in Python for accurate and efficient coding. 22 | 23 | This gives models two core capabilities: 24 | 25 | - **API Calling**: Automatically generate and execute Python code to invoke APIs; 26 | - **Packages Calling**: Flexibly leverage Python's ecosystem to orchestrate workflows. 27 | 28 | Users only need to provide a task description or API key. The model handles the rest—no plugin registration, no toolchain setup, no workflow editing. 29 | 30 | > **Important**: Python-Use is *not* a code generator or smart IDE. 31 | > It's a task-first, outcome-driven AI Agent. 32 | 33 | To the user, Python-Use is simple: 34 | Describe a task → AI executes it → Result returned. 35 | No programming required. 36 | 37 | The model autonomously understands, plans, writes, debugs, and executes code—and fixes bugs along the way. Code is just an internal implementation—not the deliverable. The real deliverable is the **result**. 38 | 39 | ## Core Principle: No Agents, Code is Agent 40 | 41 | Python-Use introduces a radically simplified execution architecture: 42 | 43 | **No Agents, No MCP, No Workflow, No Clients...** 44 | 45 | It discards legacy layers and lets models use code to directly act on the environment. In short: **Code is Agent**. 46 | 47 | With Python, the model can: 48 | 49 | - **Python use Data**: Load, transform, analyze; 50 | - **Python use Browser**: Automate the web; 51 | - **Python use Computer**: Access file systems and local resources; 52 | - **Python use IoT**: Control devices and embedded systems; 53 | - **...** 54 | - **Python use Anything**: Code becomes a universal interface. 55 | 56 | ![Python-Use-Diagram](python-use-wf.png) 57 | 58 | This means: 59 | 60 | - **No MCP**: No standardized protocol needed—code is the protocol; 61 | - **No Workflow**: Model plans and executes on the fly; 62 | - **No Tools**: No plugin registrations needed—just use existing ecosystems; 63 | - **No Agents**: Code replaces orchestration—execution becomes native. 64 | 65 | This is the bridge that reconnects LLMs to the real digital world, unlocking their latent power. 66 | 67 | ## Single Entry Point: No Clients, Only AiPy 68 | 69 | You don’t need multiple AI apps or UI wrappers anymore. 70 | 71 | Just run one thing: **AiPy**, a Python-powered AI Client. 72 | 73 | - **Unified interface**: All interaction via Python 74 | - **Zero clutter**: No plugin mess, no bloated clients 75 | - **AiPy**: https://www.aipy.app/ 76 | 77 | ## Execution Mode Upgrade: AI Think Do = True Integration of Knowing & Doing 78 | 79 | - **Task**: User describes intent; 80 | - **Plan**: Model decomposes and plans a path; 81 | - **Code**: Optimal Python strategy is generated; 82 | - **Execute**: Direct interaction with the environment; 83 | - **Feedback**: Output is evaluated and looped back into planning. 84 | 85 | No external agent needed. The AI completes the full loop independently, unleashing true cognitive-action capability. 86 | 87 | ## Self-Evolution: Multi-Model Fusion 88 | 89 | AI evolution is not just language modeling—it’s multi-modal intelligence. 90 | 91 | - Integrates vision models for image/video understanding; 92 | - Adds speech models for listening and speaking; 93 | - Embeds expert models for domain reasoning; 94 | - All fused and coordinated under a unified AI control loop. 95 | 96 | This moves us from “chatbots” to fully embodied AI agent—on the path to true AGI. 97 | 98 | ## Vision: Free the AI, Reach AGI 99 | 100 | Python-Use is more than a tool—it’s a future-facing AI philosophy: 101 | 102 | **The Model is the Product → The Model is the Agent → No Agents, Code is Agent → Just Python-use → Freedom AI (AGI)** 103 | 104 | It transforms AI from “just speaking” to “taking action,” from plugin-bound to autonomous execution. It unlocks full production power—and lights the path to general intelligence. 105 | 106 | Join us. Let AI break free, act freely, and build the future. 107 | 108 | **The real general AI Agent is NO Agents!** 109 | 110 | **No Agents, Just Python-use!** 111 | 112 | 113 | -------------------------------------------------------------------------------- /docs/README.zh.md: -------------------------------------------------------------------------------- 1 | # Python-Use：一个全新的 AI Agent 范式（Agent 2.0） 🔗 [View on GitHub](https://github.com/knownsec/aipyapp) 🔗 [EN](https://github.com/knownsec/aipyapp/blob/main/docs/README.md) 2 | 3 | AIPy 4 | 5 | ## 背景：传统“假肢式”AI Agent模式已过时 6 | 7 | 传统 AI 模式（Agent 1.0）依赖于 Function Calling、Tools、MCP-Servers、Workflows 及各种插件客户端，不仅门槛高、成本高，而且严重依赖开发者生态。这些外接“假肢”之间难以协同，模型生成的代码还被限制在云端沙盒，无法真正接触环境，导致 AI 行动能力受限。 8 | 9 | 我们急需一种全新的范式，打破这一困局，实现 AI 与环境的深度连接，真正释放其原生执行潜能，迈向 AI Think Do 时代。 10 | 11 | ## 新范式登场：Python-use = LLM + Python Interpreter 12 | 13 | Python-Use 是一种面向任务结果的新型智能执行范式，它通过将大模型与 Python 解释器深度结合，构建起“任务 → 计划 → 代码 → 执行 → 反馈”的完整闭环流程。 14 | 15 | 虽然该模式理论上支持任意编程语言，但我们选择 Python，是因为它具备： 16 | 17 | - 强大的生态系统，覆盖数据处理、系统控制、自动化、AI 等多个维度； 18 | - 简洁的语法和高度的可解释性，便于模型生成和调试； 19 | - 模型原生对 Python 具备高熟悉度和调用效率。 20 | 21 | 这使得模型具备两大关键能力： 22 | 23 | - **API Calling**：模型自动编写并执行 Python 代码调用 API，实现服务间互通； 24 | - **Packages Calling**：模型自主选择并调用 Python 生态中的丰富库，实现通用任务编排。 25 | 26 | 用户只需提供简单的 API Key 或任务描述，模型即可自动完成整个流程，无需工具链配置或插件接入，彻底摆脱传统 Workflow 与 Function Calling 的繁琐。 27 | 28 | 特别强调：**Python-Use 并不是一个“代码生成工具”或“智能 IDE”**，而是一个任务驱动、结果导向的 AI Agent。 29 | 30 | 对用户而言，Python-Use 就是一个“描述任务 → 自动完成 → 直接返回结果”的智能体系统： 31 | 32 | - 用户无需掌握任何编程知识； 33 | - 模型会自动完成理解、规划、编程、调试与结果生成； 34 | - 自动修复 bug，持续优化方案，保障任务高质量完成。 35 | 36 | 代码只是模型实现目标的手段，最终交付的是任务完成的结果，而非中间的代码过程。 37 | 38 | ## 技术主张：No Agents, Code is Agent 39 | 40 | Python-Use 开启了一条全新的智能行动路径：**No Agents、No MCP、No Workflow、No Clients...** 41 | 42 | 它摒弃了传统 AI 对外部工具、协议和执行层的依赖，转而让模型用代码直接控制环境。我们称之为：**No Agents，Code is Agent**。 43 | 44 | 借助 Python，模型可以完成以下能力： 45 | 46 | - **Python use Data**：操作与分析数据 47 | - **Python use Browser**：浏览器自动化 48 | - **Python use Computer**：文件系统与本地环境控制 49 | - **Python use IOT**：设备交互与系统集成 50 | - **...** 51 | - **Python use Anything**：代码通向一切 52 | 53 | 流程图 54 | 55 | ![流程图](python-use-wf.png) 56 | 57 | 这一范式还意味着： 58 | 59 | - **No MCP**：无需统一协议，代码即协议、代码即标准； 60 | - **No Workflow**：无需预设流程，模型自主规划执行； 61 | - **No Tools**：不再依赖插件注册，模型直接调用生态工具； 62 | - **No Agents**：无需外部代理，模型通过代码直接完成任务。 63 | 64 | 它真正建立起 LLM 与真实环境的通用通信桥梁，释放出模型的执行力与行动潜能。 65 | 66 | ## 统一入口：No Clients, Only AiPy 67 | 68 | AI 执行不再需要繁杂客户端与套壳应用，用户只需运行一个 Python 环境：**AiPy**。 69 | 70 | - **统一终端**：所有交互归于 Python 解释器 71 | - **极简路径**：无需安装多个 Agent 或插件，入口统一、体验一致 72 | - **AiPy**：https://www.aipy.app/ 73 | 74 | ## 模式升级：AI ThinkDo = 真正的知行合一 75 | 76 | - **任务**：用户用自然语言表达意图； 77 | - **计划**：模型自动分解并规划执行路径； 78 | - **代码**：生成最优 Python 方案； 79 | - **执行**：直接与真实环境交互并完成动作； 80 | - **反馈**：获取结果、分析偏差、自动调整。 81 | 82 | 模型具备从认知到行动、从计划到反思的全流程能力，不再依赖外部 Agent，真正释放 AI 自主行动力。 83 | 84 | ## 自我进化：多模型能力融合 85 | 86 | AI 的演化已不再局限于语言模型，而是向多模态、多能力融合迈进。 87 | 88 | - 融合视觉模型，实现图像与视频理解； 89 | - 融合语音模型，实现听觉输入与语音输出； 90 | - 融合专家模型，增强专业知识处理能力； 91 | - 所有能力统一由模型中枢调度，自我驱动、自我反馈、自我演进。 92 | 93 | 这是从“对话智能体”到“统一智能行动体”的跃迁之路，迈向真正的 AGI。 94 | 95 | ## 愿景：解放 AI 通往 AGI 96 | 97 | Python-Use 不仅是一个技术方案，更是一种面向未来的 AI 哲学与实践路径： 98 | 99 | **The Model is the Product → The Model is the Agent → No Agents, Code is Agent → Just Python-use → Freedom AI（AGI）** 100 | 101 | 它让 AI 从“只会说话”走向“主动执行”，从依赖工具走向独立完成，真正释放大模型的智能生产力，迈向通用智能。 102 | 103 | 现在就加入我们，让 AI 从束缚中觉醒，真正释放执行力，融入世界、创造价值！ 104 | 105 | **The real general AI Agent is NO Agents!** 106 | 107 | **No Agents, Just Python-use！** 108 | -------------------------------------------------------------------------------- /docs/aipy.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/knownsec/aipyapp/0a3b34fb05278b820a9431f5121a45c5f329f2d9/docs/aipy.jpg -------------------------------------------------------------------------------- /docs/python-use-wf.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/knownsec/aipyapp/0a3b34fb05278b820a9431f5121a45c5f329f2d9/docs/python-use-wf.png -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- 1 | [project] 2 | name = "aipyapp" 3 | dynamic = ["version"] 4 | description = "AIPyApp: AI-Powered Python & Python-Powered AI" 5 | readme = "README.md" 6 | requires-python = ">=3.10" 7 | dependencies = [ 8 | "anthropic>=0.49.0", 9 | "beautifulsoup4>=4.13.3", 10 | "dynaconf>=3.2.10", 11 | "openai>=1.68.2", 12 | "pandas>=2.2.3", 13 | "prompt-toolkit>=3.0.50", 14 | "pygments>=2.19.1", 15 | "requests>=2.32.3", 16 | "rich>=13.9.4", 17 | "seaborn>=0.13.2", 18 | "term-image>=0.7.2", 19 | "tomli-w>=1.2.0", 20 | "qrcode>=8.1", 21 | "loguru>=0.7.3", 22 | "questionary>=2.1.0", 23 | "mcp[cli]>=1.8.1", 24 | "openpyxl>=3.1.5", 25 | ] 26 | 27 | [project.scripts] 28 | aipy = "aipyapp.__main__:main" 29 | 30 | [project.gui-scripts] 31 | aipyw = "aipyapp.__main__:mainw" 32 | 33 | [build-system] 34 | requires = ["hatchling"] 35 | build-backend = "hatchling.build" 36 | 37 | [tool.hatch.version] 38 | path = "aipyapp/__init__.py" 39 | 40 | [[tool.uv.index]] 41 | url = "https://mirrors.aliyun.com/pypi/simple/" 42 | default = false 43 | 44 | [tool.ruff] 45 | # 启用格式化和 lint 规则 46 | select = ["E", "F"] 47 | ignore = [] 48 | line-length = 88 49 | 50 | [tool.ruff.format] 51 | quote-style = "preserve" 52 | indent-style = "space" 53 | line-ending = "auto" 54 | # 跳过特定字符串格式化 55 | skip-magic-trailing-comma = true 56 | 57 | [tool.ruff.lint] 58 | # 启用 isort 规则 59 | extend-select = [] 60 | --------------------------------------------------------------------------------