github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-16 16:11:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix issue with runnable schema output + typed_dict

Browse Source
This commit is contained in:
William Zhu 2024-11-18 16:45:12 -05:00
parent 5b9b8fe80f
commit acffca8bb7
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/core/langchain_core/runnables/base.py
View File

@ -3593,6 +3593,12 @@ class RunnableParallel(RunnableSerializable[Input, dict[str, Any]]):
== "object"
for s in self.steps__.values()
):
for step in self.steps__.values():
for k, v in step.get_input_schema(config).model_fields.items():
if v.annotation != Any and k == "root":
return super().get_input_schema(config)
# This is correct, but pydantic typings/mypy don't think so.
return create_model_v2( # type: ignore[call-overload]
self.get_name("Input"),