如何添加自定义生命周期事件

在LangGraph平台上部署代理时，您通常需要在服务器启动时初始化资源（如数据库连接），并在服务器关闭时确保它们被正确关闭。生命周期事件允许您挂钩到服务器的启动和关闭序列，以处理这些关键的设置和拆卸任务。

这与添加自定义路由的方式相同——您只需要提供自己的Starlette应用程序（包括FastAPI、FastHTML和其他兼容的应用程序）即可。

以下是一个使用FastAPI的示例。

仅限Python

当前我们仅支持在使用langgraph-api>=0.0.26的Python部署中添加自定义生命周期事件。

创建应用

从一个**现有**的LangGraph平台应用程序开始，在你的webapp.py文件中添加以下生命周期代码。如果你是从头开始创建应用，可以使用CLI从模板创建一个新的应用。

langgraph new --template=new-langgraph-project-python my_new_project

一旦你有了一个LangGraph项目，添加以下应用代码：

# ./src/agent/webapp.py
from contextlib import asynccontextmanager
from fastapi import FastAPI
from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
from sqlalchemy.orm import sessionmaker

@asynccontextmanager
async def lifespan(app: FastAPI):
    # 例如...
    engine = create_async_engine("postgresql+asyncpg://user:pass@localhost/db")
    # 创建可重复使用的会话工厂
    async_session = sessionmaker(engine, class_=AsyncSession)
    # 存储在应用状态中
    app.state.db_session = async_session
    yield
    # 清理连接
    await engine.dispose()

app = FastAPI(lifespan=lifespan)

# ... 如果需要，可以添加自定义路由。

配置 langgraph.json

在你的 langgraph.json 文件中添加以下内容。确保路径指向上面创建的 webapp.py 文件。

{
  "dependencies": ["."],
  "graphs": {
    "agent": "./src/agent/graph.py:graph"
  },
  "env": ".env",
  "http": {
    "app": "./src/agent/webapp.py:app"
  }
  // 其他配置选项如认证、存储等
}

启动服务器

在本地测试服务器：

langgraph dev --no-browser

当服务器启动时，你应该能看到启动信息被打印出来；当你使用Ctrl+C停止服务器时，也能看到清理信息。

部署

您可以将应用程序按原样部署到托管的图谱云或您自己的自托管平台。

下一步

现在您已经在部署中添加了生命周期事件，您可以使用类似的技术来添加自定义路由或自定义中间件，以进一步定制服务器的行为。

Comments