intentkit 0.6.13.dev2__py3-none-any.whl → 0.8.17__py3-none-any.whl

intentkit/skills/venice_image/image_generation/image_generation_venice_sd35.py

@@ -1,28 +1,28 @@
-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 ImageGenerationVeniceSD35(VeniceImageGenerationBaseTool):
-    """
-    Tool for generating images using Venice AI's interface to Stable Diffusion 3.5 Large.
-    Developed by Stability AI, using MMDiT architecture. Good for art and design.
-    """
-
-    # --- Model Specific Configuration ---
-    name: str = "venice_image_generation_venice_sd35"
-    description: str = (
-        "Generate images using Stability AI's Stable Diffusion 3.5 Large model (via Venice AI).\n"
-        "Ideal for artworks, design processes, and educational use. Not for factual representations.\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 16).\n"
-        "Use must comply with Stability AI's Acceptable Use Policy."
-    )
-    # Use the specific ID provided by Venice, assuming it matches the name
-    model_id: str = "venice-sd35"
-
-    # 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 ImageGenerationVeniceSD35(VeniceImageGenerationBaseTool):
+    """
+    Tool for generating images using Venice AI's interface to Stable Diffusion 3.5 Large.
+    Developed by Stability AI, using MMDiT architecture. Good for art and design.
+    """
+
+    # --- Model Specific Configuration ---
+    name: str = "venice_image_generation_venice_sd35"
+    description: str = (
+        "Generate images using Stability AI's Stable Diffusion 3.5 Large model (via Venice AI).\n"
+        "Ideal for artworks, design processes, and educational use. Not for factual representations.\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 16).\n"
+        "Use must comply with Stability AI's Acceptable Use Policy."
+    )
+    # Use the specific ID provided by Venice, assuming it matches the name
+    model_id: str = "venice-sd35"
+
+    # args_schema and _arun are inherited from VeniceImageGenerationBaseTool

intentkit/skills/venice_image/image_upscale/image_upscale.py

@@ -1,5 +1,5 @@
 import logging
-from typing import Literal, Optional
+from typing import Literal
 
 from pydantic import HttpUrl
 
@@ -32,7 +32,7 @@ class ImageUpscale(VeniceImageUpscaleBaseTool):
         self,
         image_url: HttpUrl,
         scale: Literal[2, 4],
