Skip to content

Commit 0a90daa

Browse files
committed
chore: rename base protocol to agent interface
1 parent 075c58f commit 0a90daa

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

workflowai/core/client/_types.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)