Spaces:
Runtime error
Runtime error
File size: 559 Bytes
8a58cf3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
"""Embedding utils for LlamaIndex."""
from typing import List
def save_embedding(embedding: List[float], file_path: str) -> None:
"""Save embedding to file."""
with open(file_path, "w") as f:
f.write(",".join([str(x) for x in embedding]))
def load_embedding(file_path: str) -> List[float]:
"""Load embedding from file. Will only return first embedding in file."""
with open(file_path, "r") as f:
for line in f:
embedding = [float(x) for x in line.strip().split(",")]
break
return embedding
|