good shit

This commit is contained in:
ishaan-jaff 2023-07-26 17:31:25 -07:00
parent 918a286e12
commit e037318344
4 changed files with 153 additions and 0 deletions

19
embedding_test.py Normal file
View 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)