From fc6ea9af2b282c6be6ce19f83d093dfb08ff5e37 Mon Sep 17 00:00:00 2001 From: Stephen Lincoln Date: Thu, 7 Nov 2024 15:30:11 -0500 Subject: [PATCH] Removed unused imports --- tests/test_llm_components.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_llm_components.py b/tests/test_llm_components.py index 011790a..c9501a9 100644 --- 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