synth-ai 0.2.9.dev7__py3-none-any.whl → 0.2.10__py3-none-any.whl

synth_ai/{rl → learning/rl}/env_keys.py

@@ -1,12 +1,12 @@
-from __future__ import annotations
-
 """Helpers for uploading RL environment credentials to the backend."""
 
+from __future__ import annotations
+
 import base64
 import binascii
 import json
-from typing import Any, Dict
 import os
+from typing import Any
 
 import requests
 from nacl.public import PublicKey, SealedBox
@@ -18,14 +18,12 @@ _ALGORITHM = "libsodium.sealedbox.v1"
 
 
 def encrypt_for_backend(pubkey_b64: str, secret: str | bytes) -> str:
-    """Encrypt ``secret`` for storage by the backend using libsodium sealed boxes."""
-
     if not isinstance(pubkey_b64, str) or not pubkey_b64.strip():
         raise ValueError("public key must be a non-empty base64 string")
 
     try:
         key_bytes = base64.b64decode(pubkey_b64, validate=True)
-    except binascii.Error as exc:  # pragma: no cover - defensive guard
+    except binascii.Error as exc:
         raise ValueError("public key must be base64-encoded") from exc
 
     if len(key_bytes) != 32:
@@ -35,7 +33,7 @@ def encrypt_for_backend(pubkey_b64: str, secret: str | bytes) -> str:
         secret_bytes = secret.encode("utf-8")
     elif isinstance(secret, bytes):
         secret_bytes = secret
-    else:  # pragma: no cover - type guard
+    else:
         raise TypeError("secret must be str or bytes")
 
     if not secret_bytes:
@@ -52,20 +50,17 @@ def setup_environment_api_key(
     token: str | None = None,
     *,
     timeout: float = 15.0,
-) -> Dict[str, Any]:
-    """Upload an ENVIRONMENT_API_KEY to the backend."""
-
+) -> dict[str, Any]:
     backend = backend_base.rstrip("/")
     if not backend:
         raise ValueError("backend_base must be provided")
     if not synth_api_key:
         raise ValueError("synth_api_key must be provided")
 
-    # Require caller-provided plaintext. If not provided, read from ENVIRONMENT_API_KEY.
     plaintext = token if token is not None else os.getenv("ENVIRONMENT_API_KEY", "").strip()
     if not plaintext:
         raise ValueError("ENVIRONMENT_API_KEY must be set (or pass token=...) to upload")
-    if not isinstance(plaintext, str):  # pragma: no cover - defensive guard
+    if not isinstance(plaintext, str):
         raise TypeError("token must be a string")
 
     token_bytes = plaintext.encode("utf-8")
