Removed unused imports

AttackIQ · Nov 7, 2024 · fc6ea9a · fc6ea9a
1 parent 3e1bbf2
commit fc6ea9a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_llm_components.py b/tests/test_llm_components.py
@@ -1,8 +1,8 @@
 import pytest
-from unittest.mock import Mock, patch, create_autospec
+from unittest.mock import patch, create_autospec
 from langchain.schema import AgentAction, AgentFinish, Document, AIMessage
 from langchain.schema.vectorstore import VectorStore
-from langchain_openai import OpenAIEmbeddings, ChatOpenAI
+from langchain_openai import OpenAIEmbeddings
 from langchain.schema.language_model import BaseLanguageModel
 
 from sigmaiq.llm.base import SigmaLLM