gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: change test name

Browse source
This commit is contained in:
Igor Ilic 2025-07-21 17:10:00 +02:00
parent 5440fa766c
commit 80e2f8fd9e
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
cognee/tests/test_knowledge_graph_quality.py
View file

@ -1,7 +1,7 @@
import os import os
import asyncio import asyncio
import cognee
import pathlib import pathlib
import cognee
from cognee.modules.search.types import SearchType from cognee.modules.search.types import SearchType
from cognee.modules.users.methods import get_default_user from cognee.modules.users.methods import get_default_user
from cognee.shared.logging_utils import get_logger from cognee.shared.logging_utils import get_logger
@ -9,7 +9,7 @@ from cognee.shared.logging_utils import get_logger
logger = get_logger() logger = get_logger()
async def test_knowledge_graph_quality_with_gpt4o(): async def test_knowledge_graph_quality():
""" """
Test that verifies all main concepts and entities from a specific document are found Test that verifies all main concepts and entities from a specific document are found
in the knowledge graph using the configured LLM model for entity extraction. in the knowledge graph using the configured LLM model for entity extraction.
@ -25,7 +25,7 @@ async def test_knowledge_graph_quality_with_gpt4o():
raise ValueError("LLM_API_KEY must be set for this test") raise ValueError("LLM_API_KEY must be set for this test")
# Get model from environment variable # Get model from environment variable
current_model = os.environ.get("LLM_MODEL", "gpt-4o") current_model = os.environ.get("LLM_MODEL", "gpt-4o-mini")
print(f"Using model from environment: {current_model}") print(f"Using model from environment: {current_model}")
# Set up test directories # Set up test directories
@ -91,9 +91,9 @@ async def test_knowledge_graph_quality_with_gpt4o():
] ]
print("=" * 80) print("=" * 80)
print("KNOWLEDGE GRAPH QUALITY TEST WITH GPT-4o") print("KNOWLEDGE GRAPH QUALITY TEST")
print("=" * 80) print("=" * 80)
print(f"Using model: {os.environ.get('LLM_MODEL', 'gpt-4o')}") print(f"Using model: {current_model}")
print(f"Test document: {test_document_path}") print(f"Test document: {test_document_path}")
print() print()
@ -297,4 +297,4 @@ async def test_knowledge_graph_quality_with_gpt4o():
if __name__ == "__main__": if __name__ == "__main__":
asyncio.run(test_knowledge_graph_quality_with_gpt4o()) asyncio.run(test_knowledge_graph_quality())