Add explorer.py to push traces to explorer and cleanup code a bit.

proxy/utils/explorer.py

@@ -0,0 +1,40 @@
+"""Utility functions for the Invariant explorer."""
+
+import os
+from typing import Any, Dict, List
+
+from fastapi import HTTPException
+from invariant_sdk.client import Client
+
+DEFAULT_API_URL = "https://explorer.invariantlabs.ai"
+
+
+def push_trace(
+    messages: List[Dict[str, Any]],
+    dataset_name: str,
+    invariant_authorization: str,
+    api_url: str = DEFAULT_API_URL,
+) -> Dict[str, str]:
+    """Pushes traces to the dataset on the Invariant Explorer.
+
+    Args:
+        messages (List[Dict[str, Any]]): List of messages to push.
+        dataset_name (str): Name of the dataset.
+        invariant_authorization (str): Authorization token.
+        api_url (str): URL of the Invariant Explorer API.
+
+    Returns:
+        Dict[str, str]: Response containing the trace ID.
+    """
+    api_url = os.getenv("INVARIANT_API_URL", DEFAULT_API_URL)
+    api_key = invariant_authorization.split("Bearer ")[1]
+    client = Client(api_url=api_url, api_key=api_key)
+    try:
+        push_trace_response = client.create_request_and_push_trace(
+            messages=messages, dataset=dataset_name
+        )
+        return {"trace_id": push_trace_response.id[0]}
+    except Exception as e:
+        raise HTTPException(
+            status_code=500, detail="Failed to push traces to the dataset"
+        ) from e