Amazon Bedrock으로 구축한 EWOK Agent 재해 복구

def to_tool_spec(skill) -> dict:
    """Compile a skill's declared schema into a Bedrock toolSpec."""
    properties = {
        name: {"type": field.type, "description": field.description}
        for name, field in skill.input_schema.items()
    }
    required = [n for n, f in skill.input_schema.items() if f.required]
    return {
        "toolSpec": {
            "name": skill.name,
            "description": skill.description,
            "inputSchema": {"json": {
                "type": "object",
                "properties": properties,
                "required": required,
            }},
        }
    }

client = ChatBedrockConverse(
    model_id=config.model_id,  # pluggable via config
    guardrail_config={
        "guardrailIdentifier": config.guardrail_id,
        "guardrailVersion": config.guardrail_version,
        "trace": "enabled",
    },
)

def run(self, request: str) -> str:
    messages = [HumanMessage(content=request)]
    llm = self.client.bind_tools(
        [to_tool_spec(s) for s in self.skills]
    )
    for _ in range(MAX_ITERATIONS):
        response = llm.invoke([self.system_prompt, *messages])
        messages.append(response)
        stop_reason = response.response_metadata.get("stopReason")
        if stop_reason == "end_turn":
            return response.text()  # final answer for the engineer
        if stop_reason == "guardrail_intervened":
            return response.text()  # stop, don't retry past a guardrail
        if stop_reason == "tool_use":
            for call in response.tool_calls:
                skill = self.skills_by_name[call["name"]]
                result = skill.executor.execute(**call["args"])
                messages.append(ToolMessage(
                    tool_call_id=call["id"],
                    content=result.data,
                    status="success" if result.success else "error",
                ))
    raise MaxIterationsError(f"No resolution within {MAX_ITERATIONS} iterations")