"""State schema generated from blueprint: {{ ir.name }}"""
# Generated by agent-blueprint {{ ir.version }}

from typing import Annotated, Any
from langgraph.graph.message import add_messages
from typing_extensions import TypedDict


class AgentState(TypedDict):
{% for field_name, field_def in ir.state.fields.items() %}
{% if field_def.type == "list[message]" %}
    {{ field_name }}: Annotated[list, add_messages]
{% elif field_def.type == "string" %}
    {{ field_name }}: str | None
{% elif field_def.type == "boolean" %}
    {{ field_name }}: bool
{% elif field_def.type == "number" or field_def.type == "float" %}
    {{ field_name }}: float | None
{% elif field_def.type == "integer" %}
    {{ field_name }}: int | None
{% else %}
    {{ field_name }}: Any
{% endif %}
{% else %}
    pass
{% endfor %}
