diff --git a/ai-hub/integration_tests/conftest.py b/ai-hub/integration_tests/conftest.py index f45d069..4984094 100644 --- a/ai-hub/integration_tests/conftest.py +++ b/ai-hub/integration_tests/conftest.py @@ -121,6 +121,13 @@ "group_id": group_id, "access_level": "use" }) + + # CRITICAL FIX: Ensure the user's DB preferences points to these fresh + # nodes so that tools correctly route instead of using stale nodes from prior runs. + updated_prefs = { + "default_node_ids": [NODE_1] + } + client.patch(f"{BASE_URL}/nodes/preferences", json=updated_prefs, headers=headers) # 5. Start Node Processes is_docker_disabled = os.getenv("SKIP_DOCKER_NODES", "true").lower() == "true"