update async trigger

DB_utils.py

@@ -5,6 +5,7 @@ from prefect_utils import trigger_maintenance_request
 from prefect import task
 import pandas as pd
 import os
+import asyncio
 
 MONGODB_PASSWORD = os.getenv("MONGODB_PASSWORD")
 
@@ -81,9 +82,11 @@ def find_unused_wells():
     if len(empty_wells) == 1:
         maintenance_type =  "wellplate_maintenance"
         if check_maintenance_status(maintenance_type) == 0:
-            trigger_maintenance_request(maintenance_type)
+            asyncio.run(trigger_maintenance_request_async(maintenance_type))
             set_maintenance_status(maintenance_type,1)
-
+        else:
+            print("[DEBUG] a maintenance request has been sent")
+    
     dbclient.close()
 
     # Check if there are any empty wells

prefect_utils.py

@@ -12,11 +12,11 @@ def start_prefect_worker(work_pool_name: str = "ot2-pool"):
     worker_thread.start()
     print("Prefect Worker started in background thread.")
 
-from prefect.deployments import run_deployment
+from prefect.client import get_client
 
-def trigger_maintenance_request(maintenance_type: str):
-    deployment_name = "request-wells-maintenance/wells-maintenance"
-    run_deployment(
-        name=deployment_name,
-        parameters={"maintenance_type": maintenance_type}
-    )
+async def trigger_maintenance_request_async(maintenance_type: str):
+    async with get_client() as client:
+        await client.create_flow_run_from_deployment(
+            name="request-wells-maintenance/wells-maintenance",
+            parameters={"maintenance_type": maintenance_type}
+        )