Update retriever.py

retriever.py

@@ -6,8 +6,9 @@ import os  # Added for environment variables
 embeddings = SentenceTransformerEmbeddings(model_name="NeuML/pubmedbert-base-embeddings")
 
 # Use environment variables for cloud configuration
+
 client = QdrantClient(
-    url=os.getenv("QDRANT_URL", "https://YOUR_QDRANT_CLUSTER_URL.aws.cloud.qdrant.io"),
+    url=os.getenv("QDRANT_URL", "https://QDRANT_URL.europe-west3-0.gcp.cloud.qdrant.io"),
     api_key=os.getenv("QDRANT_API_KEY"),
     prefer_grpc=False
 )
@@ -23,10 +24,10 @@ print("######")
 query = "What is Metastatic disease?"
 
 # Updated similarity search (newer LangChain versions)
-docs = db.similarity_search_with_relevance_scores(query=query, k=3)
+docs = db.similarity_search_with_relevance_scores(query=query, k=1)
 for doc, score in docs:
     print({
         "score": score,
         "content": doc.page_content,
         "metadata": doc.metadata
-    })
+    })