aws-genai-llm-chatbot

1
import genai_core.types
2
import genai_core.workspaces
3
import genai_core.aurora.delete
4
import genai_core.opensearch.delete
5
import genai_core.kendra.delete
6
from aws_lambda_powertools import Logger
7
from aws_lambda_powertools.utilities.typing import LambdaContext
8

9
logger = Logger()
10

11

12
@logger.inject_lambda_context(log_event=True)
13
def lambda_handler(event, context: LambdaContext):
14
    workspace_id = event["workspace_id"]
15
    workspace = genai_core.workspaces.get_workspace(workspace_id)
16
    if workspace is None:
17
        raise genai_core.types.CommonError("Workspace not found")
18

19
    if workspace["engine"] == "aurora":
20
        genai_core.aurora.delete.delete_aurora_workspace(workspace)
21
    elif workspace["engine"] == "opensearch":
22
        genai_core.opensearch.delete.delete_open_search_workspace(workspace)
23
    elif workspace["engine"] == "kendra":
24
        genai_core.kendra.delete.delete_kendra_workspace(workspace)
25
    else:
26
        raise genai_core.types.CommonError("Workspace engine not supported")
27