lda.chat/lda-wf
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

type + misc

Browse Source
This commit is contained in:
lda
2026-06-11 03:31:18 +07:00 Verified
parent 3d2c78993f
commit d695fd7533
13 changed files with 146 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/wf_authoring/nodes/callables.py
+14 -17
View File
@@ -9,48 +9,45 @@ from wf_core import RuntimeContext
from .result import NodeReturn
InputT = TypeVar("InputT", bound=BaseModel)
OutputT = TypeVar("OutputT", bound=BaseModel)
InputT_contra = TypeVar("InputT_contra", bound=BaseModel, contravariant=True)
OutputT_co = TypeVar("OutputT_co", bound=BaseModel, covariant=True)
InputT = TypeVar("InputT", bound=BaseModel, infer_variance=True)
OutputT = TypeVar("OutputT", bound=BaseModel, infer_variance=True)
class ContextNodeCallable(Protocol[InputT_contra, OutputT_co]):
class ContextNodeCallable(Protocol[InputT, OutputT]):
def __call__(
self,
payload: InputT_contra,
payload: InputT,
/,
ctx: RuntimeContext,
) -> NodeReturn[OutputT_co] | OutputT_co | None: ...
) -> NodeReturn[OutputT] | OutputT | None: ...
class PlainNodeCallable(Protocol[InputT_contra, OutputT_co]):
class PlainNodeCallable(Protocol[InputT, OutputT]):
def __call__(
self,
payload: InputT_contra,
payload: InputT,
/,
) -> NodeReturn[OutputT_co] | OutputT_co | None: ...
) -> NodeReturn[OutputT] | OutputT | None: ...
NodeCallable = ContextNodeCallable[InputT, OutputT] | PlainNodeCallable[InputT, OutputT]
class AsyncContextNodeCallable(Protocol[InputT_contra, OutputT_co]):
class AsyncContextNodeCallable(Protocol[InputT, OutputT]):
def __call__(
self,
payload: InputT_contra,
payload: InputT,
/,
ctx: RuntimeContext,
) -> Awaitable[NodeReturn[OutputT_co] | OutputT_co | None]: ...
) -> Awaitable[NodeReturn[OutputT] | OutputT | None]: ...
class AsyncPlainNodeCallable(Protocol[InputT_contra, OutputT_co]):
class AsyncPlainNodeCallable(Protocol[InputT, OutputT]):
def __call__(
self,
payload: InputT_contra,
payload: InputT,
/,
) -> Awaitable[NodeReturn[OutputT_co] | OutputT_co | None]: ...
) -> Awaitable[NodeReturn[OutputT] | OutputT | None]: ...
AsyncNodeCallable = (