apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-05 19:32:11 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Embeddings q

Browse source
This commit is contained in:
Cyber MacGeddon 2024-07-23 20:38:44 +01:00
parent d8e32aee82
commit d712b55e89
3 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
trustgraph/embeddings/hf/hf.py
View file

@ -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"

5
trustgraph/embeddings/ollama/processor.py
View file

@ -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'

3
trustgraph/schema.py
View file

@ -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'