From b0dffd2d944ea0d97d7dfc8100863913b0be9adb Mon Sep 17 00:00:00 2001 From: evelynmitchell Date: Sat, 16 Dec 2023 11:10:05 -0700 Subject: [PATCH] test_LLM langchain openai import --- tests/models/test_LLM.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/models/test_LLM.py b/tests/models/test_LLM.py index 04d6a5f2..44bc29c9 100644 --- a/tests/models/test_LLM.py +++ b/tests/models/test_LLM.py @@ -1,7 +1,8 @@ import unittest import os from unittest.mock import patch -from langchain import HuggingFaceHub, ChatOpenAI +from langchain import HuggingFaceHub +from langchain.chat_models import ChatOpenAI from swarms.models.llm import LLM