dantelabs-agentic-school 1.0.0 → 1.2.0

package/plugins/common/skills/kie-image-generator/scripts/models/seedream.py

@@ -0,0 +1,51 @@
+"""Seedream 4.0 parameter handler"""
+
+from typing import Dict, Any, Optional
+
+
+def build_seedream_payload(prompt: str, **kwargs) -> Dict[str, Any]:
+    """
+    Build payload for Seedream 4.0 model.
+
+    Parameters:
+        prompt: Image description (max 5000 chars)
+        size: Image size/aspect ratio
+        resolution: Image resolution (1K, 2K, 4K) [default: 2K]
+        images: Number of images (1-6) [default: 1]
+        seed: Random seed for reproducibility
+
+    Returns:
+        API request payload
+    """
+    if len(prompt) > 5000:
+        raise ValueError("Prompt must be 5000 characters or less")
+
+    payload = {'prompt': prompt}
+
+    # Image size
+    size = kwargs.get('size')
+    if size:
+        payload['image_size'] = size
+
+    # Resolution
+    resolution = kwargs.get('resolution', '2K')
+    if resolution:
+        valid_resolutions = ['1K', '2K', '4K']
+        if resolution.upper() not in valid_resolutions:
+            raise ValueError(f"Invalid resolution. Must be one of: {', '.join(valid_resolutions)}")
+        payload['image_resolution'] = resolution.upper()
+
+    # Number of images
+    images = kwargs.get('images', 1)
+    if images:
+        images = int(images)
+        if not 1 <= images <= 6:
+            raise ValueError("images must be between 1 and 6")
+        payload['max_images'] = images
+
+    # Seed
+    seed = kwargs.get('seed')
+    if seed is not None:
+        payload['seed'] = int(seed)
+
+    return payload

package/plugins/common/skills/kie-image-generator/scripts/models/seedream_edit.py

@@ -0,0 +1,66 @@
+"""Seedream V4 Edit parameter handler"""
+
+from typing import Dict, Any, Optional, List
+
+def build_seedream_edit_payload(prompt: str, **kwargs) -> Dict[str, Any]:
+    """
+    Build payload for Seedream V4 Edit model.
+
+    Parameters:
+        prompt: Image editing instruction (max 5000 chars)
+        image_urls: List of input image URLs (required, up to 10 images)
+        image_size: Image size/aspect ratio
+        image_resolution: Image resolution (1K, 2K, 4K) [default: 1K]
+        max_images: Number of output images (1-6) [default: 1]
+        seed: Random seed for reproducibility
+
+    Returns:
+        API request payload
+    """
+    if len(prompt) > 5000:
+        raise ValueError("Prompt must be 5000 characters or less")
+
+    # image_urls is required for edit model
+    image_urls = kwargs.get('image_urls') or kwargs.get('image_url')
+    if not image_urls:
+        raise ValueError("image_urls parameter is required for seedream-edit model")
+
+    # Convert single URL to list
+    if isinstance(image_urls, str):
+        image_urls = [image_urls]
+
+    if len(image_urls) > 10:
+        raise ValueError("Maximum 10 input images allowed")
+
+    payload = {
+        'prompt': prompt,
+        'image_urls': image_urls
+    }
+
+    # Image size
+    size = kwargs.get('size')
+    if size:
+        payload['image_size'] = size
+
+    # Resolution
+    resolution = kwargs.get('resolution', '1K')
+    if resolution:
+        valid_resolutions = ['1K', '2K', '4K']
+        if resolution.upper() not in valid_resolutions:
+            raise ValueError(f"Invalid resolution. Must be one of: {', '.join(valid_resolutions)}")
+        payload['image_resolution'] = resolution.upper()
+
+    # Number of images
+    images = kwargs.get('images', 1)
+    if images:
+        images = int(images)
+        if not 1 <= images <= 6:
+            raise ValueError("images must be between 1 and 6")
+        payload['max_images'] = images
+
+    # Seed
+    seed = kwargs.get('seed')
+    if seed is not None:
+        payload['seed'] = int(seed)
+
+    return payload