From e0ba70dcf3c62b7ccef408e84a80a03bf3cd3a87 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Wed, 30 Jul 2025 22:52:03 +0100 Subject: [PATCH] Logging strategy updates --- trustgraph-embeddings-hf/trustgraph/embeddings/hf/hf.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/trustgraph-embeddings-hf/trustgraph/embeddings/hf/hf.py b/trustgraph-embeddings-hf/trustgraph/embeddings/hf/hf.py index 0ab3cef9..f1abbfae 100755 --- a/trustgraph-embeddings-hf/trustgraph/embeddings/hf/hf.py +++ b/trustgraph-embeddings-hf/trustgraph/embeddings/hf/hf.py @@ -4,10 +4,14 @@ Embeddings service, applies an embeddings model selected from HuggingFace. Input is text, output is embeddings vector. """ +import logging from ... base import EmbeddingsService from langchain_huggingface import HuggingFaceEmbeddings +# Module logger +logger = logging.getLogger(__name__) + default_ident = "embeddings" default_model="all-MiniLM-L6-v2" @@ -22,13 +26,13 @@ class Processor(EmbeddingsService): **params | { "model": model } ) - print("Get model...", flush=True) + logger.info(f"Loading HuggingFace embeddings model: {model}") self.embeddings = HuggingFaceEmbeddings(model_name=model) async def on_embeddings(self, text): embeds = self.embeddings.embed_documents([text]) - print("Done.", flush=True) + logger.debug("Embeddings generation complete") return embeds @staticmethod