Fix auto-templates (#17)

templates/components/azure.jsonnet

@@ -14,6 +14,10 @@ local url = import "url.jsonnet";
 		"${AZURE_TOKEN}",
 		"-e",
 		"${AZURE_ENDPOINT}",
+                "-x",
+                "4096",
+                "-t",
+                "0.0",
 	    ],
 	},
 
@@ -28,6 +32,10 @@ local url = import "url.jsonnet";
 		"-e",
 		"${AZURE_ENDPOINT}",
 		"-i",
+                "-x",
+                "4096",
+                "-t",
+                "0.0",
 		"non-persistent://tg/request/text-completion-rag",
 		"-o",
 		"non-persistent://tg/response/text-completion-rag-response",

templates/components/bedrock.jsonnet

@@ -29,6 +29,12 @@ local url = import "url.jsonnet";
 		"${AWS_SECRET_KEY}",
 		"-r",
 		"us-west-2",
+                "-x",
+		"4096",
+		"-t",
+		"0.0",
+		"-m",
+		"mistral.mixtral-8x7b-instruct-v0:1",
 	    ],
 	},
 
@@ -46,6 +52,12 @@ local url = import "url.jsonnet";
 		"${AWS_SECRET_KEY}",
 		"-r",
 		"us-west-2",
+                "-x",
+		"4096",
+		"-t",
+		"0.0",
+		"-m",
+		"mistral.mixtral-8x7b-instruct-v0:1",
 		"-i",
 		"non-persistent://tg/request/text-completion-rag",
 		"-o",

templates/components/claude.jsonnet

@@ -12,6 +12,10 @@ local url = import "url.jsonnet";
 		url.pulsar,
 		"-k",
 		"${CLAUDE_KEY}",
+                "-x",
+                "4096",
+                "-t",
+                "0.0",
 	    ],
 	},
 
@@ -23,6 +27,10 @@ local url = import "url.jsonnet";
 		url.pulsar,
 		"-k",
 		"${CLAUDE_KEY}",
+                "-x",
+                "4096",
+                "-t",
+                "0.0",
 		"-i",
 		"non-persistent://tg/request/text-completion-rag",
 		"-o",

templates/components/cohere.jsonnet

@@ -7,13 +7,13 @@ local url = import "url.jsonnet";
 	chunker: base + {
 	    image: images.trustgraph,
 	    command: [
-		"chunker-recursive",
+		"${CHUNKER:-chunker-token}",
 		"-p",
 		url.pulsar,
 		"--chunk-size",
-		"1000",
+		"150",
 		"--chunk-overlap",
-		"50",
+		"10",
 	    ],
 	},
 
@@ -25,6 +25,8 @@ local url = import "url.jsonnet";
 		url.pulsar,
 		"-k",
 		"${COHERE_KEY}",
+                "-t",
+                "0.0",
 	    ],
 	},
 
@@ -36,6 +38,8 @@ local url = import "url.jsonnet";
 		url.pulsar,
 		"-k",
 		"${COHERE_KEY}",
+                "-t",
+                "0.0",
 		"-i",
 		"non-persistent://tg/request/text-completion-rag",
 		"-o",

templates/components/ollama.jsonnet

@@ -10,8 +10,8 @@ local url = import "url.jsonnet";
 		"text-completion-ollama",
 		"-p",
 		url.pulsar,
-                // "-m",
-                // "llama3.1:8b",
+                "-m",
+                "gemma2:9b",
 		"-r",
 		"${OLLAMA_HOST}",
 	    ],
@@ -23,8 +23,8 @@ local url = import "url.jsonnet";
 		"text-completion-ollama",
 		"-p",
 		url.pulsar,
-                // "-m",
-                // "llama3.1:8b",
+                "-m",
+                "gemma2:9b",
 		"-r",
 		"${OLLAMA_HOST}",
 		"-i",

templates/components/openai.jsonnet

@@ -12,6 +12,10 @@ local url = import "url.jsonnet";
 		url.pulsar,
 		"-k",
 		"${OPENAI_KEY}",
+                "-x",
+                "4096",
+                "-t",
+                "0.0",
 	    ],
 	},
 
@@ -23,6 +27,10 @@ local url = import "url.jsonnet";
 		url.pulsar,
 		"-k",
 		"${OPENAI_KEY}",
+                "-x",
+                "4096",
+                "-t",
+                "0.0",
 		"-i",
 		"non-persistent://tg/request/text-completion-rag",
 		"-o",

templates/components/vertexai.jsonnet

@@ -13,7 +13,11 @@ local url = import "url.jsonnet";
 		"-k",
 		"/vertexai/private.json",
 		"-r",
-		"us-west1",
+		"us-central1",
+                "-x",
+                "4096",
+                "-t",
+                "0.0",
 	    ],
 	    volumes: [
 		"./vertexai:/vertexai"
@@ -29,7 +33,11 @@ local url = import "url.jsonnet";
 		"-k",
 		"/vertexai/private.json",
 		"-r",
-		"us-west1",
+		"us-central1",
+                "-x",
+                "4096",
+                "-t",
+                "0.0",
 		"-i",
 		"non-persistent://tg/request/text-completion-rag",
 		"-o",

templates/docker-compose.jsonnet

@@ -0,0 +1,27 @@
+local components = {
+    neo4j: import "components/neo4j.jsonnet",
+    cassandra: import "components/neo4j.jsonnet",
+    pulsar: import "components/pulsar.jsonnet",
+    milvus: import "components/milvus.jsonnet",
+    grafana: import "components/grafana.jsonnet",
+    trustgraph: import "components/trustgraph.jsonnet",
+    azure: import "components/azure.jsonnet",
+    bedrock: import "components/bedrock.jsonnet",
+    cohere: import "components/cohere.jsonnet",
+    claude: import "components/claude.jsonnet",
+    ollama: import "components/ollama.jsonnet",
+    openai: import "components/openai.jsonnet",
+    mix: import "components/mix.jsonnet",
+    vertexai: import "components/vertexai.jsonnet",
+};
+
+local options = std.split(std.extVar("options"), ",");
+
+local add = function(state, name) state + components[name];
+
+local config = std.foldl(add, options, {});
+
+std.manifestYamlDoc(config)
+
+
+