openrunner-sdk 2.0.0__tar.gz → 2.1.0__tar.gz

{openrunner_sdk-2.0.0 → openrunner_sdk-2.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrunner-sdk
-Version: 2.0.0
+Version: 2.1.0
 Summary: OpenRunner SDK - W&B-compatible ML experiment tracking client
 Project-URL: Homepage, https://github.com/jqueguiner/openrunner
 Project-URL: Repository, https://github.com/jqueguiner/openrunner

{openrunner_sdk-2.0.0 → openrunner_sdk-2.1.0}/openrunner/__init__.py

@@ -38,11 +38,16 @@ Public API:
     openrunner.trace        -> Decorator for LLM call tracing (supports sample_rate)
     openrunner.trace.patch_openai -> Auto-trace OpenAI chat completions
     openrunner.Prompt       -> Versioned prompt management (get/render/publish)
+    openrunner.StringPrompt -> Client-side string template with format()
+    openrunner.MessagesPrompt -> Multi-turn chat prompt template
+    openrunner.Model        -> Versioned model with predict/save/load/serve
     openrunner.Dataset      -> Versioned dataset for evaluations
     openrunner.add_cost     -> Register custom LLM cost per model
     openrunner.evaluate     -> Run LLM evaluation against a dataset
     openrunner.scorer       -> Decorator to mark a function as an eval scorer
-    openrunner.scorers      -> Built-in scorers (exact_match, contains, etc.)
+    openrunner.Scorer       -> Base class for class-based evaluation scorers
+    openrunner.EvaluationLogger -> Log evaluation results piecemeal
+    openrunner.scorers      -> Built-in scorers (exact_match, contains, LLM-as-judge)
     openrunner.plot.line    -> Create a line chart from a Table
     openrunner.plot.scatter -> Create a scatter chart from a Table
     openrunner.plot.bar     -> Create a bar chart from a Table
@@ -61,7 +66,8 @@ from typing import Any
 
 from openrunner.artifact import Artifact
 from openrunner.config import Config
-from openrunner.prompt import Prompt
+from openrunner.model import Model
+from openrunner.prompt import MessagesPrompt, Prompt, StringPrompt
 from openrunner.launch import LaunchJob, launch, from_run as _launch_from_run
 from openrunner.media import (
     Audio,
@@ -90,8 +96,18 @@ from openrunner.run import Run
 from openrunner.settings import SDKSettings
 from openrunner.summary import Summary
 from openrunner.sweep import agent, sweep
-from openrunner.evaluation import evaluate, scorer
-from openrunner.trace import trace, patch_openai as _patch_openai
+from openrunner.evaluation import EvaluationLogger, Scorer, evaluate, scorer
+from openrunner.guardrails import (
+    GuardrailCheckResult,
+    GuardrailResult,
+    add_guardrail,
+    check_guardrails,
+    clear_guardrails,
+    remove_guardrail,
+)
+from openrunner.feedback import add_feedback, get_feedback
+from openrunner.pii import configure_pii, redact
+from openrunner.trace import trace, patch_openai as _patch_openai, set_thread_id, set_session_id
 import openrunner.scorers as scorers
 import openrunner.plot as plot
 

{openrunner_sdk-2.0.0 → openrunner_sdk-2.1.0}/openrunner/api_client.py

@@ -594,19 +594,29 @@ class APIClient:
             return []
 
     def upload_file_to_presigned_url(
-        self, presigned_url: str, file_path: str
+        self, presigned_url: str, file_path: str, run_id: str = ""
     ) -> bool:
         """PUT raw file bytes to a presigned URL.
 
-        Uses httpx directly (not self._client) because presigned URLs
-        are absolute and should not have a base_url prefix.
+        Falls back to API proxy upload if presigned URL is unreachable
+        (e.g., MinIO not exposed publicly).
         """
         try:
             with open(file_path, "rb") as f:
                 data = f.read()
-            resp = httpx.put(presigned_url, content=data, timeout=300.0)
-            resp.raise_for_status()
-            return True
+            # Try presigned URL first
+            try:
+                resp = httpx.put(presigned_url, content=data, timeout=300.0)
+                resp.raise_for_status()
+                return True
+            except (httpx.ConnectError, httpx.ConnectTimeout):
+                # Presigned URL unreachable — use API proxy
+                if run_id:
+                    import os
+                    fname = os.path.basename(file_path)
+                    proxy_resp = self._request("PUT", f"/runs/{run_id}/files/{fname}", content=data)
+                    return proxy_resp.status_code == 200
+                return False
         except Exception as e:
             logger.warning("upload_file_to_presigned_url failed: %s", e)
             return False