Use lwa_fastapi_middleware_bedrock_agent package in bedrock-agent-fastapi example (#361)

bnusunny · web-flow · commit e052f60d8de1 · 2024-01-29T16:25:20.000+08:00
* Use `lwa_fastapi_middleware_bedrock_agent` middleware

* Add a comment about the middleware
diff --git a/examples/bedrock-agent-fastapi/app/main.py b/examples/bedrock-agent-fastapi/app/main.py
@@ -1,7 +1,8 @@
 import datetime
 
 import boto3
-from middleware import BedrockAgentMiddleware
+# This is from "lwa_fastapi_middleware_bedrock_agent" package (https://pypi.org/project/lwa-fastapi-middleware-bedrock-agent/)
+from bedrock_agent.middleware import BedrockAgentMiddleware
 from fastapi import FastAPI, Query
 from pydantic import BaseModel, Field
 
diff --git a/examples/bedrock-agent-fastapi/app/middleware.py b/examples/bedrock-agent-fastapi/app/middleware.py
diff --git a/examples/bedrock-agent-fastapi/app/requirements.txt b/examples/bedrock-agent-fastapi/app/requirements.txt
@@ -9,6 +9,7 @@ h11==0.14.0
 httptools==0.6.1
 idna==3.6
 jmespath==1.0.1
+lwa_fastapi_middleware_bedrock_agent==0.0.4
 pydantic==2.5.3
 pydantic_core==2.14.6
 python-dateutil==2.8.2
