Update checks/health_check.py

checks/health_check.py

@@ -3,6 +3,19 @@ import os
 import requests
 from typing import Dict, Tuple
 
+def check_with_prompt(endpoint_uri: str):
+    
+    # Try a direct test of your endpoint
+    test_prompt = "Hello, how are you?"
+    response = requests.post(
+        endpoint_uri,
+        json={"inputs": test_prompt},
+        headers={"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
+    )
+    print(response.json())
+
+
+
 def check_model_endpoint(endpoint_uri: str) -> Tuple[bool, Dict]:
     """Comprehensive check of model endpoint health"""
     # Basic availability check