Enable User Input in LangChain AWS Bedrock Agent

[cardutr]

Hello! I am creating and agent on AWS BedRock using LangChain and would like to enable User Input [figure] on the code. I checked the documentation and I couldn't find any information about how to enable user interaction for this scenario. It seems like this functionality may not currently be supported and could require development.

This is the code:
from langchain_aws.agents import BedrockAgentsRunnable
agent = BedrockAgentsRunnable.create_agent(
            agent_name="risk_agent",
            agent_resource_role_arn=agent_resource_role_arn,
            foundation_model=foundational_model,
            instruction=SYSTEM_INSTRUCTIONS,
            tools=list_tools,
            guardrail_configuration=None,
        )

[Zhenye-Na]

Based on my current experience with Bedrock Agent, to enable userInput just like the screenshot you included above for agent itself is not currently supported yet.

• https://docs.aws.amazon.com/cdk/api/v2/docs/aws-cdk-lib.aws_bedrock.CfnAgent.html
• https://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/aws-resource-bedrock-agent.html

---

However, there is a workaround and you could check to see if this fits your specific use cases

You can leverage CreateAgentActionGroup API with the following request to enable userInput for a specific Agent ActionGroup

CreateAgentActionGroup:
{
  "actionGroupName": "AskUserAction",
  "parentActionGroupSignature": "AMAZON.UserInput",
  "actionGroupState": "ENABLED"
}
                     

where parentActionGroupSignature allows 'AMAZON.UserInput'|'AMAZON.CodeInterpreter'

More specifically, updates could be made in the following places during create_agent_action_group API call to enable the flag

• langchain-aws/libs/aws/langchain_aws/agents/base.py

  Lines 197 to 206 in dabdd99

  	bedrock_client.create_agent_action_group(
  	actionGroupName=action_group_name,
  	actionGroupState="ENABLED",
  	actionGroupExecutor={"customControl": "RETURN_CONTROL"},
  	functionSchema={
  	"functions": [_tool_to_function(function) for function in functions]
  	},
  	agentId=agent_id,
  	agentVersion="DRAFT",
  	)

• langchain-aws/libs/aws/langchain_aws/agents/base.py

  Line 453 in dabdd99

  _create_bedrock_action_groups(bedrock_client, agent_id, tools)

---

References

• https://docs.aws.amazon.com/bedrock/latest/userguide/agents-enable-user-input.html

[3coins]

cc @kcgourishetti