Hi there!

I have a multi prompt flow set up and it works fine. I successfully get all outputs streamed to me as part of the BedrockAgentRuntimeClient responseStream However, if I add tool use to the prompt in the prompt manager, validated to work when testing the prompt the "document" is returned as an empty string.

I cant find any documentation on this. Configuration of neither prompt output nor outputNode input seem like they are still prototypes.

Am I missing something or is this just not yet supported?

Tool Spec :

{
    "toolChoice": {
        "tool": {
            "name": "apply_tags"
        }
    },
    "tools": [
        {
            "toolSpec": {
                "description": "Apply tags",
                "inputSchema": {
                    "json": {
                        "type": "object",
                        "properties": {
                            "tags": {
                                "type": "array",
                                "description": "An array of tags to apply",
                                "items": {
                                    "type": "string"
                                }
                            }
                        },
                        "required": [
                            "tags"
                        ]
                    }
                },
                "name": "apply_tags"
            }
        }
    ]
}

Flow overview

Test calling code

import {
  BedrockAgentRuntimeClient,
  InvokeFlowCommand,
} from "@aws-sdk/client-bedrock-agent-runtime";

/**
 * Invokes an alias of a flow to run the inputs that you specify and return
 * the output of each node as a stream.
 *
 * @param {{
 *  flowIdentifier: string,
 *  flowAliasIdentifier: string,
 *  prompt?: string,
 *  region?: string
 * }} options
 * @returns {Promise<Array>} An array containing all flow output and completion events.
 */
export const invokeBedrockFlow = async ({
  flowIdentifier,
  flowAliasIdentifier,
  document,
  region = "us-east-2",
}) => {
  const client = new BedrockAgentRuntimeClient({ region });

  const command = new InvokeFlowCommand({
    flowIdentifier,
    flowAliasIdentifier,
    inputs: [
      {
        content: {
          document,
        },
        nodeName: "FlowInputNode",
        nodeOutputName: "document",
      },
    ],
    enableTrace: true,
  });

  const flowEvents = [];
  const response = await client.send(command);

  for await (const chunkEvent of response.responseStream) {
    const eventToProcess = chunkEvent.flowOutputEvent || chunkEvent.flowCompletionEvent

    if (eventToProcess) {
      flowEvents.push(Object.fromEntries(Object.entries(eventToProcess)))
    }
  }


  return flowEvents;
};

const response = await invokeBedrockFlow({
  flowIdentifier: "blablabla",
  flowAliasIdentifier: "blablabla",
  document: {
    prompt: 'roses are red, violets are blue',
    metadata: {
      userName: 'John Doe'
    }
  },
  region: "us-east-2",
});

console.log(JSON.stringify(response, null, 2));

Response without tool

[
  {
    "content": {
      "document": "\"tag,roses,red,violets,blue,flowers,poetry,rhyme,classic,verse,romantic,garden,blooms,petals,color,nature,love,traditional,simple,nursery,couplet\""
    },
    "nodeName": "FlowOutputNode"
  },
  {
    "completionReason": "SUCCESS"
  }
]

# Response with tool 
```json
[
  {
    "content": {
      "document": ""
    },
    "nodeName": "FlowOutputNode"
  },
  {
    "completionReason": "SUCCESS"
  }
]