Logging strategy updates

This commit is contained in:
Cyber MacGeddon 2025-07-30 22:52:03 +01:00
parent 793d2bc77a
commit e0ba70dcf3

View file

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