From 4972a89b768053b74c7daa6e7ee929b345d6bfd5 Mon Sep 17 00:00:00 2001
From: Kyle Mistele <kyle@constellate.ai>
Date: Tue, 3 Sep 2024 23:51:42 -0500
Subject: [PATCH] fix(tests): update pytest fixture for client based on #7565

---
 tests/tool_use/conftest.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/tests/tool_use/conftest.py b/tests/tool_use/conftest.py
index 9ef90d5063e7..ab6a29eba1b3 100644
--- a/tests/tool_use/conftest.py
+++ b/tests/tool_use/conftest.py
@@ -1,4 +1,5 @@
 import pytest
+import pytest_asyncio
 from huggingface_hub import snapshot_download
 
 from tests.utils import RemoteOpenAIServer
@@ -25,6 +26,7 @@ def server(request, server_config: ServerConfig):
         yield server
 
 
-@pytest.fixture(scope="session")
-def client(server: RemoteOpenAIServer):
-    return server.get_async_client()
+@pytest_asyncio.fixture
+async def client(server: RemoteOpenAIServer):
+    async with server.get_async_client() as async_client:
+        yield async_client