-        replication: Optional[float] = 0.35,
+        replication: float | None = 0.35,
         **kwargs,
     ) -> dict:
         """
@@ -41,7 +41,7 @@ class ImageUpscale(VeniceImageUpscaleBaseTool):
         Args:
             image_url (HttpUrl): The public URL of the image to upscale.
             scale (Literal[2, 4]): The scale factor for upscaling (2x or 4x).
-            replication (Optional[float]): The replication factor for the upscale process, defaults to 0.35.
+            replication (float | None): The replication factor for the upscale process, defaults to 0.35.
             config (RunnableConfig, optional): Configuration for the runnable, if any.
             **kwargs: Additional keyword arguments.
 

intentkit/skills/venice_image/image_upscale/image_upscale_base.py

@@ -1,23 +1,21 @@
-from typing import Type
-
-from pydantic import BaseModel, Field
-
-# Import the generic base and shared input
-from intentkit.skills.venice_image.base import VeniceImageBaseTool
-from intentkit.skills.venice_image.image_upscale.image_upscale_input import (
-    VeniceImageUpscaleInput,
-)
-
-
-class VeniceImageUpscaleBaseTool(VeniceImageBaseTool):
-    """
-    Base class for Venice AI *Image Upscaling* tools.
-    Inherits from VeniceAIBaseTool and handles specifics of the
-    /image/upscale endpoint
-    """
-
-    args_schema: Type[BaseModel] = VeniceImageUpscaleInput
-    name: str = Field(description="The unique name of the image upscaling tool.")
-    description: str = Field(
-        description="A description of what the image upscaling tool does."
-    )
+from pydantic import BaseModel, Field
+
+# Import the generic base and shared input
+from intentkit.skills.venice_image.base import VeniceImageBaseTool
+from intentkit.skills.venice_image.image_upscale.image_upscale_input import (
+    VeniceImageUpscaleInput,
+)
+
+
+class VeniceImageUpscaleBaseTool(VeniceImageBaseTool):
+    """
+    Base class for Venice AI *Image Upscaling* tools.
+    Inherits from VeniceAIBaseTool and handles specifics of the
+    /image/upscale endpoint
+    """
+
+    args_schema: type[BaseModel] = VeniceImageUpscaleInput
+    name: str = Field(description="The unique name of the image upscaling tool.")
+    description: str = Field(
+        description="A description of what the image upscaling tool does."
+    )

intentkit/skills/venice_image/image_upscale/image_upscale_input.py

@@ -1,22 +1,22 @@
-from typing import Literal, Optional
-
-from pydantic import BaseModel, Field, HttpUrl
-
-
-class VeniceImageUpscaleInput(BaseModel):
-    """Input for the Image Upscale tool."""
-
-    image_url: HttpUrl = Field(
-        description="The URL of the image to upscale. Must be a publicly accessible URL.",
-    )
-    replication: Optional[float] = Field(
-        default=0.35,
-        description=(
-            'How strongly lines and noise in the base image are preserved. Higher values are noisier but less plastic/AI "generated"/hallucinated. Must be between 0.1 and 1.'
-            "Required range: 0.1 <= x <= 1"
-        ),
-    )
-    scale: Literal[2, 4] = Field(
-        default=2,
-        description="The factor by which to upscale the image (either 2 or 4). Defaults to 2.",
-    )
+from typing import Literal
+
+from pydantic import BaseModel, Field, HttpUrl
+
+
+class VeniceImageUpscaleInput(BaseModel):
+    """Input for the Image Upscale tool."""
+
+    image_url: HttpUrl = Field(
+        description="The URL of the image to upscale. Must be a publicly accessible URL.",
+    )
+    replication: float | None = Field(
+        default=0.35,
+        description=(
+            'How strongly lines and noise in the base image are preserved. Higher values are noisier but less plastic/AI "generated"/hallucinated. Must be between 0.1 and 1.'
+            "Required range: 0.1 <= x <= 1"
+        ),
+    )
+    scale: Literal[2, 4] = Field(
+        default=2,
+        description="The factor by which to upscale the image (either 2 or 4). Defaults to 2.",
+    )

intentkit/skills/venice_image/image_vision/image_vision.py

@@ -1,5 +1,5 @@
 import logging
-from typing import Any, Type
+from typing import Any
 
 from pydantic import BaseModel, HttpUrl
 
@@ -27,7 +27,7 @@ class ImageVision(VeniceImageVisionBaseTool):
         "Provide the public URL of the image to describe.\n"
         "Returns a descriptive text of the image."
     )
-    args_schema: Type[BaseModel] = VeniceImageVision
+    args_schema: type[BaseModel] = VeniceImageVision
     # No model_id needed for the generic vision endpoint currently
 
     async def _arun(

intentkit/skills/venice_image/image_vision/image_vision_base.py

@@ -1,17 +1,17 @@
-from pydantic import Field
-
-# Import the generic base and shared input
-from intentkit.skills.venice_image.base import VeniceImageBaseTool
-
-
-class VeniceImageVisionBaseTool(VeniceImageBaseTool):
-    """
-    Base class for Venice AI *Image Vision* tools.
-    Inherits from VeniceAIBaseTool and handles specifics of the
-    /chat/completions endpoint.
-    """
-
-    name: str = Field(description="The unique name of the image vision tool.")
-    description: str = Field(
-        description="A description of what the image vision tool does."
-    )
+from pydantic import Field
+
+# Import the generic base and shared input
+from intentkit.skills.venice_image.base import VeniceImageBaseTool
+
+
+class VeniceImageVisionBaseTool(VeniceImageBaseTool):
+    """
+    Base class for Venice AI *Image Vision* tools.
+    Inherits from VeniceAIBaseTool and handles specifics of the
+    /chat/completions endpoint.
+    """
+
+    name: str = Field(description="The unique name of the image vision tool.")
+    description: str = Field(
+        description="A description of what the image vision tool does."
+    )

intentkit/skills/venice_image/image_vision/image_vision_input.py

@@ -1,9 +1,9 @@
-from pydantic import BaseModel, Field, HttpUrl
-
-
-class VeniceImageVision(BaseModel):
-    """Input for the Image Vision tool."""
-
-    image_url: HttpUrl = Field(
-        description="The URL of the image to to be described by the Vision model. Must be a publicly accessible URL.",
-    )
+from pydantic import BaseModel, Field, HttpUrl
+
+
+class VeniceImageVision(BaseModel):
+    """Input for the Image Vision tool."""
+
+    image_url: HttpUrl = Field(
+        description="The URL of the image to to be described by the Vision model. Must be a publicly accessible URL.",
+    )