From 56b10d8be5337efddb9826240685293fdd3cf340 Mon Sep 17 00:00:00 2001 From: Daniel Chalef <131175+danielchalef@users.noreply.github.com> Date: Sun, 12 Oct 2025 10:05:36 -0700 Subject: [PATCH] Restore integration markers for tests that call LLM APIs MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit test_graphiti_int.py and test_entity_exclusion_int.py call graphiti.add_episode() and graphiti.search_() which require LLM API calls, so they are API integration tests, not pure database tests. Final categorization: Pure unit tests (no external dependencies): - tests/llm_client/test_*.py (except test_anthropic_client_int.py) - tests/embedder/test_*.py - tests/utils/maintenance/test_*.py (except test_temporal_operations_int.py) - tests/utils/search/search_utils_test.py - tests/test_text_utils.py Database tests (require Neo4j/FalkorDB, no API calls): - tests/test_graphiti_mock.py - tests/test_node_int.py - tests/test_edge_int.py - tests/cross_encoder/test_bge_reranker_client_int.py - tests/driver/test_falkordb_driver.py API integration tests (excluded via @pytest.mark.integration): - tests/test_graphiti_int.py - tests/test_entity_exclusion_int.py - tests/llm_client/test_anthropic_client_int.py - tests/utils/maintenance/test_temporal_operations_int.py 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude --- .github/workflows/unit_tests.yml | 2 -- tests/test_entity_exclusion_int.py | 1 + tests/test_graphiti_int.py | 1 + 3 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml index 07abebb1..c2c664a6 100644 --- a/.github/workflows/unit_tests.yml +++ b/.github/workflows/unit_tests.yml @@ -93,10 +93,8 @@ jobs: run: | uv run pytest \ tests/test_graphiti_mock.py \ - tests/test_graphiti_int.py \ tests/test_node_int.py \ tests/test_edge_int.py \ - tests/test_entity_exclusion_int.py \ tests/cross_encoder/test_bge_reranker_client_int.py \ tests/driver/test_falkordb_driver.py \ -m "not integration" diff --git a/tests/test_entity_exclusion_int.py b/tests/test_entity_exclusion_int.py index 53982bc9..0ac9897c 100644 --- a/tests/test_entity_exclusion_int.py +++ b/tests/test_entity_exclusion_int.py @@ -23,6 +23,7 @@ from graphiti_core.graphiti import Graphiti from graphiti_core.helpers import validate_excluded_entity_types from tests.helpers_test import drivers, get_driver +pytestmark = pytest.mark.integration pytest_plugins = ('pytest_asyncio',) diff --git a/tests/test_graphiti_int.py b/tests/test_graphiti_int.py index 766828e3..90033f35 100644 --- a/tests/test_graphiti_int.py +++ b/tests/test_graphiti_int.py @@ -25,6 +25,7 @@ from graphiti_core.search.search_helpers import search_results_to_context_string from graphiti_core.utils.datetime_utils import utc_now from tests.helpers_test import GraphProvider +pytestmark = pytest.mark.integration pytest_plugins = ('pytest_asyncio',)