Skip to content

Commit 5935334

Browse files
authored
Update OpenAI.jl (#67)
Add create_embedding function that receives a provider instead of API_KEY
1 parent f70f44d commit 5935334

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

src/OpenAI.jl

+14-1
Original file line numberDiff line numberDiff line change
@@ -417,7 +417,20 @@ function create_embeddings(api_key::String,
417417
input,
418418
kwargs...)
419419
end
420-
420+
function create_embeddings(provider::AbstractOpenAIProvider,
421+
input;
422+
model_id::String = DEFAULT_EMBEDDING_MODEL_ID,
423+
http_kwargs::NamedTuple=NamedTuple(),
424+
streamcallback=nothing,
425+
kwargs...)
426+
return OpenAI.openai_request("embeddings",
427+
provider;
428+
method="POST",
429+
http_kwargs=http_kwargs,
430+
model=model_id,
431+
input,
432+
kwargs...)
433+
end
421434
"""
422435
Create images
423436

0 commit comments

Comments
 (0)