mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-11 17:18:29 +00:00
daa40ed630
- Update test assertions to match new knowledge context format - Replace API-dependent test with mock-based approach - Ensure tests can run without external API access Part of #2269 Co-Authored-By: Joe Moura <joao@crewai.com>
64 lines
2.8 KiB
Python
64 lines
2.8 KiB
Python
"""Test knowledge utils functionality."""
|
|
|
|
from typing import Dict, List, Any
|
|
|
|
import pytest
|
|
|
|
from crewai.knowledge.utils.knowledge_utils import extract_knowledge_context
|
|
|
|
|
|
def test_extract_knowledge_context_with_valid_snippets():
|
|
"""Test extracting knowledge context with valid snippets."""
|
|
snippets = [
|
|
{"context": "Fact 1: The sky is blue", "score": 0.9},
|
|
{"context": "Fact 2: Water is wet", "score": 0.8},
|
|
]
|
|
result = extract_knowledge_context(snippets)
|
|
expected = "Important Context (You MUST use this information to complete your task accurately and effectively):\nFact 1: The sky is blue\nFact 2: Water is wet\n\nMake sure to incorporate the above context into your response."
|
|
assert result == expected
|
|
|
|
|
|
def test_extract_knowledge_context_with_empty_snippets():
|
|
"""Test extracting knowledge context with empty snippets."""
|
|
snippets: List[Dict[str, Any]] = []
|
|
result = extract_knowledge_context(snippets)
|
|
assert result == ""
|
|
|
|
|
|
def test_extract_knowledge_context_with_none_snippets():
|
|
"""Test extracting knowledge context with None snippets."""
|
|
snippets = [None, {"context": "Valid context"}] # type: ignore
|
|
result = extract_knowledge_context(snippets)
|
|
assert result == "Important Context (You MUST use this information to complete your task accurately and effectively):\nValid context\n\nMake sure to incorporate the above context into your response."
|
|
|
|
|
|
def test_extract_knowledge_context_with_missing_context():
|
|
"""Test extracting knowledge context with missing context."""
|
|
snippets = [{"score": 0.9}, {"context": "Valid context"}]
|
|
result = extract_knowledge_context(snippets)
|
|
assert result == "Important Context (You MUST use this information to complete your task accurately and effectively):\nValid context\n\nMake sure to incorporate the above context into your response."
|
|
|
|
|
|
def test_knowledge_effectiveness():
|
|
"""Test that knowledge is effectively used in agent execution."""
|
|
import pytest
|
|
from unittest.mock import patch, MagicMock
|
|
from crewai.knowledge.utils.knowledge_utils import extract_knowledge_context
|
|
|
|
# Create mock knowledge snippets
|
|
knowledge_snippets = [
|
|
{"context": "The capital of France is Paris. The Eiffel Tower is located in Paris.", "score": 0.9}
|
|
]
|
|
|
|
# Test that the extract_knowledge_context function formats the knowledge correctly
|
|
knowledge_context = extract_knowledge_context(knowledge_snippets)
|
|
|
|
# Verify the knowledge context contains the expected information
|
|
assert "paris" in knowledge_context.lower()
|
|
assert "capital" in knowledge_context.lower()
|
|
assert "france" in knowledge_context.lower()
|
|
|
|
# Verify the format is correct
|
|
assert knowledge_context.startswith("Important Context")
|
|
assert "Make sure to incorporate the above context" in knowledge_context
|