From f351e83badfd621decc12b91f5d1610b4b3b5776 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Fri, 5 Apr 2024 01:58:29 +0200
Subject: Fix .env load

---
 rag/db/embeddings.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'rag')

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
 
-- 
cgit v1.2.3-70-g09d2