snowglobe 0.4.10__tar.gz → 0.4.11__tar.gz

{snowglobe-0.4.10/src/snowglobe.egg-info → snowglobe-0.4.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowglobe
-Version: 0.4.10
+Version: 0.4.11
 Summary: client server for usage with snowglobe experiments
 Author-email: Guardrails AI <contact@guardrailsai.com>
 License: MIT License

{snowglobe-0.4.10 → snowglobe-0.4.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "snowglobe"
-version = "0.4.10"
+version = "0.4.11"
 authors = [
     {name = "Guardrails AI", email = "contact@guardrailsai.com"}
 ]

{snowglobe-0.4.10 → snowglobe-0.4.11}/src/snowglobe/client/src/app.py

@@ -26,7 +26,7 @@ from snowglobe.client.src.telemetry import trace_completion_fn, trace_risk_evalu
 
 from .cli_utils import info, shutdown_manager
 from .config import config, get_api_key_or_raise
-from .models import CompletionFunctionOutputs, CompletionRequest, RiskEvaluationRequest
+from .models import CompletionFunctionOutputs, CompletionRequest, RiskEvaluationRequest, SnowglobeData, SnowglobeMessage
 from .stats import initialize_stats, track_batch_completion
 from .utils import fetch_experiments, fetch_messages
 
@@ -129,7 +129,14 @@ async def process_application_heartbeat(app_id):
     }
     try:
         prompt = "Hello from Snowglobe!"
-        test_request = CompletionRequest(messages=[{"role": "user", "content": prompt}])
+        test_request = CompletionRequest(
+            messages=[SnowglobeMessage(
+                    role="user",
+                    content=prompt,
+                    snowglobe_data=SnowglobeData(
+                        conversation_id="test", test_id="test"
+                    ),
+                )])
         heartbeat_id = uuid.uuid4().hex
         agent = apps.get(app_id, {})
         agent_name = agent.get("name", "")
@@ -193,6 +200,10 @@ async def process_application_heartbeat(app_id):
         connection_test_payload["error"] = f"{str(e)}\n{traceback.format_exc()}"
         connection_test_payload["app_id"] = app_id
         connection_test_payload["applicationId"] = app_id
+        LOGGER.error(
+            f"Error processing heartbeat for application {app_id}: {connection_test_payload['error']}"
+        )
+        LOGGER.error(traceback.format_exc())
 
     connection_test_url = (
         f"{config.CONTROL_PLANE_URL}/api/successful-code-connection-tests"

{snowglobe-0.4.10 → snowglobe-0.4.11}/src/snowglobe/client/src/utils.py

@@ -16,13 +16,22 @@ async def fetch_experiments(app_id: str = None) -> list[dict]:
         list[dict]: A list of experiments.
     """
     async with httpx.AsyncClient() as client:
-        experiments_url = f"{config.CONTROL_PLANE_URL}/api/experiments?&evaluated=false"
+        experiments_url = f"{config.CONTROL_PLANE_URL}/api/experiments?evaluated=false"
+
         if app_id:
             experiments_url += f"&appId={config.APPLICATION_ID}"
-        experiments_response = await client.get(
+        try:
+            # get elapsed time for this request
+            import time
+            start_time = time.monotonic()
+            experiments_response = await client.get(
             experiments_url,
             headers={"x-api-key": get_api_key_or_raise()},
-        )
+            timeout=60.0,  # Set timeout to 60 seconds
+            )
+        except httpx.ConnectTimeout:
+            elapsed_time = time.monotonic() - start_time
+            raise Exception(f"Warning: Connection timed out while fetching experiments. Elapsed time: {elapsed_time:.2f} seconds. Polling will continue. If this persists please contact Snowglobe support.")
 
     if not experiments_response.status_code == 200:
         try:

{snowglobe-0.4.10 → snowglobe-0.4.11/src/snowglobe.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowglobe
-Version: 0.4.10
+Version: 0.4.11
 Summary: client server for usage with snowglobe experiments
 Author-email: Guardrails AI <contact@guardrailsai.com>
 License: MIT License