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

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

这与添加自定义路由的工作方式相同 - 您只需要提供您自己的 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):
    # for example...
    engine = create_async_engine("postgresql+asyncpg://user:pass@localhost/db")
    # Create reusable session factory
    async_session = sessionmaker(engine, class_=AsyncSession)
    # Store in app state
    app.state.db_session = async_session
    yield
    # Clean up connections
    await engine.dispose()

app = FastAPI(lifespan=lifespan)

# ... can add custom routes if needed.

配置 langgraph.json

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

{
  "dependencies": ["."],
  "graphs": {
    "agent": "./src/agent/graph.py:graph"
  },
  "env": ".env",
  "http": {
    "app": "./src/agent/webapp.py:app"
  }
  // Other configuration options like auth, store, etc.
}

启动服务器

在本地测试服务器

langgraph dev --no-browser

您应该在服务器启动时看到您的启动消息被打印出来，并在您使用 Ctrl+C 停止服务器时看到您的清理消息。

部署

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

下一步

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

评论