From acdd3efe51ea6499d1b6ae0b6dfd3432e847aad5 Mon Sep 17 00:00:00 2001
From: cybermaggedon <cybermaggedon@gmail.com>
Date: Wed, 15 Jan 2025 23:59:50 +0000
Subject: [PATCH] Fix argument usage not overriding defaults in Google AI
 Studio & VertexAI templates (#268)

---
 templates/components/googleaistudio.jsonnet | 5 +++++
 templates/components/vertexai.jsonnet       | 5 +++++
 2 files changed, 10 insertions(+)

diff --git a/templates/components/googleaistudio.jsonnet b/templates/components/googleaistudio.jsonnet
index 00b2bd0..ac12224 100644
--- a/templates/components/googleaistudio.jsonnet
+++ b/templates/components/googleaistudio.jsonnet
@@ -5,6 +5,11 @@ local prompts = import "prompts/mixtral.jsonnet";
 
 {
 
+    with:: function(key, value)
+        self + {
+            ["googleaistudio-" + key]:: value,
+        },
+
     "googleaistudio-max-output-tokens":: 4096,
     "googleaistudio-temperature":: 0.0,
     "googleaistudio-model":: "gemini-1.5-flash-002",
diff --git a/templates/components/vertexai.jsonnet b/templates/components/vertexai.jsonnet
index fc3192a..4924370 100644
--- a/templates/components/vertexai.jsonnet
+++ b/templates/components/vertexai.jsonnet
@@ -5,6 +5,11 @@ local prompts = import "prompts/mixtral.jsonnet";
 
 {
 
+    with:: function(key, value)
+        self + {
+            ["vertexai-" + key]:: value,
+        },
+
     "vertexai-model":: "gemini-1.0-pro-001",
     "vertexai-private-key":: "/vertexai/private.json",
     "vertexai-region":: "us-central1",