summaryrefslogtreecommitdiff
path: root/rag
diff options
context:
space:
mode:
Diffstat (limited to 'rag')
-rw-r--r--rag/retriever/encoder.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/rag/retriever/encoder.py b/rag/retriever/encoder.py
index 68d0e26..6176a37 100644
--- a/rag/retriever/encoder.py
+++ b/rag/retriever/encoder.py
@@ -7,6 +7,7 @@ import ollama
from langchain_core.documents import Document
from loguru import logger as log
from qdrant_client.http.models import StrictFloat
+from tqdm import tqdm
from .vector import Point
@@ -34,7 +35,7 @@ class Encoder:
"source": self.__get_source(chunk.metadata),
},
)
- for chunk in chunks
+ for chunk in tqdm(chunks)
]
def encode_query(self, query: str) -> List[StrictFloat]: