diff --git a/templates/components/azure.jsonnet b/templates/components/azure.jsonnet index 18e09181..fa7c4ee9 100644 --- a/templates/components/azure.jsonnet +++ b/templates/components/azure.jsonnet @@ -6,7 +6,7 @@ local prompts = import "prompts/mixtral.jsonnet"; "azure-token":: "${AZURE_TOKEN}", "azure-endpoint":: "${AZURE_ENDPOINT}", - "azure-max-output":: 4096, + "azure-max-output-tokens":: 4096, "azure-temperature":: 0.0, services +: { @@ -22,7 +22,7 @@ local prompts = import "prompts/mixtral.jsonnet"; "-e", $["azure-endpoint"], "-x", - std.toString($["azure-max-output"]), + std.toString($["azure-max-output-tokens"]), "-t", std.toString($["azure-temperature"]), ], @@ -52,7 +52,7 @@ local prompts = import "prompts/mixtral.jsonnet"; $["azure-endpoint"], "-i", "-x", - std.toString($["azure-max-output"]), + std.toString($["azure-max-output-tokens"]), "-t", std.toString($["azure-temperature"]), "non-persistent://tg/request/text-completion-rag", diff --git a/templates/config-loader.jsonnet b/templates/config-loader.jsonnet index 6555dea8..e88bf783 100644 --- a/templates/config-loader.jsonnet +++ b/templates/config-loader.jsonnet @@ -15,7 +15,7 @@ local components = { "prompt-template-relationships": import "components/null.jsonnet", "prompt-template-rows-template": import "components/null.jsonnet", "pulsar": import "components/pulsar.jsonnet", - "pulsar-manager": import "components/pulsar.jsonnet", + "pulsar-manager": import "components/pulsar-manager.jsonnet", "trustgraph-base": import "components/trustgraph.jsonnet", "vector-store-milvus": import "components/milvus.jsonnet", "vector-store-qdrant": import "components/qdrant.jsonnet", diff --git a/templates/patterns/grafana.jsonnet b/templates/patterns/grafana.jsonnet index 12827e4e..49ae692c 100644 --- a/templates/patterns/grafana.jsonnet +++ b/templates/patterns/grafana.jsonnet @@ -8,6 +8,7 @@ features: ["prometheus", "grafana"], args: [ ], + category: ["monitoring"], }, module: "components/grafana.jsonnet", } diff --git a/templates/patterns/graph-rag-cassandra.jsonnet b/templates/patterns/graph-rag-cassandra.jsonnet index 3d801572..e71a3e1c 100644 --- a/templates/patterns/graph-rag-cassandra.jsonnet +++ b/templates/patterns/graph-rag-cassandra.jsonnet @@ -7,6 +7,7 @@ requires: ["pulsar", "trustgraph"], features: ["cassandra", "rag"], args: [], + category: ["knowledge-graph"], }, module: "components/cassandra.jsonnet", } diff --git a/templates/patterns/graph-rag-neo4j.jsonnet b/templates/patterns/graph-rag-neo4j.jsonnet index c75986a2..16b1b4a9 100644 --- a/templates/patterns/graph-rag-neo4j.jsonnet +++ b/templates/patterns/graph-rag-neo4j.jsonnet @@ -7,6 +7,7 @@ requires: ["pulsar", "trustgraph"], features: ["neo4j", "rag"], args: [], + category: [ "knowledge-graph" ], }, module: "components/neo4j.jsonnet", } diff --git a/templates/patterns/llm-azure.jsonnet b/templates/patterns/llm-azure.jsonnet index 59ba9741..7a0f12ee 100644 --- a/templates/patterns/llm-azure.jsonnet +++ b/templates/patterns/llm-azure.jsonnet @@ -7,7 +7,26 @@ requires: ["pulsar", "trustgraph"], features: ["llm"], args: [ - ] + { + name: "azure-max-output-tokens", + label: "Maximum output tokens", + type: "integer", + description: "Limit on number tokens to generate", + default: 4096, + required: true, + }, + { + name: "azure-temperature", + label: "Temperature", + type: "slider", + description: "Controlling predictability / creativity balance", + min: 0, + max: 1, + step: 0.05, + default: 0.5, + }, + ], + category: [ "llm" ], }, module: "components/azure.jsonnet", } diff --git a/templates/patterns/llm-bedrock.jsonnet b/templates/patterns/llm-bedrock.jsonnet index 2122a522..e114c428 100644 --- a/templates/patterns/llm-bedrock.jsonnet +++ b/templates/patterns/llm-bedrock.jsonnet @@ -7,7 +7,26 @@ requires: ["pulsar", "trustgraph"], features: ["llm"], args: [ - ] + { + name: "bedrock-max-output-tokens", + label: "Maximum output tokens", + type: "integer", + description: "Limit on number tokens to generate", + default: 4096, + required: true, + }, + { + name: "bedrock-temperature", + label: "Temperature", + type: "slider", + description: "Controlling predictability / creativity balance", + min: 0, + max: 1, + step: 0.05, + default: 0.5, + }, + ], + category: [ "llm" ], }, module: "components/bedrock.jsonnet", } diff --git a/templates/patterns/llm-claude.jsonnet b/templates/patterns/llm-claude.jsonnet index 0b5f35c2..beefef4b 100644 --- a/templates/patterns/llm-claude.jsonnet +++ b/templates/patterns/llm-claude.jsonnet @@ -7,7 +7,26 @@ requires: ["pulsar", "trustgraph"], features: ["llm"], args: [ - ] + { + name: "claude-max-output-tokens", + label: "Maximum output tokens", + type: "integer", + description: "Limit on number tokens to generate", + default: 4096, + required: true, + }, + { + name: "claude-temperature", + label: "Temperature", + type: "slider", + description: "Controlling predictability / creativity balance", + min: 0, + max: 1, + step: 0.05, + default: 0.5, + }, + ], + category: [ "llm" ], }, module: "components/claude.jsonnet", } diff --git a/templates/patterns/llm-cohere.jsonnet b/templates/patterns/llm-cohere.jsonnet index 61b572dd..33a070e3 100644 --- a/templates/patterns/llm-cohere.jsonnet +++ b/templates/patterns/llm-cohere.jsonnet @@ -7,7 +7,26 @@ requires: ["pulsar", "trustgraph"], features: ["llm"], args: [ - ] + { + name: "cohere-max-output-tokens", + label: "Maximum output tokens", + type: "integer", + description: "Limit on number tokens to generate", + default: 4096, + required: true, + }, + { + name: "cohere-temperature", + label: "Temperature", + type: "slider", + description: "Controlling predictability / creativity balance", + min: 0, + max: 1, + step: 0.05, + default: 0.5, + }, + ], + category: [ "llm" ], }, module: "components/cohere.jsonnet", } diff --git a/templates/patterns/llm-ollama.jsonnet b/templates/patterns/llm-ollama.jsonnet index 8790ffc9..a46b9c57 100644 --- a/templates/patterns/llm-ollama.jsonnet +++ b/templates/patterns/llm-ollama.jsonnet @@ -7,7 +7,26 @@ requires: ["pulsar", "trustgraph"], features: ["llm"], args: [ - ] + { + name: "ollama-max-output-tokens", + label: "Maximum output tokens", + type: "integer", + description: "Limit on number tokens to generate", + default: 4096, + required: true, + }, + { + name: "ollama-temperature", + label: "Temperature", + type: "slider", + description: "Controlling predictability / creativity balance", + min: 0, + max: 1, + step: 0.05, + default: 0.5, + }, + ], + category: [ "llm" ], }, module: "components/ollama.jsonnet", } diff --git a/templates/patterns/llm-openai.jsonnet b/templates/patterns/llm-openai.jsonnet index 35804f27..fc1bbd51 100644 --- a/templates/patterns/llm-openai.jsonnet +++ b/templates/patterns/llm-openai.jsonnet @@ -7,7 +7,26 @@ requires: ["pulsar", "trustgraph"], features: ["llm"], args: [ - ] + { + name: "openai-max-output-tokens", + label: "Maximum output tokens", + type: "integer", + description: "Limit on number tokens to generate", + default: 4096, + required: true, + }, + { + name: "openai-temperature", + label: "Temperature", + type: "slider", + description: "Controlling predictability / creativity balance", + min: 0, + max: 1, + step: 0.05, + default: 0.5, + }, + ], + category: [ "llm" ], }, module: "components/openai.jsonnet", } diff --git a/templates/patterns/llm-vertexai.jsonnet b/templates/patterns/llm-vertexai.jsonnet index eaeab872..040aa9a1 100644 --- a/templates/patterns/llm-vertexai.jsonnet +++ b/templates/patterns/llm-vertexai.jsonnet @@ -7,7 +7,26 @@ requires: ["pulsar", "trustgraph"], features: ["llm"], args: [ - ] + { + name: "vertexai-max-output-tokens", + label: "Maximum output tokens", + type: "integer", + description: "Limit on number tokens to generate", + default: 4096, + required: true, + }, + { + name: "vertexai-temperature", + label: "Temperature", + type: "slider", + description: "Controlling predictability / creativity balance", + min: 0, + max: 1, + step: 0.05, + default: 0.5, + }, + ], + category: [ "llm" ], }, module: "components/vertexai.jsonnet", } diff --git a/templates/patterns/override-recursive-chunker.jsonnet b/templates/patterns/override-recursive-chunker.jsonnet index 1b85f2f7..65f93947 100644 --- a/templates/patterns/override-recursive-chunker.jsonnet +++ b/templates/patterns/override-recursive-chunker.jsonnet @@ -9,19 +9,20 @@ args: [ { name: "chunk-size", - type: "int", + type: "integer", description: "Chunk size value", default: 2000, required: true, }, { name: "chunk-overlap", - type: "int", + type: "integer", description: "Overlap size value", default: 100, required: true, } - ] + ], + category: [ "chunking" ], }, module: "components/cassandra.jsonnet", } diff --git a/templates/patterns/prompt-template-definitions.jsonnet b/templates/patterns/prompt-template-definitions.jsonnet index 7a7f0e00..8dc89465 100644 --- a/templates/patterns/prompt-template-definitions.jsonnet +++ b/templates/patterns/prompt-template-definitions.jsonnet @@ -9,13 +9,15 @@ args: [ { name: "prompt-definition-template", - type: "string", - width: 2000, + type: "multiline", + size: 2000, + rows: 10, description: "Definition extraction prompt", default: "\nStudy the following text and derive definitions for any discovered entities.\nDo not provide definitions for entities whose definitions are incomplete\nor unknown.\nOutput relationships in JSON format as an arary of objects with fields:\n- entity: the name of the entity\n- definition: English text which defines the entity\n\n\n\n{text}\n\n\n\nYou will respond only with raw JSON format data. Do not provide\nexplanations. Do not use special characters in the abstract text. The\nabstract will be written as plain text. Do not add markdown formatting\nor headers or prefixes. Do not include null or unknown definitions.\n", required: true, } - ] + ], + category: [ "prompting" ], }, module: "components/null.jsonnet", } diff --git a/templates/patterns/prompt-template-document-query.jsonnet b/templates/patterns/prompt-template-document-query.jsonnet index 5a46320e..f445870a 100644 --- a/templates/patterns/prompt-template-document-query.jsonnet +++ b/templates/patterns/prompt-template-document-query.jsonnet @@ -9,13 +9,15 @@ args: [ { name: "prompt-document-query-template", - type: "string", - width: 2000, + type: "multiline", + size: 2000, + rows: 10, description: "Document query prompt", default: "Study the following context. Use only the information provided in the context in your response. Do not speculate if the answer is not found in the provided set of knowledge statements.\n\nHere is the context:\n{documents}\n\nUse only the provided knowledge statements to respond to the following:\n{query}\n", required: true, } - ] + ], + category: [ "prompting" ], }, module: "components/null.jsonnet", } diff --git a/templates/patterns/prompt-template-kg-query.jsonnet b/templates/patterns/prompt-template-kg-query.jsonnet index 20a03b8e..c0a256a4 100644 --- a/templates/patterns/prompt-template-kg-query.jsonnet +++ b/templates/patterns/prompt-template-kg-query.jsonnet @@ -9,13 +9,15 @@ args: [ { name: "prompt-knowledge-query-template", - type: "string", - width: 2000, + type: "multiline", + size: 2000, + rows: 10, description: "Knowledge graph extraction prompt", default: "Study the following set of knowledge statements. The statements are written in Cypher format that has been extracted from a knowledge graph. Use only the provided set of knowledge statements in your response. Do not speculate if the answer is not found in the provided set of knowledge statements.\n\nHere's the knowledge statements:\n{graph}\n\nUse only the provided knowledge statements to respond to the following:\n{query}\n", required: true, } - ] + ], + category: [ "prompting" ], }, module: "components/null.jsonnet", } diff --git a/templates/patterns/prompt-template-relationships.jsonnet b/templates/patterns/prompt-template-relationships.jsonnet index 0664895c..1c295ca9 100644 --- a/templates/patterns/prompt-template-relationships.jsonnet +++ b/templates/patterns/prompt-template-relationships.jsonnet @@ -9,13 +9,15 @@ args: [ { name: "prompt-relationship-template", - type: "string", - width: 2000, + type: "multiline", + size: 2000, + rows: 10, description: "Relationship extraction prompt", default: "\nStudy the following text and derive entity relationships. For each\nrelationship, derive the subject, predicate and object of the relationship.\nOutput relationships in JSON format as an arary of objects with fields:\n- subject: the subject of the relationship\n- predicate: the predicate\n- object: the object of the relationship\n- object-entity: false if the object is a simple data type: name, value or date. true if it is an entity.\n\n\n\n{text}\n\n\n\nYou will respond only with raw JSON format data. Do not provide\nexplanations. Do not use special characters in the abstract text. The\nabstract must be written as plain text. Do not add markdown formatting\nor headers or prefixes.\n", required: true, } - ] + ], + category: [ "prompting" ], }, module: "components/null.jsonnet", } diff --git a/templates/patterns/prompt-template-rows-template.jsonnet b/templates/patterns/prompt-template-rows-template.jsonnet index 3ea9c0e4..09040aa1 100644 --- a/templates/patterns/prompt-template-rows-template.jsonnet +++ b/templates/patterns/prompt-template-rows-template.jsonnet @@ -9,13 +9,15 @@ args: [ { name: "prompt-rows-template", - type: "string", - width: 2000, + type: "multiline", + size: 2000, + rows: 10, description: "Row data extraction prompt", default: "\nStudy the following text and derive objects which match the schema provided.\n\nYou must output an array of JSON objects for each object you discover\nwhich matches the schema. For each object, output a JSON object whose fields\ncarry the name field specified in the schema.\n\n\n\n{schema}\n\n\n\n{text}\n\n\n\nYou will respond only with raw JSON format data. Do not provide\nexplanations. Do not add markdown formatting or headers or prefixes.\n", required: true, } - ] + ], + category: [ "prompting" ], }, module: "components/null.jsonnet", } diff --git a/templates/patterns/pulsar-manager.jsonnet b/templates/patterns/pulsar-manager.jsonnet index 93519f0f..81fa6bde 100644 --- a/templates/patterns/pulsar-manager.jsonnet +++ b/templates/patterns/pulsar-manager.jsonnet @@ -8,14 +8,16 @@ features: ["pulsar-manager"], args: [ { - name: "default-admin-password", - type: "string", - width: 20, - description: "Admin password to apply", + name: "initial-admin-password", + label: "Password", + type: "text", + width: 40, + description: "Admin password to apply at initialisation", default: "pulsaradmin", required: true, - } - ] + }, + ], + category: [ "foundation" ], }, module: "components/pulsar.jsonnet", } diff --git a/templates/patterns/pulsar.jsonnet b/templates/patterns/pulsar.jsonnet index 75247a12..4b1a1205 100644 --- a/templates/patterns/pulsar.jsonnet +++ b/templates/patterns/pulsar.jsonnet @@ -7,7 +7,8 @@ requires: [], features: ["pulsar"], args: [ - ] + ], + category: [ "foundation" ], }, module: "components/pulsar.jsonnet", } diff --git a/templates/patterns/trustgraph-base.jsonnet b/templates/patterns/trustgraph-base.jsonnet index 6ceafcf0..d129934e 100644 --- a/templates/patterns/trustgraph-base.jsonnet +++ b/templates/patterns/trustgraph-base.jsonnet @@ -7,7 +7,43 @@ requires: ["pulsar"], features: ["trustgraph"], args: [ - ] + { + name: "embeddings-model", + label: "Embeddings model", + type: "select", + description: "Embeddings model for sentence analysis", + options: [ + { id: "all-MiniLM-L6-v2", description: "all-MiniLM-L6-v2" }, + ], + default: "all-MiniLM-L6-v2", + required: true, + }, + { + name: "graph-rag-entity-limit", + label: "GraphRAG entity query limit", + type: "integer", + description: "Limit on entities to fetch from vector store", + default: 50, + required: true, + }, + { + name: "graph-rag-triple-limit", + label: "GraphRAG triple query limit", + type: "integer", + description: "Limit on triples to fetch from triple store", + default: 30, + required: true, + }, + { + name: "graph-rag-max-subgraph-size", + label: "GraphRAG maximum subgraph size", + type: "integer", + description: "Limit on size of subgraph to present to text-completion model", + default: 3000, + required: true, + }, + ], + category: [ "foundation" ], }, module: "components/trustgraph.jsonnet", } diff --git a/templates/patterns/vector-store-milvus.jsonnet b/templates/patterns/vector-store-milvus.jsonnet index 5b0f04c5..edbb4e57 100644 --- a/templates/patterns/vector-store-milvus.jsonnet +++ b/templates/patterns/vector-store-milvus.jsonnet @@ -7,7 +7,9 @@ requires: ["pulsar", "trustgraph"], features: ["milvus", "vectordb"], args: [ - ] + ], + category: [ "vector-store" ], }, module: "components/milvus.jsonnet", } + diff --git a/templates/patterns/vector-store-qdrant.jsonnet b/templates/patterns/vector-store-qdrant.jsonnet index 18ca2b1a..8d949f4b 100644 --- a/templates/patterns/vector-store-qdrant.jsonnet +++ b/templates/patterns/vector-store-qdrant.jsonnet @@ -7,7 +7,8 @@ requires: ["pulsar", "trustgraph"], features: ["qdrant", "vectordb"], args: [ - ] + ], + category: [ "vector-store" ], }, module: "components/qdrant.jsonnet", }