Source code for agentstr.agents.providers.pydantic

from typing import Callable
from pydantic_ai import Agent
from agentstr.models import ChatInput, ChatOutput




def pydantic_agent_callable(agent: Agent) -> Callable[[ChatInput], ChatOutput | str]:
    """Create a callable that can be used with the Agentstr framework.
    
    Args:
        agent: The Pydantic agent to wrap.
    
    Returns:
        A callable that can be used with the Agentstr framework.
    """
    async def agent_callable(input: ChatInput) -> str:
        result = await agent.run(input.message)
        return result.output
    return agent_callable