From cb09f17fc7580153766ae9cae879a5970ae458ed Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Moura?= Date: Sun, 4 Feb 2024 13:13:42 -0800 Subject: [PATCH] updating readme --- README.md | 8 +++++--- tests/agent_test.py | 12 ++++++------ 2 files changed, 11 insertions(+), 9 deletions(-) diff --git a/README.md b/README.md index 45eda0c82..cc94a3350 100644 --- a/README.md +++ b/README.md @@ -86,10 +86,12 @@ researcher = Agent( # model like OpenAI, Mistral, Antrophic or others (https://python.langchain.com/docs/integrations/llms/) # # Examples: + # + # from langchain.llms import Ollama # llm=ollama_llm # was defined above in the file - # llm=OpenAI(model_name="gpt-3.5", temperature=0.7) - # For the OpenAI model you would need to import - # from langchain_openai import OpenAI + # + # from langchain_openai import ChatOpenAI + # llm=ChatOpenAI(model_name="gpt-3.5", temperature=0.7) ) writer = Agent( role='Tech Content Strategist', diff --git a/tests/agent_test.py b/tests/agent_test.py index 0e4fca8c7..7cf42f8b2 100644 --- a/tests/agent_test.py +++ b/tests/agent_test.py @@ -4,7 +4,7 @@ from unittest.mock import patch import pytest from langchain.tools import tool -from langchain_openai import ChatOpenAI as OpenAI +from langchain_openai import ChatOpenAI from crewai import Agent, Crew, Task from crewai.agents.cache import CacheHandler @@ -24,7 +24,7 @@ def test_agent_creation(): def test_agent_default_values(): agent = Agent(role="test role", goal="test goal", backstory="test backstory") - assert isinstance(agent.llm, OpenAI) + assert isinstance(agent.llm, ChatOpenAI) assert agent.llm.model_name == "gpt-4" assert agent.llm.temperature == 0.7 assert agent.llm.verbose == False @@ -36,10 +36,10 @@ def test_custom_llm(): role="test role", goal="test goal", backstory="test backstory", - llm=OpenAI(temperature=0, model="gpt-4"), + llm=ChatOpenAI(temperature=0, model="gpt-4"), ) - assert isinstance(agent.llm, OpenAI) + assert isinstance(agent.llm, ChatOpenAI) assert agent.llm.model_name == "gpt-4" assert agent.llm.temperature == 0 @@ -51,7 +51,7 @@ def test_agent_without_memory(): goal="test goal", backstory="test backstory", memory=False, - llm=OpenAI(temperature=0, model="gpt-4"), + llm=ChatOpenAI(temperature=0, model="gpt-4"), ) memory_agent = Agent( @@ -59,7 +59,7 @@ def test_agent_without_memory(): goal="test goal", backstory="test backstory", memory=True, - llm=OpenAI(temperature=0, model="gpt-4"), + llm=ChatOpenAI(temperature=0, model="gpt-4"), ) result = no_memory_agent.execute_task("How much is 1 + 1?")