from __future__ import annotations
from typing import Literal, AsyncGenerator, ClassVar, Optional
import logging
from . base import Agent, registry
from .. message import Message
import inspect

class Chat(Agent):
  """
  Chat Agent
  """
  agent_type: Literal["chat"] = "chat" # type: ignore
  _agent_type: ClassVar[str] = agent_type  # Add this for registration

# Register the base agent
registry.register(Chat._agent_type, Chat)