Rename llm

1 files changed, 0 insertions, 24 deletions

diff --git a/rag/llms/encoder.py b/rag/llms/encoder.py
deleted file mode 100644
index 758b523..0000000
--- a/rag/llms/encoder.py
+++ /dev/null
@@ -1,24 +0,0 @@
-from typing import List
-import ollama
-import numpy as np
-
-# FIXME: .env
-ENCODER_MODEL = "mxbai-embed-large"
-
-
-class Encoder:
-    def __init__(self) -> None:
-        self.query_prompt = "Represent this sentence for searching relevant passages: "
-
-    def __encode(self, prompt: str) -> np.ndarray:
-        x = ollama.embeddings(model=ENCODER_MODEL, prompt=prompt)
-        x = np.array([x["embedding"]]).astype("float32")
-        return x
-
-    def encode(self, doc: List[str]) -> List[np.ndarray]:
-        return [self.__encode(chunk) for chunk in doc]
-
-    def query(self, query: str) -> np.ndarray:
-        query = self.query_prompt + query
-        return self.__encode(query)
-