[GDrive] Fix mypy error for retrieve_file_texts return type

cohere-ai · Sep 12, 2024 · 9b93785 · 9b93785
1 parent b0cd8c7
commit 9b93785
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gdrive/provider/provider.py b/gdrive/provider/provider.py
@@ -29,7 +29,7 @@
 
 
 def process_data_with_service(data, request_credentials: Credentials):
-    results = []
+    results: list[dict] = []
     files = data.get("files", [])
     if not files:
         logger.debug("No files found.")
@@ -66,7 +66,7 @@ def process_data_with_service(data, request_credentials: Credentials):
     return results
 
 
-def retrieve_file_texts(files, request_credentials: Credentials) -> [str, str]:
+def retrieve_file_texts(files, request_credentials: Credentials) -> dict[str, str]:
     missing_ids_to_urls = {}
     id_to_texts = {}