chore: implementation

Author: dreamorosi

packages/event-handler/src/bedrock-agent/BedrockAgentFunctionResolver.ts

@@ -18,10 +18,36 @@ import { assertBedrockAgentFunctionEvent } from './utils.js';
  *
  * This resolver is designed to handle function invocations from Bedrock Agents.
  *
+ * @example
+ * ```ts
+ * import {
+ *   BedrockAgentFunctionResolver
+ * } from '@aws-lambda-powertools/event-handler/bedrock-agent';
  *
+ * const app = new BedrockAgentFunctionResolver();
+ *
+ * app.tool(async (params) => {
+ *   const { name } = params;
+ *   return `Hello, ${name}!`;
+ * }, {
+ *   name: 'greeting',
+ *   description: 'Greets a person by name',
+ * });
+ *
+ * export const handler = async (event, context) =>
+ *   app.resolve(event, context);
+ * ```
  */
 export class BedrockAgentFunctionResolver {
+  /**
+   * Registry of tools added to the Bedrock Agent Function Resolver.
+   */
   readonly #tools: Map<string, Tool> = new Map();
+  /**
+   * A logger instance to be used for logging debug, warning, and error messages.
+   *
+   * When no logger is provided, we'll only log warnings and errors using the global `console` object.
+   */
   readonly #logger: Pick<GenericLogger, 'debug' | 'warn' | 'error'>;
 
   constructor(options?: ResolverOptions) {
@@ -198,6 +224,7 @@ export class BedrockAgentFunctionResolver {
       actionGroup,
       sessionAttributes,
       promptSessionAttributes,
+      // TODO: add knowledge bases
     } = event;
 
     const tool = this.#tools.get(toolName);
@@ -210,7 +237,10 @@ export class BedrockAgentFunctionResolver {
         body: `Error: tool ${toolName} has not been registered.`,
         sessionAttributes,
         promptSessionAttributes,
-      }).build();
+      }).build({
+        actionGroup,
+        func: toolName,
+      });
     }
 
     const toolParams: Record<string, ParameterValue> = {};
@@ -237,7 +267,10 @@ export class BedrockAgentFunctionResolver {
     try {
       const response = await tool.handler(toolParams, { event, context });
       if (response instanceof BedrockFunctionResponse) {
-        return response.build();
+        const res = response.build({
+          actionGroup,
+          func: toolName,
+        });
       }
       const body =
         isNullOrUndefined(response) || response === ''
@@ -249,7 +282,10 @@ export class BedrockAgentFunctionResolver {
         body,
         sessionAttributes,
         promptSessionAttributes,
-      }).build();
+      }).build({
+        actionGroup,
+        func: toolName,
+      });
     } catch (error) {
       this.#logger.error(`An error occurred in tool ${toolName}.`, error);
       return new BedrockFunctionResponse({
@@ -258,7 +294,10 @@ export class BedrockAgentFunctionResolver {
         body: `Unable to complete tool execution due to ${error instanceof Error ? `${error.name} - ${error.message}` : String(error)}`,
         sessionAttributes,
         promptSessionAttributes,
-      }).build();
+      }).build({
+        actionGroup,
+        func: toolName,
+      });
     }
   }
 }

packages/event-handler/src/bedrock-agent/BedrockFunctionResponse.ts

@@ -12,16 +12,6 @@ import type { BedrockAgentFunctionResolver } from './BedrockAgentFunctionResolve
  * when you return anything from your function handler other than an instance of this class.
  */
 class BedrockFunctionResponse {
-  /**
-   * The name of the action group, this comes from the `event.actionGroup` field
-   * in the Bedrock agent function event.
-   */
-  readonly actionGroup: string;
-  /**
-   * The name of the function returning the response, this comes from the `event.function` field
-   * in the Bedrock agent function event.
-   */
-  readonly func: string;
   /**
    * The response object that defines the response from execution of the function.
    */
@@ -60,8 +50,6 @@ class BedrockFunctionResponse {
     sessionAttributes?: Record<string, string>;
     promptSessionAttributes?: Record<string, string>;
   }) {
-    this.actionGroup = actionGroup;
-    this.func = func;
     this.body = body;
     this.responseState = responseState;
     this.sessionAttributes = sessionAttributes;
@@ -70,13 +58,20 @@ class BedrockFunctionResponse {
 
   /**
    * Builds the Bedrock function response object according to the Bedrock agent function {@link https://docs.aws.amazon.com/bedrock/latest/userguide/agents-lambda.html#agents-lambda-response | response format}.
+   *
+   * @param options - The options for building the response.
+   * @param options.actionGroup - The action group of the function, this comes from the `event.actionGroup` field in the Bedrock agent function event.
+   * @param options.func - The name of the function being invoked by the agent, this comes from the `event.function` field in the Bedrock agent function event.
    */
-  build() {
+  build(options: {
+    actionGroup: string;
+    func: string;
+  }) {
     return {
       messageVersion: '1.0',
       response: {
-        actionGroup: this.actionGroup,
-        function: this.func,
+        actionGroup: options.actionGroup,
+        function: options.func,
         functionResponse: {
           ...(this.responseState && { responseState: this.responseState }),
           responseBody: {

packages/event-handler/src/bedrock-agent/index.ts

@@ -1 +1,2 @@
 export { BedrockAgentFunctionResolver } from './BedrockAgentFunctionResolver.js';
+export { BedrockFunctionResponse } from './BedrockFunctionResponse.js';