ray-embedding 0.11.2__tar.gz → 0.11.4__tar.gz

{ray_embedding-0.11.2 → ray_embedding-0.11.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ray-embedding
-Version: 0.11.2
+Version: 0.11.4
 Summary: Deploy SentenceTransformers embedding models to a ray cluster
 Author: Crispin Almodovar
 Author-email: 

{ray_embedding-0.11.2 → ray_embedding-0.11.4}/ray_embedding/embedding_model.py

@@ -16,7 +16,7 @@ class EmbeddingModel:
                  matryoshka_dim: Optional[int] = None, trust_remote_code: Optional[bool] = False,
                  model_kwargs: Dict[str, Any] = None):
         logging.basicConfig(level=logging.INFO)
-        self.logger = logging.getLogger(__name__)
+        self.logger = logging.getLogger(self.__class__.__name__)
         self.model = model
         self.init_device = device
         if self.init_device is None or self.init_device == "auto":

{ray_embedding-0.11.2 → ray_embedding-0.11.4}/ray_embedding/embedding_service.py

@@ -14,8 +14,9 @@ web_api = FastAPI(title="Ray Embeddings - OpenAI-compatible API")
 @serve.ingress(web_api)
 class EmbeddingService:
     def __init__(self, served_models: Dict[str, DeployedModel]):
-        self.logger = logging.getLogger(self.__class__.__name__)
         assert served_models, "models cannot be empty"
+        logging.basicConfig(level=logging.INFO)
+        self.logger = logging.getLogger(self.__class__.__name__)
         self.served_models = served_models
         self.available_models = [
             {"id": str(item),
@@ -65,6 +66,7 @@ class EmbeddingService:
     async def compute_embeddings(self, request: EmbeddingRequest):
         try:
             inputs = request.input if isinstance(request.input, list) else [request.input]
+            self.logger.info(f"Received input of size {len(inputs)} text chunks")
             embeddings = await self._compute_embeddings_from_resized_batches(request.model, inputs, request.dimensions)
             response_data = [
                 {"index": idx, "embedding": emb}

{ray_embedding-0.11.2 → ray_embedding-0.11.4}/ray_embedding.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ray-embedding
-Version: 0.11.2
+Version: 0.11.4
 Summary: Deploy SentenceTransformers embedding models to a ray cluster
 Author: Crispin Almodovar
 Author-email: 

{ray_embedding-0.11.2 → ray_embedding-0.11.4}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = ray-embedding
-version = 0.11.2
+version = 0.11.4
 author = Crispin Almodovar
 author_email = 
 description = Deploy SentenceTransformers embedding models to a ray cluster