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

intentkit/skills/venice_audio/base.py

@@ -1,121 +1,117 @@
-import logging
-from typing import Dict, List, Optional, Tuple, Type
-
-from langchain.tools.base import ToolException
-from pydantic import BaseModel, Field
-
-from intentkit.abstracts.skill import SkillStoreABC
-from intentkit.skills.base import IntentKitSkill
-
-logger = logging.getLogger(__name__)
-
-
-class VeniceAudioBaseTool(IntentKitSkill):
-    """Base class for Venice Audio tools."""
-
-    name: str = Field(default="venice_base_tool", description="The name of the tool")
-    description: str = Field(description="A description of what the tool does")
-    args_schema: Type[BaseModel]  # type: ignore
-    skill_store: SkillStoreABC = Field(
-        description="The skill store for persisting data"
-    )
-
-    @property
-    def category(self) -> str:
-        return "venice_audio"
-
-    def validate_voice_model(
-        self, context, voice_model: str
-    ) -> Tuple[bool, Optional[Dict[str, object]]]:
-        config = context.config
-
-        selected_model = config.get("voice_model")
-        custom_models = config.get("voice_model_custom", [])
-
-        allowed_voice_models: List[str] = []
-
-        if selected_model == "custom":
-            allowed_voice_models = custom_models or []
-        else:
-            allowed_voice_models = [selected_model] if selected_model else []
-
-        if voice_model not in allowed_voice_models:
-            return False, {
-                "error": f'"{voice_model}" is not allowed',
-                "allowed": allowed_voice_models,
-                "suggestion": "please try again with allowed voice model",
-            }
-
-        return True, None
-
-    def get_api_key(self) -> str:
-        """
-        Retrieves the Venice AI API key based on the api_key_provider setting.
-
-        Returns:
-            The API key if found.
-
-        Raises:
-            ToolException: If the API key is not found or provider is invalid.
-        """
-        try:
-            context = self.get_context()
-            skill_config = context.agent.skill_config(self.category)
-            api_key_provider = skill_config.get("api_key_provider")
-            if api_key_provider == "agent_owner":
-                agent_api_key = skill_config.get("api_key")
-                if agent_api_key:
-                    logger.debug(
-                        f"Using agent-specific Venice API key for skill {self.name} in category {self.category}"
-                    )
-                    return agent_api_key
-                raise ToolException(
-                    f"No agent-owned Venice API key found for skill '{self.name}' in category '{self.category}'."
-                )
-
-            elif api_key_provider == "platform":
-                system_api_key = self.skill_store.get_system_config("venice_api_key")
-                if system_api_key:
-                    logger.debug(
-                        f"Using system Venice API key for skill {self.name} in category {self.category}"
-                    )
-                    return system_api_key
-                raise ToolException(
-                    f"No platform-hosted Venice API key found for skill '{self.name}' in category '{self.category}'."
-                )
-
-            else:
-                raise ToolException(
-                    f"Invalid API key provider '{api_key_provider}' for skill '{self.name}'"
-                )
-
-        except Exception as e:
-            raise ToolException(f"Failed to retrieve Venice API key: {str(e)}") from e
-
-    async def apply_rate_limit(self, context) -> None:
-        """
-        Applies rate limiting ONLY if specified in the agent's config ('skill_config').
-        Checks for 'rate_limit_number' and 'rate_limit_minutes'.
-        If not configured, NO rate limiting is applied.
-        Raises ConnectionAbortedError if the configured limit is exceeded.
-        """
-        skill_config = context.agent.skill_config(self.category)
-        user_id = context.agent.id
-
-        # Get agent-specific limits safely
-        limit_num = skill_config.get("rate_limit_number")
-        limit_min = skill_config.get("rate_limit_minutes")
-
-        # Apply limit ONLY if both values are present and valid (truthy check handles None and 0)
-        if limit_num and limit_min:
-            limit_source = "Agent"
-            logger.debug(
-                f"Applying {limit_source} rate limit ({limit_num}/{limit_min} min) for user {user_id} on {self.name}"
-            )
-            if user_id:
-                await self.user_rate_limit_by_category(user_id, limit_num, limit_min)
-        else:
-            # No valid agent configuration found, so do nothing.
-            logger.debug(
-                f"No agent rate limits configured for category '{self.category}'. Skipping rate limit for user {user_id}."
-            )
+import logging
+
+from langchain_core.tools.base import ToolException
+from pydantic import BaseModel, Field
+
+from intentkit.config.config import config
+from intentkit.skills.base import IntentKitSkill
+
+logger = logging.getLogger(__name__)
+
+
+class VeniceAudioBaseTool(IntentKitSkill):
+    """Base class for Venice Audio tools."""
+
+    name: str = Field(default="venice_base_tool", description="The name of the tool")
+    description: str = Field(description="A description of what the tool does")
+    args_schema: type[BaseModel]  # type: ignore
+
+    @property
+    def category(self) -> str:
+        return "venice_audio"
+
+    def validate_voice_model(
+        self, context, voice_model: str
+    ) -> tuple[bool, dict[str, object] | None]:
+        config = context.config
+
+        selected_model = config.get("voice_model")
+        custom_models = config.get("voice_model_custom", [])
+
+        allowed_voice_models: list[str] = []
+
+        if selected_model == "custom":
+            allowed_voice_models = custom_models or []
+        else:
+            allowed_voice_models = [selected_model] if selected_model else []
+
+        if voice_model not in allowed_voice_models:
+            return False, {
+                "error": f'"{voice_model}" is not allowed',
+                "allowed": allowed_voice_models,
+                "suggestion": "please try again with allowed voice model",
+            }
+
+        return True, None
+
+    def get_api_key(self) -> str:
+        """
+        Retrieves the Venice AI API key based on the api_key_provider setting.
+
+        Returns:
+            The API key if found.
+
+        Raises:
+            ToolException: If the API key is not found or provider is invalid.
+        """
+        try:
+            context = self.get_context()
+            skill_config = context.agent.skill_config(self.category)
+            api_key_provider = skill_config.get("api_key_provider")
+            if api_key_provider == "agent_owner":
+                agent_api_key = skill_config.get("api_key")
+                if agent_api_key:
+                    logger.debug(
+                        f"Using agent-specific Venice API key for skill {self.name} in category {self.category}"
+                    )
+                    return agent_api_key
+                raise ToolException(
+                    f"No agent-owned Venice API key found for skill '{self.name}' in category '{self.category}'."
+                )
+
+            elif api_key_provider == "platform":
+                system_api_key = config.venice_api_key
+                if system_api_key:
+                    logger.debug(
+                        f"Using system Venice API key for skill {self.name} in category {self.category}"
+                    )
+                    return system_api_key
+                raise ToolException(
+                    f"No platform-hosted Venice API key found for skill '{self.name}' in category '{self.category}'."
+                )
+
+            else:
+                raise ToolException(
+                    f"Invalid API key provider '{api_key_provider}' for skill '{self.name}'"
+                )
+
+        except Exception as e:
+            raise ToolException(f"Failed to retrieve Venice API key: {str(e)}") from e
+
+    async def apply_rate_limit(self, context) -> None:
+        """
+        Applies rate limiting ONLY if specified in the agent's config ('skill_config').
+        Checks for 'rate_limit_number' and 'rate_limit_minutes'.
+        If not configured, NO rate limiting is applied.
+        Raises ConnectionAbortedError if the configured limit is exceeded.
+        """
+        skill_config = context.agent.skill_config(self.category)
+        user_id = context.agent.id
+
+        # Get agent-specific limits safely
+        limit_num = skill_config.get("rate_limit_number")
+        limit_min = skill_config.get("rate_limit_minutes")
+
+        # Apply limit ONLY if both values are present and valid (truthy check handles None and 0)
+        if limit_num and limit_min:
+            limit_source = "Agent"
+            logger.debug(
+                f"Applying {limit_source} rate limit ({limit_num}/{limit_min} min) for user {user_id} on {self.name}"
+            )
+            if user_id:
+                await self.user_rate_limit_by_category(limit_num, limit_min * 60)
+        else:
+            # No valid agent configuration found, so do nothing.
+            logger.debug(
+                f"No agent rate limits configured for category '{self.category}'. Skipping rate limit for user {user_id}."
+            )

