apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-17 17:21:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Updated templates to for prompt-template update

Browse source
This commit is contained in:
Cyber MacGeddon 2024-10-26 15:35:22 +01:00
parent 99067cb6c0
commit 8f1a44c06e
4 changed files with 108 additions and 50 deletions
Download patch file Download diff file Expand all files Collapse all files

2
templates/components/googleaistudio.jsonnet
View file

@ -13,7 +13,7 @@ local prompts = import "prompts/mixtral.jsonnet";
create:: function(engine)
local envSecrets = engine.envSecrets("bedrock-credentials")
.with_env_var("GOOGLEAISTUDIO_KEY", "googleaistudio-key");
.with_env_var("GOOGLE_AI_STUDIO_KEY", "googleaistudio-key");
local container =
engine.container("text-completion")

55
templates/components/prompt-template.jsonnet
View file

@ -17,22 +17,38 @@ local default_prompts = import "prompts/default-prompts.jsonnet";
"prompt-template",
"-p",
url.pulsar,
"--text-completion-request-queue",
"non-persistent://tg/request/text-completion",
"--text-completion-response-queue",
"non-persistent://tg/response/text-completion-response",
"--definition-template",
"--system-prompt",
$["system-template"],
"--prompt",
"question={{question}}",
"extract-definitions=" +
$["prompt-definition-template"],
"--relationship-template",
"extract-relationships=" +
$["prompt-relationship-template"],
"--topic-template",
"extract-topics=" +
$["prompt-topic-template"],
"--knowledge-query-template",
"kg-prompt=" +
$["prompt-knowledge-query-template"],
"--document-query-template",
"document-prompt=" +
$["prompt-document-query-template"],
"--rows-template",
"extract-rows=" +
$["prompt-rows-template"],
"--prompt-response-type",
"extract-definitions=json",
"extract-relationships=json",
"extract-topics=json",
"kg-prompt=text",
"document-prompt=text",
"extract-rows=json",
])
.with_limits("0.5", "128M")
.with_reservations("0.1", "128M");
@ -71,18 +87,33 @@ local default_prompts = import "prompts/default-prompts.jsonnet";
"non-persistent://tg/request/text-completion-rag",
"--text-completion-response-queue",
"non-persistent://tg/response/text-completion-rag-response",
"--definition-template",
"--system-prompt",
$["system-template"],
"--prompt",
"question={{question}}",
"extract-definitions=" +
$["prompt-definition-template"],
"--relationship-template",
"extract-relationships=" +
$["prompt-relationship-template"],
"--topic-template",
"extract-topics=" +
$["prompt-topic-template"],
"--knowledge-query-template",
"kg-prompt=" +
$["prompt-knowledge-query-template"],
"--document-query-template",
"document-prompt=" +
$["prompt-document-query-template"],
"--rows-template",
"extract-rows=" +
$["prompt-rows-template"],
"--prompt-response-type",
"extract-definitions=json",
"extract-relationships=json",
"extract-topics=json",
"kg-prompt=text",
"document-prompt=text",
"extract-rows=json",
])
.with_limits("0.5", "128M")
.with_reservations("0.1", "128M");