From 655cfd25cf4c0aec8dc843a348a226ce4b5591c8 Mon Sep 17 00:00:00 2001 From: Yeuoly Date: Thu, 26 Jun 2025 17:00:32 +0800 Subject: [PATCH] refactor: add type hint to generator function in structured output parser - Enhanced the generator function with a return type hint for improved type safety and clarity. - Maintained existing functionality while improving code readability and maintainability. --- api/core/llm_generator/output_parser/structured_output.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/core/llm_generator/output_parser/structured_output.py b/api/core/llm_generator/output_parser/structured_output.py index 2625aa2d48..6246a9ce8e 100644 --- a/api/core/llm_generator/output_parser/structured_output.py +++ b/api/core/llm_generator/output_parser/structured_output.py @@ -143,7 +143,7 @@ def invoke_llm_with_structured_output( ) else: - def generator(): + def generator() -> Generator[LLMStructuredOutput, None, None]: result_text = "" for event in llm_result: if isinstance(event, LLMResultChunk):