ray-embedding 0.9.3__py3-none-any.whl → 0.9.5__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.
Files changed (6) hide show
  1. ray_embedding/embedding_model.py +1 -1
  2. {ray_embedding-0.9.3.dist-info → ray_embedding-0.9.5.dist-info}/METADATA +1 -1
  3. ray_embedding-0.9.5.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.5.dist-info}/WHEEL +0 -0
  6. {ray_embedding-0.9.3.dist-info → ray_embedding-0.9.5.dist-info}/top_level.txt +0 -0
ray_embedding/embedding_model.py CHANGED
@@ -71,6 +71,7 @@ class EmbeddingModel:
71
71
  )
72
72
  return await self.create_embeddings_batch(request)
73
73
  except Exception as e:
74
+ self.logger.error(e)
74
75
  raise HTTPException(status_code=500, detail=str(e))
75
76
 
76
77
  @serve.batch(max_batch_size=8, batch_wait_timeout_s=0.25)
@@ -96,7 +97,6 @@ class EmbeddingModel:
96
97
  num_inputs = len(request.input)
97
98
  batch_embeddings = embeddings[ix: ix + num_inputs]
98
99
  ix += num_inputs
99
-
100
100
  if truncate_dim is not None:
101
101
  batch_embeddings = batch_embeddings[:, :truncate_dim]
102
102
  batch_embeddings = batch_embeddings / torch.norm(batch_embeddings, dim=1, keepdim=True)
{ray_embedding-0.9.3.dist-info → ray_embedding-0.9.5.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.5
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.5.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=xV-yK24okharTuSMJPCEe1hM29KdTu6Uozqu01bSulg,5107
5
+ ray_embedding-0.9.5.dist-info/METADATA,sha256=pJtyTgr9BQMq4gF7yyOrgCUQsmjTU4sKtgBPaeTJKqk,712
6
+ ray_embedding-0.9.5.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
7
+ ray_embedding-0.9.5.dist-info/top_level.txt,sha256=ziCblpJq1YsrryshFqxTRuRMgNuO1_tgvAAkGShATNA,14
8
+ ray_embedding-0.9.5.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,,