Sarp Bilgiç commited on
Commit
76c081e
·
1 Parent(s): 3525995

sparse tuple fix

Browse files
src/clients/sparse_embedding_client.py CHANGED
@@ -11,9 +11,11 @@ class SparseEmbeddingClient:
11
  )
12
  logging.info("Sparse embedding model loaded successfully")
13
 
14
- def embed_documents(self, documents: List[str]) -> List[Tuple[List[int], List[float]]]:
15
  embeddings = list(self.model.embed(documents))
16
- return [(emb.indices.tolist(), emb.values.tolist()) for emb in embeddings]
 
 
17
 
18
  def embed_query(self, query: str) -> Tuple[List[int], List[float]]:
19
  embedding = list(self.model.query_embed(query))[0]
 
11
  )
12
  logging.info("Sparse embedding model loaded successfully")
13
 
14
+ def embed_documents(self, documents: List[str]) -> Tuple[List[List[int]], List[List[float]]]:
15
  embeddings = list(self.model.embed(documents))
16
+ all_indices = [emb.indices.tolist() for emb in embeddings]
17
+ all_values = [emb.values.tolist() for emb in embeddings]
18
+ return (all_indices, all_values)
19
 
20
  def embed_query(self, query: str) -> Tuple[List[int], List[float]]:
21
  embedding = list(self.model.query_embed(query))[0]