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

neuro_simulator/{websocket_manager.py → utils/websocket.py}

@@ -1,45 +1,49 @@
-# backend/websocket_manager.py
-from fastapi import WebSocket
-from collections import deque
+# neuro_simulator/utils/websocket.py
 import asyncio
 import json
-from starlette.websockets import WebSocketState # 确保导入 WebSocketState
+import logging
+from collections import deque
+
+from fastapi import WebSocket
+from starlette.websockets import WebSocketState
+
+logger = logging.getLogger(__name__.replace("neuro_simulator", "server", 1))
 
 class WebSocketManager:
-    """管理所有活动的 WebSocket 连接，并提供消息广播功能。"""
+    """Manages all active WebSocket connections and provides broadcasting capabilities."""
     def __init__(self):
         self.active_connections: deque[WebSocket] = deque()
-        print("WebSocketManager 初始化完成。")
+        logger.info("WebSocketManager initialized.")
 
     async def connect(self, websocket: WebSocket):
         await websocket.accept()
         self.active_connections.append(websocket)
-        print(f"WebSocket 客户端已连接。当前连接数: {len(self.active_connections)}")
+        logger.info(f"WebSocket client connected. Total connections: {len(self.active_connections)}")
 
     def disconnect(self, websocket: WebSocket):
         try:
             if websocket in self.active_connections:
                 self.active_connections.remove(websocket)
-                print(f"WebSocket 客户端已断开连接。当前连接数: {len(self.active_connections)}")
+                logger.info(f"WebSocket client disconnected. Total connections: {len(self.active_connections)}")
         except Exception as e:
-            print(f"断开 WebSocket 连接时出错: {e}")
+            logger.error(f"Error during WebSocket disconnect: {e}")
 
     async def send_personal_message(self, message: dict, websocket: WebSocket):
         if websocket.client_state == WebSocketState.CONNECTED:
             try:
                 await websocket.send_json(message)
             except Exception as e:
-                print(f"发送个人消息时出错，客户端可能已断开: {e}")
+                logger.warning(f"Could not send personal message, client likely disconnected: {e}")
                 self.disconnect(websocket)
 
     async def broadcast(self, message: dict):
         disconnected_sockets = []
-        for connection in list(self.active_connections): 
+        for connection in list(self.active_connections):
             if connection.client_state == WebSocketState.CONNECTED:
                 try:
                     await connection.send_json(message)
                 except Exception as e:
-                    print(f"广播消息时出错，客户端 {connection} 可能已断开: {e}")
+                    logger.warning(f"Could not broadcast message to client {connection}, it may have disconnected: {e}")
                     disconnected_sockets.append(connection)
             else:
                 disconnected_sockets.append(connection)
@@ -47,5 +51,5 @@ class WebSocketManager:
         for disconnected_socket in disconnected_sockets:
             self.disconnect(disconnected_socket)
 
-# --- 核心修改点：只创建一个实例 ---
-connection_manager = WebSocketManager()
+# Global singleton instance
+connection_manager = WebSocketManager()

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

@@ -1,176 +1,176 @@
-Metadata-Version: 2.4
-Name: neuro_simulator
-Version: 0.1.3
-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.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 密钥都已正确配置
+- 检查网络连接是否正常
+- 查看日志文件获取错误信息
+- 确保端口未被其他程序占用 

neuro_simulator-0.2.0.dist-info/RECORD

@@ -0,0 +1,37 @@
+neuro_simulator/__init__.py,sha256=-tposzyvg6UckPcfSvtc03UjxBa9oCe_zRvlKf8splk,31
+neuro_simulator/cli.py,sha256=nf69ABiVSLsOSVU0uL-bfxHN_N4kStb8kmCSetF1ffY,3606
+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/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
+neuro_simulator/agent/memory/manager.py,sha256=z5ZI1agkuOev5gtuGA_g6MS3atB7ufdPRVNZZ89yg1w,9259
+neuro_simulator/agent/tools/__init__.py,sha256=1WZy6PADfi6o1avyy1y-ThWBFAPJ_bBqtkobyYpf5ao,38
+neuro_simulator/agent/tools/core.py,sha256=o6Oyis-HFD-g6Z_u3T--tkmr9ylKJvybKqMRSMUwi1Q,5555
+neuro_simulator/api/__init__.py,sha256=5LWyDSayPGdQS8Rv13nmAKLyhPnMVPyTYDdvoMPB4xw,56
+neuro_simulator/api/agent.py,sha256=ABl_JoIxB4wW_J2J52bWndmTXkfGJBS5LZmbGuh7zv8,6343
+neuro_simulator/api/stream.py,sha256=Pg793dXLnJ0oydrIwWFgWD0cVdFAx_QOw6L5LjE0uIo,2069
+neuro_simulator/api/system.py,sha256=hXznMcThuFhwopYWgpzrRxwtBuFnF_b_vinkOaE5XOs,3712
+neuro_simulator/core/__init__.py,sha256=-ojq25c8XA0CU25b0OxcGjH4IWFEDHR-HXSRSZIuKe8,57
+neuro_simulator/core/agent_factory.py,sha256=qMFidwT5IrOkyNHwmpO8_fRv20KLbaIBfWF-VTFCLNA,1742
+neuro_simulator/core/agent_interface.py,sha256=r58Opcgs7SWVovYTjMWuxF8AiTy9QfRz276_YGmSei0,2791
+neuro_simulator/core/application.py,sha256=zN8KOR_nwsYsR1ZcyZU1GlW3-ijFLbjw5lz1g8DYVIU,11578
+neuro_simulator/core/config.py,sha256=brA8kiekV_995mpz04JiGj1swIWbZZuWWLNYtbroMyE,14884
+neuro_simulator/services/__init__.py,sha256=s3ZrAHg5TpJakadAAGY1h0wDw_xqN4Je4aJwJyRBmk4,61
+neuro_simulator/services/audience.py,sha256=0phlhsujh_GMXm_UMiyOntY-ZMtoseRa_FroIfc5A6w,5028
+neuro_simulator/services/audio.py,sha256=ZscQA25wVYpm9FUl4Hya7tKH8t0TjR3th9-OEZ0G7xk,2934
+neuro_simulator/services/builtin.py,sha256=7ePxEom5HIK6wGto_H5M8JOnAjyiHqUuE381DEGgzjE,3821
+neuro_simulator/services/letta.py,sha256=iXyzFyPVty3SjYCHOJAF1QqhUFmxTQbqk8lYJOwB5pc,9415
+neuro_simulator/services/stream.py,sha256=dG7RuNI_ICohPkqKZ-zlBppo54BgWm_KYBs-ezzc73E,5907
+neuro_simulator/utils/__init__.py,sha256=xSEFzjT827W81mNyQ_DLtr00TgFlttqfFgpz9pSxFXQ,58
+neuro_simulator/utils/logging.py,sha256=BO-q_cCcoeamsc8eJqq2-L3Z8nhApze_v6LnmD-O8Ww,3411
+neuro_simulator/utils/process.py,sha256=FmwIuOsp19jJBEbAj-WeUpdUsQN7B9_J1rlU0Q1TXuo,2388
+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,,