From f08486448c0543d3d91df5123a547dfddba82e87 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 6 Apr 2024 18:28:07 -0700 Subject: [PATCH] fix - test streaming --- .circleci/config.yml | 1 + litellm/tests/test_streaming.py | 3 +++ 2 files changed, 4 insertions(+) diff --git a/.circleci/config.yml b/.circleci/config.yml index 92892d3ff8..2bad708cb3 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -42,6 +42,7 @@ jobs: pip install lunary==0.2.5 pip install "langfuse==2.7.3" pip install numpydoc + pip install nest-asyncio==1.6.0 pip install traceloop-sdk==0.0.69 pip install openai pip install prisma diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py index 6041788baa..4ca61303fd 100644 --- a/litellm/tests/test_streaming.py +++ b/litellm/tests/test_streaming.py @@ -26,6 +26,9 @@ litellm.logging = False litellm.set_verbose = True litellm.num_retries = 3 litellm.cache = None +import nest_asyncio + +nest_asyncio.apply() score = 0