Lint

9 months ago · a6afd45c63
parent f7dd10b820
commit a6afd45c63
1 changed files with 2 additions and 1 deletions
--- a/libs/langchain/langchain/schema/runnable/branch.py
+++ b/libs/langchain/langchain/schema/runnable/branch.py
@ -7,6 +7,7 @@ from typing import (
    Optional,
    Sequence,
    Tuple,
+    Type,
    Union,
    cast,
 )
@ -125,7 +126,7 @@ class RunnableBranch(RunnableSerializable[Input, Output]):
        return cls.__module__.split(".")[:-1]

    @property
-    def input_schema(self) -> type[BaseModel]:
+    def input_schema(self) -> Type[BaseModel]:
        runnables = (
            [self.default]
            + [r for _, r in self.branches]