neuro-simulator 0.2.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

neuro_simulator/agent/core.py

@@ -8,6 +8,7 @@ import json
 import logging
 import re
 import sys
+from pathlib import Path
 from datetime import datetime
 from typing import Any, Dict, List, Optional
 
@@ -94,28 +95,20 @@ class Agent:
                     if tool.get("name") == "speak" and tool.get("result"):
                         assistant_responses.append(tool["result"])
 
-        # Create LLM prompt
-        prompt_parts = [
-            f"You are {self.memory_manager.init_memory.get('name', 'Neuro-Sama')}, an AI VTuber.",
-            f"Your personality: {self.memory_manager.init_memory.get('personality', 'Friendly and curious')}",
-            "\n=== CONTEXT ===", context,
-            "\n=== AVAILABLE TOOLS ===", tool_descriptions,
-            "\n=== YOUR RECENT SPEAK HISTORY (for context) ==="
-        ]
-        for response in assistant_responses[:5]: # Get last 5 responses
-            prompt_parts.append(f"- {response}")
-
-        prompt_parts.extend([
-            "\n=== INSTRUCTIONS ===",
-            "Process the user messages and respond. Use the 'speak' tool to talk to the user.",
-            "You are fully responsible for managing your own memory using the available tools.",
-            "\nUser messages to respond to:",
-        ])
+        # Create LLM prompt from template
+        template_path = Path(self.memory_manager.memory_dir).parent / "prompt_template.txt"
+        with open(template_path, 'r', encoding='utf-8') as f:
+            prompt_template = f.read()
 
