pull/11279/head
Nuno Campos 11 months ago
parent f7dd10b820
commit a6afd45c63

@ -7,6 +7,7 @@ from typing import (
Optional, Optional,
Sequence, Sequence,
Tuple, Tuple,
Type,
Union, Union,
cast, cast,
) )
@ -125,7 +126,7 @@ class RunnableBranch(RunnableSerializable[Input, Output]):
return cls.__module__.split(".")[:-1] return cls.__module__.split(".")[:-1]
@property @property
def input_schema(self) -> type[BaseModel]: def input_schema(self) -> Type[BaseModel]:
runnables = ( runnables = (
[self.default] [self.default]
+ [r for _, r in self.branches] + [r for _, r in self.branches]

Loading…
Cancel
Save