1
export const handler = async (event: any) => {
2
  const { prompt, sessionId } = JSON.parse(event.body);
3

4
  // 1. Check cache
5
  const cached = await redis.get(`exact:${hash(prompt)}`);
6
  if (cached) return respond(200, { response: cached, cached: true });
7

8
  // 2. Load session từ DynamoDB
9
  const session = await loadSession(sessionId);
10

11
  // 3. Call Bedrock
12
  const llm = await callBedrock(prompt, session?.context);
13

14
  // 4. Execute tools (nếu LLM yêu cầu)
15
  const toolResults = await executeTools(llm.toolCalls);
16

17
  // 5. Final response sau tools
18
  const final = toolResults.length
19
    ? await callBedrockWithTools(prompt, llm.text, toolResults)
20
    : { text: llm.text };
21

22
  // 6. Save session
23
  await saveSession(sessionId, {
24
    context: [...context, {role:"user",content:prompt}, {role:"assistant",content:final.text}],
25
  });
26

27
  return respond(200, { response: final.text, sessionId, cached: false });
28
};

1
const tools = [{
2
  toolSpec: {
3
    name: "github_list_issues",
4
    description: "List issues từ GitHub repository",
5
    inputSchema: { json: { type: "object", properties: { owner: {type:"string"}, repo: {type:"string"} }, required: ["owner","repo"] } },
6
  },
7
}];

1
export const wsHandler = async (event: any) => {
2
  const connectionId = event.requestContext.connectionId;
3

4
  if (event.requestContext.routeKey === "$connect") {
5
    // Lưu connection
6
    return { statusCode: 200 };
7
  }
8

9
  // Xử lý async → gửi kết quả qua WebSocket
10
  processAgentQuery(JSON.parse(event.body).prompt, connectionId);
11
  return { statusCode: 202 };
12
};

1
{
2
  "StartAt": "Analyze Request",
3
  "States": {
4
    "Analyze Request": {
5
      "Type": "Task",
6
      "Resource": "arn:aws:states:::sdk:bedrockruntime:converse",
7
      "Next": "Execute Steps"
8
    },
9
    "Execute Steps": {
10
      "Type": "Map",
11
      "ItemProcessor": {
12
        "StartAt": "Execute",
13
        "States": {
14
          "Execute": { "Type": "Task", "Resource": "arn:aws:lambda:::function:tool-executor", "End": true }
15
        }
16
      },
17
      "Next": "Summarize"
18
    },
19
    "Summarize": {
20
      "Type": "Task",
21
      "Resource": "arn:aws:states:::sdk:bedrockruntime:converse",
22
      "End": true
23
    }
24
  }
25
}