-        for msg in messages:
-            prompt_parts.append(f"{msg['username']}: {msg['text']}")
-        prompt_parts.append("\nYour response (use tools as needed):")
-        prompt = "\n".join(prompt_parts)
+        recent_speak_history_text = "\n".join([f"- {response}" for response in assistant_responses[:5]]) if assistant_responses else "You haven't said anything yet."
+        user_messages_text = "\n".join([f"{msg['username']}: {msg['text']}" for msg in messages])
+
+        prompt = prompt_template.format(
+            full_context=context,
+            tool_descriptions=tool_descriptions,
+            recent_speak_history=recent_speak_history_text,
+            user_messages=user_messages_text
+        )
         
         await self.memory_manager.add_detailed_context_entry(
             input_messages=messages, prompt=prompt, llm_response="", tool_executions=[],

neuro_simulator/cli.py

@@ -57,7 +57,13 @@ def main():
     # 2. Ensure required assets and configs exist
     copy_resource('neuro_simulator', 'core/config.yaml.example', work_dir / 'config.yaml.example')
     copy_resource('neuro_simulator', 'assets', work_dir / 'assets', is_dir=True)
-    agent_memory_dir = work_dir / "agent" / "memory"
+    # Ensure agent directory and its contents exist
+    agent_dir = work_dir / "agent"
+    agent_dir.mkdir(parents=True, exist_ok=True)
+    copy_resource('neuro_simulator', 'agent/prompt_template.txt', agent_dir / 'prompt_template.txt')
+
+    # Ensure agent memory directory and its contents exist
+    agent_memory_dir = agent_dir / "memory"
     agent_memory_dir.mkdir(parents=True, exist_ok=True)
     for filename in ["context.json", "core_memory.json", "dialog_history.json", "init_memory.json"]:
         copy_resource('neuro_simulator', f'agent/memory/{filename}', agent_memory_dir / filename)

{neuro_simulator-0.2.0.dist-info → neuro_simulator-0.2.1.dist-info}/METADATA

@@ -1,176 +1,226 @@
-Metadata-Version: 2.4
-Name: neuro_simulator
-Version: 0.2.0
-Summary: Neuro Simulator Server
-Author-email: Moha-Master <hongkongreporter@outlook.com>
-License-Expression: MIT
-Project-URL: Homepage, https://github.com/Moha-Master/neuro-simulator
-Project-URL: Repository, https://github.com/Moha-Master/neuro-simulator
-Project-URL: Issues, https://github.com/Moha-Master/neuro-simulator/issues
-Classifier: Development Status :: 4 - Beta
-Classifier: Intended Audience :: Developers
-Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.8
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Requires-Python: >=3.8
-Description-Content-Type: text/markdown
-Requires-Dist: fastapi
-Requires-Dist: uvicorn
-Requires-Dist: google-genai
-Requires-Dist: azure-cognitiveservices-speech
-Requires-Dist: letta-client
-Requires-Dist: openai
-Requires-Dist: pyyaml
-Requires-Dist: pydantic
-Requires-Dist: jinja2
-Requires-Dist: python-multipart
-Requires-Dist: mutagen
-Provides-Extra: dev
-Requires-Dist: pytest>=6.0; extra == "dev"
-Requires-Dist: pytest-cov; extra == "dev"
-Requires-Dist: black; extra == "dev"
-Requires-Dist: flake8; extra == "dev"
-
-# Neuro-Simulator 服务端
-
-*本临时README由AI自动生成*
-
-这是 Neuro Simulator 的服务端，负责处理直播逻辑、AI 交互、TTS 合成等核心功能。
-
-## 功能特性
-
-- **动态观众**：调用无状态LLM，动态生成观众聊天内容，支持 Gemini 和 OpenAI API
-- **配置管理**：支持通过 API 动态修改和热重载配置
-- **外部控制**：完全使用外部API端点操控服务端运行
-
-## 目录结构
-
-``` main
-neuro_simulator/
-├── main.py              # 应用入口和核心逻辑
-├── config.py            # 配置管理模块
-├── letta.py             # Letta Agent 集成
-├── chatbot.py           # 观众聊天生成器
-├── audio_synthesis.py   # 音频合成模块
-├── stream_chat.py       # 聊天消息处理
-├── stream_manager.py    # 直播管理器
-├── websocket_manager.py # WebSocket 连接管理
-├── process_manager.py   # 进程管理器
-├── shared_state.py      # 全局状态管理
-├── log_handler.py       # 日志处理模块
-├── requirements.txt     # Python 依赖列表
-├── pyproject.toml       # Python 包安装配置
-├── cli.py               # 命令行启动脚本
-├── config.yaml.example # 自带的备用配置模板
-└── media/               # 自带的备用媒体文件
-    └── neuro_start.mp4  # 用来计算Start Soon长度，仅读取时长
-```
-
-``` workin'dir
-working_dir_example/     # 工作目录结构，请将这个目录重命名和复制到你想要的位置（推荐放到~/.config/neuro-simulator）
-├── media/               # 媒体文件夹，如缺失会使用自带资源覆盖
-│   └── neuro_start.mp4  # 用来计算Start Soon长度，仅读取时长,请和客户端的视频保持一致
-├── config.yaml        # 由用户手工创建的配置文件
-└── config.yaml.example # 自动生成的配置文件模板，必须手动重命名和填写
-```
-
-## 安装与配置
-
-0. **在运行server前，必须有已经配置完成的Letta Agent。**
-1. 复制一份 `../docs/working_dir_example` 到你想要的位置，作为配置文件目录.
-   - 程序会在未指定 `--dir` 的情况下自动生成一个工作目录，路径为 `~/.config/neuro-simulator/`
-2. 然后进入配置文件目录，复制 `config.yaml.example` 到 `config.yaml`
-3. 编辑 `config.yaml` 文件，填入必要的 API 密钥和配置项：
-   - Letta Token 和 Agent ID
-   - Gemini/OpenAI API Key
-   - Azure TTS Key 和 Region
-
-可以执行替换media/neuro_start.mp4为其它视频文件，但记得手动替换client中的同名文件。
-
-### 直接安装方式（无需二次开发）
-
-若无需二次开发，可以直接使用pip安装：
-```bash
-python3 -m venv venv
-# Windows
-venv/Scripts/pip install neuro-simulator
-# macOS/Linux
-venv/bin/pip install neuro-simulator
-```
-
-### 二次开发方式
-
-若需要二次开发，请克隆项目，在server下建立venv，然后pip install -e ./：
-```bash
-git clone https://github.com/your-username/Neuro-Simulator.git
-cd Neuro-Simulator/server
-python3 -m venv venv
-# Windows
-venv/Scripts/pip install -e .
-# macOS/Linux
-venv/bin/pip install -e .
-```
-
-### 运行服务
-
-```bash
-# 使用默认配置 (位于~/.config/neuro-simulator/)
-neuro
-
-# 指定工作目录
-neuro -D /path/to/your/config
-
-# 指定主机和端口
-neuro -H 0.0.0.0 -P 8080
-
-# 组合使用
-neuro -D /path/to/your/config -H 0.0.0.0 -P 8080
-```
-
-服务默认运行在 `http://127.0.0.1:8000`。
-
-## API 接口
-
-后端提供丰富的 API 接口用于控制和管理：
-
-- `/api/stream/*` - 直播控制接口（启动/停止/重启/状态）
-- `/api/configs/*` - 配置管理接口（获取/更新/重载配置）
-  - `api_keys` `server` 等敏感配置项无法从接口获取和修改。
-- `/api/logs` - 日志获取接口
-- `/api/tts/synthesize` - TTS 合成接口
-- `/api/system/health` - 健康检查接口
-- `/ws/stream` - 直播内容 WebSocket 接口
-- `/ws/logs` - 日志流 WebSocket 接口
-
-详细接口说明可通过 `http://127.0.0.1:8000/docs` 访问 API 文档查看。
-
-## 配置说明
-
-配置文件 `config.yaml` 包含以下主要配置项：
-
-- `api_keys` - 各种服务的 API 密钥
-- `stream_metadata` - 直播元数据（标题、分类、标签等）
-- `neuro_behavior` - Neuro 行为设置
-- `audience_simulation` - 观众模拟设置
-- `tts` - TTS 语音合成设置
-- `performance` - 性能相关设置
-- `server` - 服务器设置（主机、端口、CORS 等）
-
-有关配置文件的完整示例，请参阅项目根目录下的 `docs/working_dir_example/` 文件夹。
-
-## 安全说明
-
-1. 通过 `panel_password` 配置项可以设置控制面板访问密码
-2. 敏感配置项（如 API 密钥）不会通过 API 接口暴露
-3. 支持 CORS，仅允许预配置的来源访问
-
-## 故障排除
-
-- 确保所有必需的 API 密钥都已正确配置
-- 检查网络连接是否正常
-- 查看日志文件获取错误信息
-- 确保端口未被其他程序占用 
+Metadata-Version: 2.4
+Name: neuro_simulator
+Version: 0.2.1
+Summary: Neuro Simulator Server
+Author-email: Moha-Master <hongkongreporter@outlook.com>
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/Moha-Master/neuro-simulator
+Project-URL: Repository, https://github.com/Moha-Master/neuro-simulator
+Project-URL: Issues, https://github.com/Moha-Master/neuro-simulator/issues
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: fastapi
+Requires-Dist: uvicorn
+Requires-Dist: google-genai
+Requires-Dist: azure-cognitiveservices-speech
+Requires-Dist: letta-client
+Requires-Dist: openai
+Requires-Dist: pyyaml
+Requires-Dist: pydantic
+Requires-Dist: jinja2
+Requires-Dist: python-multipart
+Requires-Dist: mutagen
+Provides-Extra: dev
+Requires-Dist: pytest>=6.0; extra == "dev"
+Requires-Dist: pytest-cov; extra == "dev"
+Requires-Dist: black; extra == "dev"
+Requires-Dist: flake8; extra == "dev"
+
+# Neuro-Simulator 服务端
+
+*本临时README由AI自动生成*
+
+这是 Neuro Simulator 的服务端，负责处理直播逻辑、AI 交互、TTS 合成等核心功能
+
+## 功能特性
+
+- **动态观众**：调用无状态LLM，动态生成观众聊天内容，支持 Gemini 和 OpenAI API
+- **配置管理**：支持通过 API 动态修改和热重载配置
+- **外部控制**：完全使用外部API端点操控服务端运行
+
+## 目录结构
+
+``` main
+neuro_simulator/
+├── __init__.py
+├── cli.py               # 命令行启动脚本
+├── core/                # 核心模块
+│   ├── __init__.py
+│   ├── application.py   # FastAPI应用和主要路由
+│   ├── config.py        # 配置管理模块
+│   ├── agent_factory.py # Agent工厂模式实现
+│   ├── agent_interface.py # Agent接口定义
+│   └── config.yaml.example # 自带的备用配置模板
+├── agent/               # 内建Agent模块
+│   ├── __init__.py
+│   ├── base.py          # Agent基类
+│   ├── core.py          # Agent核心实现
+│   ├── factory.py       # Agent工厂
+│   ├── llm.py           # LLM客户端
+│   ├── memory/          # 记忆管理模块
+│   │   ├── __init__.py
+│   │   ├── manager.py   # 记忆管理器
+│   │   ├── context.json # 上下文记忆文件
+│   │   ├── core_memory.json # 核心记忆文件
+│   │   ├── init_memory.json # 初始化记忆文件
+│   │   └── temp_memory.json # 临时记忆文件
+│   └── tools/           # 工具模块
+│       ├── __init__.py
+│       └── core.py      # 核心工具实现
+├── api/                 # API路由模块
+│   ├── __init__.py
+│   ├── agent.py         # Agent管理API
+│   ├── stream.py        # 直播控制API
+│   └── system.py        # 系统管理API
+├── services/            # 服务模块
+│   ├── __init__.py
+│   ├── audience.py      # 观众聊天生成器
+│   ├── audio.py         # 音频合成模块
+│   ├── builtin.py       # 内建Agent服务
+│   ├── letta.py         # Letta Agent 集成
+│   └── stream.py        # 直播管理服务
+├── utils/               # 工具模块
+│   ├── __init__.py
+│   ├── logging.py       # 日志处理模块
+│   ├── process.py       # 进程管理模块
+│   ├── queue.py         # 队列处理模块
+│   ├── state.py         # 状态管理模块
+│   └── websocket.py     # WebSocket连接管理
+├── assets/              # 自带的备用媒体文件
+│   └── neuro_start.mp4  # 用来计算Start Soon长度，仅读取时长
+├── requirements.txt     # Python 依赖列表
+└── pyproject.toml       # Python 包安装配置
+```
+
+``` workin'dir
+working_dir_example/     # 工作目录结构，请将这个目录重命名和复制到你想要的位置（推荐放到~/.config/neuro-simulator）
+├── assets/               # 媒体文件夹，如缺失会使用自带资源覆盖
+│   └── neuro_start.mp4  # 用来计算Start Soon长度，仅读取时长,请和客户端的视频保持一致
+├── config.yaml          # 由用户手工创建的配置文件
+├── config.yaml.example  # 自动生成的配置文件模板，必须手动重命名和填写
+└── agent/               # Agent相关文件夹
+    └── memory/          # Agent记忆文件夹
+        ├── context.json     # 上下文记忆文件
+        ├── core_memory.json # 核心记忆文件
+        ├── init_memory.json # 初始化记忆文件
+        └── temp_memory.json # 临时记忆文件
+```
+
+## 安装与配置
+
+1. 复制一份 `../docs/working_dir_example` 到你想要的位置，作为配置文件目录.
+   - 程序会在未指定 `--dir` 的情况下自动生成一个工作目录，路径为 `~/.config/neuro-simulator/`
+2. 然后进入配置文件目录，复制 `config.yaml.example` 到 `config.yaml`
+3. 编辑 `config.yaml` 文件，填入必要的 API 密钥和配置项：
+   - 如果使用 Letta Agent，需要配置 Letta Token 和 Agent ID
+   - Gemini/OpenAI API Key（用于观众聊天生成和 Agent）
+   - Azure TTS Key 和 Region
+
+可以自行替换 `$dir/assets/neuro_start.mp4` 为其它视频文件，但记得手动替换 client 中的同名文件
+
+### Agent配置
+
+服务端支持两种Agent类型：
+1. **Letta Agent**：需要配置 Letta Cloud 或自托管的 Letta Server
+2. **内建 Agent**：使用服务端自带的 Agent，支持 Gemini 和OpenAI API
+
+在 `config.yaml` 中通过 `agent_type` 字段选择使用的 Agent 类型：
+- `agent_type: "letta"`：使用 Letta Agent
+- `agent_type: "builtin"`：使用内建 Agent
+
+当使用内建Agent时，还需要配置：
+- `agent.agent_provider`：选择"gemini"或"openai"
+- `agent.agent_model`：指定具体的模型名称
+
+### 直接安装方式（无需二次开发）
+
+若无需二次开发，可以直接使用 pip 安装：
+```bash
+python3 -m venv venv
+# Windows
+venv/Scripts/pip install neuro-simulator
+# macOS/Linux
+venv/bin/pip install neuro-simulator
+```
+
+### 二次开发方式
+
+若需要二次开发，请克隆项目：
+```bash
+git clone https://github.com/your-username/Neuro-Simulator.git
+cd Neuro-Simulator/server
+python3 -m venv venv
+# Windows
+venv/Scripts/pip install -e .
+# macOS/Linux
+venv/bin/pip install -e .
+```
+
+### 运行服务
+
+```bash
+# 使用默认配置 (位于~/.config/neuro-simulator/)
+neuro
+
+# 指定工作目录
+neuro -D /path/to/your/config
+
+# 指定主机和端口
+neuro -H 0.0.0.0 -P 8080
+
+# 组合使用
+neuro -D /path/to/your/config -H 0.0.0.0 -P 8080
+```
+
+服务默认运行在 `http://127.0.0.1:8000`
+
+## API 接口
+
+后端提供丰富的 API 接口用于控制和管理：
+
+- `/api/stream/*` - 直播控制接口（启动/停止/重启/状态）
+- `/api/configs/*` - 配置管理接口（获取/更新/重载配置）
+  - `api_keys` `server` 等敏感配置项无法从接口获取和修改
+- `/api/logs` - 日志获取接口
+- `/api/tts/synthesize` - TTS 合成接口
+- `/api/system/health` - 健康检查接口
+- `/ws/stream` - 客户端使用的直播接口
+- `/ws/admin` - 日志和内建 Agent的 Context 流接口
+
+详细接口说明可通过 `http://127.0.0.1:8000/docs` 访问 API 文档查
+
+## 配置说明
+
+配置文件 `config.yaml` 包含以下主要配置项：
+
+- `api_keys` - 各种服务的 API 密钥
+- `stream_metadata` - 直播元数据（标题、分类、标签等）
+- `neuro_behavior` - Neuro 行为设置
+- `audience_simulation` - 观众模拟设置
+- `tts` - TTS 语音合成设置
+- `performance` - 性能相关设置
+- `server` - 服务器设置（主机、端口、CORS 等）
+
+有关配置文件的完整示例，请参阅项目根目录下的 `docs/working_dir_example/` 文件夹
+
+## 安全说明
+
+1. 通过 `panel_password` 配置项可以设置控制面板访问密码
+2. 敏感配置项（如 API 密钥）不会通过 API 接口暴露
+3. 支持 CORS，仅允许预配置的来源访问
+
+## 故障排除
+
+- 确保所有必需的 API 密钥都已正确配置
+- 检查网络连接是否正常
+- 查看日志文件获取错误信息
+- 确保端口未被其他程序占用 

{neuro_simulator-0.2.0.dist-info → neuro_simulator-0.2.1.dist-info}/RECORD

@@ -1,8 +1,8 @@
 neuro_simulator/__init__.py,sha256=-tposzyvg6UckPcfSvtc03UjxBa9oCe_zRvlKf8splk,31
-neuro_simulator/cli.py,sha256=nf69ABiVSLsOSVU0uL-bfxHN_N4kStb8kmCSetF1ffY,3606
+neuro_simulator/cli.py,sha256=p2jSNbMsQ2HYau-EX3ygmTRhECU9dbSg0v53xgYjOZ4,3894
 neuro_simulator/agent/__init__.py,sha256=t52CZlyTGWqcGjMs90qvpFpRckY2WSSlO7r_H3K_mSY,32
 neuro_simulator/agent/base.py,sha256=6v2ZO5UpGCwJEkJ23Oe96Rs510tK4ZOEpZ2DB49IZmM,1262
-neuro_simulator/agent/core.py,sha256=WbIOU15LK9YKVeY0Osp9PJEzs0NXfgCp-aj9t9daVHc,9265
+neuro_simulator/agent/core.py,sha256=CJv0We7ZjW_EKQB54Xq-y3wDu2a2HmoaOAdYR057HOg,8941
 neuro_simulator/agent/factory.py,sha256=e0IBnqJQM7OuKtglrf-pWwqwmg98wh7tOq5LxF2rV-w,1146
 neuro_simulator/agent/llm.py,sha256=vLz8hp2h2R0JaNfS1RLGYGkri_YoUdlEdNfFVbxeEuI,4261
 neuro_simulator/agent/memory/__init__.py,sha256=YJ7cynQJI6kD7vjyv3rKc-CZqmoYSuGQtRZl_XdGEps,39
@@ -30,8 +30,8 @@ neuro_simulator/utils/process.py,sha256=FmwIuOsp19jJBEbAj-WeUpdUsQN7B9_J1rlU0Q1T
 neuro_simulator/utils/queue.py,sha256=7SSnUnrqxHkqSeYVjp1S5kpr8qlo4IVdU2NsGwFQAVw,1546
 neuro_simulator/utils/state.py,sha256=E1ilecgMTOAhiw_kNvLG0akkhdZKhzKLrA3oB4NNVTA,538
 neuro_simulator/utils/websocket.py,sha256=yOdFvJzbNhcUn5EAuyS55G_R8q-snas5OvkOtS8g19E,2292
-neuro_simulator-0.2.0.dist-info/METADATA,sha256=keF5r6S8qz28e89k3fxpaDh5iUOGZlSOh_WtZevaDfU,6542
-neuro_simulator-0.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-neuro_simulator-0.2.0.dist-info/entry_points.txt,sha256=qVd5ypnRRgU8Cw7rWfZ7o0OXyS9P9hgY-cRoN_mgz9g,51
-neuro_simulator-0.2.0.dist-info/top_level.txt,sha256=V8awSKpcrFnjJDiJxSfy7jtOrnuE2BgAR9hLmfMDWK8,16
-neuro_simulator-0.2.0.dist-info/RECORD,,
+neuro_simulator-0.2.1.dist-info/METADATA,sha256=bbJ6VZxoq8fqq0qVzjm0nYUKIhwfoNPNQd8hDKPSREY,8676
+neuro_simulator-0.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+neuro_simulator-0.2.1.dist-info/entry_points.txt,sha256=qVd5ypnRRgU8Cw7rWfZ7o0OXyS9P9hgY-cRoN_mgz9g,51
+neuro_simulator-0.2.1.dist-info/top_level.txt,sha256=V8awSKpcrFnjJDiJxSfy7jtOrnuE2BgAR9hLmfMDWK8,16
+neuro_simulator-0.2.1.dist-info/RECORD,,