openrunner-sdk 2.8.0__tar.gz → 2.9.0__tar.gz

{openrunner_sdk-2.8.0 → openrunner_sdk-2.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrunner-sdk
-Version: 2.8.0
+Version: 2.9.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.8.0 → openrunner_sdk-2.9.0}/openrunner/__init__.py

@@ -120,7 +120,7 @@ launch.from_run = _launch_from_run  # type: ignore[attr-defined]
 # openrunner.trace.patch_openai() syntax
 trace.patch_openai = _patch_openai  # type: ignore[attr-defined]
 
-__version__ = "2.8.0"
+__version__ = "2.9.0"
 
 logger = logging.getLogger("openrunner")
 

{openrunner_sdk-2.8.0 → openrunner_sdk-2.9.0}/openrunner/session.py

@@ -506,80 +506,102 @@ def sync_session_to_openrunner(
 
     client = APIClient(base_url=base_url, api_key=api_key)
 
-    # Create run
+    # Resolve project_id from "org/project" format
+    project_id = None
+    try:
+        projects = client.list_projects()
+        for p in projects:
+            org_name = p.get("org_name", "")
+            proj_name = p.get("name", "")
+            full_name = f"{org_name}/{proj_name}" if org_name else proj_name
+            if full_name == project or proj_name == project:
+                project_id = p.get("id")
+                break
+    except Exception:
+        pass
+
+    if not project_id:
+        # Fallback: try creating as a run (backward compat with older servers)
+        logger.warning(f"Could not resolve project '{project}', falling back to run creation")
+        return _sync_as_run_fallback(client, parsed, project, visibility)
+
+    # Build AI session data
+    source = parsed["source"]
+    project_hint = parsed.get("project_hint", "")
+    timestamp = parsed.get("ended_at", "")[:16].replace("T", " ")
+
+    if project_hint:
+        short_hint = project_hint.rsplit("/", 1)[-1] if "/" in project_hint else project_hint
+        title = f"{source}/{short_hint}/{timestamp}"
+    else:
+        title = f"{source}/{timestamp}"
+
+    group_key = f"{source}:{project_hint}" if project_hint else f"{source}:default"
+
+    session_data = {
+        "title": title,
+        "source": source,
+        "worktree": project_hint,
+        "group_key": group_key,
+        "summary": parsed.get("summary"),
+        "notes": _format_session_notes(parsed),
+        "first_message": parsed.get("first_message"),
+        "messages": parsed.get("messages"),
+        "message_count": parsed.get("message_count", 0),
+        "user_message_count": parsed.get("user_message_count", 0),
+        "total_tokens": parsed.get("total_tokens", 0),
+        "tools_used": parsed.get("tools_used", []),
+        "files_touched": parsed.get("files_touched", []),
+        "visibility": visibility,
+        "tags": [f"source:{source}", f"worktree:{group_key}"],
+        "redacted": bool(parsed.get("_redaction")),
+        "redaction_mode": parsed.get("_redaction", {}).get("mode") if parsed.get("_redaction") else None,
+        "session_file": parsed.get("session_file"),
+        "started_at": parsed.get("started_at"),
+        "ended_at": parsed.get("ended_at"),
+    }
+
+    # POST to /projects/{project_id}/ai-sessions
+    try:
+        resp = client._request("post", f"/projects/{project_id}/ai-sessions", json=session_data)
+        if resp.status_code in (200, 201):
+            result = resp.json()
+            client.close()
+            return result.get("id")
+        else:
+            # Server doesn't have AI sessions endpoint yet — fall back to run
+            logger.info("AI sessions endpoint not available, falling back to run")
+            return _sync_as_run_fallback(client, parsed, project, visibility)
+    except Exception as e:
+        logger.warning(f"AI session sync failed: {e}")
+        client.close()
+        return None
+
+
+def _sync_as_run_fallback(client, parsed: dict, project: str, visibility: str) -> str | None:
+    """Fallback: create a run if AI sessions endpoint not available."""
     source = parsed["source"]
     timestamp = parsed.get("ended_at", "")[:16].replace("T", " ")
     project_hint = parsed.get("project_hint", "")
 
-    # Build display name: include worktree/project context
     if project_hint:
-        # Shorten: /home/ubuntu/openrun → openrun
         short_hint = project_hint.rsplit("/", 1)[-1] if "/" in project_hint else project_hint
         run_name = f"{source}/{short_hint}/{timestamp}"
     else:
         run_name = f"{source}/{timestamp}"
 
-    # Group sessions by source project (worktree)
-    group_name = f"{source}:{project_hint}" if project_hint else f"{source}:default"
-
     run_data = {
         "project": project,
         "display_name": run_name,
-        "group_name": group_name,
-        "config": {
-            "source": source,
-            "session_file": parsed.get("session_file", ""),
-            "project_hint": project_hint,
-            "message_count": parsed.get("message_count", 0),
-            "user_message_count": parsed.get("user_message_count", 0),
-            "tools_used": parsed.get("tools_used", []),
-            "total_tokens": parsed.get("total_tokens", 0),
-        },
-        "tags": [f"source:{source}", "ai-session", f"visibility:{visibility}", f"worktree:{group_name}"],
+        "group_name": f"{source}:{project_hint}" if project_hint else f"{source}:default",
+        "tags": [f"source:{source}", "ai-session"],
         "notes": _format_session_notes(parsed),
         "state": "finished",
     }
 
-    # Add redaction metadata if applied
-    if parsed.get("_redaction"):
-        run_data["config"]["redacted"] = True
-        run_data["config"]["redaction_mode"] = parsed["_redaction"]["mode"]
-        run_data["config"]["entities_redacted"] = parsed["_redaction"]["entities_redacted"]
-
     result = client.create_run(run_data)
-    if not result:
-        logger.warning("Failed to create session run")
-        client.close()
-        return None
-
-    run_id = result.get("id")
-
-    # Log metrics
-    metrics = []
-    if parsed.get("total_tokens"):
-        metrics.append({"key": "tokens", "value": parsed["total_tokens"], "step": 1})
-    if parsed.get("message_count"):
-        metrics.append({"key": "messages", "value": parsed["message_count"], "step": 1})
-    if parsed.get("user_message_count"):
-        metrics.append({"key": "user_messages", "value": parsed["user_message_count"], "step": 1})
-    if parsed.get("files_touched"):
-        metrics.append({"key": "files_touched", "value": len(parsed["files_touched"]), "step": 1})
-
-    if metrics and run_id:
-        client.post_metrics(run_id, metrics)
-
-    # Log files touched as summary
-    if parsed.get("files_touched") and run_id:
-        client.update_run(run_id, {
-            "summary": {
-                "files_touched": len(parsed["files_touched"]),
-                "tools_used": len(parsed.get("tools_used", [])),
-                "tokens": parsed.get("total_tokens", 0),
-            }
-        })
-
     client.close()
-    return run_id
+    return result.get("id") if result else None
 
 
 def _format_session_notes(parsed: dict) -> str:

{openrunner_sdk-2.8.0 → openrunner_sdk-2.9.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "openrunner-sdk"
-version = "2.8.0"
+version = "2.9.0"
 description = "OpenRunner SDK - W&B-compatible ML experiment tracking client"
 readme = "README.md"
 license = {text = "MIT"}