aws-genai-llm-chatbot

1
import boto3
2
import os
3
import json
4
from datetime import datetime
5
from aws_lambda_powertools import Logger, Tracer
6
from aws_lambda_powertools.utilities.typing import LambdaContext
7

8
tracer = Tracer()
9
logger = Logger(log_uncaught_exceptions=True)
10

11
sns = boto3.client("sns")
12
TOPIC_ARN=os.environ.get("SNS_TOPIC_ARN", "")
13

14
@tracer.capture_lambda_handler
15
@logger.inject_lambda_context(log_event=True)
16
def handler(event, context: LambdaContext): 
17
    print(event["arguments"]["data"])
18
    print(event["identity"])
19
    request = json.loads(event["arguments"]["data"])
20
    message = {
21
        "action": request["action"],
22
        "modelInterface": request["modelInterface"],
23
        "direction": "IN",
24
        "timestamp": str(int(round(datetime.now().timestamp()))),
25
        "userId": event["identity"]["sub"],
26
        "data": request.get("data", {}),
27
    }
28
    print(message)
29

30
    response = sns.publish(
31
        TopicArn=TOPIC_ARN, Message=json.dumps(message)
32
        )
33
    
34
    return response
35