aws-genai-llm-chatbot

1
import genai_core.workspaces
2
import genai_core.aurora.create
3
from aws_lambda_powertools import Logger
4
from aws_lambda_powertools.utilities.typing import LambdaContext
5

6
logger = Logger()
7

8

9
@logger.inject_lambda_context(log_event=True)
10
def lambda_handler(event, context: LambdaContext):
11
    workspace_id = event["workspace_id"]
12
    logger.info(f"Creating workspace {workspace_id}")
13

14
    workspace = genai_core.workspaces.get_workspace(workspace_id)
15
    if not workspace:
16
        raise Exception(f"Workspace {workspace_id} does not exist")
17

18
    genai_core.aurora.create.create_workspace_table(workspace)
19

20
    return {"ok": True}
21