diff --git a/tests/test_agents/test_agents.py b/tests/test_agents/test_agents.py index 68094dc5..0ae04a5f 100644 --- a/tests/test_agents/test_agents.py +++ b/tests/test_agents/test_agents.py @@ -3,7 +3,7 @@ from langchain_community.tools import DuckDuckGoSearchRun from motleycrew.agents.crewai.crewai_agent import CrewAIMotleyAgent -from motleycrew.agents.langchain.react import ReactMotleyAgent +from motleycrew.agents.langchain.openai_tools_react import ReactOpenAIToolsAgent from motleycrew.agents.llama_index.llama_index_react import ReActLlamaIndexMotleyAgent from motleycrew.common.exceptions import AgentNotMaterialized, CannotModifyMaterializedAgent from motleycrew.tools.python_repl import create_repl_tool @@ -32,11 +32,8 @@ def crewai_agent(self): @pytest.fixture(scope="class") def langchain_agent(self): - agent = ReactMotleyAgent( + agent = ReactOpenAIToolsAgent( name="AI writer agent", - description="""Conduct a comprehensive analysis of the latest advancements in AI in 2024. - Identify key trends, breakthrough technologies, and potential industry impacts. - Your final answer MUST be a full analysis report""", tools=[DuckDuckGoSearchRun()], verbose=True, ) diff --git a/tests/test_tasks/test_simple_task.py b/tests/test_tasks/test_simple_task.py index ea390870..d7836d5e 100644 --- a/tests/test_tasks/test_simple_task.py +++ b/tests/test_tasks/test_simple_task.py @@ -2,7 +2,7 @@ from langchain_community.tools import DuckDuckGoSearchRun from motleycrew.crew import MotleyCrew -from motleycrew.agents.langchain import ReactMotleyAgent +from motleycrew.agents.langchain.openai_tools_react import ReactOpenAIToolsAgent from motleycrew.tasks.simple import ( SimpleTask, SimpleTaskUnit, @@ -18,11 +18,8 @@ def crew(self): @pytest.fixture(scope="class") def agent(self): - agent = ReactMotleyAgent( + agent = ReactOpenAIToolsAgent( name="AI writer agent", - description="""Conduct a comprehensive analysis of the latest advancements in AI in 2024. - Identify key trends, breakthrough technologies, and potential industry impacts. - Your final answer MUST be a full analysis report""", tools=[DuckDuckGoSearchRun()], verbose=True, )