internomega-terrablue commited on
Commit
0b6760c
·
1 Parent(s): 44b27f4

ingestion split

Browse files
ingestion_engine/embedding_generator.py CHANGED
@@ -30,7 +30,7 @@ def generate(texts: list[str]) -> list[list[float]]:
30
  all_embeddings.extend(result)
31
 
32
  logger.info("Generated %d embeddings via HF Inference API", len(all_embeddings))
33
- return [list(emb) for emb in all_embeddings]
34
 
35
 
36
  def generate_query(query: str) -> list[float]:
 
30
  all_embeddings.extend(result)
31
 
32
  logger.info("Generated %d embeddings via HF Inference API", len(all_embeddings))
33
+ return [[float(x) for x in emb] for emb in all_embeddings]
34
 
35
 
36
  def generate_query(query: str) -> list[float]: