Skip to content

Commit 69a3b29

Browse files
authored
Updated deps for latest langchain version (#3092)
* Updated deps and imports for latest langchain version
1 parent eed9d48 commit 69a3b29

File tree

2 files changed

+16
-5
lines changed

2 files changed

+16
-5
lines changed

tests/integrations/langchain/test_langchain.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,14 @@
22
from unittest.mock import Mock
33

44
import pytest
5-
from langchain_community.chat_models import ChatOpenAI
5+
6+
try:
7+
# Langchain >= 0.2
8+
from langchain_openai import ChatOpenAI
9+
except ImportError:
10+
# Langchain < 0.2
11+
from langchain_community.chat_models import ChatOpenAI
12+
613
from langchain_core.callbacks import CallbackManagerForLLMRun
714
from langchain_core.messages import BaseMessage, AIMessageChunk
815
from langchain_core.outputs import ChatGenerationChunk

tox.ini

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,7 @@ envlist =
152152
{py3.9,py3.11,py3.12}-huggingface_hub-{v0.22,latest}
153153

154154
# Langchain
155-
{py3.9,py3.11,py3.12}-langchain-0.1
155+
{py3.9,py3.11,py3.12}-langchain-v0.1
156156
{py3.9,py3.11,py3.12}-langchain-latest
157157
{py3.9,py3.11,py3.12}-langchain-notiktoken
158158

@@ -462,12 +462,16 @@ deps =
462462
huggingface_hub-latest: huggingface_hub
463463

464464
# Langchain
465-
langchain: openai~=1.0.0
466-
langchain-0.1: langchain~=0.1.11
467-
langchain-0.1: tiktoken~=0.6.0
465+
langchain-v0.1: openai~=1.0.0
466+
langchain-v0.1: langchain~=0.1.11
467+
langchain-v0.1: tiktoken~=0.6.0
468468
langchain-latest: langchain
469+
langchain-latest: langchain-openai
470+
langchain-latest: openai>=1.6.1
469471
langchain-latest: tiktoken~=0.6.0
470472
langchain-notiktoken: langchain
473+
langchain-notiktoken: langchain-openai
474+
langchain-notiktoken: openai>=1.6.1
471475

472476
# Loguru
473477
loguru-v0.5: loguru~=0.5.0

0 commit comments

Comments
 (0)