Skip to content

Commit 8e2cec9

Browse files
fix legacy import
1 parent 9beb3f4 commit 8e2cec9

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

tests/integrations/langchain/test_langchain.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,8 @@
1212
from langchain_openai import ChatOpenAI, OpenAI
1313
except ImportError:
1414
# Langchain < 0.2
15-
from langchain_community.chat_models import ChatOpenAI, OpenAI
15+
from langchain_community.llms import OpenAI
16+
from langchain_community.chat_models import ChatOpenAI
1617

1718
from langchain_core.callbacks import BaseCallbackManager, CallbackManagerForLLMRun
1819
from langchain_core.messages import BaseMessage, AIMessageChunk

0 commit comments

Comments
 (0)