Update checks/endpoint_check.py

checks/endpoint_check.py

@@ -7,11 +7,21 @@ def is_public_endpoint(endpoint: str):
     try:
         # No Authorization header required for public models
         response = requests.get(endpoint)
-
+         
+        # Log the response status and headers for better insight
+        print(f"Response status code: {response.status_code}")
+        print(f"Response headers: {response.headers}")
+        
         # Check if the response status code is 200 and it returns inference data
         if response.status_code == 200:
             # Public models will return inference data without needing an API key
-            if "model" in response.json() or "error" in response.json():
+            # Attempt to parse JSON response
+                response_json = response.json()
+                # Print the first few keys of the response JSON for debugging
+                print(f"Response JSON keys: {list(response_json.keys())[:5]}")
+
+                # Public models will return inference data without needing an API key
+                if "model" in response_json or "error" in response_json:
                 return True
             else:
                 print("The response does not contain inference-related data.")