Fix Google AI Studio model (#139)

trustgraph-ai · Nov 5, 2024 · 45091e0 · 45091e0
1 parent ee9373d
commit 45091e0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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",