@@ -50,7 +50,7 @@ def __call__(self, _: AgentInputContra, /) -> AsyncIterator[AgentOutputCov]: ...
5050]
5151
5252
53- class _BaseProtocol (_BaseObject , Generic [AgentInputContra , AgentOutput ], Protocol ):
53+ class AgentInterface (_BaseObject , Generic [AgentInputContra , AgentOutput ], Protocol ):
5454 __kwdefaults__ : Optional [dict [str , Any ]]
5555 __code__ : Any
5656
@@ -79,7 +79,7 @@ async def reply(
7979 async def list_models (self ) -> list [ModelInfo ]: ...
8080
8181
82- class RunnableAgent (_BaseProtocol [AgentInputContra , AgentOutput ], Protocol ):
82+ class RunnableAgent (AgentInterface [AgentInputContra , AgentOutput ], Protocol ):
8383 async def __call__ (
8484 self ,
8585 _ : AgentInputContra ,
@@ -88,7 +88,7 @@ async def __call__(
8888 ) -> Run [AgentOutput ]: ...
8989
9090
91- class RunnableOutputAgent (_BaseProtocol [AgentInputContra , AgentOutput ], Protocol ):
91+ class RunnableOutputAgent (AgentInterface [AgentInputContra , AgentOutput ], Protocol ):
9292 async def __call__ (
9393 self ,
9494 _ : AgentInputContra ,
@@ -97,7 +97,7 @@ async def __call__(
9797 ) -> AgentOutput : ...
9898
9999
100- class StreamableAgent (_BaseProtocol [AgentInputContra , AgentOutput ], Protocol ):
100+ class StreamableAgent (AgentInterface [AgentInputContra , AgentOutput ], Protocol ):
101101 def __call__ (
102102 self ,
103103 _ : AgentInputContra ,
@@ -106,7 +106,7 @@ def __call__(
106106 ) -> AsyncIterator [Run [AgentOutput ]]: ...
107107
108108
109- class StreamableOutputAgent (_BaseProtocol [AgentInputContra , AgentOutput ], Protocol ):
109+ class StreamableOutputAgent (AgentInterface [AgentInputContra , AgentOutput ], Protocol ):
110110 def __call__ (
111111 self ,
112112 _ : AgentInputContra ,
0 commit comments