From 805e8d6d0195f1d6c0974e64b39f8e45dfae71cc Mon Sep 17 00:00:00 2001 From: Fahad Date: Mon, 16 Jun 2025 06:08:58 +0400 Subject: [PATCH] Fix remaining TestGenRequest reference in format_response method MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Fixed NameError that was causing Docker container crashes: - Updated type annotation in format_response method from TestGenRequest to TestGenerationRequest - This was the last missing reference from the class rename 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude --- tools/testgen.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/testgen.py b/tools/testgen.py index 75b6ff2..b9aa372 100644 --- a/tools/testgen.py +++ b/tools/testgen.py @@ -418,7 +418,7 @@ class TestGenerationTool(BaseTool): return full_prompt - def format_response(self, response: str, request: TestGenRequest, model_info: Optional[dict] = None) -> str: + def format_response(self, response: str, request: TestGenerationRequest, model_info: Optional[dict] = None) -> str: """ Format the test generation response.