intentkit/skills/venice_audio/input.py

@@ -1,41 +1,41 @@
-from typing import Literal, Optional
-
-from pydantic import BaseModel, Field
-
-# Define the allowed format literals based on the API documentation
-AllowedAudioFormat = Literal["mp3", "opus", "aac", "flac", "wav", "pcm"]
-
-
-class VeniceAudioInput(BaseModel):
-    """
-    Input schema for Venice AI Text-to-Speech (/audio/speech endpoint).
-    Defines parameters controllable by the user when invoking the tool.
-    """
-
-    input: str = Field(
-        ...,  # Ellipsis (...) indicates this field is required
-        description="The text to generate audio for. Maximum length is 4096 characters.",
-        min_length=1,  # As per API docs: Required string length: 1
-        max_length=4096,  # As per API docs: The maximum length is 4096 characters.
-    )
-
-    voice_model: str = Field(
-        description="voice model to used to generate voice from text_to_speech tool."
-    )
-
-    speed: Optional[float] = Field(
-        default=1.0,  # As per API docs: default: 1 (using float for consistency)
-        description="The speed of the generated audio. 1.0 is normal speed. Allowed range: 0.25 to 4.0.",
-        ge=0.25,  # As per API docs: Required range: 0.25 <= x
-        le=4.0,  # As per API docs: Required range: x <= 4
-    )
-
-    response_format: Optional[AllowedAudioFormat] = Field(
-        default="mp3",  # As per API docs: default: mp3
-        description="The desired audio format for the output file.",
-    )
-
-    # --- Note on other API parameters ---
-    # 'model': Currently hardcoded to 'tts-kokoro' in VeniceAudioBaseTool._arun. Could be added here if needed.
-    # 'voice': Handled by the 'voice_model' attribute of the specific VeniceAudioBaseTool instance. Not typically set via input schema.
-    # 'streaming': Currently hardcoded to False in VeniceAudioBaseTool._arun. Could be added here if streaming support is implemented.
+from typing import Literal
+
+from pydantic import BaseModel, Field
+
+# Define the allowed format literals based on the API documentation
+AllowedAudioFormat = Literal["mp3", "opus", "aac", "flac", "wav", "pcm"]
+
+
+class VeniceAudioInput(BaseModel):
+    """
+    Input schema for Venice AI Text-to-Speech (/audio/speech endpoint).
+    Defines parameters controllable by the user when invoking the tool.
+    """
+
+    voice_input: str = Field(
+        ...,  # Ellipsis (...) indicates this field is required
+        description="The text to generate audio for. Maximum length is 4096 characters.",
+        min_length=1,  # As per API docs: Required string length: 1
+        max_length=4096,  # As per API docs: The maximum length is 4096 characters.
+    )
+
+    voice_model: str = Field(
+        description="voice model to used to generate voice from text_to_speech tool."
+    )
+
+    speed: float | None = Field(
+        default=1.0,  # As per API docs: default: 1 (using float for consistency)
+        description="The speed of the generated audio. 1.0 is normal speed. Allowed range: 0.25 to 4.0.",
+        ge=0.25,  # As per API docs: Required range: 0.25 <= x
+        le=4.0,  # As per API docs: Required range: x <= 4
+    )
+
+    response_format: AllowedAudioFormat | None = Field(
+        default="mp3",  # As per API docs: default: mp3
+        description="The desired audio format for the output file.",
+    )
+
+    # --- Note on other API parameters ---
+    # 'model': Currently hardcoded to 'tts-kokoro' in VeniceAudioBaseTool._arun. Could be added here if needed.
+    # 'voice': Handled by the 'voice_model' attribute of the specific VeniceAudioBaseTool instance. Not typically set via input schema.
+    # 'streaming': Currently hardcoded to False in VeniceAudioBaseTool._arun. Could be added here if streaming support is implemented.

