knowledge2 0.4.0__py3-none-any.whl

sdk/types/training.py

@@ -0,0 +1,55 @@
+from __future__ import annotations
+
+from typing import TypedDict
+
+
+class TrainingDataBuildResponse(TypedDict):
+    job_id: str
+    dataset_id: str
+
+
+class TrainingDatasetResponse(TypedDict, total=False):
+    id: str
+    corpus_id: str
+    sample_count: int
+    dataset_hash: str
+    created_at: str | None
+
+
+class TrainingDatasetListResponse(TypedDict):
+    datasets: list[TrainingDatasetResponse]
+
+
+class TuningRunResponse(TypedDict):
+    run_id: str
+    status: str
+    job_id: str
+
+
+class TuningRunBuildResponse(TypedDict):
+    run_id: str
+    status: str
+    build_job_id: str
+    dataset_id: str
+
+
+class TuningRunDetailResponse(TypedDict, total=False):
+    run_id: str
+    status: str
+    metrics: dict
+
+
+class TuningRunLogsResponse(TypedDict):
+    lines: list[str]
+
+
+class TuningRunListResponse(TypedDict):
+    runs: list[TuningRunDetailResponse]
+
+
+class PromoteResponse(TypedDict):
+    model_id: str
+
+
+class CancelTuningRunResponse(TypedDict):
+    status: str

sdk/types/usage.py

@@ -0,0 +1,37 @@
+from __future__ import annotations
+
+from typing import List, Optional, TypedDict
+
+
+class UsageSeriesPoint(TypedDict):
+    date: str
+    count: int
+
+
+class UsageSummaryResponse(TypedDict, total=False):
+    range: str
+    total_requests: int
+    daily: list[UsageSeriesPoint]
+    latency_p50_ms: float | None
+    latency_p95_ms: float | None
+    error_rate: float | None
+
+
+class UsageByCorpusItem(TypedDict, total=False):
+    corpus_id: str
+    corpus_name: str | None
+    requests: int
+
+
+class UsageByCorpusResponse(TypedDict):
+    items: list[UsageByCorpusItem]
+
+
+class UsageByKeyItem(TypedDict, total=False):
+    key_id: str
+    name: str | None
+    requests: int
+
+
+class UsageByKeyResponse(TypedDict):
+    items: list[UsageByKeyItem]