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"], ])