Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
BeatrixCohere committed Aug 15, 2023
1 parent 4162ab4 commit 3a1f34b
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion tests/async/test_async_chat.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand All @@ -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,
)

Expand Down

0 comments on commit 3a1f34b

Please sign in to comment.