From e5d364d04360d00a5c92ce60da81037fb31cb8bc Mon Sep 17 00:00:00 2001 From: cthomas Date: Mon, 29 Sep 2025 17:26:04 -0700 Subject: [PATCH] fix: test sdk client fixtures (#4997) * fix: test sdk client fixtures * make server fixture not async --- tests/test_sdk_client.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/tests/test_sdk_client.py b/tests/test_sdk_client.py index 7b78f3bb..98b81d05 100644 --- a/tests/test_sdk_client.py +++ b/tests/test_sdk_client.py @@ -1,3 +1,4 @@ +import asyncio import io import json import os @@ -50,8 +51,7 @@ def run_server(): start_server(debug=True) -# @pytest.fixture(scope="module") -@pytest.fixture(scope="function") +@pytest.fixture(scope="module") def client() -> LettaSDKClient: # Get URL from environment or start server server_url = os.getenv("LETTA_SERVER_URL", f"http://localhost:{SERVER_PORT}") @@ -66,8 +66,8 @@ def client() -> LettaSDKClient: yield client -@pytest.fixture(scope="function") -async def server(): +@pytest.fixture(scope="module") +def server(): """ Creates a SyncServer instance for testing. @@ -76,7 +76,7 @@ async def server(): config = LettaConfig.load() config.save() server = SyncServer() - await server.init_async() + asyncio.run(server.init_async()) return server @@ -2250,6 +2250,7 @@ def test_run_list(client: LettaSDKClient): run = client.runs.retrieve(runs[0].id) assert run.agent_id == agent.id + @pytest.mark.asyncio async def test_create_batch(client: LettaSDKClient, server: SyncServer): # create agents