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 55632be8e..6ef27e51d 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 @@ -53,7 +53,7 @@ async def acreate_structured_output( # Transform BAML response to proper pydantic reponse model if response_model is str: - # Note: when a response model is set to string in python result is stored in text property in the BAML response model + # Note: when a response model is set to string in python, result is stored in text property in the BAML response model return str(result.text) return response_model.model_validate(result.dict())