From d487ef8b04cc7f5ac1491f0638f902fe2abe5ac5 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Mon, 8 Apr 2024 22:28:47 +0200
Subject: Wip refactor

---
 rag/llm/generator.py | 33 ---------------------------------
 1 file changed, 33 deletions(-)
 delete mode 100644 rag/llm/generator.py

(limited to 'rag/llm/generator.py')

diff --git a/rag/llm/generator.py b/rag/llm/generator.py
deleted file mode 100644
index 8c7702f..0000000
--- a/rag/llm/generator.py
+++ /dev/null
@@ -1,33 +0,0 @@
-import os
-from dataclasses import dataclass
-
-import ollama
-from loguru import logger as log
-
-
-@dataclass
-class Prompt:
-    query: str
-    context: str
-
-
-class Generator:
-    def __init__(self) -> None:
-        self.model = os.environ["GENERATOR_MODEL"]
-
-    def __metaprompt(self, prompt: Prompt) -> str:
-        metaprompt = (
-            "Answer the following question using the provided context.\n"
-            "If you can't find the answer, do not pretend you know it,"
-            'but answer "I don\'t know".\n\n'
-            f"Question: {prompt.query.strip()}\n\n"
-            "Context:\n"
-            f"{prompt.context.strip()}\n\n"
-            "Answer:\n"
-        )
-        return metaprompt
-
-    def generate(self, prompt: Prompt) -> str:
-        log.debug("Generating answer...")
-        metaprompt = self.__metaprompt(prompt)
-        return ollama.generate(model=self.model, prompt=metaprompt)
-- 
cgit v1.2.3-70-g09d2