From 3f447bff69c20109474c455f1ad52bd547ab49e9 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Tue, 9 Apr 2024 00:41:55 +0200
Subject: Update

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

(limited to 'rag/rag.py')

diff --git a/rag/rag.py b/rag/rag.py
deleted file mode 100644
index c95d93a..0000000
--- a/rag/rag.py
+++ /dev/null
@@ -1,52 +0,0 @@
-from dataclasses import dataclass
-from io import BytesIO
-from pathlib import Path
-from typing import List, Optional, Type
-
-from dotenv import load_dotenv
-from loguru import logger as log
-
-
-try:
-    from rag.retriever.vector import Document
-    from rag.generator.abstract import AbstractGenerator
-    from rag.retriever.retriever import Retriever
-    from rag.generator.prompt import Prompt
-except ModuleNotFoundError:
-    from retriever.vector import Document
-    from generator.abstract import AbstractGenerator
-    from retriever.retriever import Retriever
-    from generator.prompt import Prompt
-
-
-@dataclass
-class Response:
-    query: str
-    context: List[str]
-    answer: str
-
-
-class RAG:
-    def __init__(self) -> None:
-        # FIXME: load this somewhere else?
-        load_dotenv()
-        self.retriever = Retriever()
-
-    def add_pdf(
-        self,
-        path: Optional[Path],
-        blob: Optional[BytesIO],
-        source: Optional[str] = None,
-    ):
-        if path:
-            self.retriever.add_pdf_from_path(path)
-        elif blob:
-            self.retriever.add_pdf_from_blob(blob, source)
-        else:
-            log.error("Both path and blob was None, no pdf added!")
-
-    def retrieve(self, query: str, limit: int = 5) -> List[Document]:
-        return self.retriever.retrieve(query, limit)
-
-    def generate(self, generator: Type[AbstractGenerator], prompt: Prompt):
-        yield from generator.generate(prompt)
-- 
cgit v1.2.3-70-g09d2