Use lwa_fastapi_middleware_bedrock_agent package in `bedrock-agent-…

…fastapi` example (#361) * Use `lwa_fastapi_middleware_bedrock_agent` middleware * Add a comment about the middleware
awslabs · Jan 29, 2024 · e052f60 · e052f60
1 parent e092a60
commit e052f60
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 83 deletions.
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