intentkit/skills/venice_audio/schema.json

@@ -1,152 +1,151 @@
-{
-    "$schema": "http://json-schema.org/draft-07/schema#",
-    "title": "Venice Audio Skills",
-    "x-icon": "https://ai.service.crestal.dev/skills/venice_audio/venice_audio.jpg",
-    "description": "Configuration for the Venice Audio skill.",
-    "type": "object",
-    "x-tags": [
-        "AI",
-        "Audio",
-        "Text to Speech"
-    ],
-    "properties": {
-        "enabled": {
-            "type": "boolean",
-            "title": "Enabled",
-            "description": "Enable or disable the Venice Audio skill.",
-            "default": false
-        },
-        "voice_model": {
-            "type": "string",
-            "title": "Voice Model",
-            "x-link": "[Listen Voice Example](https://huggingface.co/spaces/hexgrad/Kokoro-TTS)",
-            "enum": [
-                "af_heart",
-                "bm_lewis",
-                "custom"
-            ],
-            "x-enum-title": [
-                "af_heart (default female)",
-                "bm_lewis (default male)",
-                "Custom"
-            ],
-            "description": "Text to speech tool",
-            "default": "af_heart"
-        },
-        "states": {
-            "type": "object",
-            "title": "Skill States",
-            "description": "Enable/disable specific voice models. Only enable one if you want a consistent characteristic for your agent. See docs for voice details and quality grades.",
-            "properties": {
-                "text_to_speech": {
-                    "type": "string",
-                    "title": "Text to Speech",
-                    "enum": [
-                        "disabled",
-                        "public",
-                        "private"
-                    ],
-                    "x-enum-title": [
-                        "Disabled",
-                        "Agent Owner + All Users",
-                        "Agent Owner Only"
-                    ],
-                    "description": "Text to speech tool",
-                    "default": "disabled"
-                }
-            }
-        },
-        "api_key_provider": {
-            "type": "string",
-            "title": "API Key Provider",
-            "description": "Provider of the API key",
-            "enum": [
-                "agent_owner"
-            ],
-            "x-enum-title": [
-                "Owner Provided"
-            ],
-            "default": "agent_owner"
-        }
-    },
-    "required": [
-        "states",
-        "enabled"
-    ],
-    "allOf": [
-        {
-            "if": {
-                "properties": {
-                    "voice_model": {
-                        "const": "custom"
-                    }
-                }
-            },
-            "then": {
-                "properties": {
-                    "voice_model_custom": {
-                        "type": "array",
-                        "items": {
-                            "type": "string"
-                        },
-                        "title": "Voice Model (Custom)",
-                        "x-link": "[Supported Voice Model](https://docs.venice.ai/api-reference/endpoint/audio/speech#body-voice)",
-                        "description": "You can add one or more custom voice models.",
-                        "default": [
-                            "af_heart",
-                            "bm_lewis"
-                        ]
-                    }
-                },
-                "required": [
-                    "voice_model_custom"
-                ]
-            }
-        },
-        {
-            "if": {
-                "allOf": [
-                    {
-                        "properties": {
-                            "enabled": {
-                                "const": true
-                            }
-                        }
-                    },
-                    {
-                        "properties": {
-                            "api_key_provider": {
-                                "const": "agent_owner"
-                            }
-                        }
-                    }
-                ]
-            },
-            "then": {
-                "properties": {
-                    "api_key": {
-                        "type": "string",
-                        "title": "Venice API Key",
-                        "x-link": "[Get your API key](https://venice.ai/)",
-                        "x-sensitive": true,
-                        "description": "API Key for authenticating with the Venice AI API."
-                    },
-                    "rate_limit_number": {
-                        "type": "integer",
-                        "title": "Rate Limit Number",
-                        "description": "Number of requests allowed per time window."
-                    },
-                    "rate_limit_minutes": {
-                        "type": "integer",
-                        "title": "Rate Limit Minutes",
-                        "description": "Time window in minutes for rate limiting."
-                    }
-                },
-                "required": [
-                    "api_key"
-                ]
-            }
-        }
-    ],
-    "additionalProperties": true
-}
+{
+  "$schema": "http://json-schema.org/draft-07/schema#",
+  "title": "Venice Audio Skills",
+  "x-icon": "https://ai.service.crestal.dev/skills/venice_audio/venice_audio.jpg",
+  "description": "Configuration for the Venice Audio skill.",
+  "type": "object",
+  "x-tags": [
+    "AI",
+    "Audio"
+  ],
+  "properties": {
+    "enabled": {
+      "type": "boolean",
+      "title": "Enabled",
+      "description": "Enable or disable the Venice Audio skill.",
+      "default": false
+    },
+    "voice_model": {
+      "type": "string",
+      "title": "Voice Model",
+      "x-link": "[Listen Voice Example](https://huggingface.co/spaces/hexgrad/Kokoro-TTS)",
+      "enum": [
+        "af_heart",
+        "bm_lewis",
+        "custom"
+      ],
+      "x-enum-title": [
+        "af_heart (default female)",
+        "bm_lewis (default male)",
+        "Custom"
+      ],
+      "description": "Text to speech tool",
+      "default": "af_heart"
+    },
+    "states": {
+      "type": "object",
+      "title": "Skill States",
+      "description": "Enable/disable specific voice models. Only enable one if you want a consistent characteristic for your agent. See docs for voice details and quality grades.",
+      "properties": {
+        "text_to_speech": {
+          "type": "string",
+          "title": "Text to Speech",
+          "enum": [
+            "disabled",
+            "public",
+            "private"
+          ],
+          "x-enum-title": [
+            "Disabled",
+            "Agent Owner + All Users",
+            "Agent Owner Only"
+          ],
+          "description": "Text to speech tool",
+          "default": "disabled"
+        }
+      }
+    },
+    "api_key_provider": {
+      "type": "string",
+      "title": "API Key Provider",
+      "description": "Provider of the API key",
+      "enum": [
+        "agent_owner"
+      ],
+      "x-enum-title": [
+        "Owner Provided"
+      ],
+      "default": "agent_owner"
+    }
+  },
+  "required": [
+    "states",
+    "enabled"
+  ],
+  "allOf": [
+    {
+      "if": {
+        "properties": {
+          "voice_model": {
+            "const": "custom"
+          }
+        }
+      },
+      "then": {
+        "properties": {
+          "voice_model_custom": {
+            "type": "array",
+            "items": {
+              "type": "string"
+            },
+            "title": "Voice Model (Custom)",
+            "x-link": "[Supported Voice Model](https://docs.venice.ai/api-reference/endpoint/audio/speech#body-voice)",
+            "description": "You can add one or more custom voice models.",
+            "default": [
+              "af_heart",
+              "bm_lewis"
+            ]
+          }
+        },
+        "required": [
+          "voice_model_custom"
+        ]
+      }
+    },
+    {
+      "if": {
+        "allOf": [
+          {
+            "properties": {
+              "enabled": {
+                "const": true
+              }
+            }
+          },
+          {
+            "properties": {
+              "api_key_provider": {
+                "const": "agent_owner"
+              }
+            }
+          }
+        ]
+      },
+      "then": {
+        "properties": {
+          "api_key": {
+            "type": "string",
+            "title": "Venice API Key",
+            "x-link": "[Get your API key](https://venice.ai/)",
+            "x-sensitive": true,
+            "description": "API Key for authenticating with the Venice AI API."
+          },
+          "rate_limit_number": {
+            "type": "integer",
+            "title": "Rate Limit Number",
+            "description": "Number of requests allowed per time window."
+          },
+          "rate_limit_minutes": {
+            "type": "integer",
+            "title": "Rate Limit Minutes",
+            "description": "Time window in minutes for rate limiting."
+          }
+        },
+        "required": [
+          "api_key"
+        ]
+      }
+    }
+  ],
+  "additionalProperties": true
+}