forked from phoenix/litellm-mirror
good shit
This commit is contained in:
parent
918a286e12
commit
e037318344
4 changed files with 153 additions and 0 deletions
19
embedding_test.py
Normal file
19
embedding_test.py
Normal file
|
@ -0,0 +1,19 @@
|
|||
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)
|
Loading…
Add table
Add a link
Reference in a new issue