diff options
Diffstat (limited to 'rag/db')
| -rw-r--r-- | rag/db/embeddings.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/rag/db/embeddings.py b/rag/db/embeddings.py index f9d0735..1a97d16 100644 --- a/rag/db/embeddings.py +++ b/rag/db/embeddings.py @@ -8,7 +8,7 @@ import numpy as np # TODO: inner product distance metric? class Embeddings: def __init__(self): - self.dim = os.environ["EMBEDDING_DIM"] + self.dim = int(os.environ["EMBEDDING_DIM"]) self.index = faiss.IndexFlatL2(self.dim) # TODO: load from file |