dict -> Dict

fkodom · Dec 1, 2023 · b388618 · b388618
1 parent 865e2d8
commit b388618
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/document_rag/vector_db/qdrant.py b/document_rag/vector_db/qdrant.py
@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import os
-from typing import Any, List, Sequence, Tuple, cast
+from typing import Any, Dict, List, Sequence, Tuple, cast
 
 from qdrant_client import QdrantClient
 from typing_extensions import Self
@@ -32,7 +32,7 @@ def add_documents(self, documents: Sequence[Tuple[str, TextMetadata]]) -> None:
         self.client.add(
             collection_name=COLLECTION_NAME,
             documents=_documents,
-            metadata=cast(List[dict[str, Any]], metadata),  # For mypy
+            metadata=cast(List[Dict[str, Any]], metadata),  # For mypy
         )
 
     def search(self, query: str, limit: int = 10) -> List[SearchResult]: