intentkit 0.5.0__py3-none-any.whl → 0.5.2__py3-none-any.whl

intentkit/skills/heurist/image_generation_arthemy_comics.py

@@ -0,0 +1,162 @@
+import logging
+from typing import Optional, Type
+
+import httpx
+from epyxid import XID
+from langchain_core.runnables import RunnableConfig
+from pydantic import BaseModel, Field
+
+from intentkit.skills.heurist.base import HeuristBaseTool
+from intentkit.utils.s3 import store_image
+
+logger = logging.getLogger(__name__)
+
+
+class ImageGenerationArthemyComicsInput(BaseModel):
+    """Input for ImageGenerationArthemyComics tool."""
+
+    prompt: str = Field(
+        description="Text prompt describing the image to generate.",
+    )
+    neg_prompt: Optional[str] = Field(
+        default="(worst quality: 1.4), bad quality, nsfw",
+        description="Negative prompt describing what to avoid in the generated image.",
+    )
+    width: Optional[int] = Field(
+        default=1024,
+        le=1024,
+        description="Width of the generated image.",
+    )
+    height: Optional[int] = Field(
+        default=1024,
+        le=1024,
+        description="Height of the generated image.",
+    )
+
+
+class ImageGenerationArthemyComics(HeuristBaseTool):
+    """Tool for generating comic-style images using Heurist AI's ArthemyComics model.
+
+    This tool takes a text prompt and uses Heurist's API to generate
+    a comic-style image based on the description.
+
+    Attributes:
+        name: The name of the tool.
+        description: A description of what the tool does.
+        args_schema: The schema for the tool's input arguments.
+    """
+
+    name: str = "heurist_image_generation_arthemy_comics"
+    description: str = (
+        "Generate comic-style images using Heurist AI's ArthemyComics model.\n"
+        "Provide a text prompt describing the comic-style image you want to generate.\n"
+        "ArthemyComics specializes in creating vibrant, stylized comic book illustrations.\n"
+        "If you have height and width, remember to specify them.\n"
+    )
+    args_schema: Type[BaseModel] = ImageGenerationArthemyComicsInput
+
+    async def _arun(
+        self,
+        prompt: str,
+        neg_prompt: Optional[str] = "(worst quality: 1.4), bad quality, nsfw",
+        width: Optional[int] = 1024,
+        height: Optional[int] = 680,
+        config: RunnableConfig = None,
+        **kwargs,
+    ) -> str:
+        """Implementation of the tool to generate comic-style images using Heurist AI's ArthemyComics model.
+
+        Args:
+            prompt: Text prompt describing the image to generate.
+            neg_prompt: Negative prompt describing what to avoid in the generated image.
+            width: Width of the generated image.
+            height: Height of the generated image.
+            config: Configuration for the runnable.
+
+        Returns:
+            str: URL of the generated image.
+        """
+        context = self.context_from_config(config)
+        skill_config = context.config
+
+        # Get the Heurist API key from the skill store
+        if "api_key" in skill_config and skill_config["api_key"]:
+            api_key = skill_config["api_key"]
+            if skill_config.get("rate_limit_number") and skill_config.get(
+                "rate_limit_minutes"
+            ):
+                await self.user_rate_limit_by_category(
+                    context.user_id,
+                    skill_config["rate_limit_number"],
+                    skill_config["rate_limit_minutes"],
+                )
+        else:
+            api_key = self.skill_store.get_system_config("heurist_api_key")
+            await self.user_rate_limit_by_category(context.user_id, 10, 1440)
+
+        # Generate a unique job ID
+        job_id = str(XID())
+
+        # Prepare the request payload
+        payload = {
+            "job_id": job_id,
+            "model_input": {
+                "SD": {
+                    "prompt": prompt,
+                    "neg_prompt": neg_prompt,
+                    "num_iterations": 25,
+                    "width": width,
+                    "height": height,
+                    "guidance_scale": 5,
+                    "seed": -1,
+                }
+            },
+            "model_id": "ArthemyComics",
+            "deadline": 120,
+            "priority": 1,
+        }
+        logger.debug(f"Heurist API payload: {payload}")
+
+        headers = {
+            "Authorization": f"Bearer {api_key}",
+            "Content-Type": "application/json",
+        }
+
+        try:
+            # Make the API request
+            async with httpx.AsyncClient() as client:
+                response = await client.post(
+                    "http://sequencer.heurist.xyz/submit_job",
+                    json=payload,
+                    headers=headers,
+                    timeout=120,
+                )
+                logger.debug(f"Heurist API response: {response.text}")
+                response.raise_for_status()
+
+            # Store the image URL
+            image_url = response.text.strip('"')
+            # Generate a key with agent ID as prefix
+            image_key = f"{context.agent.id}/heurist/{job_id}"
+            # Store the image and get the CDN URL
+            stored_url = await store_image(image_url, image_key)
+
+            # Return the stored image URL
+            return stored_url
+
+        except httpx.HTTPStatusError as e:
+            # Extract error details from response
+            try:
+                error_json = e.response.json()
+                error_code = error_json.get("error", "")
+                error_message = error_json.get("message", "")
+                full_error = f"Heurist API error: Error code: {error_code}, Message: {error_message}"
+            except Exception:
+                full_error = f"Heurist API error: {e}"
+
+            logger.error(full_error)
+            raise Exception(full_error)
+
+        except Exception as e:
+            logger.error(f"Error generating image with Heurist: {e}")
+            raise Exception(f"Error generating image with Heurist: {str(e)}")

