fix: flake8

provos · Nov 27, 2024 · 122ffaf · 122ffaf
1 parent ef66a75
commit 122ffaf
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/planai/test_llm_interface.py b/tests/planai/test_llm_interface.py
@@ -1,7 +1,5 @@
 import json
 import unittest
-from dataclasses import dataclass
-from typing import Any, Dict
 from unittest.mock import Mock, patch
 
 from pydantic import BaseModel
@@ -409,7 +407,7 @@ def test_chat_with_invalid_tool(self):
 
         # Make the chat request
         with self.assertLogs(level="ERROR") as log:
-            response = self.llm_interface.chat(
+            _ = self.llm_interface.chat(
                 messages=[{"role": "user", "content": "Use an invalid tool"}],
                 tools=[mock_tool],
             )
@@ -448,7 +446,7 @@ def error_execute(self, **kwargs) -> str:
 
         # Make the chat request
         with self.assertLogs(level="ERROR") as log:
-            response = self.llm_interface.chat(
+            _ = self.llm_interface.chat(
                 messages=[{"role": "user", "content": "Use the error tool"}],
                 tools=[error_tool],
             )