Spaces:
Running
Running
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 [
|
| 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]:
|