Update OpenAI.jl (#67)

Add create_embedding function that receives a provider instead of API_KEY
JuliaML · Nov 29, 2024 · 5935334 · 5935334
1 parent f70f44d
commit 5935334
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/src/OpenAI.jl b/src/OpenAI.jl
@@ -417,7 +417,20 @@ function create_embeddings(api_key::String,
         input,
         kwargs...)
 end
-
+function create_embeddings(provider::AbstractOpenAIProvider,
+    input;
+    model_id::String = DEFAULT_EMBEDDING_MODEL_ID,   
+    http_kwargs::NamedTuple=NamedTuple(),
+    streamcallback=nothing,
+    kwargs...)
+    return OpenAI.openai_request("embeddings",
+        provider;
+        method="POST",
+        http_kwargs=http_kwargs,
+        model=model_id,
+        input,
+        kwargs...)
+end
 """
 Create images