@@ -81,7 +76,7 @@ def setup_environment_api_key(
 
     try:
         doc = response.json()
-    except ValueError as exc:  # pragma: no cover - backend invariant
+    except ValueError as exc:
         raise RuntimeError("backend returned invalid JSON for public key") from exc
 
     if not isinstance(doc, dict):
@@ -91,16 +86,45 @@ def setup_environment_api_key(
     if not isinstance(pubkey, str) or not pubkey:
         raise RuntimeError("backend response missing public_key")
 
-    # The backend currently returns a single algorithm identifier; keep a guard in
-    # case future versions change the value and we need to surface that to callers.
     alg = doc.get("alg")
     if alg is not None and alg != _ALGORITHM:
         raise RuntimeError(f"unsupported sealed box algorithm: {alg}")
 
+    # Diagnostics: safe previews and hashes to correlate with backend logs
+    try:
+        import hashlib as _hash
+
+        pk_bytes = base64.b64decode(pubkey, validate=True)
+        pk_sha256 = _hash.sha256(pk_bytes).hexdigest()
+        print(
+            f"[env-keys] public_key: b64_len={len(pubkey)} sha256={pk_sha256} head={pubkey[:16]} tail={pubkey[-16:]}"
+        )
+        _plen = len(plaintext)
+        _ppref = (plaintext[:6] + "…") if _plen > 10 else plaintext
+        _psuf = ("…" + plaintext[-4:]) if _plen > 10 else ""
+        _has_ws = any(ch.isspace() for ch in plaintext)
+        print(
+            f"[env-keys] plaintext: len={_plen} preview={_ppref}{_psuf} has_ws={bool(_has_ws)}"
+        )
+    except Exception:
+        pass
+
     ciphertext_b64 = encrypt_for_backend(pubkey, token_bytes)
 
     body = {"name": "ENVIRONMENT_API_KEY", "ciphertext_b64": ciphertext_b64}
     post_url = f"{backend}/api/v1/env-keys"
+    # Ciphertext diagnostics
+    try:
+        import hashlib as _hash
+
+        _ct_bytes = base64.b64decode(ciphertext_b64, validate=True)
+        _ct_sha = _hash.sha256(_ct_bytes).hexdigest()
+        print(
+            f"[env-keys] ciphertext: b64_len={len(ciphertext_b64)} sha256={_ct_sha} head={ciphertext_b64[:16]} tail={ciphertext_b64[-16:]}"
+        )
+    except Exception:
+        pass
+
     response2 = requests.post(
         post_url,
         headers={**headers, "Content-Type": "application/json"},

synth_ai/learning/rl/secrets.py

@@ -0,0 +1,13 @@
+"""Helpers for generating RL environment credentials."""
+
+from __future__ import annotations
+
+import secrets
+
+__all__ = ["mint_environment_api_key"]
+
+
+def mint_environment_api_key() -> str:
+    """Mint a random ENVIRONMENT_API_KEY value."""
+
+    return secrets.token_hex(32)

synth_ai/learning/rl_client.py

@@ -1,284 +1,5 @@
 from __future__ import annotations
 
-from typing import Any, Dict, List, Optional, Callable
-import os
-import time
+from .rl.client import RlClient
 
-from ..http import AsyncHttpClient, HTTPError, sleep
-
-
-def _api_base(b: str) -> str:
-    b = (b or "").rstrip("/")
-    return b if b.endswith("/api") else f"{b}/api"
-
-
-class RlClient:
-    """Lightweight RL client for provider-agnostic job control.
-
-    Notes:
-    - Uses learning/* for status/events/metrics and rl/* for creation/start.
-    - Trainer endpoints are resolved server-side via trainer_id.
-    """
-
-    def __init__(self, base_url: str, api_key: str, *, timeout: float = 600.0) -> None:
-        self._base_url = base_url.rstrip("/")
-        self._api_key = api_key
-        self._timeout = timeout
-
-    async def resolve_trainer_start_url(self, trainer_id: str) -> str:
-        """GET /api/rl/services/{id} → { training_start_url }"""
-        path = f"/api/rl/services/{trainer_id}"
-        async with AsyncHttpClient(self._base_url, self._api_key, timeout=30.0) as http:
-            js = await http.get(path)
-        if not isinstance(js, dict):
-            raise HTTPError(
-                status=500, url=path, message="invalid_service_response", body_snippet=str(js)[:200]
-            )
-        start_url = js.get("training_start_url")
-        if not isinstance(start_url, str) or not start_url:
-            raise HTTPError(
-                status=500,
-                url=path,
-                message="missing_training_start_url",
-                body_snippet=str(js)[:200],
-            )
-        return start_url
-
-    async def create_job(
-        self,
-        *,
-        model: str,
-        task_app_url: str,
-        trainer: Dict[str, Any],
-        trainer_id: Optional[str] = None,
-        job_config_id: Optional[str] = None,
-        inline_config: Optional[Dict[str, Any]] = None,
-    ) -> Dict[str, Any]:
-        body = {
-            "job_type": "rl",
-            "data": {
-                "model": model,
-                "endpoint_base_url": task_app_url,
-                **({"job_config_id": job_config_id} if job_config_id else {}),
-                **({"config": inline_config} if inline_config else {}),
-                "trainer": {
-                    "batch_size": int(trainer.get("batch_size", 1)),
-                    "group_size": max(2, int(trainer.get("group_size", 2))),
-                },
-            },
-        }
-        async with AsyncHttpClient(self._base_url, self._api_key, timeout=self._timeout) as http:
-            js = await http.post_json(f"{_api_base(self._base_url)}/rl/jobs", json=body)
-        if not isinstance(js, dict):
-            raise HTTPError(
-                status=500,
-                url="/api/rl/jobs",
-                message="invalid_create_response",
-                body_snippet=str(js)[:200],
-            )
-        return js
-
-    async def start_job_if_supported(self, job_id: str) -> Optional[Dict[str, Any]]:
-        path = f"{_api_base(self._base_url)}/rl/jobs/{job_id}/start"
-        try:
-            async with AsyncHttpClient(self._base_url, self._api_key, timeout=30.0) as http:
-                return await http.post_json(path, json={})
-        except HTTPError as he:  # noqa: PERF203
-            if he.status == 404:
-                return None
-            raise
-
-    async def get_job(self, job_id: str) -> Dict[str, Any]:
-        async with AsyncHttpClient(self._base_url, self._api_key, timeout=30.0) as http:
-            return await http.get(f"{_api_base(self._base_url)}/learning/jobs/{job_id}")
-
-    async def get_events(
-        self, job_id: str, *, since_seq: int = 0, limit: int = 200
-    ) -> List[Dict[str, Any]]:
-        params = {"since_seq": since_seq, "limit": limit}
-        async with AsyncHttpClient(self._base_url, self._api_key, timeout=30.0) as http:
-            try:
-                js = await http.get(
-                    f"{_api_base(self._base_url)}/learning/jobs/{job_id}/events", params=params
-                )
-            except HTTPError as he:
-                try:
-                    print(
-                        f"[poll] events HTTPError status={he.status} url={he.url} since_seq={since_seq} body={(he.body_snippet or '')[:200]}"
-                    )
-                except Exception:
-                    pass
-                raise
-        if isinstance(js, dict):
-            evs = js.get("events") or js.get("data")
-            if isinstance(evs, list):
-                return evs
-        return []
-
-    async def get_metrics(
-        self, job_id: str, *, after_step: int = -1, limit: int = 200
-    ) -> List[Dict[str, Any]]:
-        params = {"after_step": after_step, "limit": limit}
-        async with AsyncHttpClient(self._base_url, self._api_key, timeout=30.0) as http:
-            js = await http.get(
-                f"{_api_base(self._base_url)}/learning/jobs/{job_id}/metrics", params=params
-            )
-        if isinstance(js, dict) and isinstance(js.get("points"), list):
-            return js["points"]
-        return []
-
-    async def poll_until_terminal(
-        self,
-        job_id: str,
-        *,
-        interval_seconds: float = 2.0,
-        max_seconds: float | None = None,
-        empty_polls_threshold: int = 5,
-        startup_deadline_s: int = 45,
-        on_event: Optional[Callable[[Dict[str, Any]], None]] = None,
-        on_metric: Optional[Callable[[Dict[str, Any]], None]] = None,
-    ) -> Dict[str, Any]:
-        last_seq_by_stream: Dict[str, int] = {}
-        events_job_id: Optional[str] = None
-        last_status: Optional[str] = None
-        last_step_by_name: Dict[str, int] = {}
-        empty_polls = 0
-        saw_any_event = False
-        start_t = time.time()
-        terminal = {"succeeded", "failed", "cancelled", "canceled", "error", "completed"}
-
-        while True:
-            status_data: Optional[Dict[str, Any]] = None
-            try:
-                status_data = await self.get_job(job_id)
-            except Exception:
-                status_data = None
-            if status_data is None:
-                try:
-                    print(f"[poll] get_job returned None base={self._base_url} job_id={job_id}")
-                except Exception:
-                    pass
-            status = str((status_data or {}).get("status") or "").lower()
-            if status_data:
-                linked = status_data.get("linked_job_id")
-                if isinstance(linked, str) and linked and linked != events_job_id:
-                    events_job_id = linked
-                    try:
-                        print(f"[poll] discovered linked_job_id stream={events_job_id}")
-                    except Exception:
-                        pass
-            if status and status != last_status:
-                last_status = status
-                # Status transitions only to avoid log spam
-                if on_event:
-                    try:
-                        on_event({"type": "rl.status", "message": status})
-                    except Exception:
-                        pass
-
-            # Events
-            stream_ids = [job_id]
-            if events_job_id and events_job_id not in stream_ids:
-                stream_ids.append(events_job_id)
-            try:
-                print(
-                    f"[poll] streams={stream_ids} intervals={interval_seconds}s since_map={last_seq_by_stream} empty_polls={empty_polls}"
-                )
-            except Exception:
-                pass
-            total_events_this_cycle = 0
-            terminal_event_seen = False
-            terminal_event_status: Optional[str] = None
-            for ev_id in stream_ids:
-                since = last_seq_by_stream.get(ev_id, 0)
-                try:
-                    events = await self.get_events(ev_id, since_seq=since, limit=200)
-                except HTTPError as he:
-                    try:
-                        print(
-                            f"[poll] get_events error status={he.status} url={he.url} since={since} body={(he.body_snippet or '')[:200]}"
-                        )
-                    except Exception:
-                        pass
-                    events = []
-                except Exception as e:
-                    try:
-                        print(
-                            f"[poll] get_events unexpected error ev_id={ev_id} since={since} err={type(e).__name__}: {e}"
-                        )
-                    except Exception:
-                        pass
-                    events = []
-                total_events_this_cycle += len(events)
-                if events:
-                    saw_any_event = True
-                for e in events:
-                    seq_val = int(e.get("seq") or 0)
-                    if seq_val <= last_seq_by_stream.get(ev_id, 0):
-                        continue
-                    last_seq_by_stream[ev_id] = seq_val
-                    if on_event:
-                        try:
-                            on_event(e)
-                        except Exception:
-                            pass
-                    et = str(e.get("type") or e.get("event_type") or "").lower()
-                    if et in ("rl.job.completed", "workflow.completed", "rl.train.completed"):
-                        terminal_event_seen = True
-                        terminal_event_status = "succeeded"
-                    elif et in ("rl.job.failed", "workflow.failed"):
-                        terminal_event_seen = True
-                        terminal_event_status = "failed"
-
-            # Metrics
-            try:
-                after = max(last_step_by_name.values()) if last_step_by_name else -1
-                points = await self.get_metrics(job_id, after_step=after, limit=200)
-                for p in points:
-                    name = str(p.get("name") or "")
-                    step = int(p.get("step") or -1)
-                    if step <= last_step_by_name.get(name, -1):
-                        continue
-                    last_step_by_name[name] = step
-                    if on_metric:
-                        try:
-                            on_metric(p)
-                        except Exception:
-                            pass
-            except Exception:
-                pass
-
-            if terminal_event_seen:
-                return {"status": terminal_event_status or status or "completed", "job_id": job_id}
-            if status and status in terminal:
-                return {"status": status, "job_id": job_id}
-
-            if total_events_this_cycle == 0:
-                empty_polls += 1
-            else:
-                empty_polls = 0
-            if empty_polls >= max(1, int(empty_polls_threshold)):
-                try:
-                    print(
-                        f"[poll] threshold hit: empty_polls={empty_polls} >= {empty_polls_threshold} streams={stream_ids} last_seq_map={last_seq_by_stream}"
-                    )
-                except Exception:
-                    pass
-                raise AssertionError(
-                    f"No new events detected for {empty_polls_threshold} consecutive polls. Check event ingestion."
-                )
-
-            if not saw_any_event and (time.time() - start_t) > int(startup_deadline_s):
-                try:
-                    print(
-                        f"[poll] startup window exceeded: {startup_deadline_s}s base={self._base_url} job={job_id} streams={stream_ids} last_seq_map={last_seq_by_stream}"
-                    )
-                except Exception:
-                    pass
-                raise AssertionError(
-                    f"No events observed within startup window ({startup_deadline_s}s). Investigate event streaming."
-                )
-
-            await sleep(interval_seconds)
-            if max_seconds is not None and (time.time() - start_t) >= max_seconds:
-                raise TimeoutError(f"Polling timed out after {max_seconds}s for job {job_id}")
+__all__ = ["RlClient"]

synth_ai/learning/sft/__init__.py

@@ -0,0 +1,29 @@
+from .client import FtClient
+from .data import (
+    SFTDataError,
+    SFTExample,
+    SFTMessage,
+    SFTToolCall,
+    SFTToolDefinition,
+    coerce_example,
+    collect_sft_jsonl_errors,
+    iter_sft_examples,
+    load_jsonl,
+    parse_jsonl_line,
+    validate_jsonl_or_raise,
+)
+
+__all__ = [
+    "FtClient",
+    "SFTDataError",
+    "SFTExample",
+    "SFTMessage",
+    "SFTToolCall",
+    "SFTToolDefinition",
+    "collect_sft_jsonl_errors",
+    "coerce_example",
+    "iter_sft_examples",
+    "load_jsonl",
+    "parse_jsonl_line",
+    "validate_jsonl_or_raise",
+]

synth_ai/learning/sft/client.py

@@ -0,0 +1,68 @@
+from __future__ import annotations
+
+from pathlib import Path
+from typing import Any
+
+from ...http import AsyncHttpClient, HTTPError
+from .config import prepare_sft_job_payload
+from .data import validate_jsonl_or_raise
+
+
+class FtClient:
+    def __init__(self, base_url: str, api_key: str, *, timeout: float = 30.0) -> None:
+        self._base_url = base_url.rstrip("/")
+        self._api_key = api_key
+        self._timeout = timeout
+
+    async def upload_training_file(self, path: str | Path, *, purpose: str = "fine-tune") -> str:
+        p = Path(path)
+        if p.suffix.lower() == ".jsonl" and purpose == "fine-tune":
+            validate_jsonl_or_raise(p, min_messages=2)
+        content = p.read_bytes()
+        async with AsyncHttpClient(self._base_url, self._api_key, timeout=self._timeout) as http:
+            data = {"purpose": purpose}
+            files = {"file": (p.name, content, _infer_content_type(p.name))}
+            js = await http.post_multipart("/api/learning/files", data=data, files=files)
+        if not isinstance(js, dict) or "id" not in js:
+            raise HTTPError(
+                status=500,
+                url="/api/learning/files",
+                message="invalid_upload_response",
+                body_snippet=str(js)[:200],
+            )
+        return str(js["id"])
+
+    async def create_sft_job(
+        self,
+        *,
+        model: str,
+        training_file_id: str,
+        hyperparameters: dict[str, Any],
+        metadata: dict[str, Any] | None = None,
+    ) -> dict[str, Any]:
+        body = prepare_sft_job_payload(
+            model=model,
+            training_file=training_file_id,
+            hyperparameters=hyperparameters,
+            metadata=metadata,
+            training_type="sft_offline",
+            training_file_field="training_file_id",
+            require_training_file=True,
+        )
+        async with AsyncHttpClient(self._base_url, self._api_key, timeout=self._timeout) as http:
+            return await http.post_json("/api/learning/jobs", json=body)
+
+    async def start_job(self, job_id: str) -> dict[str, Any]:
+        async with AsyncHttpClient(self._base_url, self._api_key, timeout=self._timeout) as http:
+            return await http.post_json(f"/api/learning/jobs/{job_id}/start", json={})
+
+
+def _infer_content_type(filename: str) -> str:
+    name = filename.lower()
+    if name.endswith(".jsonl"):
+        return "application/jsonl"
+    if name.endswith(".json"):
+        return "application/json"
+    if name.endswith(".txt"):
+        return "text/plain"
+    return "application/octet-stream"