diff --git a/lib/open_ai/api.ex b/lib/open_ai/api.ex index 0acb8ee1..2dc2499b 100644 --- a/lib/open_ai/api.ex +++ b/lib/open_ai/api.ex @@ -33,7 +33,7 @@ defmodule OpenAI.API do Req.post!("https://api.openai.com/v1/chat/completions", json: %{ - model: "gpt-4o-mini", + model: completion.model, messages: [Map.take(completion, [:content, :role])], response_format: %{type: "json_object"}, stream: true, diff --git a/lib/open_ai/completion.ex b/lib/open_ai/completion.ex index 7f238230..3577957e 100644 --- a/lib/open_ai/completion.ex +++ b/lib/open_ai/completion.ex @@ -1,5 +1,6 @@ defmodule OpenAI.Completion do defstruct content: "", temperature: 0.2, - role: "user" + role: "user", + model: "gpt-4o-mini" end