Merge pull request #180 from ecmwf-projects/execute-model

Execute model

Author: francesconazzaro

cads_processing_api_service/clients.py

@@ -167,7 +167,7 @@ def get_process(
     def post_process_execution(
         self,
         process_id: str = fastapi.Path(...),
-        execution_content: ogc_api_processes_fastapi.models.Execute = fastapi.Body(...),
+        execution_content: models.Execute = fastapi.Body(...),
         auth_header: tuple[str, str] = fastapi.Depends(auth.get_auth_header),
         portal_header: str | None = fastapi.Header(
             None, alias=config.PORTAL_HEADER_NAME

cads_processing_api_service/costing.py

@@ -19,14 +19,13 @@
 import cads_adaptors
 import cads_catalogue
 import fastapi
-import ogc_api_processes_fastapi.models
 
 from . import adaptors, costing, db_utils, models, utils
 
 
 def estimate_costs(
     process_id: str = fastapi.Path(...),
-    request: ogc_api_processes_fastapi.models.Execute = fastapi.Body(...),
+    request: models.Execute = fastapi.Body(...),
 ) -> models.Costing:
     table = cads_catalogue.database.Resource
     catalogue_sessionmaker = db_utils.get_catalogue_sessionmaker(

cads_processing_api_service/models.py

@@ -63,3 +63,7 @@ class Costing(pydantic.BaseModel):
     costs: dict[str, float] | None = None
     max_costs: dict[str, float] | None = None
     max_costs_exceeded: dict[str, float] | None = None
+
+
+class Execute(ogc_api_processes_fastapi.models.Execute):
+    inputs: dict[str, Any] | None = None