From 4746c3f12b4ce8a1cd47e2f45e0ea8ab3258ca9d Mon Sep 17 00:00:00 2001 From: cthomas Date: Fri, 16 May 2025 02:14:20 -0700 Subject: [PATCH] test(asyncify): fix batch tests (#2216) --- tests/integration_test_batch_api_cron_jobs.py | 11 ++--------- 1 file changed, 2 insertions(+), 9 deletions(-) diff --git a/tests/integration_test_batch_api_cron_jobs.py b/tests/integration_test_batch_api_cron_jobs.py index 4479b0dd..786bad13 100644 --- a/tests/integration_test_batch_api_cron_jobs.py +++ b/tests/integration_test_batch_api_cron_jobs.py @@ -16,7 +16,6 @@ from anthropic.types.beta.messages import ( BetaMessageBatchSucceededResult, ) from dotenv import load_dotenv -from letta_client import Letta from letta.config import LettaConfig from letta.helpers import ToolRulesSolver @@ -75,12 +74,6 @@ def server(): return SyncServer() -@pytest.fixture(scope="session") -def client(server_url): - """Creates a REST client for testing.""" - return Letta(base_url=server_url) - - # --- Dummy Response Factories --- # @@ -263,7 +256,7 @@ def mock_anthropic_client(server, batch_a_resp, batch_b_resp, agent_b_id, agent_ # End-to-End Test # ----------------------------- @pytest.mark.asyncio(loop_scope="session") -async def test_polling_simple_real_batch(client, default_user, server): +async def test_polling_simple_real_batch(default_user, server): # --- Step 1: Prepare test data --- # Create batch responses with different statuses # NOTE: This is a REAL batch id! @@ -404,7 +397,7 @@ async def test_polling_simple_real_batch(client, default_user, server): @pytest.mark.asyncio(loop_scope="session") -async def test_polling_mixed_batch_jobs(client, default_user, server): +async def test_polling_mixed_batch_jobs(default_user, server): """ End-to-end test for polling batch jobs with mixed statuses and idempotency.