cleanup webapp initialization code

Author: jgbradley1

backend/graphrag_app/main.py

@@ -7,8 +7,7 @@
 from pathlib import Path
 
 import yaml
-
-# from azure.cosmos import PartitionKey, ThroughputProperties
+from azure.cosmos import PartitionKey, ThroughputProperties
 from fastapi import (
     FastAPI,
     Request,
@@ -29,8 +28,7 @@
 from graphrag_app.api.query import query_route
 from graphrag_app.api.source import source_route
 from graphrag_app.logger.load_logger import load_pipeline_logger
-
-# from graphrag_app.utils.azure_clients import AzureClientManager
+from graphrag_app.utils.azure_clients import AzureClientManager
 
 
 async def catch_all_exceptions_middleware(request: Request, call_next):
@@ -48,19 +46,25 @@ async def catch_all_exceptions_middleware(request: Request, call_next):
         return Response("Unexpected internal server error.", status_code=500)
 
 
-# def intialize_cosmosdb_setup():
-#     """Initialise database setup (if necessary) and configure CosmosDB containers that are expected at startup time if they do not exist."""
-#     azure_client_manager = AzureClientManager()
-#     client = azure_client_manager.get_cosmos_client()
-#     db_client = client.create_database_if_not_exists("graphrag")
-#     # create containers with default settings
-#     db_client.create_container_if_not_exists(
-#         id="jobs", partition_key=PartitionKey(path="/id")
-#     )
-#     db_client.create_container_if_not_exists(
-#         id="container-store",
-#         partition_key=PartitionKey(path="/id"),
-#     )
+# NOTE: this function is not currently used, but it is a placeholder for future use once RBAC issues have been resolved
+def intialize_cosmosdb_setup():
+    """Initialise database setup (if necessary) and configure CosmosDB containers that are expected at startup time if they do not exist."""
+    azure_client_manager = AzureClientManager()
+    client = azure_client_manager.get_cosmos_client()
+    throughput = ThroughputProperties(
+        auto_scale_max_throughput=1000, auto_scale_increment_percent=1
+    )
+    db_client = client.create_database_if_not_exists(
+        "graphrag", offer_throughput=throughput
+    )
+    # create containers with default settings
+    db_client.create_container_if_not_exists(
+        id="jobs", partition_key=PartitionKey(path="/id")
+    )
+    db_client.create_container_if_not_exists(
+        id="container-store",
+        partition_key=PartitionKey(path="/id"),
+    )
 
 
 @asynccontextmanager