From 11aefadb81f9e3b1dc1f1b90104c2cf22f3db1c6 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Tue, 9 Sep 2025 14:44:24 +0200 Subject: [PATCH] docs: Add comments to baml structured outputs creator --- .../baml/baml_src/extraction/acreate_structured_output.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/acreate_structured_output.py b/cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/acreate_structured_output.py index 08f12c6f2..16bf7c3d7 100644 --- a/cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/acreate_structured_output.py +++ b/cognee/infrastructure/llm/structured_output_framework/baml/baml_src/extraction/acreate_structured_output.py @@ -152,8 +152,9 @@ async def acreate_structured_output( BaseModel: The summarized content in the specified format """ config = get_llm_config() - tb = TypeBuilder() + # Dynamically create BAML response model + tb = TypeBuilder() type_builder = create_dynamic_baml_type(tb, tb.ResponseModel, response_model) result = await b.AcreateStructuredOutput( @@ -162,8 +163,9 @@ async def acreate_structured_output( baml_options={"client_registry": config.baml_registry, "tb": type_builder}, ) + # Transform BAML response to proper pydantic reponse model if response_model is str: - return result + return str(result) return response_model.model_validate(result.dict())