Fix ability to specify Claude model

This commit is contained in:
Cyber MacGeddon 2025-01-10 16:39:00 +00:00
parent a595217aa4
commit 7565532e38
2 changed files with 6 additions and 0 deletions

View file

@ -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",

View file

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