From f10aba100766871bd9e373f1d81522ea23182ea3 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 6 Apr 2024 18:23:10 -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 92892d3ff..355e57e4b 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -43,6 +43,7 @@ jobs: pip install "langfuse==2.7.3" pip install numpydoc pip install traceloop-sdk==0.0.69 + pip install nest-asyncio==1.6.0 pip install openai pip install prisma pip install "httpx==0.24.1" diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py index ed6298752..81c021093 100644 --- a/litellm/tests/test_streaming.py +++ b/litellm/tests/test_streaming.py @@ -5,6 +5,9 @@ import sys, os, asyncio import traceback import time, pytest from pydantic import BaseModel +import nest_asyncio + +nest_asyncio.apply() sys.path.insert( 0, os.path.abspath("../..")