KingOchoa/help-ngo-bot-fastapi-poc
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

udpated

Browse Source
This commit is contained in:
KingOchoa 2025-05-30 02:47:30 +08:00
parent 62a8a9c833
commit 5ad5cbab65
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
main.py
View File

@ -7,17 +7,23 @@ import time
app = FastAPI() app = FastAPI()
os.environ['AWS_REGION'] = 'us-east-1'
# Define request body # Define request body
class BedrockRequest(BaseModel): class BedrockRequest(BaseModel):
question: str question: str
requestSessionId: str = 'user' requestSessionId: str = 'user'
agentId: str = "ROJCGWHSC0" agentId: str = "1"
agentAliasId: str ="TQ8VDTVQII" agentAliasId: str ="1"
# AWS Bedrock client setup # AWS Bedrock client setup
def get_bedrock_client(): def get_bedrock_client():
return boto3.client( return boto3.client(
'bedrock-agent-runtime' 'bedrock-agent-runtime',
region_name=os.getenv('AWS_REGION', 'us-east-1'),
aws_access_key_id=os.getenv('AWS_ACCESS_KEY_ID'),
aws_secret_access_key=os.getenv('AWS_SECRET_ACCESS_KEY'),
aws_session_token=os.getenv('AWS_SESSION_TOKEN')
) )
@app.get("/health") @app.get("/health")