Bump to langchain0.1.0 (#108)

* Bump `langchain`, `openai`; add `langchain-openai` * Update imports to fix warnings
crewAIInc · Jan 11, 2024 · 8cc51d5 · 8cc51d5
1 parent fdd36b0
commit 8cc51d5
Show file tree

Hide file tree

Showing 4 changed files with 84 additions and 14 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,8 +18,9 @@ Repository = "https://github.com/joaomdmoura/crewai"
 [tool.poetry.dependencies]
 python = ">=3.9,<4.0"
 pydantic = "^2.4.2"
-langchain = "^0.0.354"
-openai = "^1.5.0"
+langchain = "^0.1.0"
+openai = "^1.7.1"
+langchain-openai = "^0.0.2"
 
 [tool.poetry.group.dev.dependencies]
 isort = "^5.13.2"

diff --git a/src/crewai/agent.py b/src/crewai/agent.py
@@ -2,10 +2,10 @@
 from typing import Any, List, Optional
 
 from langchain.agents.format_scratchpad import format_log_to_str
-from langchain.chat_models import ChatOpenAI
 from langchain.memory import ConversationSummaryMemory
 from langchain.tools.render import render_text_description
 from langchain_core.runnables.config import RunnableConfig
+from langchain_openai import ChatOpenAI
 from pydantic import (
     UUID4,
     BaseModel,

diff --git a/tests/agent_test.py b/tests/agent_test.py
@@ -1,7 +1,7 @@
 """Test Agent creation and execution basic functionality."""
 
 import pytest
-from langchain.chat_models import ChatOpenAI as OpenAI
+from langchain_openai import ChatOpenAI as OpenAI
 
 from crewai.agent import Agent
 from crewai.agents.cache import CacheHandler