Move ChatGPT model specification to an environment variable

identity-research-lab · Jun 17, 2024 · 2d77f23 · 2d77f23
1 parent 065e90f
commit 2d77f23
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.env.example b/.env.example
@@ -1,4 +1,5 @@
 OPENAI_ACCESS_TOKEN=
+OPENAI_MODEL="gpt-3.5-turbo"
 GENERAL_ADMISSION_USERNAME=
 GENERAL_ADMISSION_PASSWORD=
 NEO4J_URI=neo4j://127.0.0.1:7687/neo4j

diff --git a/lib/clients/open_ai.rb b/lib/clients/open_ai.rb
@@ -4,7 +4,7 @@ def self.request(prompt)
     client = OpenAI::Client.new
     response = client.chat(
       parameters: {
-        model: "gpt-4o",
+        model: ENV.fetch("OPENAI_MODEL"),
         response_format: { type: "json_object" },
         messages: [{ role: "user", content: prompt }],
         temperature: 0.7,