From 6d57d3f1dd3081de23aaf3e7c35528df8feb0b78 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 8 Aug 2023 11:32:31 -0700
Subject: [PATCH] fix circle ci test

---
 .circleci/config.yml            | 1 +
 litellm/tests/test_embedding.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/.circleci/config.yml b/.circleci/config.yml
index edefc4b7e..397031de7 100644
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -15,6 +15,7 @@ jobs:
             python -m pip install -r requirements.txt
             pip install infisical
             pip install pytest
+            pip install openai[datalib]
 
       # Run pytest and generate JUnit XML report
       - run:
diff --git a/litellm/tests/test_embedding.py b/litellm/tests/test_embedding.py
index be2b30a81..ce83ffc70 100644
--- a/litellm/tests/test_embedding.py
+++ b/litellm/tests/test_embedding.py
@@ -17,4 +17,4 @@ def test_openai_embedding():
         # Add any assertions here to check the response
         print(f"response: {str(response)}")
     except Exception as e:
-        pytest.fail(f"Error occurred: {e}")
+        pytest.fail(f"Error occurred: {e}")
\ No newline at end of file