udpated
This commit is contained in:
parent
62a8a9c833
commit
5ad5cbab65
12
main.py
12
main.py
@ -7,17 +7,23 @@ import time
|
||||
|
||||
app = FastAPI()
|
||||
|
||||
os.environ['AWS_REGION'] = 'us-east-1'
|
||||
|
||||
# Define request body
|
||||
class BedrockRequest(BaseModel):
|
||||
question: str
|
||||
requestSessionId: str = 'user'
|
||||
agentId: str = "ROJCGWHSC0"
|
||||
agentAliasId: str ="TQ8VDTVQII"
|
||||
agentId: str = "1"
|
||||
agentAliasId: str ="1"
|
||||
|
||||
# AWS Bedrock client setup
|
||||
def get_bedrock_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")
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user