From b1ff0c55422d7b0af2c379679b8721014ef36926 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Tue, 18 Jun 2024 01:37:32 +0200
Subject: Wip rewrite

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

(limited to 'rag/retriever/memory.py')

diff --git a/rag/retriever/memory.py b/rag/retriever/memory.py
deleted file mode 100644
index c4455ed..0000000
--- a/rag/retriever/memory.py
+++ /dev/null
@@ -1,51 +0,0 @@
-from dataclasses import dataclass
-from typing import Dict, List
-
-
-@dataclass
-class Log:
-    user: Message
-    bot: Message
-
-    def get():
-        return (user, bot)
-
-
-@dataclass
-class Message:
-    role: str
-    message: str
-
-    def as_dict(self, model: str) -> Dict[str, str]:
-        if model == "cohere":
-            match self.role:
-                case "user":
-                    role = "USER"
-                case _:
-                    role = "CHATBOT"
-
-            return {"role": role, "message": self.message}
-        else:
-            return {"role": self.role, "content": self.message}
-
-
-class Memory:
-    def __init__(self, reranker) -> None:
-        self.history = []
-        self.reranker = reranker
-        self.user = "user"
-        self.bot = "assistant"
-
-    def add(self, prompt: str, response: str):
-        self.history.append(
-            Log(
-                user=Message(role=self.user, message=prompt),
-                bot=Message(role=self.bot, message=response),
-            )
-        )
-
-    def get(self) -> List[Log]:
-        return [m.as_dict() for log in self.history for m in log.get()]
-
-    def reset(self):
-        self.history = []
-- 
cgit v1.2.3-70-g09d2