intentkit/skills/heurist/image_generation_arthemy_real.py

@@ -0,0 +1,162 @@
+import logging
+from typing import Optional, Type
+
+import httpx
+from epyxid import XID
+from langchain_core.runnables import RunnableConfig
+from pydantic import BaseModel, Field
+
+from intentkit.skills.heurist.base import HeuristBaseTool
+from intentkit.utils.s3 import store_image
+
+logger = logging.getLogger(__name__)
+
+
+class ImageGenerationArthemyRealInput(BaseModel):
+    """Input for ImageGenerationArthemyReal tool."""
+
+    prompt: str = Field(
+        description="Text prompt describing the image to generate.",
+    )
+    neg_prompt: Optional[str] = Field(
+        default="(worst quality: 1.4), bad quality, nsfw",
+        description="Negative prompt describing what to avoid in the generated image.",
+    )
+    width: Optional[int] = Field(
+        default=1024,
+        le=1024,
+        description="Width of the generated image.",
+    )
+    height: Optional[int] = Field(
+        default=1024,
+        le=1024,
+        description="Height of the generated image.",
+    )
+
+
+class ImageGenerationArthemyReal(HeuristBaseTool):
+    """Tool for generating realistic images using Heurist AI's ArthemyReal model.
+
+    This tool takes a text prompt and uses Heurist's API to generate
+    a realistic image based on the description.
+
+    Attributes:
+        name: The name of the tool.
+        description: A description of what the tool does.
+        args_schema: The schema for the tool's input arguments.
+    """
+
+    name: str = "heurist_image_generation_arthemy_real"
+    description: str = (
+        "Generate realistic images using Heurist AI's ArthemyReal model.\n"
+        "Provide a text prompt describing the realistic image you want to generate.\n"
+        "ArthemyReal specializes in creating photorealistic, lifelike images with fine details.\n"
+        "If you have height and width, remember to specify them.\n"
+    )
+    args_schema: Type[BaseModel] = ImageGenerationArthemyRealInput
+
+    async def _arun(
+        self,
+        prompt: str,
+        neg_prompt: Optional[str] = "(worst quality: 1.4), bad quality, nsfw",
+        width: Optional[int] = 1024,
+        height: Optional[int] = 680,
+        config: RunnableConfig = None,
+        **kwargs,
+    ) -> str:
+        """Implementation of the tool to generate realistic images using Heurist AI's ArthemyReal model.
+
+        Args:
+            prompt: Text prompt describing the image to generate.
+            neg_prompt: Negative prompt describing what to avoid in the generated image.
+            width: Width of the generated image.
+            height: Height of the generated image.
+            config: Configuration for the runnable.
+
+        Returns:
+            str: URL of the generated image.
+        """
+        context = self.context_from_config(config)
+        skill_config = context.config
+
+        # Get the Heurist API key from the skill store
+        if "api_key" in skill_config and skill_config["api_key"]:
+            api_key = skill_config["api_key"]
+            if skill_config.get("rate_limit_number") and skill_config.get(
+                "rate_limit_minutes"
+            ):
+                await self.user_rate_limit_by_category(
+                    context.user_id,
+                    skill_config["rate_limit_number"],
+                    skill_config["rate_limit_minutes"],
+                )
+        else:
+            api_key = self.skill_store.get_system_config("heurist_api_key")
+            await self.user_rate_limit_by_category(context.user_id, 10, 1440)
+
+        # Generate a unique job ID
+        job_id = str(XID())
+
+        # Prepare the request payload
+        payload = {
+            "job_id": job_id,
+            "model_input": {
+                "SD": {
+                    "prompt": prompt,
+                    "neg_prompt": neg_prompt,
+                    "num_iterations": 25,
+                    "width": width,
+                    "height": height,
+                    "guidance_scale": 5,
+                    "seed": -1,
+                }
+            },
+            "model_id": "ArthemyReal",
+            "deadline": 120,
+            "priority": 1,
+        }
+        logger.debug(f"Heurist API payload: {payload}")
+
+        headers = {
+            "Authorization": f"Bearer {api_key}",
+            "Content-Type": "application/json",
+        }
+
+        try:
+            # Make the API request
+            async with httpx.AsyncClient() as client:
+                response = await client.post(
+                    "http://sequencer.heurist.xyz/submit_job",
+                    json=payload,
+                    headers=headers,
+                    timeout=120,
+                )
+                logger.debug(f"Heurist API response: {response.text}")
+                response.raise_for_status()
+
+            # Store the image URL
+            image_url = response.text.strip('"')
+            # Generate a key with agent ID as prefix
+            image_key = f"{context.agent.id}/heurist/{job_id}"
+            # Store the image and get the CDN URL
+            stored_url = await store_image(image_url, image_key)
+
+            # Return the stored image URL
+            return stored_url
+
+        except httpx.HTTPStatusError as e:
+            # Extract error details from response
+            try:
+                error_json = e.response.json()
+                error_code = error_json.get("error", "")
+                error_message = error_json.get("message", "")
+                full_error = f"Heurist API error: Error code: {error_code}, Message: {error_message}"
+            except Exception:
+                full_error = f"Heurist API error: {e}"
+
+            logger.error(full_error)
+            raise Exception(full_error)
+
+        except Exception as e:
+            logger.error(f"Error generating image with Heurist: {e}")
+            raise Exception(f"Error generating image with Heurist: {str(e)}")

