|
16 | 16 | BetaMessageBatchSucceededResult, |
17 | 17 | ) |
18 | 18 | from dotenv import load_dotenv |
19 | | -from letta_client import Letta |
20 | 19 |
|
21 | 20 | from letta.config import LettaConfig |
22 | 21 | from letta.helpers import ToolRulesSolver |
@@ -75,12 +74,6 @@ def server(): |
75 | 74 | return SyncServer() |
76 | 75 |
|
77 | 76 |
|
78 | | -@pytest.fixture(scope="session") |
79 | | -def client(server_url): |
80 | | - """Creates a REST client for testing.""" |
81 | | - return Letta(base_url=server_url) |
82 | | - |
83 | | - |
84 | 77 | # --- Dummy Response Factories --- # |
85 | 78 |
|
86 | 79 |
|
@@ -263,7 +256,7 @@ async def dummy_results(batch_resp_id: str): |
263 | 256 | # End-to-End Test |
264 | 257 | # ----------------------------- |
265 | 258 | @pytest.mark.asyncio(loop_scope="session") |
266 | | -async def test_polling_simple_real_batch(client, default_user, server): |
| 259 | +async def test_polling_simple_real_batch(default_user, server): |
267 | 260 | # --- Step 1: Prepare test data --- |
268 | 261 | # Create batch responses with different statuses |
269 | 262 | # NOTE: This is a REAL batch id! |
@@ -404,7 +397,7 @@ async def test_polling_simple_real_batch(client, default_user, server): |
404 | 397 |
|
405 | 398 |
|
406 | 399 | @pytest.mark.asyncio(loop_scope="session") |
407 | | -async def test_polling_mixed_batch_jobs(client, default_user, server): |
| 400 | +async def test_polling_mixed_batch_jobs(default_user, server): |
408 | 401 | """ |
409 | 402 | End-to-end test for polling batch jobs with mixed statuses and idempotency. |
410 | 403 |
|
|
0 commit comments