diff --git a/main.py b/main.py
index b55280c..91987a6 100644
--- a/main.py
+++ b/main.py
@@ -11,7 +11,8 @@ from langchain_core.tools import tool
 from langchain_ollama import ChatOllama, OllamaEmbeddings
 from langchain_qdrant import QdrantVectorStore
 from langchain_text_splitters import RecursiveCharacterTextSplitter
-from qdrant_client import QdrantClient, Distance, VectorParams
+from qdrant_client import QdrantClient
+from qdrant_client.models import Distance, VectorParams
 
 QDRANT_URL = os.getenv("QDRANT_URL", "http://localhost:6333")
 QDRANT_COLLECTION = os.getenv("QDRANT_COLLECTION", "rag_memory")