GONATURELLE for AWS Bedrock

import os
import boto3
from gonaturelle.integrations.bedrock import BedrockConverseLoop

from app.tools import TOOL_SPECS, call_tool  # [{'toolSpec': {'name', 'description', 'inputSchema': {'json': {...}}}}]

client = boto3.client('bedrock-runtime', region_name='us-east-1')
MODEL_ID = os.environ['UNDERWRITING_MODEL_ID']

def run_agent(user_input: str) -> str:
    loop = BedrockConverseLoop(client, agent='underwriting')
    messages = [{'role': 'user', 'content': [{'text': user_input}]}]
    for _ in range(8):
        response = loop.converse(modelId=MODEL_ID, messages=messages, toolConfig={'tools': TOOL_SPECS})
        message = response['output']['message']
        messages.append(message)
        if response['stopReason'] != 'tool_use':
            return ''.join(c.get('text', '') for c in message['content'])
        results = []
        for content in message['content']:
            if 'toolUse' in content:
                use = content['toolUse']
                output = call_tool(use['name'], use['input'])
                results.append(loop.tool_result(use['toolUseId'], {'json': output}))
        messages.append({'role': 'user', 'content': results})
    raise RuntimeError('tool loop exceeded 8 turns')