intentkit 0.7.5.dev3__py3-none-any.whl → 0.8.34.dev7__py3-none-any.whl

intentkit/skills/venice_image/image_generation/image_generation_base.py

@@ -1,17 +1,18 @@
 import base64
 import hashlib
 import logging
-from typing import Any, Dict, Literal, Optional, Type
+from typing import Any, Literal
 
 from pydantic import BaseModel, Field
 
+from intentkit.clients.s3 import store_image_bytes
+
 # Import the generic base
 from intentkit.skills.base import ToolException
 from intentkit.skills.venice_image.base import VeniceImageBaseTool
 from intentkit.skills.venice_image.image_generation.image_generation_input import (
     VeniceImageGenerationInput,
 )
-from intentkit.utils.s3 import store_image_bytes
 
 logger = logging.getLogger(__name__)
 
@@ -24,7 +25,7 @@ class VeniceImageGenerationBaseTool(VeniceImageBaseTool):
     """
 
     # --- Attributes specific to Image Generation ---
-    args_schema: Type[BaseModel] = VeniceImageGenerationInput
+    args_schema: type[BaseModel] = VeniceImageGenerationInput
 
     # --- Attributes Subclasses MUST Define ---
     name: str = Field(description="The unique name of the image generation tool/model.")
@@ -38,15 +39,15 @@ class VeniceImageGenerationBaseTool(VeniceImageBaseTool):
     async def _arun(
         self,
         prompt: str,
-        seed: Optional[int] = None,
-        negative_prompt: Optional[str] = None,
-        width: Optional[int] = 1024,
-        height: Optional[int] = 1024,
+        seed: int | None = None,
+        negative_prompt: str | None = None,
+        width: int | None = 1024,
+        height: int | None = 1024,
         format: Literal["png", "jpeg", "webp"] = "png",
-        cfg_scale: Optional[float] = 7.5,
-        style_preset: Optional[str] = "Photographic",
+        cfg_scale: float | None = 7.5,
+        style_preset: str | None = "Photographic",
         **kwargs,
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         try:
             context = self.get_context()
             skillConfig = self.getSkillConfig(context)

intentkit/skills/venice_image/image_generation/image_generation_fluently_xl.py

@@ -1,26 +1,26 @@
-from intentkit.skills.venice_image.image_generation.image_generation_base import (
-    VeniceImageGenerationBaseTool,
-)
-from intentkit.skills.venice_image.image_generation.image_generation_input import (
-    STYLE_PRESETS,
-)
-
-
-class ImageGenerationFluentlyXL(VeniceImageGenerationBaseTool):
-    """
-    Tool for generating images using the Fluently-XL model via Venice AI.
-    Known for aesthetics, lighting, realism, and correct anatomy.
-    """
-
-    # --- Model Specific Configuration ---
-    name: str = "venice_image_generation_fluently_xl"
-    description: str = (
-        "Generate images using the Fluently-XL model (via Venice AI).\n"
-        "Aims for improved aesthetics, lighting, realism, and anatomy. Good for professional-quality images.\n"
-        "Provide a text prompt describing the image (up to 1500 chars).\n"
-        f"Optionally specify a style preset from the list: {', '.join(STYLE_PRESETS)}.\n"
-        "Supports dimensions up to 2048x2048 (multiple of 8)."
-    )
-    model_id: str = "fluently-xl"
-
-    # args_schema and _arun are inherited from VeniceImageGenerationBaseTool
+from intentkit.skills.venice_image.image_generation.image_generation_base import (
+    VeniceImageGenerationBaseTool,
+)
+from intentkit.skills.venice_image.image_generation.image_generation_input import (
+    STYLE_PRESETS,
+)
+
+
+class ImageGenerationFluentlyXL(VeniceImageGenerationBaseTool):
+    """
+    Tool for generating images using the Fluently-XL model via Venice AI.
+    Known for aesthetics, lighting, realism, and correct anatomy.
+    """
+
+    # --- Model Specific Configuration ---
+    name: str = "venice_image_generation_fluently_xl"
+    description: str = (
+        "Generate images using the Fluently-XL model (via Venice AI).\n"
+        "Aims for improved aesthetics, lighting, realism, and anatomy. Good for professional-quality images.\n"
+        "Provide a text prompt describing the image (up to 1500 chars).\n"
+        f"Optionally specify a style preset from the list: {', '.join(STYLE_PRESETS)}.\n"
+        "Supports dimensions up to 2048x2048 (multiple of 8)."
+    )
+    model_id: str = "fluently-xl"
+
+    # args_schema and _arun are inherited from VeniceImageGenerationBaseTool

intentkit/skills/venice_image/image_generation/image_generation_flux_dev.py

@@ -1,27 +1,27 @@
-from intentkit.skills.venice_image.image_generation.image_generation_base import (
-    VeniceImageGenerationBaseTool,
-)
-from intentkit.skills.venice_image.image_generation.image_generation_input import (
-    STYLE_PRESETS,
-)
-
-
-class ImageGenerationFluxDev(VeniceImageGenerationBaseTool):
-    """
-    Tool for generating images using Venice AI's Flux Dev model.
-    Developed by Black Forest Labs, this is a 12 billion parameter rectified flow transformer.
-    """
-
-    # --- Model Specific Configuration ---
-    name: str = "venice_image_generation_flux_dev"
-    description: str = (
-        "Generate images using Venice AI's Flux Dev model (by Black Forest Labs).\n"
-        "This 12B parameter model is good for research and innovative art workflows.\n"
-        "Provide a text prompt describing the image (up to 2048 chars).\n"
-        f"Optionally specify a style preset from the list: {', '.join(STYLE_PRESETS)}.\n"
-        "Supports dimensions up to 2048x2048 (multiple of 8).\n"
-        "Use complies with FLUX.1 [dev] Non-Commercial License."
-    )
-    model_id: str = "flux-dev"
-
-    # args_schema and _arun are inherited from VeniceImageGenerationBaseTool
+from intentkit.skills.venice_image.image_generation.image_generation_base import (
+    VeniceImageGenerationBaseTool,
+)
+from intentkit.skills.venice_image.image_generation.image_generation_input import (
+    STYLE_PRESETS,
+)
+
+
+class ImageGenerationFluxDev(VeniceImageGenerationBaseTool):
+    """
+    Tool for generating images using Venice AI's Flux Dev model.
+    Developed by Black Forest Labs, this is a 12 billion parameter rectified flow transformer.
+    """
+
+    # --- Model Specific Configuration ---
+    name: str = "venice_image_generation_flux_dev"
+    description: str = (
+        "Generate images using Venice AI's Flux Dev model (by Black Forest Labs).\n"
+        "This 12B parameter model is good for research and innovative art workflows.\n"
+        "Provide a text prompt describing the image (up to 2048 chars).\n"
+        f"Optionally specify a style preset from the list: {', '.join(STYLE_PRESETS)}.\n"
+        "Supports dimensions up to 2048x2048 (multiple of 8).\n"
+        "Use complies with FLUX.1 [dev] Non-Commercial License."
+    )
+    model_id: str = "flux-dev"
+
+    # args_schema and _arun are inherited from VeniceImageGenerationBaseTool

intentkit/skills/venice_image/image_generation/image_generation_flux_dev_uncensored.py

@@ -1,26 +1,26 @@
-from intentkit.skills.venice_image.image_generation.image_generation_base import (
-    VeniceImageGenerationBaseTool,
-)
-from intentkit.skills.venice_image.image_generation.image_generation_input import (
-    STYLE_PRESETS,
-)
-
-
-class ImageGenerationFluxDevUncensored(VeniceImageGenerationBaseTool):
-    """
-    Tool for generating images using Venice AI's Flux Dev Uncensored model.
-    An uncensored version of the flux-dev model for unrestricted generation.
-    """
-
-    # --- Model Specific Configuration ---
-    name: str = "venice_image_generation_flux_dev_uncensored"
-    description: str = (
-        "Generate images using Venice AI's Flux Dev Uncensored model.\n"
-        "This is an uncensored version of flux-dev, suitable for unrestricted content including NSFW.\n"
-        "Provide a text prompt describing the image (up to 2048 chars).\n"
-        f"Optionally specify a style preset from the list: {', '.join(STYLE_PRESETS)}.\n"
-        "Supports dimensions up to 2048x2048 (multiple of 8)."
-    )
-    model_id: str = "flux-dev-uncensored"
-
-    # args_schema and _arun are inherited from VeniceImageGenerationBaseTool
+from intentkit.skills.venice_image.image_generation.image_generation_base import (
+    VeniceImageGenerationBaseTool,
+)
+from intentkit.skills.venice_image.image_generation.image_generation_input import (
+    STYLE_PRESETS,
+)
+
+
+class ImageGenerationFluxDevUncensored(VeniceImageGenerationBaseTool):
+    """
+    Tool for generating images using Venice AI's Flux Dev Uncensored model.
+    An uncensored version of the flux-dev model for unrestricted generation.
+    """
+
+    # --- Model Specific Configuration ---
+    name: str = "venice_image_generation_flux_dev_uncensored"
+    description: str = (
+        "Generate images using Venice AI's Flux Dev Uncensored model.\n"
+        "This is an uncensored version of flux-dev, suitable for unrestricted content including NSFW.\n"
+        "Provide a text prompt describing the image (up to 2048 chars).\n"
+        f"Optionally specify a style preset from the list: {', '.join(STYLE_PRESETS)}.\n"
+        "Supports dimensions up to 2048x2048 (multiple of 8)."
+    )
+    model_id: str = "flux-dev-uncensored"
+
+    # args_schema and _arun are inherited from VeniceImageGenerationBaseTool

intentkit/skills/venice_image/image_generation/image_generation_input.py

@@ -1,158 +1,158 @@
-from typing import Literal, Optional
-
-from pydantic import BaseModel, Field, HttpUrl
-
-STYLE_PRESETS = [
-    "3D Model",
-    "Analog Film",
-    "Anime",
-    "Cinematic",
-    "Comic Book",
-    "Craft Clay",
-    "Digital Art",
-    "Enhance",
-    "Fantasy Art",
-    "Isometric Style",
-    "Line Art",
-    "Lowpoly",
-    "Neon Punk",
-    "Origami",
-    "Photographic",
-    "Pixel Art",
-    "Texture",
-    "Advertising",
-    "Food Photography",
-    "Real Estate",
-    "Abstract",
-    "Cubist",
-    "Graffiti",
-    "Hyperrealism",
-    "Impressionist",
-    "Pointillism",
-    "Pop Art",
-    "Psychedelic",
-    "Renaissance",
-    "Steampunk",
-    "Surrealist",
-    "Typography",
-    "Watercolor",
-    "Fighting Game",
-    "GTA",
-    "Super Mario",
-    "Minecraft",
-    "Pokemon",
-    "Retro Arcade",
-    "Retro Game",
-    "RPG Fantasy Game",
-    "Strategy Game",
-    "Street Fighter",
-    "Legend of Zelda",
-    "Architectural",
-    "Disco",
-    "Dreamscape",
-    "Dystopian",
-    "Fairy Tale",
-    "Gothic",
-    "Grunge",
-    "Horror",
-    "Minimalist",
-    "Monochrome",
-    "Nautical",
-    "Space",
-    "Stained Glass",
-    "Techwear Fashion",
-    "Tribal",
-    "Zentangle",
-    "Collage",
-    "Flat Papercut",
-    "Kirigami",
-    "Paper Mache",
-    "Paper Quilling",
-    "Papercut Collage",
-    "Papercut Shadow Box",
-    "Stacked Papercut",
-    "Thick Layered Papercut",
-    "Alien",
-    "Film Noir",
-    "HDR",
-    "Long Exposure",
-    "Neon Noir",
-    "Silhouette",
-    "Tilt-Shift",
-]
-
-STYLE_PRESETS_DESCRIPTION = (
-    "Optional style preset to apply. Available options: "
-    + ", ".join([f"'{s}'" for s in STYLE_PRESETS])
-    + ". Defaults to 'Photographic'."
-)
-
-
-class InpaintMask(BaseModel):
-    image_prompt: str = Field(
-        ...,
-        description="A text prompt describing the original input image that an image model would use to produce a similar/identical image, including the changed features the user will be inpainting.",
-    )
-    inferred_object: str = Field(
-        ..., description="The content being added via inpainting."
-    )
-    object_target: str = Field(
-        ..., description="Element(s) in the original image to be inpainted over."
-    )
-
-
-class Inpaint(BaseModel):
-    image_url: HttpUrl = Field(
-        ...,
-        description="Image target to inpaint",
-    )
-    strength: int = Field(
-        ..., ge=0, le=100, description="Strength of the inpainting (0-100).", example=50
-    )
-    mask: InpaintMask = Field(..., description="Mask settings for inpainting.")
-
-
-class VeniceImageGenerationInput(BaseModel):
-    """Model representing input parameters for Venice Image Generation."""
-
-    prompt: str = Field(
-        description="The main text prompt describing what should be included in the generated image."
-    )
-    seed: Optional[int] = Field(
-        default=None,
-        description="Random seed value to control image generation randomness. "
-        "Use the same seed to reproduce identical results. If not set, a random seed will be used.",
-    )
-    negative_prompt: Optional[str] = Field(
-        default=None,
-        description="Text describing what should be excluded from the generated image. "
-        "If not provided, the default agent configuration will be used.",
-    )
-    width: Optional[int] = Field(
-        default=1024,
-        le=2048,
-        description="Width of the generated image in pixels. Maximum allowed is 2048. Default is 1024.",
-    )
-    height: Optional[int] = Field(
-        default=1024,
-        le=2048,
-        description="Height of the generated image in pixels. Maximum allowed is 2048. Default is 1024.",
-    )
-    format: Literal["png", "jpeg", "webp"] = Field(
-        default="png",
-        description="Output image format. Options are 'png', 'jpeg', or 'webp'. Defaults to 'png'.",
-    )
-    cfg_scale: Optional[float] = Field(
-        default=7.5,
-        description="Classifier-Free Guidance (CFG) scale controls how closely the image follows the prompt. "
-        "Higher values (1-20) result in more adherence. Default is 7.5.",
-    )
-    style_preset: Optional[str] = Field(
-        default="Photographic", description=STYLE_PRESETS_DESCRIPTION
-    )
-    inpainting: Optional[Inpaint] = Field(
-        default=None,
-        description="Optional inpainting operation that allows modification of specific objects within an image. "
-        "Requires an original image url, a strength value (0-100), and detailed mask instructions "
-        "to define which part of the image should be edited and what should replace it.",
-    )
+from typing import Literal
+
+from pydantic import BaseModel, Field, HttpUrl
+
+STYLE_PRESETS = [
+    "3D Model",
+    "Analog Film",
+    "Anime",
+    "Cinematic",
+    "Comic Book",
+    "Craft Clay",
+    "Digital Art",
+    "Enhance",
+    "Fantasy Art",
+    "Isometric Style",
+    "Line Art",
+    "Lowpoly",
+    "Neon Punk",
+    "Origami",
+    "Photographic",
+    "Pixel Art",
+    "Texture",
+    "Advertising",
+    "Food Photography",
+    "Real Estate",
+    "Abstract",
+    "Cubist",
+    "Graffiti",
+    "Hyperrealism",
+    "Impressionist",
+    "Pointillism",
+    "Pop Art",
+    "Psychedelic",
+    "Renaissance",
+    "Steampunk",
+    "Surrealist",
+    "Typography",
+    "Watercolor",
+    "Fighting Game",
+    "GTA",
+    "Super Mario",
+    "Minecraft",
+    "Pokemon",
+    "Retro Arcade",
+    "Retro Game",
+    "RPG Fantasy Game",
+    "Strategy Game",
+    "Street Fighter",
+    "Legend of Zelda",
+    "Architectural",
+    "Disco",
+    "Dreamscape",
+    "Dystopian",
+    "Fairy Tale",
+    "Gothic",
+    "Grunge",
+    "Horror",
+    "Minimalist",
+    "Monochrome",
+    "Nautical",
+    "Space",
+    "Stained Glass",
+    "Techwear Fashion",
+    "Tribal",
+    "Zentangle",
+    "Collage",
+    "Flat Papercut",
+    "Kirigami",
+    "Paper Mache",
+    "Paper Quilling",
+    "Papercut Collage",
+    "Papercut Shadow Box",
+    "Stacked Papercut",
+    "Thick Layered Papercut",
+    "Alien",
+    "Film Noir",
+    "HDR",
+    "Long Exposure",
+    "Neon Noir",
+    "Silhouette",
+    "Tilt-Shift",
+]
+
+STYLE_PRESETS_DESCRIPTION = (
+    "Optional style preset to apply. Available options: "
+    + ", ".join([f"'{s}'" for s in STYLE_PRESETS])
+    + ". Defaults to 'Photographic'."
+)
+
+
+class InpaintMask(BaseModel):
+    image_prompt: str = Field(
+        ...,
+        description="A text prompt describing the original input image that an image model would use to produce a similar/identical image, including the changed features the user will be inpainting.",
+    )
+    inferred_object: str = Field(
+        ..., description="The content being added via inpainting."
+    )
+    object_target: str = Field(
+        ..., description="Element(s) in the original image to be inpainted over."
+    )
+
+
+class Inpaint(BaseModel):
+    image_url: HttpUrl = Field(
+        ...,
+        description="Image target to inpaint",
+    )
+    strength: int = Field(
+        ..., ge=0, le=100, description="Strength of the inpainting (0-100).", example=50
+    )
+    mask: InpaintMask = Field(..., description="Mask settings for inpainting.")
+
+
+class VeniceImageGenerationInput(BaseModel):
+    """Model representing input parameters for Venice Image Generation."""
+
+    prompt: str = Field(
+        description="The main text prompt describing what should be included in the generated image."
+    )
+    seed: int | None = Field(
+        default=None,
+        description="Random seed value to control image generation randomness. "
+        "Use the same seed to reproduce identical results. If not set, a random seed will be used.",
+    )
+    negative_prompt: str | None = Field(
+        default=None,
+        description="Text describing what should be excluded from the generated image. "
+        "If not provided, the default agent configuration will be used.",
+    )
+    width: int | None = Field(
+        default=1024,
+        le=2048,
+        description="Width of the generated image in pixels. Maximum allowed is 2048. Default is 1024.",
+    )
+    height: int | None = Field(
+        default=1024,
+        le=2048,
+        description="Height of the generated image in pixels. Maximum allowed is 2048. Default is 1024.",
+    )
+    format: Literal["png", "jpeg", "webp"] = Field(
+        default="png",
+        description="Output image format. Options are 'png', 'jpeg', or 'webp'. Defaults to 'png'.",
+    )
+    cfg_scale: float | None = Field(
+        default=7.5,
+        description="Classifier-Free Guidance (CFG) scale controls how closely the image follows the prompt. "
+        "Higher values (1-20) result in more adherence. Default is 7.5.",
+    )
+    style_preset: str | None = Field(
+        default="Photographic", description=STYLE_PRESETS_DESCRIPTION
+    )
+    inpainting: Inpaint | None = Field(
+        default=None,
+        description="Optional inpainting operation that allows modification of specific objects within an image. "
+        "Requires an original image url, a strength value (0-100), and detailed mask instructions "
+        "to define which part of the image should be edited and what should replace it.",
+    )

intentkit/skills/venice_image/image_generation/image_generation_lustify_sdxl.py

@@ -1,26 +1,26 @@
-from intentkit.skills.venice_image.image_generation.image_generation_base import (
-    VeniceImageGenerationBaseTool,
-)
-from intentkit.skills.venice_image.image_generation.image_generation_input import (
-    STYLE_PRESETS,
-)
-
-
-class ImageGenerationLustifySDXL(VeniceImageGenerationBaseTool):
-    """
-    Tool for generating images using the Lustify SDXL model via Venice AI.
-    A photorealistic SDXL checkpoint primarily focused on NSFW content, but can do SFW.
-    """
-
-    # --- Model Specific Configuration ---
-    name: str = "venice_image_generation_lustify_sdxl"
-    description: str = (
-        "Generate images using the Lustify SDXL model (via Venice AI).\n"
-        "A photorealistic SDXL model focused on NSFW scenes, but can generate SFW images (objects, animals, fantasy).\n"
-        "Provide a text prompt describing the image (up to 1500 chars).\n"
-        f"Optionally specify a style preset from the list: {', '.join(STYLE_PRESETS)}.\n"
-        "Supports dimensions up to 2048x2048 (multiple of 8)."
-    )
-    model_id: str = "lustify-sdxl"
-
-    # args_schema and _arun are inherited from VeniceImageGenerationBaseTool
+from intentkit.skills.venice_image.image_generation.image_generation_base import (
+    VeniceImageGenerationBaseTool,
+)
+from intentkit.skills.venice_image.image_generation.image_generation_input import (
+    STYLE_PRESETS,
+)
+
+
+class ImageGenerationLustifySDXL(VeniceImageGenerationBaseTool):
+    """
+    Tool for generating images using the Lustify SDXL model via Venice AI.
+    A photorealistic SDXL checkpoint primarily focused on NSFW content, but can do SFW.
+    """
+
+    # --- Model Specific Configuration ---
+    name: str = "venice_image_generation_lustify_sdxl"
+    description: str = (
+        "Generate images using the Lustify SDXL model (via Venice AI).\n"
+        "A photorealistic SDXL model focused on NSFW scenes, but can generate SFW images (objects, animals, fantasy).\n"
+        "Provide a text prompt describing the image (up to 1500 chars).\n"
+        f"Optionally specify a style preset from the list: {', '.join(STYLE_PRESETS)}.\n"
+        "Supports dimensions up to 2048x2048 (multiple of 8)."
+    )
+    model_id: str = "lustify-sdxl"
+
+    # args_schema and _arun are inherited from VeniceImageGenerationBaseTool

intentkit/skills/venice_image/image_generation/image_generation_pony_realism.py

@@ -1,26 +1,26 @@
-from intentkit.skills.venice_image.image_generation.image_generation_base import (
-    VeniceImageGenerationBaseTool,
-)
-from intentkit.skills.venice_image.image_generation.image_generation_input import (
-    STYLE_PRESETS,
-)
-
-
-class ImageGenerationPonyRealism(VeniceImageGenerationBaseTool):
-    """
-    Tool for generating images using the Pony Realism model via Venice AI.
-    Focused on high-detail, realistic images, especially anime/character designs. Uses Danbooru tags.
-    """
-
-    # --- Model Specific Configuration ---
-    name: str = "venice_image_generation_pony_realism"
-    description: str = (
-        "Generate images using the Pony Realism model (via Venice AI).\n"
-        "Creates high-detail, realistic images, good for anime/character designs. Benefits from Danbooru tags (e.g., 'score_9', 'female'/'male').\n"
-        "Provide a text prompt describing the image (up to 1500 chars).\n"
-        f"Optionally specify a style preset from the list: {', '.join(STYLE_PRESETS)}.\n"
-        "Supports dimensions up to 2048x2048 (multiple of 8). Marked as 'most_uncensored'."
-    )
-    model_id: str = "pony-realism"
-
-    # args_schema and _arun are inherited from VeniceImageGenerationBaseTool
+from intentkit.skills.venice_image.image_generation.image_generation_base import (
+    VeniceImageGenerationBaseTool,
+)
+from intentkit.skills.venice_image.image_generation.image_generation_input import (
+    STYLE_PRESETS,
+)
+
+
+class ImageGenerationPonyRealism(VeniceImageGenerationBaseTool):
+    """
+    Tool for generating images using the Pony Realism model via Venice AI.
+    Focused on high-detail, realistic images, especially anime/character designs. Uses Danbooru tags.
+    """
+
+    # --- Model Specific Configuration ---
+    name: str = "venice_image_generation_pony_realism"
+    description: str = (
+        "Generate images using the Pony Realism model (via Venice AI).\n"
+        "Creates high-detail, realistic images, good for anime/character designs. Benefits from Danbooru tags (e.g., 'score_9', 'female'/'male').\n"
+        "Provide a text prompt describing the image (up to 1500 chars).\n"
+        f"Optionally specify a style preset from the list: {', '.join(STYLE_PRESETS)}.\n"
+        "Supports dimensions up to 2048x2048 (multiple of 8). Marked as 'most_uncensored'."
+    )
+    model_id: str = "pony-realism"
+
+    # args_schema and _arun are inherited from VeniceImageGenerationBaseTool