From 3a1f34bc44bb46a3598f4c21d24a81b0e8237cac Mon Sep 17 00:00:00 2001 From: BeatrixCohere <128378696+BeatrixCohere@users.noreply.github.com> Date: Tue, 15 Aug 2023 13:38:38 +0100 Subject: [PATCH] fix --- tests/async/test_async_chat.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/async/test_async_chat.py b/tests/async/test_async_chat.py index da72609fc..db3ac538c 100644 --- a/tests/async/test_async_chat.py +++ b/tests/async/test_async_chat.py @@ -4,7 +4,9 @@ @pytest.mark.asyncio async def test_async_multi_replies(async_client): num_replies = 3 - prediction = await async_client.chat("Yo what's up?", return_chatlog=True, max_tokens=5) + prediction = await async_client.chat( + "Yo what's up?", return_chatlog=True, max_tokens=5, conversation_id="test_conv_id" + ) assert prediction.chatlog is not None for _ in range(num_replies): prediction = await prediction.respond("oh that's cool", max_tokens=5) @@ -21,6 +23,7 @@ async def test_async_chat_stream(async_client): res = await async_client.chat( message="wagmi", max_tokens=5, + conversation_id="test_conv_id", stream=True, )