trustgraph/templates/docker-compose-openai-neo4j.jsonnet

69 lines
1.3 KiB
Jsonnet
Raw Normal View History

local base = import "components/base.jsonnet";
local url = import "components/url.jsonnet";
local images = import "components/images.jsonnet";
local neo4j = import "components/neo4j.jsonnet";
local pulsar = import "components/pulsar.jsonnet";
local milvus = import "components/milvus.jsonnet";
local grafana = import "components/grafana.jsonnet";
local trustgraph = import "components/trustgraph.jsonnet";
local config = neo4j + pulsar + milvus + grafana + trustgraph + {
services: std.mergePatch(super.services, {
"text-completion": base + {
image: images.trustgraph,
command: [
"text-completion-openai",
"-p",
url.pulsar,
"-k",
"${OPENAI_KEY}",
],
},
"text-completion-rag": base + {
image: images.trustgraph,
command: [
"text-completion-openai",
"-p",
url.pulsar,
"-k",
"${OPENAI_KEY}",
"-i",
"non-persistent://tg/request/text-completion-rag",
"-o",
"non-persistent://tg/response/text-completion-rag-response",
],
},
"query-triples": {
image: images.trustgraph,
command: [
"triples-query-neo4j",
"-p",
url.pulsar,
"-g",
"bolt://neo4j:7687",
],
},
"store-triples": {
image: images.trustgraph,
command: [
"triples-write-neo4j",
"-p",
url.pulsar,
"-g",
"bolt://neo4j:7687",
],
}
})
};
std.manifestYamlDoc(config)