ray-embedding 0.9.3__py3-none-any.whl → 0.9.4__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of ray-embedding might be problematic. Click here for more details.

Files changed (6) hide show
  1. ray_embedding/embedding_model.py +3 -2
  2. {ray_embedding-0.9.3.dist-info → ray_embedding-0.9.4.dist-info}/METADATA +1 -1
  3. ray_embedding-0.9.4.dist-info/RECORD +8 -0
  4. ray_embedding-0.9.3.dist-info/RECORD +0 -8
  5. {ray_embedding-0.9.3.dist-info → ray_embedding-0.9.4.dist-info}/WHEEL +0 -0
  6. {ray_embedding-0.9.3.dist-info → ray_embedding-0.9.4.dist-info}/top_level.txt +0 -0
ray_embedding/embedding_model.py CHANGED
@@ -71,7 +71,9 @@ class EmbeddingModel:
71
71
  )
72
72
  return await self.create_embeddings_batch(request)
73
73
  except Exception as e:
74
- raise HTTPException(status_code=500, detail=str(e))
74
+ err_msg = str(e)
75
+ self.logger.error(err_msg)
76
+ raise HTTPException(status_code=500, detail=err_msg)
75
77
 
76
78
  @serve.batch(max_batch_size=8, batch_wait_timeout_s=0.25)
77
79
  async def create_embeddings_batch(self, requests: List[EmbeddingRequest]) -> List[EmbeddingResponse]:
@@ -96,7 +98,6 @@ class EmbeddingModel:
96
98
  num_inputs = len(request.input)
97
99
  batch_embeddings = embeddings[ix: ix + num_inputs]
98
100
  ix += num_inputs
99
-
100
101
  if truncate_dim is not None:
101
102
  batch_embeddings = batch_embeddings[:, :truncate_dim]
102
103
  batch_embeddings = batch_embeddings / torch.norm(batch_embeddings, dim=1, keepdim=True)
{ray_embedding-0.9.3.dist-info → ray_embedding-0.9.4.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: ray-embedding
3
- Version: 0.9.3
3
+ Version: 0.9.4
4
4
  Summary: Deploy SentenceTransformers embedding models to a ray cluster
5
5
  Author: Crispin Almodovar
6
6
  Author-email: crispin.almodovar@docorto.ai
ray_embedding-0.9.4.dist-info/RECORD ADDED
@@ -0,0 +1,8 @@
1
+ ray_embedding/__init__.py,sha256=OYJT0rVaaGzY613JqgfktsCgroDnBkGOHxR2FE9UtRU,49
2
+ ray_embedding/deploy.py,sha256=YD_udSm13QbFPgSAkCrTQso15DmtIn0QEhErOFNg7jM,1841
3
+ ray_embedding/dto.py,sha256=e91ejZbM_NB9WTjF1YnfuV71cajYIh0vOX8oV_g2OwM,595
4
+ ray_embedding/embedding_model.py,sha256=-34BD2dCVmQdhtymwxJtfT7mXnUO4AmpliV3GTyIEuw,5144
5
+ ray_embedding-0.9.4.dist-info/METADATA,sha256=emy9XYOJHvGU0-fP0Zfe05U9OmwgGe4WS91HmC4ND6E,712
6
+ ray_embedding-0.9.4.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
7
+ ray_embedding-0.9.4.dist-info/top_level.txt,sha256=ziCblpJq1YsrryshFqxTRuRMgNuO1_tgvAAkGShATNA,14
8
+ ray_embedding-0.9.4.dist-info/RECORD,,
ray_embedding-0.9.3.dist-info/RECORD DELETED
@@ -1,8 +0,0 @@
1
- ray_embedding/__init__.py,sha256=OYJT0rVaaGzY613JqgfktsCgroDnBkGOHxR2FE9UtRU,49
2
- ray_embedding/deploy.py,sha256=YD_udSm13QbFPgSAkCrTQso15DmtIn0QEhErOFNg7jM,1841
3
- ray_embedding/dto.py,sha256=e91ejZbM_NB9WTjF1YnfuV71cajYIh0vOX8oV_g2OwM,595
4
- ray_embedding/embedding_model.py,sha256=dYuf5dqeHqWx1s0pc_Cy8jm33BrTli2j6op_b1HcWcA,5075
5
- ray_embedding-0.9.3.dist-info/METADATA,sha256=Dr94gTWvkUAHMG6fOy1posRct0y1IHDFYGMjoks3w1Q,712
6
- ray_embedding-0.9.3.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
7
- ray_embedding-0.9.3.dist-info/top_level.txt,sha256=ziCblpJq1YsrryshFqxTRuRMgNuO1_tgvAAkGShATNA,14
8
- ray_embedding-0.9.3.dist-info/RECORD,,