ray-embedding 0.11.3__py3-none-any.whl → 0.11.5__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),
@@ -29,6 +30,7 @@ class EmbeddingService:
     async def _compute_embeddings_from_resized_batches(self, model: str, inputs: List[str], dimensions: Optional[int] = None):
         assert model in self.served_models
         model_handle = self.served_models[model].deployment_handle
+        self.logger.info(f"Model handle: {model_handle} ")
         batch_size = self.served_models[model].batch_size
         num_retries = self.served_models[model].num_retries
 

{ray_embedding-0.11.3.dist-info → ray_embedding-0.11.5.dist-info}/METADATA

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

ray_embedding-0.11.5.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=yjPKW1qZ1JWyox2ociIutTL_hNJadtZW2DbkbniS9CM,3911
+ray_embedding-0.11.5.dist-info/METADATA,sha256=3QYPjE8inGEe-QVdqDNF9GLuXoJkBRSTgeLo-u9tiuA,1094
+ray_embedding-0.11.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ray_embedding-0.11.5.dist-info/top_level.txt,sha256=ziCblpJq1YsrryshFqxTRuRMgNuO1_tgvAAkGShATNA,14
+ray_embedding-0.11.5.dist-info/RECORD,,

ray_embedding-0.11.3.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=Dn--DW1bK1xKzkn3qUOFNKq2FKLqX55v8_DoeqivzvU,3804
-ray_embedding-0.11.3.dist-info/METADATA,sha256=5QanHoPTnrq6aADrQmoI3Jm_AoG5jzj-bvlYr_U2NNg,1094
-ray_embedding-0.11.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ray_embedding-0.11.3.dist-info/top_level.txt,sha256=ziCblpJq1YsrryshFqxTRuRMgNuO1_tgvAAkGShATNA,14
-ray_embedding-0.11.3.dist-info/RECORD,,