diff options
Diffstat (limited to 'rag/retriever/rerank/abstract.py')
-rw-r--r-- | rag/retriever/rerank/abstract.py | 2 |
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 |