Skip to content

Commit 676d335

Browse files
committed
fix: move MilvusSingleton import to the correct location in agent_retriever.py
1 parent 11f8512 commit 676d335

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/chatbot/agents/utils/agent_retriever.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
import os
22
from typing import List, NamedTuple
33

4-
from src.chatbot.db.milvus_client import MilvusSingleton
54
from src.chatbot.db.ragflow_client import RAGFlowSingleton
65
from src.chatbot.embeddings.main import get_embeddings
76
from src.chatbot_log.chatbot_logger import logger
@@ -44,6 +43,8 @@ def retrieve_from_infinity_ragflow(collection_name: str, query: str):
4443

4544
def retrieve_from_milvus(collection_name: str, query: str, doc_search_params: dict):
4645

46+
from src.chatbot.db.milvus_client import MilvusSingleton
47+
4748
search_params = {
4849
"metric_type": "L2",
4950
"offset": 0,

0 commit comments

Comments
 (0)