ray-embedding 0.11.2__py3-none-any.whl → 0.11.4__py3-none-any.whl

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/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.dist-info → ray_embedding-0.11.4.dist-info}/METADATA

@@ -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.4.dist-info/RECORD

@@ -0,0 +1,9 @@
+ray_embedding/__init__.py,sha256=YS5LAZfRIwwVvE3C9g7hsauvjgIkqKtHyxkwMFFfAGY,46
+ray_embedding/deploy.py,sha256=YRTh6jj6fbJhS3NQg34YbE59zwzFwp_39fDF71H5Ixw,2190
+ray_embedding/dto.py,sha256=QlduDoqkFHaeF_KgsFeUKq2XWiPMmrgRPy_QjCTSCRE,1399
+ray_embedding/embedding_model.py,sha256=9QPbvcRYrRF5Wi67r2htqPDSL1zh2KMZiH1UVAYsjb8,3387
+ray_embedding/embedding_service.py,sha256=ixqyCSdCy1ykRifMf0c3JBeLBqhQAmhUEALyBleq1fg,3852
+ray_embedding-0.11.4.dist-info/METADATA,sha256=eNawM5DjPYTE9WIQpa6sw5J5e7qLP6fDy5svivyIs34,1094
+ray_embedding-0.11.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ray_embedding-0.11.4.dist-info/top_level.txt,sha256=ziCblpJq1YsrryshFqxTRuRMgNuO1_tgvAAkGShATNA,14
+ray_embedding-0.11.4.dist-info/RECORD,,

ray_embedding-0.11.2.dist-info/RECORD

@@ -1,9 +0,0 @@
-ray_embedding/__init__.py,sha256=YS5LAZfRIwwVvE3C9g7hsauvjgIkqKtHyxkwMFFfAGY,46
-ray_embedding/deploy.py,sha256=YRTh6jj6fbJhS3NQg34YbE59zwzFwp_39fDF71H5Ixw,2190
-ray_embedding/dto.py,sha256=QlduDoqkFHaeF_KgsFeUKq2XWiPMmrgRPy_QjCTSCRE,1399
-ray_embedding/embedding_model.py,sha256=d_gqqcKK3B2nF8qMDk7NKW8gHBbtPwvPFArKTBKNdFA,3372
-ray_embedding/embedding_service.py,sha256=OFy1yWyWodbSuSfz4r9LZIbyRo4ngQqUvMp0tnIqQwo,3722
-ray_embedding-0.11.2.dist-info/METADATA,sha256=C9E1lps5BrKDOvPRk2c_12u4Z_yYfq4TDMA9FD0wfVI,1094
-ray_embedding-0.11.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ray_embedding-0.11.2.dist-info/top_level.txt,sha256=ziCblpJq1YsrryshFqxTRuRMgNuO1_tgvAAkGShATNA,14
-ray_embedding-0.11.2.dist-info/RECORD,,