From 7565532e38e6f126d135a934d0e5dad7170deae2 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Fri, 10 Jan 2025 16:39:00 +0000 Subject: [PATCH] Fix ability to specify Claude model --- templates/components/claude-rag.jsonnet | 3 +++ templates/components/claude.jsonnet | 3 +++ 2 files changed, 6 insertions(+) diff --git a/templates/components/claude-rag.jsonnet b/templates/components/claude-rag.jsonnet index 9dcc7615..71a214fc 100644 --- a/templates/components/claude-rag.jsonnet +++ b/templates/components/claude-rag.jsonnet @@ -10,6 +10,7 @@ local prompts = import "prompts/mixtral.jsonnet"; ["claude-rag-" + key]:: value, }, + "claude-rag-model":: "claude-3-sonnet-20240229", "claude-rag-max-output-tokens":: 4096, "claude-rag-temperature":: 0.0, @@ -29,6 +30,8 @@ local prompts = import "prompts/mixtral.jsonnet"; url.pulsar, "-x", std.toString($["claude-rag-max-output-tokens"]), + "-m", + $["claude-rag-model"], "-t", "%0.3f" % $["claude-rag-temperature"], "-i", diff --git a/templates/components/claude.jsonnet b/templates/components/claude.jsonnet index 560da870..fd1650d2 100644 --- a/templates/components/claude.jsonnet +++ b/templates/components/claude.jsonnet @@ -10,6 +10,7 @@ local prompts = import "prompts/mixtral.jsonnet"; ["claude-" + key]:: value, }, + "claude-model":: "claude-3-sonnet-20240229", "claude-max-output-tokens":: 4096, "claude-temperature":: 0.0, @@ -29,6 +30,8 @@ local prompts = import "prompts/mixtral.jsonnet"; url.pulsar, "-x", std.toString($["claude-max-output-tokens"]), + "-m", + $["claude-model"], "-t", "%0.3f" % $["claude-temperature"], ])