Fix argument usage not overriding defaults in Google AI Studio & Vert…

…exAI templates (#268)
trustgraph-ai · Jan 15, 2025 · acdd3ef · acdd3ef
1 parent 1280af3
commit acdd3ef
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git 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
@@ -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",