add condition to reflect gateway response as good when onprem (#1757)

Author: acosferreira

ansible_ai_connect/healthcheck/tests/test_healthcheck.py

@@ -186,6 +186,18 @@ def test_liveness_probe(self):
             data = json.loads(r.content)
             self.assert_common_data(data, "ok", settings.DEPLOYED_REGION)
 
+    @override_settings(DEPLOYMENT_MODE="onprem")
+    def test_liveness_probe_onprem(self):
+        with patch.object(
+            apps.get_app_config("ai"),
+            "get_model_pipeline",
+            Mock(return_value=DummyCompletionsPipeline(mock_pipeline_config("dummy"))),
+        ):
+            r = self.client.get(reverse("liveness_probe"), format="json")
+            self.assertEqual(r.status_code, HTTPStatus.OK)
+            data = json.loads(r.content)
+            self.assert_common_data(data, "good", settings.DEPLOYED_REGION)
+
     def test_health_check_all_healthy(self):
         cache.clear()
         with patch.object(

ansible_ai_connect/healthcheck/views.py

@@ -162,7 +162,10 @@ class WisdomServiceLivenessProbeView(APIView):
     @method_decorator(never_cache)
     def get(self, request, *args, **kwargs):
         data = common_data()
-        data["status"] = "ok"
+        if settings.DEPLOYMENT_MODE == "onprem":
+            data["status"] = "good"
+        else:
+            data["status"] = "ok"
         data_json = json.dumps(data)
         return HttpResponse(data_json, content_type="application/json")