From 45091e01ec2e51704ffef1d8ca5c9a13271fb0a9 Mon Sep 17 00:00:00 2001
From: cybermaggedon <cybermaggedon@gmail.com>
Date: Tue, 5 Nov 2024 23:49:24 +0000
Subject: [PATCH] Fix Google AI Studio model (#139)

---
 templates/components/googleaistudio.jsonnet | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/templates/components/googleaistudio.jsonnet b/templates/components/googleaistudio.jsonnet
index ae880e36..6fa01115 100644
--- a/templates/components/googleaistudio.jsonnet
+++ b/templates/components/googleaistudio.jsonnet
@@ -26,6 +26,8 @@ local prompts = import "prompts/mixtral.jsonnet";
                         std.toString($["googleaistudio-max-output-tokens"]),
                         "-t",
                         "%0.3f" % $["googleaistudio-temperature"],
+                        "-m",
+                        $["googleaistudio-model"],
                     ])
                     .with_env_var_secrets(envSecrets)
                     .with_limits("0.5", "128M")
@@ -42,6 +44,8 @@ local prompts = import "prompts/mixtral.jsonnet";
                         std.toString($["googleaistudio-max-output-tokens"]),
                         "-t",
                         "%0.3f" % $["googleaistudio-temperature"],
+                        "-m",
+                        $["googleaistudio-model"],
                         "-i",
                         "non-persistent://tg/request/text-completion-rag",
                         "-o",