From 122ffaf185cba63c9f539dcee45369b249a2674d Mon Sep 17 00:00:00 2001 From: Niels Provos Date: Wed, 27 Nov 2024 15:25:17 -0800 Subject: [PATCH] fix: flake8 --- tests/planai/test_llm_interface.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) 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], )