diff --git a/templates/components/googleaistudio.jsonnet b/templates/components/googleaistudio.jsonnet index 00b2bd0e..ac122243 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 fc3192a4..49243701 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",