jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: StructuredOutputComponent tool name generation for multiple=True mode (#5476)

Browse source 
StructuredOutputComponent: fix tool name generation for multiple=True mode
This commit is contained in:
Andrey Tatarinov 2024-12-29 17:52:48 +04:00 • committed by GitHub
commit 5640f84f54
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/base/langflow/components/helpers/structured_output.py
View file

@ -93,6 +93,8 @@ class StructuredOutputComponent(Component):
] ]
def build_structured_output(self) -> Data: def build_structured_output(self) -> Data:
schema_name = self.schema_name or "OutputModel"
if not hasattr(self.llm, "with_structured_output"): if not hasattr(self.llm, "with_structured_output"):
msg = "Language model does not support structured output." msg = "Language model does not support structured output."
raise TypeError(msg) raise TypeError(msg)
@ -103,8 +105,8 @@ class StructuredOutputComponent(Component):
output_model_ = build_model_from_schema(self.output_schema) output_model_ = build_model_from_schema(self.output_schema)
if self.multiple: if self.multiple:
output_model = create_model( output_model = create_model(
self.schema_name, schema_name,
objects=(list[output_model_], Field(description=f"A list of {self.schema_name}.")), # type: ignore[valid-type] objects=(list[output_model_], Field(description=f"A list of {schema_name}.")), # type: ignore[valid-type]
) )
else: else:
output_model = output_model_ output_model = output_model_