diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-05-29 00:53:39 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-05-29 00:53:39 +0200 |
commit | 716e3fe58adee5b8a6bfa91de4b3ba6cf204d172 (patch) | |
tree | 778da9011d21051006fc206ce0978f0fc114b77b /rag/ui.py | |
parent | 2d91c118d71a8dd7fbd7f9cf21f86e92da33827e (diff) |
Wip memory
Diffstat (limited to 'rag/ui.py')
-rw-r--r-- | rag/ui.py | 14 |
1 files changed, 0 insertions, 14 deletions
@@ -13,18 +13,6 @@ from rag.retriever.retriever import Retriever from rag.retriever.vector import Document -@dataclass -class Message: - role: str - message: str - - def as_dict(self, model: str) -> Dict[str, str]: - if model == "cohere": - return {"role": self.role, "message": self.message} - else: - return {"role": self.role, "content": self.message} - - def set_chat_users(): log.debug("Setting user and bot value") ss = st.session_state @@ -38,13 +26,11 @@ def load_retriever(): st.session_state.retriever = Retriever() -# @st.cache_resource def load_generator(model: str): log.debug("Loading generator model") st.session_state.generator = get_generator(model) -# @st.cache_resource def load_reranker(model: str): log.debug("Loading reranker model") st.session_state.reranker = get_reranker(model) |