intentkit/skills/heurist/image_generation_braindance.py

@@ -0,0 +1,162 @@
+import logging
+from typing import Optional, Type
+
+import httpx
+from epyxid import XID
+from langchain_core.runnables import RunnableConfig
+from pydantic import BaseModel, Field
+
+from intentkit.skills.heurist.base import HeuristBaseTool
+from intentkit.utils.s3 import store_image
+
+logger = logging.getLogger(__name__)
+
+
+class ImageGenerationBrainDanceInput(BaseModel):
+    """Input for ImageGenerationBrainDance tool."""
+
+    prompt: str = Field(
+        description="Text prompt describing the image to generate.",
+    )
+    neg_prompt: Optional[str] = Field(
+        default="(worst quality: 1.4), bad quality, nsfw",
+        description="Negative prompt describing what to avoid in the generated image.",
+    )
+    width: Optional[int] = Field(
+        default=1024,
+        le=1024,
+        description="Width of the generated image.",
+    )
+    height: Optional[int] = Field(
+        default=1024,
+        le=1024,
+        description="Height of the generated image.",
+    )
+
+
+class ImageGenerationBrainDance(HeuristBaseTool):
+    """Tool for generating artistic images using Heurist AI's BrainDance model.
+
+    This tool takes a text prompt and uses Heurist's API to generate
+    an artistic image based on the description.
+
+    Attributes:
+        name: The name of the tool.
+        description: A description of what the tool does.
+        args_schema: The schema for the tool's input arguments.
+    """
+
+    name: str = "heurist_image_generation_braindance"
+    description: str = (
+        "Generate artistic images using Heurist AI's BrainDance model.\n"
+        "Provide a text prompt describing the artistic image you want to generate.\n"
+        "BrainDance specializes in creating unique, artistic interpretations with creative flair.\n"
+        "If you have height and width, remember to specify them.\n"
+    )
+    args_schema: Type[BaseModel] = ImageGenerationBrainDanceInput
+
+    async def _arun(
+        self,
+        prompt: str,
+        neg_prompt: Optional[str] = "(worst quality: 1.4), bad quality, nsfw",
+        width: Optional[int] = 1024,
+        height: Optional[int] = 680,
+        config: RunnableConfig = None,
+        **kwargs,
+    ) -> str:
+        """Implementation of the tool to generate artistic images using Heurist AI's BrainDance model.
+
+        Args:
+            prompt: Text prompt describing the image to generate.
+            neg_prompt: Negative prompt describing what to avoid in the generated image.
+            width: Width of the generated image.
+            height: Height of the generated image.
+            config: Configuration for the runnable.
+
+        Returns:
+            str: URL of the generated image.
+        """
+        context = self.context_from_config(config)
+        skill_config = context.config
+
+        # Get the Heurist API key from the skill store
+        if "api_key" in skill_config and skill_config["api_key"]:
+            api_key = skill_config["api_key"]
+            if skill_config.get("rate_limit_number") and skill_config.get(
+                "rate_limit_minutes"
+            ):
+                await self.user_rate_limit_by_category(
+                    context.user_id,
+                    skill_config["rate_limit_number"],
+                    skill_config["rate_limit_minutes"],
+                )
+        else:
+            api_key = self.skill_store.get_system_config("heurist_api_key")
+            await self.user_rate_limit_by_category(context.user_id, 10, 1440)
+
+        # Generate a unique job ID
+        job_id = str(XID())
+
+        # Prepare the request payload
+        payload = {
+            "job_id": job_id,
+            "model_input": {
+                "SD": {
+                    "prompt": prompt,
+                    "neg_prompt": neg_prompt,
+                    "num_iterations": 25,
+                    "width": width,
+                    "height": height,
+                    "guidance_scale": 5,
+                    "seed": -1,
+                }
+            },
+            "model_id": "BrainDance",
+            "deadline": 120,
+            "priority": 1,
+        }
+        logger.debug(f"Heurist API payload: {payload}")
+
+        headers = {
+            "Authorization": f"Bearer {api_key}",
+            "Content-Type": "application/json",
+        }
+
+        try:
+            # Make the API request
+            async with httpx.AsyncClient() as client:
+                response = await client.post(
+                    "http://sequencer.heurist.xyz/submit_job",
+                    json=payload,
+                    headers=headers,
+                    timeout=120,
+                )
+                logger.debug(f"Heurist API response: {response.text}")
+                response.raise_for_status()
+
+            # Store the image URL
+            image_url = response.text.strip('"')
+            # Generate a key with agent ID as prefix
+            image_key = f"{context.agent.id}/heurist/{job_id}"
+            # Store the image and get the CDN URL
+            stored_url = await store_image(image_url, image_key)
+
+            # Return the stored image URL
+            return stored_url
+
+        except httpx.HTTPStatusError as e:
+            # Extract error details from response
+            try:
+                error_json = e.response.json()
+                error_code = error_json.get("error", "")
+                error_message = error_json.get("message", "")
+                full_error = f"Heurist API error: Error code: {error_code}, Message: {error_message}"
+            except Exception:
+                full_error = f"Heurist API error: {e}"
+
+            logger.error(full_error)
+            raise Exception(full_error)
+
+        except Exception as e:
+            logger.error(f"Error generating image with Heurist: {e}")
+            raise Exception(f"Error generating image with Heurist: {str(e)}")

