summaryrefslogtreecommitdiff
path: root/rag/retriever/rerank/abstract.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2024-06-19 02:07:06 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2024-06-19 02:07:06 +0200
commitaac821b148c6c0d35b940609dc9b0ddcb053b28e (patch)
tree5c125045b2b60ead39e093327d664adf43d1d35b /rag/retriever/rerank/abstract.py
parentf2846429310452bebbf0d07203b1e53978c439c7 (diff)
Still wip on rewrite
Diffstat (limited to 'rag/retriever/rerank/abstract.py')
-rw-r--r--rag/retriever/rerank/abstract.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/rag/retriever/rerank/abstract.py b/rag/retriever/rerank/abstract.py
index f32ee77..015a60d 100644
--- a/rag/retriever/rerank/abstract.py
+++ b/rag/retriever/rerank/abstract.py
@@ -1,7 +1,7 @@
from abc import abstractmethod
from typing import List
-from rag.memory import Message
+from rag.message import Message
from rag.retriever.vector import Document