Parameters, Parsing, renaming YAMLs and Neo4j YAMLS (#15)

* Added some params
* Parameter updates
* Fixed Neo4j issue
This commit is contained in:
Jack Colquitt 2024-08-21 16:03:56 -07:00 committed by GitHub
parent 7d00a53711
commit c4bfd9fc8c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
25 changed files with 2258 additions and 200 deletions

View file

@ -1,6 +1,6 @@
# VERSION=$(shell git describe | sed 's/^v//') # VERSION=$(shell git describe | sed 's/^v//')
VERSION=0.6.8 VERSION=0.6.9
DOCKER=podman DOCKER=podman

View file

@ -4,7 +4,7 @@ import os
with open("README.md", "r") as fh: with open("README.md", "r") as fh:
long_description = fh.read() long_description = fh.read()
version = "0.6.8" version = "0.6.9"
setuptools.setup( setuptools.setup(
name="trustgraph", name="trustgraph",

271
tg-launch-azure-neo4j.yaml Normal file
View file

@ -0,0 +1,271 @@
"services":
"neo4j":
"environment":
"NEO4J_AUTH": "neo4j/password"
"image": "docker.io/neo4j:5.22.0-community-bullseye"
"ports":
- "7474:7474"
- "7687:7687"
"restart": "on-failure:100"
"volumes":
- "neo4j:/data"
"chunker":
"command":
- "chunker-recursive"
- "-p"
- "pulsar://pulsar:6650"
- "--chunk-size"
- "2000"
- "--chunk-overlap"
- "100"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"embeddings":
"command":
- "embeddings-hf"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"etcd":
"command":
- "etcd"
- "-advertise-client-urls=http://127.0.0.1:2379"
- "-listen-client-urls"
- "http://0.0.0.0:2379"
- "--data-dir"
- "/etcd"
"environment":
"ETCD_AUTO_COMPACTION_MODE": "revision"
"ETCD_AUTO_COMPACTION_RETENTION": "1000"
"ETCD_QUOTA_BACKEND_BYTES": "4294967296"
"ETCD_SNAPSHOT_COUNT": "50000"
"image": "quay.io/coreos/etcd:v3.5.5"
"ports":
- "2379:2379"
"restart": "on-failure:100"
"volumes":
- "etcd:/etcd"
"grafana":
"environment":
"GF_ORG_NAME": "trustgraph.ai"
"image": "docker.io/grafana/grafana:10.0.0"
"ports":
- "3000:3000"
"restart": "on-failure:100"
"volumes":
- "grafana-storage:/var/lib/grafana"
- "./grafana/dashboard.yml:/etc/grafana/provisioning/dashboards/dashboard.yml"
- "./grafana/datasource.yml:/etc/grafana/provisioning/datasources/datasource.yml"
- "./grafana/dashboard.json:/var/lib/grafana/dashboards/dashboard.json"
"graph-rag":
"command":
- "graph-rag"
- "-p"
- "pulsar://pulsar:6650"
- "--prompt-request-queue"
- "non-persistent://tg/request/prompt-rag"
- "--prompt-response-queue"
- "non-persistent://tg/response/prompt-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"init-pulsar":
"command":
- "sh"
- "-c"
- "pulsar-admin --admin-url http://pulsar:8080 tenants create tg && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/flow && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/request && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/response && pulsar-admin --admin-url http://pulsar:8080 namespaces set-retention --size -1 --time 3m tg/response"
"depends_on":
"pulsar":
"condition": "service_started"
"image": "docker.io/apachepulsar/pulsar:3.3.0"
"restart": "on-failure:100"
"kg-extract-definitions":
"command":
- "kg-extract-definitions"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"kg-extract-relationships":
"command":
- "kg-extract-relationships"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"milvus":
"command":
- "milvus"
- "run"
- "standalone"
"environment":
"ETCD_ENDPOINTS": "etcd:2379"
"MINIO_ADDRESS": "minio:9000"
"image": "docker.io/milvusdb/milvus:v2.4.5"
"ports":
- "9091:9091"
- "19530:19530"
"restart": "on-failure:100"
"volumes":
- "milvus:/var/lib/milvus"
"minio":
"command":
- "minio"
- "server"
- "/minio_data"
- "--console-address"
- ":9001"
"environment":
"MINIO_ROOT_PASSWORD": "minioadmin"
"MINIO_ROOT_USER": "minioadmin"
"image": "docker.io/minio/minio:RELEASE.2024-07-04T14-25-45Z"
"ports":
- "9001:9001"
"restart": "on-failure:100"
"volumes":
- "minio-data:/minio_data"
"pdf-decoder":
"command":
- "pdf-decoder"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"prometheus":
"image": "docker.io/prom/prometheus:v2.53.1"
"ports":
- "9090:9090"
"restart": "on-failure:100"
"volumes":
- "./prometheus:/etc/prometheus"
- "prometheus-data:/prometheus"
"prompt":
"command":
- "prompt-generic"
- "-p"
- "pulsar://pulsar:6650"
- "--text-completion-request-queue"
- "non-persistent://tg/request/text-completion"
- "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"prompt-rag":
"command":
- "prompt-generic"
- "-p"
- "pulsar://pulsar:6650"
- "-i"
- "non-persistent://tg/request/prompt-rag"
- "-o"
- "non-persistent://tg/response/prompt-rag-response"
- "--text-completion-request-queue"
- "non-persistent://tg/request/text-completion-rag"
- "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"pulsar":
"command": "bin/pulsar standalone"
"image": "docker.io/apachepulsar/pulsar:3.3.0"
"ports":
- "6650:6650"
- "8080:8080"
"restart": "on-failure:100"
"volumes":
- "pulsar-conf:/pulsar/conf"
- "pulsar-data:/pulsar/data"
"pulsar-manager":
"environment":
"SPRING_CONFIGURATION_FILE": "/pulsar-manager/pulsar-manager/application.properties"
"image": "docker.io/apachepulsar/pulsar-manager:v0.3.0"
"ports":
- "9527:9527"
- "7750:7750"
"restart": "on-failure:100"
"query-graph-embeddings":
"command":
- "ge-query-milvus"
- "-p"
- "pulsar://pulsar:6650"
- "-t"
- "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"query-triples":
"command":
- "triples-query-neo4j"
- "-p"
- "pulsar://pulsar:6650"
- "-g"
- "bolt://neo4j:7687"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"store-graph-embeddings":
"command":
- "ge-write-milvus"
- "-p"
- "pulsar://pulsar:6650"
- "-t"
- "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"store-triples":
"command":
- "triples-write-neo4j"
- "-p"
- "pulsar://pulsar:6650"
- "-g"
- "bolt://neo4j:7687"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"text-completion":
"command":
- "text-completion-azure"
- "-p"
- "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
- "-k"
- "${AZURE_TOKEN}"
- "-e"
- "${AZURE_ENDPOINT}"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"text-completion-rag":
"command":
- "text-completion-azure"
- "-p"
- "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
- "-k"
- "${AZURE_TOKEN}"
- "-e"
- "${AZURE_ENDPOINT}"
- "-i"
- "non-persistent://tg/request/text-completion-rag"
- "-o"
- "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"vectorize":
"command":
- "embeddings-vectorize"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"volumes":
"neo4j": {}
"etcd": {}
"grafana-storage": {}
"milvus": {}
"minio-data": {}
"prometheus-data": {}
"pulsar-conf": {}
"pulsar-data": {}

View file

@ -15,7 +15,7 @@
- "250" - "250"
- "--chunk-overlap" - "--chunk-overlap"
- "15" - "15"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"embeddings": "embeddings":
"command": "command":
@ -24,7 +24,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-m" - "-m"
- "all-MiniLM-L6-v2" - "all-MiniLM-L6-v2"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"etcd": "etcd":
"command": "command":
@ -72,7 +72,7 @@
- "30" - "30"
- "--max-subgraph-size" - "--max-subgraph-size"
- "3000" - "3000"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"init-pulsar": "init-pulsar":
"command": "command":
@ -89,14 +89,14 @@
- "kg-extract-definitions" - "kg-extract-definitions"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"kg-extract-relationships": "kg-extract-relationships":
"command": "command":
- "kg-extract-relationships" - "kg-extract-relationships"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"milvus": "milvus":
"command": "command":
@ -134,7 +134,7 @@
- "pdf-decoder" - "pdf-decoder"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"prometheus": "prometheus":
"image": "docker.io/prom/prometheus:v2.53.1" "image": "docker.io/prom/prometheus:v2.53.1"
@ -153,7 +153,7 @@
- "non-persistent://tg/request/text-completion" - "non-persistent://tg/request/text-completion"
- "--text-completion-response-queue" - "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-response" - "non-persistent://tg/response/text-completion-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"prompt-rag": "prompt-rag":
"command": "command":
@ -168,7 +168,7 @@
- "non-persistent://tg/request/text-completion-rag" - "non-persistent://tg/request/text-completion-rag"
- "--text-completion-response-queue" - "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-rag-response" - "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"pulsar": "pulsar":
"command": "bin/pulsar standalone" "command": "bin/pulsar standalone"
@ -195,7 +195,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-t" - "-t"
- "http://milvus:19530" - "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"query-triples": "query-triples":
"command": "command":
@ -204,7 +204,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-g" - "-g"
- "cassandra" - "cassandra"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"store-graph-embeddings": "store-graph-embeddings":
"command": "command":
@ -213,7 +213,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-t" - "-t"
- "http://milvus:19530" - "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"store-triples": "store-triples":
"command": "command":
@ -222,24 +222,32 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-g" - "-g"
- "cassandra" - "cassandra"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"text-completion": "text-completion":
"command": "command":
- "text-completion-azure" - "text-completion-azure"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
- "-k" - "-k"
- "${AZURE_TOKEN}" - "${AZURE_TOKEN}"
- "-e" - "-e"
- "${AZURE_ENDPOINT}" - "${AZURE_ENDPOINT}"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"text-completion-rag": "text-completion-rag":
"command": "command":
- "text-completion-azure" - "text-completion-azure"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
- "-k" - "-k"
- "${AZURE_TOKEN}" - "${AZURE_TOKEN}"
- "-e" - "-e"
@ -248,14 +256,14 @@
- "non-persistent://tg/request/text-completion-rag" - "non-persistent://tg/request/text-completion-rag"
- "-o" - "-o"
- "non-persistent://tg/response/text-completion-rag-response" - "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"vectorize": "vectorize":
"command": "command":
- "embeddings-vectorize" - "embeddings-vectorize"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"volumes": "volumes":
"cassandra": {} "cassandra": {}

View file

@ -0,0 +1,279 @@
"services":
"neo4j":
"environment":
"NEO4J_AUTH": "neo4j/password"
"image": "docker.io/neo4j:5.22.0-community-bullseye"
"ports":
- "7474:7474"
- "7687:7687"
"restart": "on-failure:100"
"volumes":
- "neo4j:/data"
"chunker":
"command":
- "chunker-recursive"
- "-p"
- "pulsar://pulsar:6650"
- "--chunk-size"
- "500"
- "--chunk-overlap"
- "25"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"embeddings":
"command":
- "embeddings-hf"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"etcd":
"command":
- "etcd"
- "-advertise-client-urls=http://127.0.0.1:2379"
- "-listen-client-urls"
- "http://0.0.0.0:2379"
- "--data-dir"
- "/etcd"
"environment":
"ETCD_AUTO_COMPACTION_MODE": "revision"
"ETCD_AUTO_COMPACTION_RETENTION": "1000"
"ETCD_QUOTA_BACKEND_BYTES": "4294967296"
"ETCD_SNAPSHOT_COUNT": "50000"
"image": "quay.io/coreos/etcd:v3.5.5"
"ports":
- "2379:2379"
"restart": "on-failure:100"
"volumes":
- "etcd:/etcd"
"grafana":
"environment":
"GF_ORG_NAME": "trustgraph.ai"
"image": "docker.io/grafana/grafana:10.0.0"
"ports":
- "3000:3000"
"restart": "on-failure:100"
"volumes":
- "grafana-storage:/var/lib/grafana"
- "./grafana/dashboard.yml:/etc/grafana/provisioning/dashboards/dashboard.yml"
- "./grafana/datasource.yml:/etc/grafana/provisioning/datasources/datasource.yml"
- "./grafana/dashboard.json:/var/lib/grafana/dashboards/dashboard.json"
"graph-rag":
"command":
- "graph-rag"
- "-p"
- "pulsar://pulsar:6650"
- "--prompt-request-queue"
- "non-persistent://tg/request/prompt-rag"
- "--prompt-response-queue"
- "non-persistent://tg/response/prompt-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"init-pulsar":
"command":
- "sh"
- "-c"
- "pulsar-admin --admin-url http://pulsar:8080 tenants create tg && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/flow && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/request && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/response && pulsar-admin --admin-url http://pulsar:8080 namespaces set-retention --size -1 --time 3m tg/response"
"depends_on":
"pulsar":
"condition": "service_started"
"image": "docker.io/apachepulsar/pulsar:3.3.0"
"restart": "on-failure:100"
"kg-extract-definitions":
"command":
- "kg-extract-definitions"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"kg-extract-relationships":
"command":
- "kg-extract-relationships"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"milvus":
"command":
- "milvus"
- "run"
- "standalone"
"environment":
"ETCD_ENDPOINTS": "etcd:2379"
"MINIO_ADDRESS": "minio:9000"
"image": "docker.io/milvusdb/milvus:v2.4.5"
"ports":
- "9091:9091"
- "19530:19530"
"restart": "on-failure:100"
"volumes":
- "milvus:/var/lib/milvus"
"minio":
"command":
- "minio"
- "server"
- "/minio_data"
- "--console-address"
- ":9001"
"environment":
"MINIO_ROOT_PASSWORD": "minioadmin"
"MINIO_ROOT_USER": "minioadmin"
"image": "docker.io/minio/minio:RELEASE.2024-07-04T14-25-45Z"
"ports":
- "9001:9001"
"restart": "on-failure:100"
"volumes":
- "minio-data:/minio_data"
"pdf-decoder":
"command":
- "pdf-decoder"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"prometheus":
"image": "docker.io/prom/prometheus:v2.53.1"
"ports":
- "9090:9090"
"restart": "on-failure:100"
"volumes":
- "./prometheus:/etc/prometheus"
- "prometheus-data:/prometheus"
"prompt":
"command":
- "prompt-generic"
- "-p"
- "pulsar://pulsar:6650"
- "--text-completion-request-queue"
- "non-persistent://tg/request/text-completion"
- "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"prompt-rag":
"command":
- "prompt-generic"
- "-p"
- "pulsar://pulsar:6650"
- "-i"
- "non-persistent://tg/request/prompt-rag"
- "-o"
- "non-persistent://tg/response/prompt-rag-response"
- "--text-completion-request-queue"
- "non-persistent://tg/request/text-completion-rag"
- "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"pulsar":
"command": "bin/pulsar standalone"
"image": "docker.io/apachepulsar/pulsar:3.3.0"
"ports":
- "6650:6650"
- "8080:8080"
"restart": "on-failure:100"
"volumes":
- "pulsar-conf:/pulsar/conf"
- "pulsar-data:/pulsar/data"
"pulsar-manager":
"environment":
"SPRING_CONFIGURATION_FILE": "/pulsar-manager/pulsar-manager/application.properties"
"image": "docker.io/apachepulsar/pulsar-manager:v0.3.0"
"ports":
- "9527:9527"
- "7750:7750"
"restart": "on-failure:100"
"query-graph-embeddings":
"command":
- "ge-query-milvus"
- "-p"
- "pulsar://pulsar:6650"
- "-t"
- "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"query-triples":
"command":
- "triples-query-neo4j"
- "-p"
- "pulsar://pulsar:6650"
- "-g"
- "bolt://neo4j:7687"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"store-graph-embeddings":
"command":
- "ge-write-milvus"
- "-p"
- "pulsar://pulsar:6650"
- "-t"
- "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"store-triples":
"command":
- "triples-write-neo4j"
- "-p"
- "pulsar://pulsar:6650"
- "-g"
- "bolt://neo4j:7687"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"text-completion":
"command":
- "text-completion-bedrock"
- "-p"
- "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
# - "-m"
# - "mistral.mixtral-8x7b-instruct-v0:1"
- "-z"
- "${AWS_ID_KEY}"
- "-k"
- "${AWS_SECRET_KEY}"
- "-r"
- "us-west-2"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"text-completion-rag":
"command":
- "text-completion-bedrock"
- "-p"
- "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
# - "-m"
# - "mistral.mixtral-8x7b-instruct-v0:1"
- "-z"
- "${AWS_ID_KEY}"
- "-k"
- "${AWS_SECRET_KEY}"
- "-r"
- "us-west-2"
- "-i"
- "non-persistent://tg/request/text-completion-rag"
- "-o"
- "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"vectorize":
"command":
- "embeddings-vectorize"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"volumes":
"neo4j": {}
"etcd": {}
"grafana-storage": {}
"milvus": {}
"minio-data": {}
"prometheus-data": {}
"pulsar-conf": {}
"pulsar-data": {}

View file

@ -15,7 +15,7 @@
- "2000" - "2000"
- "--chunk-overlap" - "--chunk-overlap"
- "100" - "100"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"embeddings": "embeddings":
"command": "command":
@ -24,7 +24,6 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-m" - "-m"
- "all-MiniLM-L6-v2" - "all-MiniLM-L6-v2"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8"
"restart": "on-failure:100" "restart": "on-failure:100"
"etcd": "etcd":
"command": "command":
@ -72,7 +71,7 @@
- "30" - "30"
- "--max-subgraph-size" - "--max-subgraph-size"
- "3000" - "3000"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"init-pulsar": "init-pulsar":
"command": "command":
@ -89,14 +88,14 @@
- "kg-extract-definitions" - "kg-extract-definitions"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"kg-extract-relationships": "kg-extract-relationships":
"command": "command":
- "kg-extract-relationships" - "kg-extract-relationships"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"milvus": "milvus":
"command": "command":
@ -134,7 +133,7 @@
- "pdf-decoder" - "pdf-decoder"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"prometheus": "prometheus":
"image": "docker.io/prom/prometheus:v2.53.1" "image": "docker.io/prom/prometheus:v2.53.1"
@ -153,7 +152,7 @@
- "non-persistent://tg/request/text-completion" - "non-persistent://tg/request/text-completion"
- "--text-completion-response-queue" - "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-response" - "non-persistent://tg/response/text-completion-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"prompt-rag": "prompt-rag":
"command": "command":
@ -168,7 +167,7 @@
- "non-persistent://tg/request/text-completion-rag" - "non-persistent://tg/request/text-completion-rag"
- "--text-completion-response-queue" - "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-rag-response" - "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"pulsar": "pulsar":
"command": "bin/pulsar standalone" "command": "bin/pulsar standalone"
@ -195,7 +194,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-t" - "-t"
- "http://milvus:19530" - "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"query-triples": "query-triples":
"command": "command":
@ -204,7 +203,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-g" - "-g"
- "cassandra" - "cassandra"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"store-graph-embeddings": "store-graph-embeddings":
"command": "command":
@ -213,7 +212,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-t" - "-t"
- "http://milvus:19530" - "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"store-triples": "store-triples":
"command": "command":
@ -222,26 +221,38 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-g" - "-g"
- "cassandra" - "cassandra"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"text-completion": "text-completion":
"command": "command":
- "text-completion-bedrock" - "text-completion-bedrock"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
# - "-m"
# - "mistral.mixtral-8x7b-instruct-v0:1"
- "-z" - "-z"
- "${AWS_ID_KEY}" - "${AWS_ID_KEY}"
- "-k" - "-k"
- "${AWS_SECRET_KEY}" - "${AWS_SECRET_KEY}"
- "-r" - "-r"
- "us-west-2" - "us-west-2"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"text-completion-rag": "text-completion-rag":
"command": "command":
- "text-completion-bedrock" - "text-completion-bedrock"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-x"
# - "0.0"
# - "-m"
# - "mistral.mixtral-8x7b-instruct-v0:1"
- "-z" - "-z"
- "${AWS_ID_KEY}" - "${AWS_ID_KEY}"
- "-k" - "-k"
@ -252,14 +263,14 @@
- "non-persistent://tg/request/text-completion-rag" - "non-persistent://tg/request/text-completion-rag"
- "-o" - "-o"
- "non-persistent://tg/response/text-completion-rag-response" - "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"vectorize": "vectorize":
"command": "command":
- "embeddings-vectorize" - "embeddings-vectorize"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"volumes": "volumes":
"cassandra": {} "cassandra": {}

267
tg-launch-claude-neo4j.yaml Normal file
View file

@ -0,0 +1,267 @@
"services":
"neo4j":
"environment":
"NEO4J_AUTH": "neo4j/password"
"image": "docker.io/neo4j:5.22.0-community-bullseye"
"ports":
- "7474:7474"
- "7687:7687"
"restart": "on-failure:100"
"volumes":
- "neo4j:/data"
"chunker":
"command":
- "chunker-recursive"
- "-p"
- "pulsar://pulsar:6650"
- "--chunk-size"
- "2000"
- "--chunk-overlap"
- "100"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"embeddings":
"command":
- "embeddings-hf"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"etcd":
"command":
- "etcd"
- "-advertise-client-urls=http://127.0.0.1:2379"
- "-listen-client-urls"
- "http://0.0.0.0:2379"
- "--data-dir"
- "/etcd"
"environment":
"ETCD_AUTO_COMPACTION_MODE": "revision"
"ETCD_AUTO_COMPACTION_RETENTION": "1000"
"ETCD_QUOTA_BACKEND_BYTES": "4294967296"
"ETCD_SNAPSHOT_COUNT": "50000"
"image": "quay.io/coreos/etcd:v3.5.5"
"ports":
- "2379:2379"
"restart": "on-failure:100"
"volumes":
- "etcd:/etcd"
"grafana":
"environment":
"GF_ORG_NAME": "trustgraph.ai"
"image": "docker.io/grafana/grafana:10.0.0"
"ports":
- "3000:3000"
"restart": "on-failure:100"
"volumes":
- "grafana-storage:/var/lib/grafana"
- "./grafana/dashboard.yml:/etc/grafana/provisioning/dashboards/dashboard.yml"
- "./grafana/datasource.yml:/etc/grafana/provisioning/datasources/datasource.yml"
- "./grafana/dashboard.json:/var/lib/grafana/dashboards/dashboard.json"
"graph-rag":
"command":
- "graph-rag"
- "-p"
- "pulsar://pulsar:6650"
- "--prompt-request-queue"
- "non-persistent://tg/request/prompt-rag"
- "--prompt-response-queue"
- "non-persistent://tg/response/prompt-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"init-pulsar":
"command":
- "sh"
- "-c"
- "pulsar-admin --admin-url http://pulsar:8080 tenants create tg && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/flow && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/request && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/response && pulsar-admin --admin-url http://pulsar:8080 namespaces set-retention --size -1 --time 3m tg/response"
"depends_on":
"pulsar":
"condition": "service_started"
"image": "docker.io/apachepulsar/pulsar:3.3.0"
"restart": "on-failure:100"
"kg-extract-definitions":
"command":
- "kg-extract-definitions"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"kg-extract-relationships":
"command":
- "kg-extract-relationships"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"milvus":
"command":
- "milvus"
- "run"
- "standalone"
"environment":
"ETCD_ENDPOINTS": "etcd:2379"
"MINIO_ADDRESS": "minio:9000"
"image": "docker.io/milvusdb/milvus:v2.4.5"
"ports":
- "9091:9091"
- "19530:19530"
"restart": "on-failure:100"
"volumes":
- "milvus:/var/lib/milvus"
"minio":
"command":
- "minio"
- "server"
- "/minio_data"
- "--console-address"
- ":9001"
"environment":
"MINIO_ROOT_PASSWORD": "minioadmin"
"MINIO_ROOT_USER": "minioadmin"
"image": "docker.io/minio/minio:RELEASE.2024-07-04T14-25-45Z"
"ports":
- "9001:9001"
"restart": "on-failure:100"
"volumes":
- "minio-data:/minio_data"
"pdf-decoder":
"command":
- "pdf-decoder"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"prometheus":
"image": "docker.io/prom/prometheus:v2.53.1"
"ports":
- "9090:9090"
"restart": "on-failure:100"
"volumes":
- "./prometheus:/etc/prometheus"
- "prometheus-data:/prometheus"
"prompt":
"command":
- "prompt-generic"
- "-p"
- "pulsar://pulsar:6650"
- "--text-completion-request-queue"
- "non-persistent://tg/request/text-completion"
- "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"prompt-rag":
"command":
- "prompt-generic"
- "-p"
- "pulsar://pulsar:6650"
- "-i"
- "non-persistent://tg/request/prompt-rag"
- "-o"
- "non-persistent://tg/response/prompt-rag-response"
- "--text-completion-request-queue"
- "non-persistent://tg/request/text-completion-rag"
- "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"pulsar":
"command": "bin/pulsar standalone"
"image": "docker.io/apachepulsar/pulsar:3.3.0"
"ports":
- "6650:6650"
- "8080:8080"
"restart": "on-failure:100"
"volumes":
- "pulsar-conf:/pulsar/conf"
- "pulsar-data:/pulsar/data"
"pulsar-manager":
"environment":
"SPRING_CONFIGURATION_FILE": "/pulsar-manager/pulsar-manager/application.properties"
"image": "docker.io/apachepulsar/pulsar-manager:v0.3.0"
"ports":
- "9527:9527"
- "7750:7750"
"restart": "on-failure:100"
"query-graph-embeddings":
"command":
- "ge-query-milvus"
- "-p"
- "pulsar://pulsar:6650"
- "-t"
- "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"query-triples":
"command":
- "triples-query-neo4j"
- "-p"
- "pulsar://pulsar:6650"
- "-g"
- "bolt://neo4j:7687"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"store-graph-embeddings":
"command":
- "ge-write-milvus"
- "-p"
- "pulsar://pulsar:6650"
- "-t"
- "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"store-triples":
"command":
- "triples-write-neo4j"
- "-p"
- "pulsar://pulsar:6650"
- "-g"
- "bolt://neo4j:7687"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"text-completion":
"command":
- "text-completion-claude"
- "-p"
- "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
- "-k"
- "${CLAUDE_KEY}"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"text-completion-rag":
"command":
- "text-completion-claude"
- "-p"
- "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
- "-k"
- "${CLAUDE_KEY}"
- "-i"
- "non-persistent://tg/request/text-completion-rag"
- "-o"
- "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"vectorize":
"command":
- "embeddings-vectorize"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"volumes":
"neo4j": {}
"etcd": {}
"grafana-storage": {}
"milvus": {}
"minio-data": {}
"prometheus-data": {}
"pulsar-conf": {}
"pulsar-data": {}

View file

@ -15,7 +15,7 @@
- "250" - "250"
- "--chunk-overlap" - "--chunk-overlap"
- "15" - "15"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"embeddings": "embeddings":
"command": "command":
@ -24,7 +24,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-m" - "-m"
- "all-MiniLM-L6-v2" - "all-MiniLM-L6-v2"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"etcd": "etcd":
"command": "command":
@ -72,7 +72,7 @@
- "30" - "30"
- "--max-subgraph-size" - "--max-subgraph-size"
- "3000" - "3000"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"init-pulsar": "init-pulsar":
"command": "command":
@ -89,14 +89,14 @@
- "kg-extract-definitions" - "kg-extract-definitions"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"kg-extract-relationships": "kg-extract-relationships":
"command": "command":
- "kg-extract-relationships" - "kg-extract-relationships"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"milvus": "milvus":
"command": "command":
@ -134,7 +134,7 @@
- "pdf-decoder" - "pdf-decoder"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"prometheus": "prometheus":
"image": "docker.io/prom/prometheus:v2.53.1" "image": "docker.io/prom/prometheus:v2.53.1"
@ -153,7 +153,7 @@
- "non-persistent://tg/request/text-completion" - "non-persistent://tg/request/text-completion"
- "--text-completion-response-queue" - "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-response" - "non-persistent://tg/response/text-completion-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"prompt-rag": "prompt-rag":
"command": "command":
@ -168,7 +168,7 @@
- "non-persistent://tg/request/text-completion-rag" - "non-persistent://tg/request/text-completion-rag"
- "--text-completion-response-queue" - "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-rag-response" - "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"pulsar": "pulsar":
"command": "bin/pulsar standalone" "command": "bin/pulsar standalone"
@ -195,7 +195,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-t" - "-t"
- "http://milvus:19530" - "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"query-triples": "query-triples":
"command": "command":
@ -204,7 +204,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-g" - "-g"
- "cassandra" - "cassandra"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"store-graph-embeddings": "store-graph-embeddings":
"command": "command":
@ -213,7 +213,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-t" - "-t"
- "http://milvus:19530" - "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"store-triples": "store-triples":
"command": "command":
@ -222,36 +222,44 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-g" - "-g"
- "cassandra" - "cassandra"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"text-completion": "text-completion":
"command": "command":
- "text-completion-claude" - "text-completion-claude"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
- "-k" - "-k"
- "${CLAUDE_KEY}" - "${CLAUDE_KEY}"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"text-completion-rag": "text-completion-rag":
"command": "command":
- "text-completion-claude" - "text-completion-claude"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
- "-k" - "-k"
- "${CLAUDE_KEY}" - "${CLAUDE_KEY}"
- "-i" - "-i"
- "non-persistent://tg/request/text-completion-rag" - "non-persistent://tg/request/text-completion-rag"
- "-o" - "-o"
- "non-persistent://tg/response/text-completion-rag-response" - "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"vectorize": "vectorize":
"command": "command":
- "embeddings-vectorize" - "embeddings-vectorize"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"volumes": "volumes":
"cassandra": {} "cassandra": {}

267
tg-launch-cohere-neo4j.yaml Normal file
View file

@ -0,0 +1,267 @@
"services":
"neo4j":
"environment":
"NEO4J_AUTH": "neo4j/password"
"image": "docker.io/neo4j:5.22.0-community-bullseye"
"ports":
- "7474:7474"
- "7687:7687"
"restart": "on-failure:100"
"volumes":
- "neo4j:/data"
"chunker":
"command":
- "chunker-recursive"
- "-p"
- "pulsar://pulsar:6650"
- "--chunk-size"
- "1000"
- "--chunk-overlap"
- "50"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"embeddings":
"command":
- "embeddings-hf"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"etcd":
"command":
- "etcd"
- "-advertise-client-urls=http://127.0.0.1:2379"
- "-listen-client-urls"
- "http://0.0.0.0:2379"
- "--data-dir"
- "/etcd"
"environment":
"ETCD_AUTO_COMPACTION_MODE": "revision"
"ETCD_AUTO_COMPACTION_RETENTION": "1000"
"ETCD_QUOTA_BACKEND_BYTES": "4294967296"
"ETCD_SNAPSHOT_COUNT": "50000"
"image": "quay.io/coreos/etcd:v3.5.5"
"ports":
- "2379:2379"
"restart": "on-failure:100"
"volumes":
- "etcd:/etcd"
"grafana":
"environment":
"GF_ORG_NAME": "trustgraph.ai"
"image": "docker.io/grafana/grafana:10.0.0"
"ports":
- "3000:3000"
"restart": "on-failure:100"
"volumes":
- "grafana-storage:/var/lib/grafana"
- "./grafana/dashboard.yml:/etc/grafana/provisioning/dashboards/dashboard.yml"
- "./grafana/datasource.yml:/etc/grafana/provisioning/datasources/datasource.yml"
- "./grafana/dashboard.json:/var/lib/grafana/dashboards/dashboard.json"
"graph-rag":
"command":
- "graph-rag"
- "-p"
- "pulsar://pulsar:6650"
- "--prompt-request-queue"
- "non-persistent://tg/request/prompt-rag"
- "--prompt-response-queue"
- "non-persistent://tg/response/prompt-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"init-pulsar":
"command":
- "sh"
- "-c"
- "pulsar-admin --admin-url http://pulsar:8080 tenants create tg && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/flow && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/request && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/response && pulsar-admin --admin-url http://pulsar:8080 namespaces set-retention --size -1 --time 3m tg/response"
"depends_on":
"pulsar":
"condition": "service_started"
"image": "docker.io/apachepulsar/pulsar:3.3.0"
"restart": "on-failure:100"
"kg-extract-definitions":
"command":
- "kg-extract-definitions"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"kg-extract-relationships":
"command":
- "kg-extract-relationships"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"milvus":
"command":
- "milvus"
- "run"
- "standalone"
"environment":
"ETCD_ENDPOINTS": "etcd:2379"
"MINIO_ADDRESS": "minio:9000"
"image": "docker.io/milvusdb/milvus:v2.4.5"
"ports":
- "9091:9091"
- "19530:19530"
"restart": "on-failure:100"
"volumes":
- "milvus:/var/lib/milvus"
"minio":
"command":
- "minio"
- "server"
- "/minio_data"
- "--console-address"
- ":9001"
"environment":
"MINIO_ROOT_PASSWORD": "minioadmin"
"MINIO_ROOT_USER": "minioadmin"
"image": "docker.io/minio/minio:RELEASE.2024-07-04T14-25-45Z"
"ports":
- "9001:9001"
"restart": "on-failure:100"
"volumes":
- "minio-data:/minio_data"
"pdf-decoder":
"command":
- "pdf-decoder"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"prometheus":
"image": "docker.io/prom/prometheus:v2.53.1"
"ports":
- "9090:9090"
"restart": "on-failure:100"
"volumes":
- "./prometheus:/etc/prometheus"
- "prometheus-data:/prometheus"
"prompt":
"command":
- "prompt-generic"
- "-p"
- "pulsar://pulsar:6650"
- "--text-completion-request-queue"
- "non-persistent://tg/request/text-completion"
- "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"prompt-rag":
"command":
- "prompt-generic"
- "-p"
- "pulsar://pulsar:6650"
- "-i"
- "non-persistent://tg/request/prompt-rag"
- "-o"
- "non-persistent://tg/response/prompt-rag-response"
- "--text-completion-request-queue"
- "non-persistent://tg/request/text-completion-rag"
- "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"pulsar":
"command": "bin/pulsar standalone"
"image": "docker.io/apachepulsar/pulsar:3.3.0"
"ports":
- "6650:6650"
- "8080:8080"
"restart": "on-failure:100"
"volumes":
- "pulsar-conf:/pulsar/conf"
- "pulsar-data:/pulsar/data"
"pulsar-manager":
"environment":
"SPRING_CONFIGURATION_FILE": "/pulsar-manager/pulsar-manager/application.properties"
"image": "docker.io/apachepulsar/pulsar-manager:v0.3.0"
"ports":
- "9527:9527"
- "7750:7750"
"restart": "on-failure:100"
"query-graph-embeddings":
"command":
- "ge-query-milvus"
- "-p"
- "pulsar://pulsar:6650"
- "-t"
- "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"query-triples":
"command":
- "triples-query-neo4j"
- "-p"
- "pulsar://pulsar:6650"
- "-g"
- "bolt://neo4j:7687"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"store-graph-embeddings":
"command":
- "ge-write-milvus"
- "-p"
- "pulsar://pulsar:6650"
- "-t"
- "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"store-triples":
"command":
- "triples-write-neo4j"
- "-p"
- "pulsar://pulsar:6650"
- "-g"
- "bolt://neo4j:7687"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"text-completion":
"command":
- "text-completion-cohere"
- "-p"
- "pulsar://pulsar:6650"
# - "-l"
# - "4096"
# - "-t"
# - "0.0"
- "-k"
- "${COHERE_KEY}"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"text-completion-rag":
"command":
- "text-completion-cohere"
- "-p"
- "pulsar://pulsar:6650"
# - "-l"
# - "4096"
# - "-t"
# - "0.0"
- "-k"
- "${COHERE_KEY}"
- "-i"
- "non-persistent://tg/request/text-completion-rag"
- "-o"
- "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"vectorize":
"command":
- "embeddings-vectorize"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"volumes":
"neo4j": {}
"etcd": {}
"grafana-storage": {}
"milvus": {}
"minio-data": {}
"prometheus-data": {}
"pulsar-conf": {}
"pulsar-data": {}

View file

@ -15,7 +15,7 @@
- "1000" - "1000"
- "--chunk-overlap" - "--chunk-overlap"
- "50" - "50"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"embeddings": "embeddings":
"command": "command":
@ -24,7 +24,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-m" - "-m"
- "all-MiniLM-L6-v2" - "all-MiniLM-L6-v2"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"etcd": "etcd":
"command": "command":
@ -72,7 +72,7 @@
- "30" - "30"
- "--max-subgraph-size" - "--max-subgraph-size"
- "3000" - "3000"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"init-pulsar": "init-pulsar":
"command": "command":
@ -89,14 +89,14 @@
- "kg-extract-definitions" - "kg-extract-definitions"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"kg-extract-relationships": "kg-extract-relationships":
"command": "command":
- "kg-extract-relationships" - "kg-extract-relationships"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"milvus": "milvus":
"command": "command":
@ -134,7 +134,7 @@
- "pdf-decoder" - "pdf-decoder"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"prometheus": "prometheus":
"image": "docker.io/prom/prometheus:v2.53.1" "image": "docker.io/prom/prometheus:v2.53.1"
@ -153,7 +153,7 @@
- "non-persistent://tg/request/text-completion" - "non-persistent://tg/request/text-completion"
- "--text-completion-response-queue" - "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-response" - "non-persistent://tg/response/text-completion-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"prompt-rag": "prompt-rag":
"command": "command":
@ -168,7 +168,7 @@
- "non-persistent://tg/request/text-completion-rag" - "non-persistent://tg/request/text-completion-rag"
- "--text-completion-response-queue" - "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-rag-response" - "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"pulsar": "pulsar":
"command": "bin/pulsar standalone" "command": "bin/pulsar standalone"
@ -195,7 +195,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-t" - "-t"
- "http://milvus:19530" - "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"query-triples": "query-triples":
"command": "command":
@ -204,7 +204,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-g" - "-g"
- "cassandra" - "cassandra"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"store-graph-embeddings": "store-graph-embeddings":
"command": "command":
@ -213,7 +213,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-t" - "-t"
- "http://milvus:19530" - "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"store-triples": "store-triples":
"command": "command":
@ -222,36 +222,40 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-g" - "-g"
- "cassandra" - "cassandra"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"text-completion": "text-completion":
"command": "command":
- "text-completion-cohere" - "text-completion-cohere"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
# - "-t"
# - "0.0"
- "-k" - "-k"
- "${COHERE_KEY}" - "${COHERE_KEY}"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"text-completion-rag": "text-completion-rag":
"command": "command":
- "text-completion-cohere" - "text-completion-cohere"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
# - "-t"
# - "0.0"
- "-k" - "-k"
- "${COHERE_KEY}" - "${COHERE_KEY}"
- "-i" - "-i"
- "non-persistent://tg/request/text-completion-rag" - "non-persistent://tg/request/text-completion-rag"
- "-o" - "-o"
- "non-persistent://tg/response/text-completion-rag-response" - "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"vectorize": "vectorize":
"command": "command":
- "embeddings-vectorize" - "embeddings-vectorize"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"volumes": "volumes":
"cassandra": {} "cassandra": {}

271
tg-launch-mix-neo4j.yaml Normal file
View file

@ -0,0 +1,271 @@
"services":
"neo4j":
"environment":
"NEO4J_AUTH": "neo4j/password"
"image": "docker.io/neo4j:5.22.0-community-bullseye"
"ports":
- "7474:7474"
- "7687:7687"
"restart": "on-failure:100"
"volumes":
- "neo4j:/data"
"chunker":
"command":
- "chunker-recursive"
- "-p"
- "pulsar://pulsar:6650"
- "--chunk-size"
- "2000"
- "--chunk-overlap"
- "100"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"embeddings":
"command":
- "embeddings-hf"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"etcd":
"command":
- "etcd"
- "-advertise-client-urls=http://127.0.0.1:2379"
- "-listen-client-urls"
- "http://0.0.0.0:2379"
- "--data-dir"
- "/etcd"
"environment":
"ETCD_AUTO_COMPACTION_MODE": "revision"
"ETCD_AUTO_COMPACTION_RETENTION": "1000"
"ETCD_QUOTA_BACKEND_BYTES": "4294967296"
"ETCD_SNAPSHOT_COUNT": "50000"
"image": "quay.io/coreos/etcd:v3.5.5"
"ports":
- "2379:2379"
"restart": "on-failure:100"
"volumes":
- "etcd:/etcd"
"grafana":
"environment":
"GF_ORG_NAME": "trustgraph.ai"
"image": "docker.io/grafana/grafana:10.0.0"
"ports":
- "3000:3000"
"restart": "on-failure:100"
"volumes":
- "grafana-storage:/var/lib/grafana"
- "./grafana/dashboard.yml:/etc/grafana/provisioning/dashboards/dashboard.yml"
- "./grafana/datasource.yml:/etc/grafana/provisioning/datasources/datasource.yml"
- "./grafana/dashboard.json:/var/lib/grafana/dashboards/dashboard.json"
"graph-rag":
"command":
- "graph-rag"
- "-p"
- "pulsar://pulsar:6650"
- "--prompt-request-queue"
- "non-persistent://tg/request/prompt-rag"
- "--prompt-response-queue"
- "non-persistent://tg/response/prompt-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"init-pulsar":
"command":
- "sh"
- "-c"
- "pulsar-admin --admin-url http://pulsar:8080 tenants create tg && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/flow && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/request && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/response && pulsar-admin --admin-url http://pulsar:8080 namespaces set-retention --size -1 --time 3m tg/response"
"depends_on":
"pulsar":
"condition": "service_started"
"image": "docker.io/apachepulsar/pulsar:3.3.0"
"restart": "on-failure:100"
"kg-extract-definitions":
"command":
- "kg-extract-definitions"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"kg-extract-relationships":
"command":
- "kg-extract-relationships"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"milvus":
"command":
- "milvus"
- "run"
- "standalone"
"environment":
"ETCD_ENDPOINTS": "etcd:2379"
"MINIO_ADDRESS": "minio:9000"
"image": "docker.io/milvusdb/milvus:v2.4.5"
"ports":
- "9091:9091"
- "19530:19530"
"restart": "on-failure:100"
"volumes":
- "milvus:/var/lib/milvus"
"minio":
"command":
- "minio"
- "server"
- "/minio_data"
- "--console-address"
- ":9001"
"environment":
"MINIO_ROOT_PASSWORD": "minioadmin"
"MINIO_ROOT_USER": "minioadmin"
"image": "docker.io/minio/minio:RELEASE.2024-07-04T14-25-45Z"
"ports":
- "9001:9001"
"restart": "on-failure:100"
"volumes":
- "minio-data:/minio_data"
"pdf-decoder":
"command":
- "pdf-decoder"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"prometheus":
"image": "docker.io/prom/prometheus:v2.53.1"
"ports":
- "9090:9090"
"restart": "on-failure:100"
"volumes":
- "./prometheus:/etc/prometheus"
- "prometheus-data:/prometheus"
"prompt":
"command":
- "prompt-generic"
- "-p"
- "pulsar://pulsar:6650"
- "--text-completion-request-queue"
- "non-persistent://tg/request/text-completion"
- "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"prompt-rag":
"command":
- "prompt-generic"
- "-p"
- "pulsar://pulsar:6650"
- "-i"
- "non-persistent://tg/request/prompt-rag"
- "-o"
- "non-persistent://tg/response/prompt-rag-response"
- "--text-completion-request-queue"
- "non-persistent://tg/request/text-completion-rag"
- "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"pulsar":
"command": "bin/pulsar standalone"
"image": "docker.io/apachepulsar/pulsar:3.3.0"
"ports":
- "6650:6650"
- "8080:8080"
"restart": "on-failure:100"
"volumes":
- "pulsar-conf:/pulsar/conf"
- "pulsar-data:/pulsar/data"
"pulsar-manager":
"environment":
"SPRING_CONFIGURATION_FILE": "/pulsar-manager/pulsar-manager/application.properties"
"image": "docker.io/apachepulsar/pulsar-manager:v0.3.0"
"ports":
- "9527:9527"
- "7750:7750"
"restart": "on-failure:100"
"query-graph-embeddings":
"command":
- "ge-query-milvus"
- "-p"
- "pulsar://pulsar:6650"
- "-t"
- "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"query-triples":
"command":
- "triples-query-neo4j"
- "-p"
- "pulsar://pulsar:6650"
- "-g"
- "bolt://neo4j:7687"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"store-graph-embeddings":
"command":
- "ge-write-milvus"
- "-p"
- "pulsar://pulsar:6650"
- "-t"
- "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"store-triples":
"command":
- "triples-write-neo4j"
- "-p"
- "pulsar://pulsar:6650"
- "-g"
- "bolt://neo4j:7687"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"text-completion":
"command":
- "text-completion-cohere"
- "-p"
- "pulsar://pulsar:6650"
# - "-l"
# - "4096"
# - "-t"
# - "0.0"
- "-k"
- "${COHERE_KEY}"
- "-m"
- "c4ai-aya-23-35b"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"text-completion-rag":
"command":
- "text-completion-cohere"
- "-p"
- "pulsar://pulsar:6650"
# - "-l"
# - "4096"
# - "-t"
# - "0.0"
- "-k"
- "${COHERE_KEY}"
- "-i"
- "non-persistent://tg/request/text-completion-rag"
- "-o"
- "non-persistent://tg/response/text-completion-rag-response"
- "-m"
- "c4ai-aya-23-8b"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"vectorize":
"command":
- "embeddings-vectorize"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"volumes":
"neo4j": {}
"etcd": {}
"grafana-storage": {}
"milvus": {}
"minio-data": {}
"prometheus-data": {}
"pulsar-conf": {}
"pulsar-data": {}

View file

@ -12,10 +12,10 @@
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "--chunk-size" - "--chunk-size"
- "4000" - "2000"
- "--chunk-overlap" - "--chunk-overlap"
- "120" - "100"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"embeddings": "embeddings":
"command": "command":
@ -24,7 +24,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-m" - "-m"
- "all-MiniLM-L6-v2" - "all-MiniLM-L6-v2"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"etcd": "etcd":
"command": "command":
@ -72,7 +72,7 @@
- "30" - "30"
- "--max-subgraph-size" - "--max-subgraph-size"
- "3000" - "3000"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"init-pulsar": "init-pulsar":
"command": "command":
@ -89,14 +89,14 @@
- "kg-extract-definitions" - "kg-extract-definitions"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"kg-extract-relationships": "kg-extract-relationships":
"command": "command":
- "kg-extract-relationships" - "kg-extract-relationships"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"milvus": "milvus":
"command": "command":
@ -134,7 +134,7 @@
- "pdf-decoder" - "pdf-decoder"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"prometheus": "prometheus":
"image": "docker.io/prom/prometheus:v2.53.1" "image": "docker.io/prom/prometheus:v2.53.1"
@ -153,7 +153,7 @@
- "non-persistent://tg/request/text-completion" - "non-persistent://tg/request/text-completion"
- "--text-completion-response-queue" - "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-response" - "non-persistent://tg/response/text-completion-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"prompt-rag": "prompt-rag":
"command": "command":
@ -168,7 +168,7 @@
- "non-persistent://tg/request/text-completion-rag" - "non-persistent://tg/request/text-completion-rag"
- "--text-completion-response-queue" - "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-rag-response" - "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"pulsar": "pulsar":
"command": "bin/pulsar standalone" "command": "bin/pulsar standalone"
@ -195,7 +195,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-t" - "-t"
- "http://milvus:19530" - "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"query-triples": "query-triples":
"command": "command":
@ -204,7 +204,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-g" - "-g"
- "cassandra" - "cassandra"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"store-graph-embeddings": "store-graph-embeddings":
"command": "command":
@ -213,7 +213,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-t" - "-t"
- "http://milvus:19530" - "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"store-triples": "store-triples":
"command": "command":
@ -222,24 +222,32 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-g" - "-g"
- "cassandra" - "cassandra"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"text-completion": "text-completion":
"command": "command":
- "text-completion-cohere" - "text-completion-cohere"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
- "-k" - "-k"
- "${COHERE_KEY}" - "${COHERE_KEY}"
- "-m" - "-m"
- "c4ai-aya-23-35b" - "c4ai-aya-23-35b"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"text-completion-rag": "text-completion-rag":
"command": "command":
- "text-completion-cohere" - "text-completion-cohere"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
- "-k" - "-k"
- "${COHERE_KEY}" - "${COHERE_KEY}"
- "-i" - "-i"
@ -248,14 +256,14 @@
- "non-persistent://tg/response/text-completion-rag-response" - "non-persistent://tg/response/text-completion-rag-response"
- "-m" - "-m"
- "c4ai-aya-23-8b" - "c4ai-aya-23-8b"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"vectorize": "vectorize":
"command": "command":
- "embeddings-vectorize" - "embeddings-vectorize"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"volumes": "volumes":
"cassandra": {} "cassandra": {}

259
tg-launch-ollama-neo4j.yaml Normal file
View file

@ -0,0 +1,259 @@
"services":
"neo4j":
"environment":
"NEO4J_AUTH": "neo4j/password"
"image": "docker.io/neo4j:5.22.0-community-bullseye"
"ports":
- "7474:7474"
- "7687:7687"
"restart": "on-failure:100"
"volumes":
- "neo4j:/data"
"chunker":
"command":
- "chunker-recursive"
- "-p"
- "pulsar://pulsar:6650"
- "--chunk-size"
- "2000"
- "--chunk-overlap"
- "100"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"embeddings":
"command":
- "embeddings-hf"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"etcd":
"command":
- "etcd"
- "-advertise-client-urls=http://127.0.0.1:2379"
- "-listen-client-urls"
- "http://0.0.0.0:2379"
- "--data-dir"
- "/etcd"
"environment":
"ETCD_AUTO_COMPACTION_MODE": "revision"
"ETCD_AUTO_COMPACTION_RETENTION": "1000"
"ETCD_QUOTA_BACKEND_BYTES": "4294967296"
"ETCD_SNAPSHOT_COUNT": "50000"
"image": "quay.io/coreos/etcd:v3.5.5"
"ports":
- "2379:2379"
"restart": "on-failure:100"
"volumes":
- "etcd:/etcd"
"grafana":
"environment":
"GF_ORG_NAME": "trustgraph.ai"
"image": "docker.io/grafana/grafana:10.0.0"
"ports":
- "3000:3000"
"restart": "on-failure:100"
"volumes":
- "grafana-storage:/var/lib/grafana"
- "./grafana/dashboard.yml:/etc/grafana/provisioning/dashboards/dashboard.yml"
- "./grafana/datasource.yml:/etc/grafana/provisioning/datasources/datasource.yml"
- "./grafana/dashboard.json:/var/lib/grafana/dashboards/dashboard.json"
"graph-rag":
"command":
- "graph-rag"
- "-p"
- "pulsar://pulsar:6650"
- "--prompt-request-queue"
- "non-persistent://tg/request/prompt-rag"
- "--prompt-response-queue"
- "non-persistent://tg/response/prompt-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"init-pulsar":
"command":
- "sh"
- "-c"
- "pulsar-admin --admin-url http://pulsar:8080 tenants create tg && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/flow && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/request && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/response && pulsar-admin --admin-url http://pulsar:8080 namespaces set-retention --size -1 --time 3m tg/response"
"depends_on":
"pulsar":
"condition": "service_started"
"image": "docker.io/apachepulsar/pulsar:3.3.0"
"restart": "on-failure:100"
"kg-extract-definitions":
"command":
- "kg-extract-definitions"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"kg-extract-relationships":
"command":
- "kg-extract-relationships"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"milvus":
"command":
- "milvus"
- "run"
- "standalone"
"environment":
"ETCD_ENDPOINTS": "etcd:2379"
"MINIO_ADDRESS": "minio:9000"
"image": "docker.io/milvusdb/milvus:v2.4.5"
"ports":
- "9091:9091"
- "19530:19530"
"restart": "on-failure:100"
"volumes":
- "milvus:/var/lib/milvus"
"minio":
"command":
- "minio"
- "server"
- "/minio_data"
- "--console-address"
- ":9001"
"environment":
"MINIO_ROOT_PASSWORD": "minioadmin"
"MINIO_ROOT_USER": "minioadmin"
"image": "docker.io/minio/minio:RELEASE.2024-07-04T14-25-45Z"
"ports":
- "9001:9001"
"restart": "on-failure:100"
"volumes":
- "minio-data:/minio_data"
"pdf-decoder":
"command":
- "pdf-decoder"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"prometheus":
"image": "docker.io/prom/prometheus:v2.53.1"
"ports":
- "9090:9090"
"restart": "on-failure:100"
"volumes":
- "./prometheus:/etc/prometheus"
- "prometheus-data:/prometheus"
"prompt":
"command":
- "prompt-generic"
- "-p"
- "pulsar://pulsar:6650"
- "--text-completion-request-queue"
- "non-persistent://tg/request/text-completion"
- "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"prompt-rag":
"command":
- "prompt-generic"
- "-p"
- "pulsar://pulsar:6650"
- "-i"
- "non-persistent://tg/request/prompt-rag"
- "-o"
- "non-persistent://tg/response/prompt-rag-response"
- "--text-completion-request-queue"
- "non-persistent://tg/request/text-completion-rag"
- "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"pulsar":
"command": "bin/pulsar standalone"
"image": "docker.io/apachepulsar/pulsar:3.3.0"
"ports":
- "6650:6650"
- "8080:8080"
"restart": "on-failure:100"
"volumes":
- "pulsar-conf:/pulsar/conf"
- "pulsar-data:/pulsar/data"
"pulsar-manager":
"environment":
"SPRING_CONFIGURATION_FILE": "/pulsar-manager/pulsar-manager/application.properties"
"image": "docker.io/apachepulsar/pulsar-manager:v0.3.0"
"ports":
- "9527:9527"
- "7750:7750"
"restart": "on-failure:100"
"query-graph-embeddings":
"command":
- "ge-query-milvus"
- "-p"
- "pulsar://pulsar:6650"
- "-t"
- "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"query-triples":
"command":
- "triples-query-neo4j"
- "-p"
- "pulsar://pulsar:6650"
- "-g"
- "bolt://neo4j:7687"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"store-graph-embeddings":
"command":
- "ge-write-milvus"
- "-p"
- "pulsar://pulsar:6650"
- "-t"
- "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"store-triples":
"command":
- "triples-write-neo4j"
- "-p"
- "pulsar://pulsar:6650"
- "-g"
- "bolt://neo4j:7687"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"text-completion":
"command":
- "text-completion-ollama"
- "-p"
- "pulsar://pulsar:6650"
- "-r"
- "${OLLAMA_HOST}"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"text-completion-rag":
"command":
- "text-completion-ollama"
- "-p"
- "pulsar://pulsar:6650"
- "-r"
- "${OLLAMA_HOST}"
- "-i"
- "non-persistent://tg/request/text-completion-rag"
- "-o"
- "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"vectorize":
"command":
- "embeddings-vectorize"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"volumes":
"neo4j": {}
"etcd": {}
"grafana-storage": {}
"milvus": {}
"minio-data": {}
"prometheus-data": {}
"pulsar-conf": {}
"pulsar-data": {}

View file

@ -15,7 +15,7 @@
- "250" - "250"
- "--chunk-overlap" - "--chunk-overlap"
- "15" - "15"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"embeddings": "embeddings":
"command": "command":
@ -24,7 +24,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-m" - "-m"
- "all-MiniLM-L6-v2" - "all-MiniLM-L6-v2"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"etcd": "etcd":
"command": "command":
@ -72,7 +72,7 @@
- "30" - "30"
- "--max-subgraph-size" - "--max-subgraph-size"
- "3000" - "3000"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"init-pulsar": "init-pulsar":
"command": "command":
@ -89,14 +89,14 @@
- "kg-extract-definitions" - "kg-extract-definitions"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"kg-extract-relationships": "kg-extract-relationships":
"command": "command":
- "kg-extract-relationships" - "kg-extract-relationships"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"milvus": "milvus":
"command": "command":
@ -134,7 +134,7 @@
- "pdf-decoder" - "pdf-decoder"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"prometheus": "prometheus":
"image": "docker.io/prom/prometheus:v2.53.1" "image": "docker.io/prom/prometheus:v2.53.1"
@ -153,7 +153,7 @@
- "non-persistent://tg/request/text-completion" - "non-persistent://tg/request/text-completion"
- "--text-completion-response-queue" - "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-response" - "non-persistent://tg/response/text-completion-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"prompt-rag": "prompt-rag":
"command": "command":
@ -168,7 +168,7 @@
- "non-persistent://tg/request/text-completion-rag" - "non-persistent://tg/request/text-completion-rag"
- "--text-completion-response-queue" - "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-rag-response" - "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"pulsar": "pulsar":
"command": "bin/pulsar standalone" "command": "bin/pulsar standalone"
@ -195,7 +195,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-t" - "-t"
- "http://milvus:19530" - "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"query-triples": "query-triples":
"command": "command":
@ -204,7 +204,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-g" - "-g"
- "cassandra" - "cassandra"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"store-graph-embeddings": "store-graph-embeddings":
"command": "command":
@ -213,7 +213,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-t" - "-t"
- "http://milvus:19530" - "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"store-triples": "store-triples":
"command": "command":
@ -222,7 +222,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-g" - "-g"
- "cassandra" - "cassandra"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"text-completion": "text-completion":
"command": "command":
@ -231,7 +231,9 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-r" - "-r"
- "${OLLAMA_HOST}" - "${OLLAMA_HOST}"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" # - "-m"
# - "phi3.5:latest"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"text-completion-rag": "text-completion-rag":
"command": "command":
@ -240,18 +242,20 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-r" - "-r"
- "${OLLAMA_HOST}" - "${OLLAMA_HOST}"
# - "-m"
# - "phi3.5:latest"
- "-i" - "-i"
- "non-persistent://tg/request/text-completion-rag" - "non-persistent://tg/request/text-completion-rag"
- "-o" - "-o"
- "non-persistent://tg/response/text-completion-rag-response" - "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"vectorize": "vectorize":
"command": "command":
- "embeddings-vectorize" - "embeddings-vectorize"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"volumes": "volumes":
"cassandra": {} "cassandra": {}

View file

@ -8,7 +8,7 @@
- "250" - "250"
- "--chunk-overlap" - "--chunk-overlap"
- "15" - "15"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"embeddings": "embeddings":
"command": "command":
@ -17,7 +17,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-m" - "-m"
- "all-MiniLM-L6-v2" - "all-MiniLM-L6-v2"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"etcd": "etcd":
"command": "command":
@ -65,7 +65,7 @@
- "30" - "30"
- "--max-subgraph-size" - "--max-subgraph-size"
- "3000" - "3000"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"init-pulsar": "init-pulsar":
"command": "command":
@ -82,14 +82,14 @@
- "kg-extract-definitions" - "kg-extract-definitions"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"kg-extract-relationships": "kg-extract-relationships":
"command": "command":
- "kg-extract-relationships" - "kg-extract-relationships"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"milvus": "milvus":
"command": "command":
@ -137,7 +137,7 @@
- "pdf-decoder" - "pdf-decoder"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"prometheus": "prometheus":
"image": "docker.io/prom/prometheus:v2.53.1" "image": "docker.io/prom/prometheus:v2.53.1"
@ -156,7 +156,7 @@
- "non-persistent://tg/request/text-completion" - "non-persistent://tg/request/text-completion"
- "--text-completion-response-queue" - "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-response" - "non-persistent://tg/response/text-completion-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"prompt-rag": "prompt-rag":
"command": "command":
@ -171,7 +171,7 @@
- "non-persistent://tg/request/text-completion-rag" - "non-persistent://tg/request/text-completion-rag"
- "--text-completion-response-queue" - "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-rag-response" - "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"pulsar": "pulsar":
"command": "bin/pulsar standalone" "command": "bin/pulsar standalone"
@ -198,7 +198,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-t" - "-t"
- "http://milvus:19530" - "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"query-triples": "query-triples":
"command": "command":
@ -207,7 +207,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-g" - "-g"
- "bolt://neo4j:7687" - "bolt://neo4j:7687"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"store-graph-embeddings": "store-graph-embeddings":
"command": "command":
@ -216,7 +216,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-t" - "-t"
- "http://milvus:19530" - "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"store-triples": "store-triples":
"command": "command":
@ -225,36 +225,44 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-g" - "-g"
- "bolt://neo4j:7687" - "bolt://neo4j:7687"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"text-completion": "text-completion":
"command": "command":
- "text-completion-openai" - "text-completion-openai"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
- "-k" - "-k"
- "${OPENAI_KEY}" - "${OPENAI_KEY}"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"text-completion-rag": "text-completion-rag":
"command": "command":
- "text-completion-openai" - "text-completion-openai"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
- "-k" - "-k"
- "${OPENAI_KEY}" - "${OPENAI_KEY}"
- "-i" - "-i"
- "non-persistent://tg/request/text-completion-rag" - "non-persistent://tg/request/text-completion-rag"
- "-o" - "-o"
- "non-persistent://tg/response/text-completion-rag-response" - "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"vectorize": "vectorize":
"command": "command":
- "embeddings-vectorize" - "embeddings-vectorize"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"volumes": "volumes":
"etcd": {} "etcd": {}

View file

@ -15,7 +15,7 @@
- "250" - "250"
- "--chunk-overlap" - "--chunk-overlap"
- "15" - "15"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"embeddings": "embeddings":
"command": "command":
@ -24,7 +24,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-m" - "-m"
- "all-MiniLM-L6-v2" - "all-MiniLM-L6-v2"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"etcd": "etcd":
"command": "command":
@ -72,7 +72,7 @@
- "30" - "30"
- "--max-subgraph-size" - "--max-subgraph-size"
- "3000" - "3000"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"init-pulsar": "init-pulsar":
"command": "command":
@ -89,14 +89,14 @@
- "kg-extract-definitions" - "kg-extract-definitions"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"kg-extract-relationships": "kg-extract-relationships":
"command": "command":
- "kg-extract-relationships" - "kg-extract-relationships"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"milvus": "milvus":
"command": "command":
@ -134,7 +134,7 @@
- "pdf-decoder" - "pdf-decoder"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"prometheus": "prometheus":
"image": "docker.io/prom/prometheus:v2.53.1" "image": "docker.io/prom/prometheus:v2.53.1"
@ -153,7 +153,7 @@
- "non-persistent://tg/request/text-completion" - "non-persistent://tg/request/text-completion"
- "--text-completion-response-queue" - "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-response" - "non-persistent://tg/response/text-completion-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"prompt-rag": "prompt-rag":
"command": "command":
@ -168,7 +168,7 @@
- "non-persistent://tg/request/text-completion-rag" - "non-persistent://tg/request/text-completion-rag"
- "--text-completion-response-queue" - "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-rag-response" - "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"pulsar": "pulsar":
"command": "bin/pulsar standalone" "command": "bin/pulsar standalone"
@ -195,7 +195,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-t" - "-t"
- "http://milvus:19530" - "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"query-triples": "query-triples":
"command": "command":
@ -204,7 +204,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-g" - "-g"
- "cassandra" - "cassandra"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"store-graph-embeddings": "store-graph-embeddings":
"command": "command":
@ -213,7 +213,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-t" - "-t"
- "http://milvus:19530" - "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"store-triples": "store-triples":
"command": "command":
@ -222,36 +222,44 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-g" - "-g"
- "cassandra" - "cassandra"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"text-completion": "text-completion":
"command": "command":
- "text-completion-openai" - "text-completion-openai"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
- "-k" - "-k"
- "${OPENAI_KEY}" - "${OPENAI_KEY}"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"text-completion-rag": "text-completion-rag":
"command": "command":
- "text-completion-openai" - "text-completion-openai"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
- "-k" - "-k"
- "${OPENAI_KEY}" - "${OPENAI_KEY}"
- "-i" - "-i"
- "non-persistent://tg/request/text-completion-rag" - "non-persistent://tg/request/text-completion-rag"
- "-o" - "-o"
- "non-persistent://tg/response/text-completion-rag-response" - "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"vectorize": "vectorize":
"command": "command":
- "embeddings-vectorize" - "embeddings-vectorize"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"volumes": "volumes":
"cassandra": {} "cassandra": {}

View file

@ -0,0 +1,275 @@
"services":
"neo4j":
"environment":
"NEO4J_AUTH": "neo4j/password"
"image": "docker.io/neo4j:5.22.0-community-bullseye"
"ports":
- "7474:7474"
- "7687:7687"
"restart": "on-failure:100"
"volumes":
- "neo4j:/data"
"chunker":
"command":
- "chunker-recursive"
- "-p"
- "pulsar://pulsar:6650"
- "--chunk-size"
- "2000"
- "--chunk-overlap"
- "100"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"embeddings":
"command":
- "embeddings-hf"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"etcd":
"command":
- "etcd"
- "-advertise-client-urls=http://127.0.0.1:2379"
- "-listen-client-urls"
- "http://0.0.0.0:2379"
- "--data-dir"
- "/etcd"
"environment":
"ETCD_AUTO_COMPACTION_MODE": "revision"
"ETCD_AUTO_COMPACTION_RETENTION": "1000"
"ETCD_QUOTA_BACKEND_BYTES": "4294967296"
"ETCD_SNAPSHOT_COUNT": "50000"
"image": "quay.io/coreos/etcd:v3.5.5"
"ports":
- "2379:2379"
"restart": "on-failure:100"
"volumes":
- "etcd:/etcd"
"grafana":
"environment":
"GF_ORG_NAME": "trustgraph.ai"
"image": "docker.io/grafana/grafana:10.0.0"
"ports":
- "3000:3000"
"restart": "on-failure:100"
"volumes":
- "grafana-storage:/var/lib/grafana"
- "./grafana/dashboard.yml:/etc/grafana/provisioning/dashboards/dashboard.yml"
- "./grafana/datasource.yml:/etc/grafana/provisioning/datasources/datasource.yml"
- "./grafana/dashboard.json:/var/lib/grafana/dashboards/dashboard.json"
"graph-rag":
"command":
- "graph-rag"
- "-p"
- "pulsar://pulsar:6650"
- "--prompt-request-queue"
- "non-persistent://tg/request/prompt-rag"
- "--prompt-response-queue"
- "non-persistent://tg/response/prompt-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"init-pulsar":
"command":
- "sh"
- "-c"
- "pulsar-admin --admin-url http://pulsar:8080 tenants create tg && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/flow && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/request && pulsar-admin --admin-url http://pulsar:8080 namespaces create tg/response && pulsar-admin --admin-url http://pulsar:8080 namespaces set-retention --size -1 --time 3m tg/response"
"depends_on":
"pulsar":
"condition": "service_started"
"image": "docker.io/apachepulsar/pulsar:3.3.0"
"restart": "on-failure:100"
"kg-extract-definitions":
"command":
- "kg-extract-definitions"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"kg-extract-relationships":
"command":
- "kg-extract-relationships"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"milvus":
"command":
- "milvus"
- "run"
- "standalone"
"environment":
"ETCD_ENDPOINTS": "etcd:2379"
"MINIO_ADDRESS": "minio:9000"
"image": "docker.io/milvusdb/milvus:v2.4.5"
"ports":
- "9091:9091"
- "19530:19530"
"restart": "on-failure:100"
"volumes":
- "milvus:/var/lib/milvus"
"minio":
"command":
- "minio"
- "server"
- "/minio_data"
- "--console-address"
- ":9001"
"environment":
"MINIO_ROOT_PASSWORD": "minioadmin"
"MINIO_ROOT_USER": "minioadmin"
"image": "docker.io/minio/minio:RELEASE.2024-07-04T14-25-45Z"
"ports":
- "9001:9001"
"restart": "on-failure:100"
"volumes":
- "minio-data:/minio_data"
"pdf-decoder":
"command":
- "pdf-decoder"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"prometheus":
"image": "docker.io/prom/prometheus:v2.53.1"
"ports":
- "9090:9090"
"restart": "on-failure:100"
"volumes":
- "./prometheus:/etc/prometheus"
- "prometheus-data:/prometheus"
"prompt":
"command":
- "prompt-generic"
- "-p"
- "pulsar://pulsar:6650"
- "--text-completion-request-queue"
- "non-persistent://tg/request/text-completion"
- "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"prompt-rag":
"command":
- "prompt-generic"
- "-p"
- "pulsar://pulsar:6650"
- "-i"
- "non-persistent://tg/request/prompt-rag"
- "-o"
- "non-persistent://tg/response/prompt-rag-response"
- "--text-completion-request-queue"
- "non-persistent://tg/request/text-completion-rag"
- "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"pulsar":
"command": "bin/pulsar standalone"
"image": "docker.io/apachepulsar/pulsar:3.3.0"
"ports":
- "6650:6650"
- "8080:8080"
"restart": "on-failure:100"
"volumes":
- "pulsar-conf:/pulsar/conf"
- "pulsar-data:/pulsar/data"
"pulsar-manager":
"environment":
"SPRING_CONFIGURATION_FILE": "/pulsar-manager/pulsar-manager/application.properties"
"image": "docker.io/apachepulsar/pulsar-manager:v0.3.0"
"ports":
- "9527:9527"
- "7750:7750"
"restart": "on-failure:100"
"query-graph-embeddings":
"command":
- "ge-query-milvus"
- "-p"
- "pulsar://pulsar:6650"
- "-t"
- "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"query-triples":
"command":
- "triples-query-neo4j"
- "-p"
- "pulsar://pulsar:6650"
- "-g"
- "bolt://neo4j:7687"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"store-graph-embeddings":
"command":
- "ge-write-milvus"
- "-p"
- "pulsar://pulsar:6650"
- "-t"
- "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"store-triples":
"command":
- "triples-write-neo4j"
- "-p"
- "pulsar://pulsar:6650"
- "-g"
- "bolt://neo4j:7687"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"text-completion":
"command":
- "text-completion-vertexai"
- "-p"
- "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
- "-k"
- "/vertexai/private.json"
- "-r"
- "us-central1"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"volumes":
- "./vertexai:/vertexai"
"text-completion-rag":
"command":
- "text-completion-vertexai"
- "-p"
- "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
- "-k"
- "/vertexai/private.json"
- "-r"
- "us-central1"
- "-i"
- "non-persistent://tg/request/text-completion-rag"
- "-o"
- "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"volumes":
- "./vertexai:/vertexai"
"vectorize":
"command":
- "embeddings-vectorize"
- "-p"
- "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100"
"volumes":
"neo4j": {}
"etcd": {}
"grafana-storage": {}
"milvus": {}
"minio-data": {}
"prometheus-data": {}
"pulsar-conf": {}
"pulsar-data": {}

View file

@ -15,7 +15,7 @@
- "250" - "250"
- "--chunk-overlap" - "--chunk-overlap"
- "15" - "15"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"embeddings": "embeddings":
"command": "command":
@ -24,7 +24,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-m" - "-m"
- "all-MiniLM-L6-v2" - "all-MiniLM-L6-v2"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"etcd": "etcd":
"command": "command":
@ -72,7 +72,7 @@
- "30" - "30"
- "--max-subgraph-size" - "--max-subgraph-size"
- "3000" - "3000"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"init-pulsar": "init-pulsar":
"command": "command":
@ -89,14 +89,14 @@
- "kg-extract-definitions" - "kg-extract-definitions"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"kg-extract-relationships": "kg-extract-relationships":
"command": "command":
- "kg-extract-relationships" - "kg-extract-relationships"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"milvus": "milvus":
"command": "command":
@ -134,7 +134,7 @@
- "pdf-decoder" - "pdf-decoder"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"prometheus": "prometheus":
"image": "docker.io/prom/prometheus:v2.53.1" "image": "docker.io/prom/prometheus:v2.53.1"
@ -153,7 +153,7 @@
- "non-persistent://tg/request/text-completion" - "non-persistent://tg/request/text-completion"
- "--text-completion-response-queue" - "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-response" - "non-persistent://tg/response/text-completion-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"prompt-rag": "prompt-rag":
"command": "command":
@ -168,7 +168,7 @@
- "non-persistent://tg/request/text-completion-rag" - "non-persistent://tg/request/text-completion-rag"
- "--text-completion-response-queue" - "--text-completion-response-queue"
- "non-persistent://tg/response/text-completion-rag-response" - "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"pulsar": "pulsar":
"command": "bin/pulsar standalone" "command": "bin/pulsar standalone"
@ -195,7 +195,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-t" - "-t"
- "http://milvus:19530" - "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"query-triples": "query-triples":
"command": "command":
@ -204,7 +204,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-g" - "-g"
- "cassandra" - "cassandra"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"store-graph-embeddings": "store-graph-embeddings":
"command": "command":
@ -213,7 +213,7 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-t" - "-t"
- "http://milvus:19530" - "http://milvus:19530"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"store-triples": "store-triples":
"command": "command":
@ -222,18 +222,22 @@
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
- "-g" - "-g"
- "cassandra" - "cassandra"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"text-completion": "text-completion":
"command": "command":
- "text-completion-vertexai" - "text-completion-vertexai"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
- "-k" - "-k"
- "/vertexai/private.json" - "/vertexai/private.json"
- "-r" - "-r"
- "us-west1" - "us-central1"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"volumes": "volumes":
- "./vertexai:/vertexai" - "./vertexai:/vertexai"
@ -242,15 +246,19 @@
- "text-completion-vertexai" - "text-completion-vertexai"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
# - "-x"
# - "4096"
# - "-t"
# - "0.0"
- "-k" - "-k"
- "/vertexai/private.json" - "/vertexai/private.json"
- "-r" - "-r"
- "us-west1" - "us-central1"
- "-i" - "-i"
- "non-persistent://tg/request/text-completion-rag" - "non-persistent://tg/request/text-completion-rag"
- "-o" - "-o"
- "non-persistent://tg/response/text-completion-rag-response" - "non-persistent://tg/response/text-completion-rag-response"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"volumes": "volumes":
- "./vertexai:/vertexai" - "./vertexai:/vertexai"
@ -259,7 +267,7 @@
- "embeddings-vectorize" - "embeddings-vectorize"
- "-p" - "-p"
- "pulsar://pulsar:6650" - "pulsar://pulsar:6650"
"image": "docker.io/trustgraph/trustgraph-flow:0.6.8" "image": "docker.io/trustgraph/trustgraph-flow:0.6.9"
"restart": "on-failure:100" "restart": "on-failure:100"
"volumes": "volumes":
"cassandra": {} "cassandra": {}

View file

@ -19,6 +19,8 @@ module = ".".join(__name__.split(".")[1:-1])
default_input_queue = text_completion_request_queue default_input_queue = text_completion_request_queue
default_output_queue = text_completion_response_queue default_output_queue = text_completion_response_queue
default_subscriber = module default_subscriber = module
default_temperature = 0.0
default_max_output = 4192
class Processor(ConsumerProducer): class Processor(ConsumerProducer):
@ -29,6 +31,8 @@ class Processor(ConsumerProducer):
subscriber = params.get("subscriber", default_subscriber) subscriber = params.get("subscriber", default_subscriber)
endpoint = params.get("endpoint") endpoint = params.get("endpoint")
token = params.get("token") token = params.get("token")
temperature = params.get("temperature", default_temperature)
max_output = params.get("max_output", default_max_output)
super(Processor, self).__init__( super(Processor, self).__init__(
**params | { **params | {
@ -37,11 +41,15 @@ class Processor(ConsumerProducer):
"subscriber": subscriber, "subscriber": subscriber,
"input_schema": TextCompletionRequest, "input_schema": TextCompletionRequest,
"output_schema": TextCompletionResponse, "output_schema": TextCompletionResponse,
"temperature": temperature,
"max_output": max_output,
} }
) )
self.endpoint = endpoint self.endpoint = endpoint
self.token = token self.token = token
self.temperature = temperature
self.max_output = max_output
def build_prompt(self, system, content): def build_prompt(self, system, content):
@ -54,8 +62,8 @@ class Processor(ConsumerProducer):
"role": "user", "content": content "role": "user", "content": content
} }
], ],
"max_tokens": 4192, "max_tokens": self.max_output,
"temperature": 0.2, "temperature": self.temperature,
"top_p": 1 "top_p": 1
} }
@ -105,7 +113,11 @@ class Processor(ConsumerProducer):
response = self.call_llm(prompt) response = self.call_llm(prompt)
print("Send response...", flush=True) print("Send response...", flush=True)
r = TextCompletionResponse(response=response)
resp = response.replace("```json", "")
resp = response.replace("```", "")
r = TextCompletionResponse(response=resp)
self.producer.send(r, properties={"id": id}) self.producer.send(r, properties={"id": id})
print("Done.", flush=True) print("Done.", flush=True)
@ -128,6 +140,20 @@ class Processor(ConsumerProducer):
help=f'LLM model token' help=f'LLM model token'
) )
parser.add_argument(
'-t', '--temperature',
type=float,
default=default_temperature,
help=f'LLM temperature parameter (default: {default_temperature})'
)
parser.add_argument(
'-x', '--max-output',
type=int,
default=default_max_output,
help=f'LLM max output tokens (default: {default_max_output})'
)
def run(): def run():
Processor.start(module, __doc__) Processor.start(module, __doc__)

