From 85b403817f88a6aa2ad04e1792d004255586988c Mon Sep 17 00:00:00 2001 From: Claudio Ortolina Date: Sun, 29 Dec 2024 20:58:09 +0000 Subject: [PATCH] Move model definition into completion struct --- lib/open_ai/api.ex | 2 +- lib/open_ai/completion.ex | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) 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