Skip to content

startup.py

作用

初始化帮助函数模块,供 chat_server.py 和外部 host app 共用,避免重复代码。

核心函数

函数说明
load_memory_bench_env()加载 memory_bench/.env.benchmark 到环境变量
resolve_memory_bench_config()解析配置(env + CLI overrides),返回配置对象
init_router_state()初始化 router 状态(OpenAI client、mem0 client 等)

使用方式

该模块不是独立 CLI,而是被其他模块导入使用:

python
from memory_bench.server.startup import (
    load_memory_bench_env,
    resolve_memory_bench_config,
    init_router_state,
)

# 在 lifespan 里初始化
load_memory_bench_env()
cfg = resolve_memory_bench_config()
init_router_state(state, cfg)

设计决策

  • 配置隔离:memory_bench 的配置与 host app 完全隔离
  • 可覆盖resolve_memory_bench_config(overrides={...}) 可覆盖 env 配置
  • 复用性:外部 app 可轻松集成 memory_bench router

魔女の实验室