1919 CreateElicitationResponse ,
2020)
2121from ._router import MethodRouter
22- from .interfaces import Agent
2322from .meta import CLIENT_METHODS
2423
2524__all__ = ["AgentSideConnection" , "run_agent" ]
@@ -30,7 +29,7 @@ def _dump(model: BaseModel) -> dict[str, Any]:
3029
3130
3231class _AgentRouter :
33- def __init__ (self , agent : Agent , state : InitializationState ) -> None :
32+ def __init__ (self , agent : object , state : InitializationState ) -> None :
3433 self ._router = MethodRouter (agent , AGENT_REQUESTS , AGENT_NOTIFICATIONS )
3534 self ._state = state
3635
@@ -56,7 +55,7 @@ class AgentSideConnection:
5655
5756 def __init__ (
5857 self ,
59- agent : Agent ,
58+ agent : object ,
6059 input_stream : Any ,
6160 output_stream : Any = None ,
6261 * ,
@@ -78,7 +77,7 @@ def __init__(
7877 @classmethod
7978 def _attach (
8079 cls ,
81- agent_factory : Callable [[AgentSideConnection ], Agent ],
80+ agent_factory : Callable [[AgentSideConnection ], object ],
8281 connection : Connection ,
8382 ) -> tuple [AgentSideConnection , _AgentRouter ]:
8483 self = cls .__new__ (cls )
@@ -161,7 +160,7 @@ def _extension_method(method: str) -> str:
161160
162161
163162async def run_agent (
164- agent : Agent ,
163+ agent : object ,
165164 input_stream : Any = None ,
166165 output_stream : Any = None ,
167166 * ,
0 commit comments