Use AKS_NAMESPACE env var when creating indexing manager (#145)

Co-authored-by: Josh Bradley <[email protected]>

Author: clarked-msft

backend/src/api/index.py

@@ -127,7 +127,7 @@ async def setup_indexing_pipeline(
             )
         # if indexing job is in a failed state, delete the associated K8s job and pod to allow for a new job to be scheduled
         if PipelineJobState(existing_job.status) == PipelineJobState.FAILED:
-            _delete_k8s_job(f"indexing-job-{sanitized_index_name}", "graphrag")
+            _delete_k8s_job(f"indexing-job-{sanitized_index_name}", os.environ["AKS_NAMESPACE"])
         # reset the pipeline job details
         existing_job._status = PipelineJobState.SCHEDULED
         existing_job._percent_complete = 0

backend/src/main.py

@@ -68,11 +68,11 @@ async def lifespan(app: FastAPI):
         ] = pod.spec.service_account_name
         # retrieve list of existing cronjobs
         batch_v1 = client.BatchV1Api()
-        namespace_cronjobs = batch_v1.list_namespaced_cron_job(namespace="graphrag")
+        namespace_cronjobs = batch_v1.list_namespaced_cron_job(namespace=os.environ["AKS_NAMESPACE"])
         cronjob_names = [cronjob.metadata.name for cronjob in namespace_cronjobs.items]
         # create cronjob if it does not exist
         if manifest["metadata"]["name"] not in cronjob_names:
-            batch_v1.create_namespaced_cron_job(namespace="graphrag", body=manifest)
+            batch_v1.create_namespaced_cron_job(namespace=os.environ["AKS_NAMESPACE"], body=manifest)
     except Exception as e:
         print("Failed to create graphrag cronjob.")
         reporter = ReporterSingleton().get_instance()