Fixing issues brought about by merge

tests/agent_test.py

@@ -4,10 +4,6 @@ from unittest import mock
 from unittest.mock import patch
 
 import pytest
-from langchain.tools import tool
-from langchain_core.exceptions import OutputParserException
-from langchain_openai import ChatOpenAI
-
 from crewai import Agent, Crew, Task
 from crewai.agents.cache import CacheHandler
 from crewai.agents.executor import CrewAgentExecutor
@@ -15,6 +11,9 @@ from crewai.agents.parser import CrewAgentParser
 from crewai.tools.tool_calling import InstructorToolCalling
 from crewai.tools.tool_usage import ToolUsage
 from crewai.utilities import RPMController
+from langchain.tools import tool
+from langchain_core.exceptions import OutputParserException
+from langchain_openai import ChatOpenAI
 
 
 def test_agent_creation():
@@ -631,8 +630,8 @@ def test_agent_use_specific_tasks_output_as_context(capsys):
 
     crew = Crew(agents=[agent1, agent2], tasks=tasks)
     result = crew.kickoff()
-    assert "bye" not in result.lower()
-    assert "hi" in result.lower() or "hello" in result.lower()
+    assert "bye" not in result.raw_output().lower()
+    assert "hi" in result.raw_output().lower() or "hello" in result.raw_output().lower()
 
 
 @pytest.mark.vcr(filter_headers=["authorization"])
@@ -750,7 +749,8 @@ def test_tool_result_as_answer_is_the_final_answer_for_the_agent():
     crew = Crew(agents=[agent1], tasks=tasks)
 
     result = crew.kickoff()
-    assert result == "Howdy!"
+    print("RESULT: ", result.raw_output())
+    assert result.raw_output() == "Howdy!"
 
 
 @pytest.mark.vcr(filter_headers=["authorization"])