litellm-mirror/litellm/proxy/tests/test_openai_simple_embedding.py
2024-04-10 16:46:56 -07:00

10 lines
289 B
Python

import openai
client = openai.OpenAI(api_key="sk-1234", base_url="http://0.0.0.0:4000")
# # request sent to model set on litellm proxy, `litellm --model`
response = client.embeddings.create(
model="text-embedding-ada-002", input=["test"], encoding_format="base64"
)
print(response)