From 0941c64ba711db83c4e2e930b38543a26212b9ef Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Tue, 5 Aug 2025 21:24:58 +0200 Subject: [PATCH] fix: Add upper to BAML structured output name --- cognee/infrastructure/llm/LLMGateway.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cognee/infrastructure/llm/LLMGateway.py b/cognee/infrastructure/llm/LLMGateway.py index e7941dcce..a88cfb85d 100644 --- a/cognee/infrastructure/llm/LLMGateway.py +++ b/cognee/infrastructure/llm/LLMGateway.py @@ -107,7 +107,7 @@ class LLMGateway: @staticmethod def extract_code_summary(content: str) -> Coroutine: llm_config = get_llm_config() - if llm_config.structured_output_framework == "BAML": + if llm_config.structured_output_framework.upper() == "BAML": from cognee.infrastructure.llm.structured_output_framework.baml.baml_src.extraction import ( extract_code_summary, ) @@ -123,7 +123,7 @@ class LLMGateway: @staticmethod def extract_summary(content: str, response_model: Type[BaseModel]) -> Coroutine: llm_config = get_llm_config() - if llm_config.structured_output_framework == "BAML": + if llm_config.structured_output_framework.upper() == "BAML": from cognee.infrastructure.llm.structured_output_framework.baml.baml_src.extraction import ( extract_summary, )