remove deps datalib, pytest, tenacity, infisical

litellm/tests/test_embedding.py

@@ -0,0 +1,20 @@
+
+import sys, os
+import traceback
+import pytest
+
+sys.path.insert(0, os.path.abspath('../..'))  # Adds the parent directory to the system path
+import litellm
+from litellm import embedding, completion
+from infisical import InfisicalClient
+
+# litellm.set_verbose = True
+litellm.secret_manager_client = InfisicalClient(token=os.environ["INFISICAL_TOKEN"])
+
+def test_openai_embedding():
+    try:
+        response = embedding(model='text-embedding-ada-002', input=["good morning from litellm"])
+        # Add any assertions here to check the response
+        print(f"response: {str(response)}")
+    except Exception as e:
+        pytest.fail(f"Error occurred: {e}")