feat: associate generation calls with devices

2026-07-21 09:40:40 +08:00
parent 69f4517958
commit 11c0d63ac8
16 changed files with 398 additions and 19 deletions
@@ -38,6 +38,7 @@ def create_image_generation_task(
    *,
    user,
    api_key,
+    client_device=None,
    request_data: Mapping[str, Any],
    idempotency_key: str = "",
 ) -> tuple[ImageGenerationTask, bool]:
@@ -53,6 +54,7 @@ def create_image_generation_task(
        GenerationInput(
            user=user,
            api_key=api_key,
+            client_device=client_device,
            operation_type=CallRecord.OperationType.IMAGE,
            prompt=request_data["prompt"],
            alias=request_data.get("model") or None,