From 0e5e9f3f51ed95e4ed3e516e22e534e90d5cda53 Mon Sep 17 00:00:00 2001 From: hargunsiingh Date: Wed, 4 Feb 2026 01:11:38 +0530 Subject: [PATCH] Fix type annotation errors in llm_generator.py Added explicit list[DatasetCase] type annotations to the cases variable in both LlamaDatasetGenerator.generate() and OpenAIDatasetGenerator.generate() methods to satisfy Mypy type checking. Closes #67 --- src/agentunit/generators/llm_generator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/agentunit/generators/llm_generator.py b/src/agentunit/generators/llm_generator.py index 836f011..24c6f67 100644 --- a/src/agentunit/generators/llm_generator.py +++ b/src/agentunit/generators/llm_generator.py @@ -141,7 +141,7 @@ async def generate( cases_data = json.loads(response_text) - cases = [] + cases: list[DatasetCase] = [] for case_data in cases_data: case = DatasetCase( id=case_data.get("id", f"generated_{len(cases)}"), @@ -288,7 +288,7 @@ async def generate( cases_data = json.loads(response_text) - cases = [] + cases: list[DatasetCase] = [] for case_data in cases_data: case = DatasetCase( id=case_data.get("id", f"generated_{len(cases)}"),