github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-02-21 14:43:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix output schema

Browse Source
This commit is contained in:
Eugene Yurtsev
2024-08-07 17:23:44 -04:00
parent 1ef3fa54fc
commit 61cdb9ccce
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
libs/core/langchain_core/runnables/passthrough.py
View File

@@ -434,14 +434,17 @@ class RunnableAssign(RunnableSerializable[Dict[str, Any], Dict[str, Any]]):
if not issubclass(map_input_schema, RootModel) and not issubclass(
map_output_schema, RootModel
):
# ie. both are dicts
fields = {}
for name, field_info in map_input_schema.model_fields.items():
fields[name] = (field_info.annotation, field_info.default)
for name, field_info in map_output_schema.model_fields.items():
fields[name] = (field_info.annotation, field_info.default)
return create_model( # type: ignore[call-overload]
"RunnableAssignOutput",
**{
k: (v.type_, v.default)
for s in (map_input_schema, map_output_schema)
for k, v in s.model_fields.items()
},
**fields,
)
elif not issubclass(map_output_schema, RootModel):
# ie. only map output is a dict