diff --git a/tests/planai/test_llm_interface.py b/tests/planai/test_llm_interface.py index d6b084c..7b6d914 100644 --- 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], )