trustgraph/templates/components/vertexai.jsonnet
cybermaggedon a01a72ba00
Set resource limits (#22)
* Added resource limits to resources.

* Boost version number, rebuild YAMLs
2024-08-22 17:54:00 +01:00

76 lines
1.3 KiB
Jsonnet

local base = import "base.jsonnet";
local images = import "images.jsonnet";
local url = import "url.jsonnet";
{
services +: {
"text-completion": base + {
image: images.trustgraph,
command: [
"text-completion-vertexai",
"-p",
url.pulsar,
"-k",
"/vertexai/private.json",
"-r",
"us-central1",
"-x",
"4096",
"-t",
"0.0",
],
volumes: [
"./vertexai:/vertexai"
],
deploy: {
resources: {
limits: {
cpus: '0.1',
memory: '128M'
},
reservations: {
cpus: '0.1',
memory: '128M'
}
}
},
},
"text-completion-rag": base + {
image: images.trustgraph,
command: [
"text-completion-vertexai",
"-p",
url.pulsar,
"-k",
"/vertexai/private.json",
"-r",
"us-central1",
"-x",
"4096",
"-t",
"0.0",
"-i",
"non-persistent://tg/request/text-completion-rag",
"-o",
"non-persistent://tg/response/text-completion-rag-response",
],
volumes: [
"./vertexai:/vertexai"
],
deploy: {
resources: {
limits: {
cpus: '0.1',
memory: '128M'
},
reservations: {
cpus: '0.1',
memory: '128M'
}
}
},
},
},
}