From d712b55e8929e1306028e4b2f975e802942ae821 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Tue, 23 Jul 2024 20:38:44 +0100 Subject: [PATCH] Embeddings q --- trustgraph/embeddings/hf/hf.py | 5 +++-- trustgraph/embeddings/ollama/processor.py | 5 +++-- trustgraph/schema.py | 3 +++ 3 files changed, 9 insertions(+), 4 deletions(-) diff --git a/trustgraph/embeddings/hf/hf.py b/trustgraph/embeddings/hf/hf.py index 24f536ab..20e57d53 100755 --- a/trustgraph/embeddings/hf/hf.py +++ b/trustgraph/embeddings/hf/hf.py @@ -7,13 +7,14 @@ Input is text, output is embeddings vector. from langchain_huggingface import HuggingFaceEmbeddings from ... schema import EmbeddingsRequest, EmbeddingsResponse +from ... schema import embeddings_request_queue, embeddings_response_queue from ... log_level import LogLevel from ... base import ConsumerProducer module = ".".join(__name__.split(".")[1:-1]) -default_input_queue = 'embeddings' -default_output_queue = 'embeddings-response' +default_input_queue = embeddings_request_queue +default_output_queue = embeddings_response_queue default_subscriber = module default_model="all-MiniLM-L6-v2" diff --git a/trustgraph/embeddings/ollama/processor.py b/trustgraph/embeddings/ollama/processor.py index ff6a4c59..6682a79f 100755 --- a/trustgraph/embeddings/ollama/processor.py +++ b/trustgraph/embeddings/ollama/processor.py @@ -6,13 +6,14 @@ Input is text, output is embeddings vector. from langchain_community.embeddings import OllamaEmbeddings from ... schema import EmbeddingsRequest, EmbeddingsResponse +from ... schema import embeddings_request_queue, embeddings_response_queue from ... log_level import LogLevel from ... base import ConsumerProducer module = ".".join(__name__.split(".")[1:-1]) -default_input_queue = 'embeddings' -default_output_queue = 'embeddings-response' +default_input_queue = embeddings_request_queue +default_output_queue = embeddings_response_queue default_subscriber = module default_model="mxbai-embed-large" default_ollama = 'http://localhost:11434' diff --git a/trustgraph/schema.py b/trustgraph/schema.py index 190f55c1..fac85fff 100644 --- a/trustgraph/schema.py +++ b/trustgraph/schema.py @@ -79,3 +79,6 @@ class GraphRagResponse(Record): graph_rag_request_queue = 'graph-rag' graph_rag_response_queue = 'graph-rag-response' +embeddings_request_queue = 'embeddings' +embeddings_response_queue = 'embeddings-response' +