diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-05 01:58:29 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-05 01:58:29 +0200 |
| commit | f351e83badfd621decc12b91f5d1610b4b3b5776 (patch) | |
| tree | e0ab3ee9725093d90bfe3c0f7e21648ab3091d12 /rag/db/embeddings.py | |
| parent | 07741e09f4569d27aca3ffa111178bc324af1eab (diff) | |
Fix .env load
Diffstat (limited to 'rag/db/embeddings.py')
| -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 |