polly-python 4.2.0__tar.gz → 4.3.0__tar.gz

{polly_python-4.2.0/polly_python.egg-info → polly_python-4.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: polly_python
-Version: 4.2.0
+Version: 4.3.0
 Summary: Polly SDK
 Home-page: https://github.com/ElucidataInc/polly-python
 Project-URL: Documentation, https://docs.elucidata.io

polly_python-4.3.0/polly/__init__.py

@@ -0,0 +1 @@
+__version__ = "4.3.0"

{polly_python-4.2.0 → polly_python-4.3.0}/polly/polly_kg.py

@@ -277,13 +277,30 @@ class PollyKG:
                 result_json = requests.get(result_s3_signed_url)
                 return result_json.json()
             elif status == "FAILED":
-                raise QueryFailedException(
+                metadata = (
                     status_response.json()
                     .get("data", {})
                     .get("attributes", {})
                     .get("metadata", {})
-                    .get("error")
                 )
+                # Try new format first: metadata.results.error and metadata.results.details
+                results = metadata.get("results", {})
+                if results and ("error" in results or "details" in results):
+                    error_type = results.get("error", "Unknown error")
+                    error_details = results.get("details", "")
+                    error_message = (
+                        f"{error_type}: {error_details}"
+                        if error_details
+                        else error_type
+                    )
+                else:
+                    # Fallback to old format: metadata.error
+                    error_message = metadata.get(
+                        "error", "Query failed without a specific error message."
+                    )
+
+                print(f"Query failed. Query ID: {query_id}. Error: {error_message}")
+                raise QueryFailedException(error_message)
         except Exception as e:
             print(e)
             raise e
@@ -386,11 +403,25 @@ class PollyKG:
 
             status = data.get("attributes", {}).get("status")
             if status == "FAILED":
-                raise QueryFailedException(
-                    data.get("attributes", {})
-                    .get("metadata", {})
-                    .get("error", "Query failed without a specific error message.")
-                )
+                metadata = data.get("attributes", {}).get("metadata", {})
+                # Try new format first: metadata.results.error and metadata.results.details
+                results = metadata.get("results", {})
+                if results and ("error" in results or "details" in results):
+                    error_type = results.get("error", "Unknown error")
+                    error_details = results.get("details", "")
+                    error_message = (
+                        f"{error_type}: {error_details}"
+                        if error_details
+                        else error_type
+                    )
+                else:
+                    # Fallback to old format: metadata.error
+                    error_message = metadata.get(
+                        "error", "Query failed without a specific error message."
+                    )
+
+                print(f"Query failed. Query ID: {query_id}. Error: {error_message}")
+                raise QueryFailedException(error_message)
             result_s3_signed_url = (
                 data.get("attributes", {})
                 .get("metadata", {})
@@ -436,11 +467,25 @@ class PollyKG:
 
             status = data.get("attributes", {}).get("status")
             if status == "FAILED":
-                raise QueryFailedException(
-                    data.get("attributes", {})
-                    .get("metadata", {})
-                    .get("error", "Query failed without a specific error message.")
-                )
+                metadata = data.get("attributes", {}).get("metadata", {})
+                # Try new format first: metadata.results.error and metadata.results.details
+                results = metadata.get("results", {})
+                if results and ("error" in results or "details" in results):
+                    error_type = results.get("error", "Unknown error")
+                    error_details = results.get("details", "")
+                    error_message = (
+                        f"{error_type}: {error_details}"
+                        if error_details
+                        else error_type
+                    )
+                else:
+                    # Fallback to old format: metadata.error
+                    error_message = metadata.get(
+                        "error", "Query failed without a specific error message."
+                    )
+
+                print(f"Query failed. Query ID: {query_id}. Error: {error_message}")
+                raise QueryFailedException(error_message)
             result_s3_signed_url = (
                 data.get("attributes", {})
                 .get("metadata", {})

{polly_python-4.2.0 → polly_python-4.3.0/polly_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: polly_python
-Version: 4.2.0
+Version: 4.3.0
 Summary: Polly SDK
 Home-page: https://github.com/ElucidataInc/polly-python
 Project-URL: Documentation, https://docs.elucidata.io

polly_python-4.2.0/polly/__init__.py

@@ -1 +0,0 @@
-__version__ = "4.2.0"