View file

@ -6,7 +6,6 @@ Input is prompt, output is response. Mistral is default.
import boto3 import boto3
import json import json
import re
from .... schema import TextCompletionRequest, TextCompletionResponse from .... schema import TextCompletionRequest, TextCompletionResponse
from .... schema import text_completion_request_queue from .... schema import text_completion_request_queue
@ -22,6 +21,9 @@ default_output_queue = text_completion_response_queue
default_subscriber = module default_subscriber = module
default_model = 'mistral.mistral-large-2407-v1:0' default_model = 'mistral.mistral-large-2407-v1:0'
default_region = 'us-west-2' default_region = 'us-west-2'
default_temperature = 0.0
default_max_output = 2048
class Processor(ConsumerProducer): class Processor(ConsumerProducer):
@ -34,6 +36,8 @@ class Processor(ConsumerProducer):
aws_id = params.get("aws_id_key") aws_id = params.get("aws_id_key")
aws_secret = params.get("aws_secret") aws_secret = params.get("aws_secret")
aws_region = params.get("aws_region", default_region) aws_region = params.get("aws_region", default_region)
temperature = params.get("temperature", default_temperature)
max_output = params.get("max_output", default_max_output)
super(Processor, self).__init__( super(Processor, self).__init__(
**params | { **params | {
@ -43,10 +47,14 @@ class Processor(ConsumerProducer):
"input_schema": TextCompletionRequest, "input_schema": TextCompletionRequest,
"output_schema": TextCompletionResponse, "output_schema": TextCompletionResponse,
"model": model, "model": model,
"temperature": temperature,
"max_output": max_output,
} }
) )
self.model = model self.model = model
self.temperature = temperature
self.max_output = max_output
self.session = boto3.Session( self.session = boto3.Session(
aws_access_key_id=aws_id, aws_access_key_id=aws_id,
@ -74,8 +82,8 @@ class Processor(ConsumerProducer):
if self.model.startswith("mistral"): if self.model.startswith("mistral"):
promptbody = json.dumps({ promptbody = json.dumps({
"prompt": prompt, "prompt": prompt,
"max_tokens": 8192, "max_tokens": self.max_output,
"temperature": 0.0, "temperature": self.temperature,
"top_p": 0.99, "top_p": 0.99,
"top_k": 40 "top_k": 40
}) })
@ -84,8 +92,8 @@ class Processor(ConsumerProducer):
elif self.model.startswith("meta"): elif self.model.startswith("meta"):
promptbody = json.dumps({ promptbody = json.dumps({
"prompt": prompt, "prompt": prompt,
"max_gen_len": 2048, "max_gen_len": self.max_output,
"temperature": 0.0, "temperature": self.temperature,
"top_p": 0.95, "top_p": 0.95,
}) })
@ -93,8 +101,8 @@ class Processor(ConsumerProducer):
elif self.model.startswith("anthropic"): elif self.model.startswith("anthropic"):
promptbody = json.dumps({ promptbody = json.dumps({
"anthropic_version": "bedrock-2023-05-31", "anthropic_version": "bedrock-2023-05-31",
"max_tokens": 8192, "max_tokens": self.max_output,
"temperature": 0, "temperature": self.temperature,
"top_p": 0.999, "top_p": 0.999,
"messages": [ "messages": [
{ {
@ -113,8 +121,8 @@ class Processor(ConsumerProducer):
else: else:
promptbody = json.dumps({ promptbody = json.dumps({
"prompt": prompt, "prompt": prompt,
"max_tokens": 8192, "max_tokens": self.max_output,
"temperature": 0.0, "temperature": self.temperature,
"top_p": 0.99, "top_p": 0.99,
"top_k": 40 "top_k": 40
}) })
@ -148,21 +156,11 @@ class Processor(ConsumerProducer):
print(outputtext, flush=True) print(outputtext, flush=True)
# Parse output for ```json``` delimiters resp = outputtext.replace("```json", "")
pattern = r'```json\s*([\s\S]*?)\s*```' resp = outputtext.replace("```", "")
match = re.search(pattern, outputtext)
if match:
# If delimiters are found, extract the JSON content
json_content = match.group(1)
json_resp = json_content.strip()
else:
# If no delimiters are found, return the original text
json_resp = outputtext.strip()
print("Send response...", flush=True) print("Send response...", flush=True)
r = TextCompletionResponse(response=json_resp) r = TextCompletionResponse(response=resp)
self.send(r, properties={"id": id}) self.send(r, properties={"id": id})
print("Done.", flush=True) print("Done.", flush=True)
@ -196,6 +194,20 @@ class Processor(ConsumerProducer):
help=f'AWS Region (default: us-west-2)' help=f'AWS Region (default: us-west-2)'
) )
parser.add_argument(
'-t', '--temperature',
type=float,
default=default_temperature,
help=f'LLM temperature parameter (default: {default_temperature})'
)
parser.add_argument(
'-x', '--max-output',
type=int,
default=default_max_output,
help=f'LLM max output tokens (default: {default_max_output})'
)
def run(): def run():
Processor.start(module, __doc__) Processor.start(module, __doc__)

View file

@ -18,6 +18,8 @@ default_input_queue = text_completion_request_queue
default_output_queue = text_completion_response_queue default_output_queue = text_completion_response_queue
default_subscriber = module default_subscriber = module
default_model = 'claude-3-5-sonnet-20240620' default_model = 'claude-3-5-sonnet-20240620'
default_temperature = 0.0
default_max_output = 8192
class Processor(ConsumerProducer): class Processor(ConsumerProducer):
@ -28,6 +30,8 @@ class Processor(ConsumerProducer):
subscriber = params.get("subscriber", default_subscriber) subscriber = params.get("subscriber", default_subscriber)
model = params.get("model", default_model) model = params.get("model", default_model)
api_key = params.get("api_key") api_key = params.get("api_key")
temperature = params.get("temperature", default_temperature)
max_output = params.get("max_output", default_max_output)
super(Processor, self).__init__( super(Processor, self).__init__(
**params | { **params | {
@ -37,12 +41,15 @@ class Processor(ConsumerProducer):
"input_schema": TextCompletionRequest, "input_schema": TextCompletionRequest,
"output_schema": TextCompletionResponse, "output_schema": TextCompletionResponse,
"model": model, "model": model,
"temperature": temperature,
"max_output": max_output,
} }
) )
self.model = model self.model = model
self.claude = anthropic.Anthropic(api_key=api_key) self.claude = anthropic.Anthropic(api_key=api_key)
self.temperature = temperature
self.max_output = max_output
print("Initialised", flush=True) print("Initialised", flush=True)
@ -61,8 +68,8 @@ class Processor(ConsumerProducer):
# FIXME: Rate limits? # FIXME: Rate limits?
response = message = self.claude.messages.create( response = message = self.claude.messages.create(
model=self.model, model=self.model,
max_tokens=1000, max_tokens=self.max_output,
temperature=0.1, temperature=self.temperature,
system = "You are a helpful chatbot.", system = "You are a helpful chatbot.",
messages=[ messages=[
{ {
@ -105,6 +112,20 @@ class Processor(ConsumerProducer):
help=f'Claude API key' help=f'Claude API key'
) )
parser.add_argument(
'-t', '--temperature',
type=float,
default=default_temperature,
help=f'LLM temperature parameter (default: {default_temperature})'
)
parser.add_argument(
'-x', '--max-output',
type=int,
default=default_max_output,
help=f'LLM max output tokens (default: {default_max_output})'
)
def run(): def run():
Processor.start(module, __doc__) Processor.start(module, __doc__)

View file

@ -5,7 +5,6 @@ Input is prompt, output is response.
""" """
import cohere import cohere
import re
from .... schema import TextCompletionRequest, TextCompletionResponse from .... schema import TextCompletionRequest, TextCompletionResponse
from .... schema import text_completion_request_queue from .... schema import text_completion_request_queue
@ -19,6 +18,7 @@ default_input_queue = text_completion_request_queue
default_output_queue = text_completion_response_queue default_output_queue = text_completion_response_queue
default_subscriber = module default_subscriber = module
default_model = 'c4ai-aya-23-8b' default_model = 'c4ai-aya-23-8b'
default_temperature = 0.0
class Processor(ConsumerProducer): class Processor(ConsumerProducer):
@ -29,6 +29,7 @@ class Processor(ConsumerProducer):
subscriber = params.get("subscriber", default_subscriber) subscriber = params.get("subscriber", default_subscriber)
model = params.get("model", default_model) model = params.get("model", default_model)
api_key = params.get("api_key") api_key = params.get("api_key")
temperature = params.get("temperature", default_temperature)
super(Processor, self).__init__( super(Processor, self).__init__(
**params | { **params | {
@ -38,11 +39,12 @@ class Processor(ConsumerProducer):
"input_schema": TextCompletionRequest, "input_schema": TextCompletionRequest,
"output_schema": TextCompletionResponse, "output_schema": TextCompletionResponse,
"model": model, "model": model,
"temperature": temperature,
} }
) )
self.model = model self.model = model
self.temperature = temperature
self.cohere = cohere.Client(api_key=api_key) self.cohere = cohere.Client(api_key=api_key)
print("Initialised", flush=True) print("Initialised", flush=True)
@ -64,7 +66,7 @@ class Processor(ConsumerProducer):
model=self.model, model=self.model,
message=prompt, message=prompt,
preamble = "You are a helpful AI-assistant.", preamble = "You are a helpful AI-assistant.",
temperature=0.0, temperature=self.temperature,
chat_history=[], chat_history=[],
prompt_truncation='auto', prompt_truncation='auto',
connectors=[] connectors=[]
@ -73,21 +75,11 @@ class Processor(ConsumerProducer):
resp = output.text resp = output.text
print(resp, flush=True) print(resp, flush=True)
# Parse output for ```json``` delimiters resp = resp.replace("```json", "")
pattern = r'```json\s*([\s\S]*?)\s*```' resp = resp.replace("```", "")
match = re.search(pattern, resp)
if match:
# If delimiters are found, extract the JSON content
json_content = match.group(1)
json_resp = json_content.strip()
else:
# If no delimiters are found, return the original text
json_resp = resp.strip()
print("Send response...", flush=True) print("Send response...", flush=True)
r = TextCompletionResponse(response=json_resp) r = TextCompletionResponse(response=resp)
self.send(r, properties={"id": id}) self.send(r, properties={"id": id})
print("Done.", flush=True) print("Done.", flush=True)
@ -111,6 +103,13 @@ class Processor(ConsumerProducer):
help=f'Cohere API key' help=f'Cohere API key'
) )
parser.add_argument(
'-t', '--temperature',
type=float,
default=default_temperature,
help=f'LLM temperature parameter (default: {default_temperature})'
)
def run(): def run():
Processor.start(module, __doc__) Processor.start(module, __doc__)

View file

@ -71,7 +71,10 @@ class Processor(ConsumerProducer):
print("Send response...", flush=True) print("Send response...", flush=True)
r = TextCompletionResponse(response=response) resp = response.replace("```json", "")
resp = response.replace("```", "")
r = TextCompletionResponse(response=resp)
self.send(r, properties={"id": id}) self.send(r, properties={"id": id})

View file

@ -18,6 +18,8 @@ default_input_queue = text_completion_request_queue
default_output_queue = text_completion_response_queue default_output_queue = text_completion_response_queue
default_subscriber = module default_subscriber = module
default_model = 'gpt-3.5-turbo' default_model = 'gpt-3.5-turbo'
default_temperature = 0.0
default_max_output = 4096
class Processor(ConsumerProducer): class Processor(ConsumerProducer):
@ -28,6 +30,8 @@ class Processor(ConsumerProducer):
subscriber = params.get("subscriber", default_subscriber) subscriber = params.get("subscriber", default_subscriber)
model = params.get("model", default_model) model = params.get("model", default_model)
api_key = params.get("api_key") api_key = params.get("api_key")
temperature = params.get("temperature", default_temperature)
max_output = params.get("max_output", default_max_output)
super(Processor, self).__init__( super(Processor, self).__init__(
**params | { **params | {
@ -37,11 +41,14 @@ class Processor(ConsumerProducer):
"input_schema": TextCompletionRequest, "input_schema": TextCompletionRequest,
"output_schema": TextCompletionResponse, "output_schema": TextCompletionResponse,
"model": model, "model": model,
"temperature": temperature,
"max_output": max_output,
} }
) )
self.model = model self.model = model
self.temperature = temperature
self.max_output = max_output
self.openai = OpenAI(api_key=api_key) self.openai = OpenAI(api_key=api_key)
print("Initialised", flush=True) print("Initialised", flush=True)
@ -72,8 +79,8 @@ class Processor(ConsumerProducer):
] ]
} }
], ],
temperature=0, temperature=self.temperature,
max_tokens=4096, max_tokens=self.max_output,
top_p=1, top_p=1,
frequency_penalty=0, frequency_penalty=0,
presence_penalty=0, presence_penalty=0,
@ -109,6 +116,20 @@ class Processor(ConsumerProducer):
help=f'OpenAI API key' help=f'OpenAI API key'
) )
parser.add_argument(
'-t', '--temperature',
type=float,
default=default_temperature,
help=f'LLM temperature parameter (default: {default_temperature})'
)
parser.add_argument(
'-x', '--max-output',
type=int,
default=default_max_output,
help=f'LLM max output tokens (default: {default_max_output})'
)
def run(): def run():
Processor.start(module, __doc__) Processor.start(module, __doc__)

View file

@ -33,6 +33,10 @@ module = ".".join(__name__.split(".")[1:-1])
default_input_queue = text_completion_request_queue default_input_queue = text_completion_request_queue
default_output_queue = text_completion_response_queue default_output_queue = text_completion_response_queue
default_subscriber = module default_subscriber = module
default_model = 'gemini-1.0-pro-001'
default_region = 'us-central1'
default_temperature = 0.0
default_max_output = 8192
class Processor(ConsumerProducer): class Processor(ConsumerProducer):
@ -41,9 +45,11 @@ class Processor(ConsumerProducer):
input_queue = params.get("input_queue", default_input_queue) input_queue = params.get("input_queue", default_input_queue)
output_queue = params.get("output_queue", default_output_queue) output_queue = params.get("output_queue", default_output_queue)
subscriber = params.get("subscriber", default_subscriber) subscriber = params.get("subscriber", default_subscriber)
region = params.get("region", "us-west1") region = params.get("region", default_region)
model = params.get("model", "gemini-1.0-pro-001") model = params.get("model", default_model)
private_key = params.get("private_key") private_key = params.get("private_key")
temperature = params.get("temperature", default_temperature)
max_output = params.get("max_output", default_max_output)
super(Processor, self).__init__( super(Processor, self).__init__(
**params | { **params | {
@ -56,19 +62,19 @@ class Processor(ConsumerProducer):
) )
self.parameters = { self.parameters = {
"temperature": 0.2, "temperature": temperature,
"top_p": 1.0, "top_p": 1.0,
"top_k": 32, "top_k": 32,
"candidate_count": 1, "candidate_count": 1,
"max_output_tokens": 8192, "max_output_tokens": max_output,
} }
self.generation_config = GenerationConfig( self.generation_config = GenerationConfig(
temperature=0.2, temperature=temperature,
top_p=1.0, top_p=1.0,
top_k=10, top_k=10,
candidate_count=1, candidate_count=1,
max_output_tokens=8191, max_output_tokens=max_output,
) )
# Block none doesn't seem to work # Block none doesn't seem to work
@ -155,8 +161,8 @@ class Processor(ConsumerProducer):
parser.add_argument( parser.add_argument(
'-m', '--model', '-m', '--model',
default="gemini-1.0-pro-001", default=default_model,
help=f'LLM model (default: gemini-1.0-pro-001)' help=f'LLM model (default: {default_model})'
) )
# Also: text-bison-32k # Also: text-bison-32k
@ -167,8 +173,22 @@ class Processor(ConsumerProducer):
parser.add_argument( parser.add_argument(
'-r', '--region', '-r', '--region',
default='us-west1', default=default_region,
help=f'Google Cloud region (default: us-west1)', help=f'Google Cloud region (default: {default_region})',
)
parser.add_argument(
'-t', '--temperature',
type=float,
default=default_temperature,
help=f'LLM temperature parameter (default: {default_temperature})'
)
parser.add_argument(
'-x', '--max-output',
type=int,
default=default_max_output,
help=f'LLM max output tokens (default: {default_max_output})'
) )
def run(): def run():