diff --git a/laborious/activities/api.py b/laborious/activities/api.py index 05d2d3c..d152345 100644 --- a/laborious/activities/api.py +++ b/laborious/activities/api.py @@ -77,7 +77,7 @@ class API(SientiaMonitoring): async def process_pi_web_api_response( self, - response_data: dict[str, Any], + response_data: list[dict[str, Any]], tags: dict[str, str], core_labels: dict[str, str], metadata: dict[str, Any], @@ -110,8 +110,7 @@ class API(SientiaMonitoring): # Evaluate response for each tag written_tags = [] - response_items = response_data.get('Items', []) - for item in response_items: + for item in response_data: web_id = item.get('WebId') if not web_id: self.error('The response did not contain some WebIds', metadata) @@ -233,6 +232,8 @@ class API(SientiaMonitoring): attachment_content=trace, ) + self.error(trace, metadata) + data['prediction_confidence'] = PI_WEB_API_PREDICTION_ERROR_CONFIDENCE data['comments'] = str(e) @@ -266,4 +267,4 @@ class API(SientiaMonitoring): attachment_content=trace, ) - return data.to_dict() + return data.to_dict() \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index 8be4007..c9f111c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,7 @@ psycopg2-binary sqlalchemy asyncua redis -git+ssh://git@github.com/Aignosi/sientia-dataops-library.git@1.8.1 +git+ssh://git@github.com/Aignosi/sientia-dataops-library.git@1.8.2 git+ssh://git@github.com/Aignosi/sientia-mlops-library.git@0.40.7 prometheus-client botocore