From 25441f3a14164459fc29b449a50ff349b031a666 Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Wed, 26 Feb 2025 04:52:39 +0000
Subject: [PATCH] Fix test mock to use 'context' key instead of 'content' key

Co-Authored-By: Joe Moura <joao@crewai.com>
---
 tests/agent_test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/agent_test.py b/tests/agent_test.py
index 87c32b5e8..f96cd0f23 100644
--- a/tests/agent_test.py
+++ b/tests/agent_test.py
@@ -1673,7 +1673,7 @@ def test_agent_uses_task_knowledge():
         try:
             # Configure the mock
             mock_knowledge = MockKnowledge.return_value
-            mock_knowledge.query.return_value = [{"content": content}]
+            mock_knowledge.query.return_value = [{"context": content}]
             
             # Create a real LLM but patch its call method
             agent = Agent(