trustgraph-embeddings-hf 1.2.3__tar.gz → 1.2.4__tar.gz

{trustgraph_embeddings_hf-1.2.3 → trustgraph_embeddings_hf-1.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-embeddings-hf
-Version: 1.2.3
+Version: 1.2.4
 Summary: HuggingFace embeddings support for TrustGraph.
 Author-email: "trustgraph.ai" <security@trustgraph.ai>
 Project-URL: Homepage, https://github.com/trustgraph-ai/trustgraph

{trustgraph_embeddings_hf-1.2.3 → trustgraph_embeddings_hf-1.2.4}/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

trustgraph_embeddings_hf-1.2.4/trustgraph/embeddings_hf_version.py

@@ -0,0 +1 @@
+__version__ = "1.2.4"

{trustgraph_embeddings_hf-1.2.3 → trustgraph_embeddings_hf-1.2.4}/trustgraph_embeddings_hf.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-embeddings-hf
-Version: 1.2.3
+Version: 1.2.4
 Summary: HuggingFace embeddings support for TrustGraph.
 Author-email: "trustgraph.ai" <security@trustgraph.ai>
 Project-URL: Homepage, https://github.com/trustgraph-ai/trustgraph

trustgraph_embeddings_hf-1.2.3/trustgraph/embeddings_hf_version.py

@@ -1 +0,0 @@
-__version__ = "1.2.3"