neuro-simulator 0.1.3__py3-none-any.whl → 0.2.1__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.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: neuro_simulator
-Version: 0.1.3
+Version: 0.2.1
 Summary: Neuro Simulator Server
 Author-email: Moha-Master <hongkongreporter@outlook.com>
 License-Expression: MIT
@@ -39,7 +39,7 @@ Requires-Dist: flake8; extra == "dev"
 
 *本临时README由AI自动生成*
 
-这是 Neuro Simulator 的服务端，负责处理直播逻辑、AI 交互、TTS 合成等核心功能。
+这是 Neuro Simulator 的服务端，负责处理直播逻辑、AI 交互、TTS 合成等核心功能
 
 ## 功能特性
 
@@ -51,49 +51,99 @@ Requires-Dist: flake8; extra == "dev"
 
 ``` 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 包安装配置
+├── __init__.py
 ├── cli.py               # 命令行启动脚本
-├── config.yaml.example # 自带的备用配置模板
-└── media/               # 自带的备用媒体文件
-    └── neuro_start.mp4  # 用来计算Start Soon长度，仅读取时长
+├── 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）
-├── media/               # 媒体文件夹，如缺失会使用自带资源覆盖
+├── assets/               # 媒体文件夹，如缺失会使用自带资源覆盖
 │   └── neuro_start.mp4  # 用来计算Start Soon长度，仅读取时长,请和客户端的视频保持一致
-├── config.yaml        # 由用户手工创建的配置文件
-└── config.yaml.example # 自动生成的配置文件模板，必须手动重命名和填写
+├── config.yaml          # 由用户手工创建的配置文件
+├── config.yaml.example  # 自动生成的配置文件模板，必须手动重命名和填写
+└── agent/               # Agent相关文件夹
+    └── memory/          # Agent记忆文件夹
+        ├── context.json     # 上下文记忆文件
+        ├── core_memory.json # 核心记忆文件
+        ├── init_memory.json # 初始化记忆文件
+        └── temp_memory.json # 临时记忆文件
 ```
 
 ## 安装与配置
 
-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
+   - 如果使用 Letta Agent，需要配置 Letta Token 和 Agent ID
+   - Gemini/OpenAI API Key（用于观众聊天生成和 Agent）
    - Azure TTS Key 和 Region
 
-可以执行替换media/neuro_start.mp4为其它视频文件，但记得手动替换client中的同名文件。
+可以自行替换 `$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安装：
+若无需二次开发，可以直接使用 pip 安装：
 ```bash
 python3 -m venv venv
 # Windows
@@ -104,7 +154,7 @@ 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
@@ -131,7 +181,7 @@ 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`。
+服务默认运行在 `http://127.0.0.1:8000`
 
 ## API 接口
 
@@ -139,14 +189,14 @@ neuro -D /path/to/your/config -H 0.0.0.0 -P 8080
 
 - `/api/stream/*` - 直播控制接口（启动/停止/重启/状态）
 - `/api/configs/*` - 配置管理接口（获取/更新/重载配置）
-  - `api_keys` `server` 等敏感配置项无法从接口获取和修改。
+  - `api_keys` `server` 等敏感配置项无法从接口获取和修改
 - `/api/logs` - 日志获取接口
 - `/api/tts/synthesize` - TTS 合成接口
 - `/api/system/health` - 健康检查接口
-- `/ws/stream` - 直播内容 WebSocket 接口
-- `/ws/logs` - 日志流 WebSocket 接口
+- `/ws/stream` - 客户端使用的直播接口
+- `/ws/admin` - 日志和内建 Agent的 Context 流接口
 
-详细接口说明可通过 `http://127.0.0.1:8000/docs` 访问 API 文档查看。
+详细接口说明可通过 `http://127.0.0.1:8000/docs` 访问 API 文档查
 
 ## 配置说明
 
@@ -160,7 +210,7 @@ neuro -D /path/to/your/config -H 0.0.0.0 -P 8080
 - `performance` - 性能相关设置
 - `server` - 服务器设置（主机、端口、CORS 等）
 
-有关配置文件的完整示例，请参阅项目根目录下的 `docs/working_dir_example/` 文件夹。
+有关配置文件的完整示例，请参阅项目根目录下的 `docs/working_dir_example/` 文件夹
 
 ## 安全说明
 

neuro_simulator-0.2.1.dist-info/RECORD

@@ -0,0 +1,37 @@
+neuro_simulator/__init__.py,sha256=-tposzyvg6UckPcfSvtc03UjxBa9oCe_zRvlKf8splk,31
+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=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
+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.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,,