trustgraph/templates/patterns/embeddings-hf.jsonnet
cybermaggedon 208c219962
Template rejig (#48)
* document-rag / graph-rag refactor of templates

* Tweaking the docs and categories

* Clarify triple store vs RAG

* Tweak knowledge graph linkage

* Doc embedding for Qdrant

* Fix document RAG on Qdrant

* Fix templates

* Bump version

* Updated templates
2024-09-03 00:09:15 +01:00

26 lines
875 B
Jsonnet

{
pattern: {
name: "embeddings-hf",
icon: "🤖💬",
title: "Add embeddings model which uses HuggingFace models",
description: "This pattern integrates an embeddings model based on HuggingFace sentence-transformer library.",
requires: ["pulsar", "trustgraph"],
features: ["llm"],
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" },
{ id: "mixedbread-ai/mxbai-embed-large-v1", description: "mxbai-embed-large-v1" },
],
default: "all-MiniLM-L6-v2",
required: true,
},
],
category: [ "embeddings" ],
},
module: "components/embeddings-hf.jsonnet",
}