litellm-mirror/embedding_test.py
2023-07-26 17:31:25 -07:00

19 lines
399 B
Python

from main import embedding
import os
## Configs for Models ##
# OpenAI Configs
## End Configs for Models ##
input = ["Who is ishaan"]
# openai call
response = embedding(model="text-embedding-ada-002", input=input)
print("\nOpenAI call")
print(response)
# azure openai call
response = embedding(model="azure-embedding-mode", input=input, azure=True)
print("\nAzure OpenAI call")
print(response)