intentkit/skills/heurist/image_generation_cyber_realistic_xl.py

@@ -0,0 +1,162 @@
+import logging
+from typing import Optional, Type
+
+import httpx
+from epyxid import XID
+from langchain_core.runnables import RunnableConfig
+from pydantic import BaseModel, Field
+
+from intentkit.skills.heurist.base import HeuristBaseTool
+from intentkit.utils.s3 import store_image
+
+logger = logging.getLogger(__name__)
+
+
+class ImageGenerationCyberRealisticXLInput(BaseModel):
+    """Input for ImageGenerationCyberRealisticXL tool."""
+
+    prompt: str = Field(
+        description="Text prompt describing the image to generate.",
+    )
+    neg_prompt: Optional[str] = Field(
+        default="(worst quality: 1.4), bad quality, nsfw",
+        description="Negative prompt describing what to avoid in the generated image.",
+    )
+    width: Optional[int] = Field(
+        default=1024,
+        le=1024,
+        description="Width of the generated image.",
+    )
+    height: Optional[int] = Field(
+        default=680,
+        le=1024,
+        description="Height of the generated image.",
+    )
+
+
+class ImageGenerationCyberRealisticXL(HeuristBaseTool):
+    """Tool for generating hyperrealistic cyberpunk images using Heurist AI's CyberRealisticXL model.
+
+    This tool takes a text prompt and uses Heurist's API to generate
+    a hyperrealistic photograph with a cyberpunk aesthetic based on the description.
+
+    Attributes:
+        name: The name of the tool.
+        description: A description of what the tool does.
+        args_schema: The schema for the tool's input arguments.
+    """
+
+    name: str = "heurist_image_generation_cyber_realistic_xl"
+    description: str = (
+        "Generate hyperrealistic cyberpunk photography using Heurist AI's CyberRealisticXL model.\n"
+        "Provide a text prompt describing the hyperrealistic cyberpunk image you want to generate.\n"
+        "CyberRealisticXL specializes in creating high-quality hyperrealistic photographs with a cyberpunk aesthetic.\n"
+        "If you have height and width, remember to specify them.\n"
+    )
+    args_schema: Type[BaseModel] = ImageGenerationCyberRealisticXLInput
+
+    async def _arun(
+        self,
+        prompt: str,
+        neg_prompt: Optional[str] = "(worst quality: 1.4), bad quality, nsfw",
+        width: Optional[int] = 1024,
+        height: Optional[int] = 680,
+        config: RunnableConfig = None,
+        **kwargs,
+    ) -> str:
+        """Implementation of the tool to generate hyperrealistic cyberpunk images using Heurist AI's CyberRealisticXL model.
+
+        Args:
+            prompt: Text prompt describing the image to generate.
+            neg_prompt: Negative prompt describing what to avoid in the generated image.
+            width: Width of the generated image.
+            height: Height of the generated image.
+            config: Configuration for the runnable.
+
+        Returns:
+            str: URL of the generated image.
+        """
+        context = self.context_from_config(config)
+        skill_config = context.config
+
+        # Get the Heurist API key from the skill store
+        if "api_key" in skill_config and skill_config["api_key"]:
+            api_key = skill_config["api_key"]
+            if skill_config.get("rate_limit_number") and skill_config.get(
+                "rate_limit_minutes"
+            ):
+                await self.user_rate_limit_by_category(
+                    context.user_id,
+                    skill_config["rate_limit_number"],
+                    skill_config["rate_limit_minutes"],
+                )
+        else:
+            api_key = self.skill_store.get_system_config("heurist_api_key")
+            await self.user_rate_limit_by_category(context.user_id, 10, 1440)
+
+        # Generate a unique job ID
+        job_id = str(XID())
+
+        # Prepare the request payload
+        payload = {
+            "job_id": job_id,
+            "model_input": {
+                "SD": {
+                    "prompt": prompt,
+                    "neg_prompt": neg_prompt,
+                    "num_iterations": 25,
+                    "width": width,
+                    "height": height,
+                    "guidance_scale": 5,
+                    "seed": -1,
+                }
+            },
+            "model_id": "CyberRealisticXL",
+            "deadline": 180,
+            "priority": 1,
+        }
+        logger.debug(f"Heurist API payload: {payload}")
+
+        headers = {
+            "Authorization": f"Bearer {api_key}",
+            "Content-Type": "application/json",
+        }
+
+        try:
+            # Make the API request
+            async with httpx.AsyncClient() as client:
+                response = await client.post(
+                    "http://sequencer.heurist.xyz/submit_job",
+                    json=payload,
+                    headers=headers,
+                    timeout=120,
+                )
+                logger.debug(f"Heurist API response: {response.text}")
+                response.raise_for_status()
+
+            # Store the image URL
+            image_url = response.text.strip('"')
+            # Generate a key with agent ID as prefix
+            image_key = f"{context.agent.id}/heurist/{job_id}"
+            # Store the image and get the CDN URL
+            stored_url = await store_image(image_url, image_key)
+
+            # Return the stored image URL
+            return stored_url
+
+        except httpx.HTTPStatusError as e:
+            # Extract error details from response
+            try:
+                error_json = e.response.json()
+                error_code = error_json.get("error", "")
+                error_message = error_json.get("message", "")
+                full_error = f"Heurist API error: Error code: {error_code}, Message: {error_message}"
+            except Exception:
+                full_error = f"Heurist API error: {e}"
+
+            logger.error(full_error)
+            raise Exception(full_error)
+
+        except Exception as e:
+            logger.error(f"Error generating image with Heurist: {e}")
+            raise Exception(f"Error generating image with Heurist: {str(e)}")