如何添加自定义中间件

在LangGraph平台上部署代理时，您可以向服务器添加自定义中间件来处理诸如日志记录请求指标、注入或检查头信息以及强制执行安全策略等跨切面关注点，而无需修改核心服务器逻辑。这与添加自定义路由的方式相同——您只需提供自己的Starlette应用（包括FastAPI、FastHTML和其他兼容的应用）即可。

添加中间件可以让您拦截并全局修改请求和响应，无论它们是命中您的自定义端点还是内置的LangGraph平台API。

以下是一个使用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 fastapi import FastAPI, Request
from starlette.middleware.base import BaseHTTPMiddleware

app = FastAPI()

class CustomHeaderMiddleware(BaseHTTPMiddleware):
    async def dispatch(self, request: Request, call_next):
        response = await call_next(request)
        response.headers['X-Custom-Header'] = 'Hello from middleware!'
        return response

# 将中间件添加到应用中
app.add_middleware(CustomHeaderMiddleware)

配置 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

现在，对你的服务器发出的任何请求都将包含自定义头X-Custom-Header。

请注意，代码块中的命令未进行翻译。

部署

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

下一步

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

Comments