diff --git a/semantic_router/index/qdrant.py b/semantic_router/index/qdrant.py index 8c739d7d..e1339afc 100644 --- a/semantic_router/index/qdrant.py +++ b/semantic_router/index/qdrant.py @@ -222,7 +222,7 @@ def query(self, vector: np.ndarray, top_k: int = 5) -> Tuple[np.ndarray, List[st self.index_name, query_vector=vector, limit=top_k, with_payload=True ) scores = [result.score for result in results] - route_names = [result.payload["sr_route"] for result in results] + route_names = [result.payload[SR_ROUTE_PAYLOAD_KEY] for result in results] return np.array(scores), route_names def delete_index(self):