This commit is contained in:
Nuno Campos
2024-05-03 11:01:34 -07:00
parent 2ab84d1562
commit 0e30723684
+1 -1
View File
@@ -180,7 +180,7 @@ class CompiledStateGraph(CompiledGraph):
return super().get_input_schema(config)
def get_output_schema(self, config: RunnableConfig | None = None) -> BaseModel:
def get_output_schema(self, config: Optional[RunnableConfig] = None) -> BaseModel:
if isinstance(self.builder.schema, BaseModel):
return self.builder.schema