Address mypy issues

Author: pamelafox

app/backend/chat_history/cosmosdb.py

@@ -187,6 +187,6 @@ async def setup_clients():
 
 @chat_history_cosmosdb_bp.after_app_serving
 async def close_clients():
-    cosmos_client: CosmosClient = current_app.config.get(CONFIG_COSMOS_HISTORY_CLIENT)
-    if cosmos_client:
+    if current_app.config.get(CONFIG_COSMOS_HISTORY_CLIENT):
+        cosmos_client: CosmosClient = current_app.config[CONFIG_COSMOS_HISTORY_CLIENT]
         await cosmos_client.close()

app/backend/decorators.py

@@ -1,6 +1,6 @@
 import logging
 from functools import wraps
-from typing import Any, Callable, Dict
+from typing import Any, Callable, Dict, TypeVar, cast
 
 from quart import abort, current_app, request
 
@@ -37,7 +37,10 @@ async def auth_handler(path=""):
     return auth_handler
 
 
-def authenticated(route_fn: Callable[[Dict[str, Any]], Any]):
+_C = TypeVar("_C", bound=Callable[..., Any])
+
+
+def authenticated(route_fn: _C) -> _C:
     """
     Decorator for routes that might require access control. Unpacks Authorization header information into an auth_claims dictionary
     """
@@ -52,4 +55,4 @@ async def auth_handler(*args, **kwargs):
 
         return await route_fn(auth_claims, *args, **kwargs)
 
-    return auth_handler
+    return cast(_C, auth_handler)