Rename llm

3 files changed, 4 insertions, 5 deletions

diff --git a/rag/llms/__init__.py b/rag/llm/__init__.py
index e69de29..e69de29 100644
--- a/rag/llms/__init__.py
+++ b/rag/llm/__init__.py
diff --git a/rag/llms/encoder.py b/rag/llm/encoder.py
index 758b523..d5e0566 100644
--- a/rag/llms/encoder.py
+++ b/rag/llm/encoder.py
@@ -1,13 +1,13 @@
+import os
 from typing import List
-import ollama
-import numpy as np
 
-# FIXME: .env
-ENCODER_MODEL = "mxbai-embed-large"
+import numpy as np
+import ollama
 
 
 class Encoder:
     def __init__(self) -> None:
+        self.model = os.environ["ENCODER_MODEL"]
         self.query_prompt = "Represent this sentence for searching relevant passages: "
 
     def __encode(self, prompt: str) -> np.ndarray:
@@ -21,4 +21,3 @@ class Encoder:
     def query(self, query: str) -> np.ndarray:
         query = self.query_prompt + query
         return self.__encode(query)
-
diff --git a/rag/llms/generator.py b/rag/llm/generator.py
index e69de29..e69de29 100644
--- a/rag/llms/generator.py
+++ b/rag/llm/generator.py