intentkit 0.8.17.dev1__py3-none-any.whl → 0.8.17.dev2__py3-none-any.whl

intentkit/skills/twitter/get_timeline.py

@@ -1,5 +1,4 @@
 import logging
-from typing import Type
 
 from pydantic import BaseModel
 
@@ -34,7 +33,7 @@ class TwitterGetTimeline(TwitterBaseTool):
 
     name: str = NAME
     description: str = PROMPT
-    args_schema: Type[BaseModel] = TwitterGetTimelineInput
+    args_schema: type[BaseModel] = TwitterGetTimelineInput
 
     async def _arun(self, **kwargs):
         context = self.get_context()

intentkit/skills/twitter/get_user_by_username.py

@@ -1,5 +1,4 @@
 import logging
-from typing import Type
 
 from pydantic import BaseModel, Field
 
@@ -35,7 +34,7 @@ class TwitterGetUserByUsername(TwitterBaseTool):
 
     name: str = NAME
     description: str = PROMPT
-    args_schema: Type[BaseModel] = TwitterGetUserByUsernameInput
+    args_schema: type[BaseModel] = TwitterGetUserByUsernameInput
 
     async def _arun(self, username: str, **kwargs):
         context = self.get_context()

intentkit/skills/twitter/get_user_tweets.py

@@ -1,5 +1,4 @@
 import logging
-from typing import List, Optional, Type
 
 from pydantic import BaseModel, Field
 
@@ -21,7 +20,7 @@ class TwitterGetUserTweetsInput(BaseModel):
     """Input for TwitterGetUserTweets tool."""
 
     user_id: str = Field(description="The Twitter user ID to fetch tweets from")
-    exclude: Optional[List[str]] = Field(
+    exclude: list[str] | None = Field(
         default=["replies", "retweets"],
         description="Types of tweets to exclude (e.g., 'replies', 'retweets')",
     )
@@ -41,7 +40,7 @@ class TwitterGetUserTweets(TwitterBaseTool):
 
     name: str = NAME
     description: str = PROMPT
-    args_schema: Type[BaseModel] = TwitterGetUserTweetsInput
+    args_schema: type[BaseModel] = TwitterGetUserTweetsInput
 
     async def _arun(self, **kwargs):
         context = self.get_context()

intentkit/skills/twitter/like_tweet.py

@@ -1,5 +1,4 @@
 import logging
-from typing import Type
 
 from langchain_core.tools import ToolException
 from pydantic import BaseModel, Field
@@ -32,7 +31,7 @@ class TwitterLikeTweet(TwitterBaseTool):
 
     name: str = NAME
     description: str = PROMPT
-    args_schema: Type[BaseModel] = TwitterLikeTweetInput
+    args_schema: type[BaseModel] = TwitterLikeTweetInput
 
     async def _arun(self, tweet_id: str, **kwargs):
         context = self.get_context()

intentkit/skills/twitter/post_tweet.py

@@ -1,5 +1,4 @@
 import logging
-from typing import Optional, Type
 
 from langchain_core.tools import ToolException
 from pydantic import BaseModel, Field
@@ -24,7 +23,7 @@ class TwitterPostTweetInput(BaseModel):
         description="Tweet text (280 chars for regular users, 25,000 bytes for verified)",
         max_length=25000,
     )
-    image: Optional[str] = Field(
+    image: str | None = Field(
         default=None, description="Optional URL of an image to attach to the tweet"
     )
 
@@ -42,12 +41,12 @@ class TwitterPostTweet(TwitterBaseTool):
 
     name: str = NAME
     description: str = PROMPT
-    args_schema: Type[BaseModel] = TwitterPostTweetInput
+    args_schema: type[BaseModel] = TwitterPostTweetInput
 
     async def _arun(
         self,
         text: str,
-        image: Optional[str] = None,
+        image: str | None = None,
         **kwargs,
     ):
         context = self.get_context()

intentkit/skills/twitter/reply_tweet.py

@@ -1,5 +1,4 @@
 import logging
-from typing import Optional, Type
 
 from langchain_core.tools import ToolException
 from pydantic import BaseModel, Field
@@ -25,7 +24,7 @@ class TwitterReplyTweetInput(BaseModel):
         description="Tweet text (280 chars for regular users, 25,000 bytes for verified)",
         max_length=25000,
     )
-    image: Optional[str] = Field(
+    image: str | None = Field(
         default=None, description="Optional URL of an image to attach to the reply"
     )
 
@@ -43,13 +42,13 @@ class TwitterReplyTweet(TwitterBaseTool):
 
     name: str = NAME
     description: str = PROMPT
-    args_schema: Type[BaseModel] = TwitterReplyTweetInput
+    args_schema: type[BaseModel] = TwitterReplyTweetInput
 
     async def _arun(
         self,
         tweet_id: str,
         text: str,
-        image: Optional[str] = None,
+        image: str | None = None,
         **kwargs,
     ):
         context = self.get_context()

intentkit/skills/twitter/retweet.py

@@ -1,5 +1,4 @@
 import logging
-from typing import Type
 
 from langchain_core.tools import ToolException
 from pydantic import BaseModel, Field
@@ -32,7 +31,7 @@ class TwitterRetweet(TwitterBaseTool):
 
     name: str = NAME
     description: str = PROMPT
-    args_schema: Type[BaseModel] = TwitterRetweetInput
+    args_schema: type[BaseModel] = TwitterRetweetInput
 
     async def _arun(self, tweet_id: str, **kwargs):
         context = self.get_context()

intentkit/skills/twitter/search_tweets.py

@@ -1,6 +1,5 @@
 import datetime
 import logging
-from typing import Type
 
 from pydantic import BaseModel, Field
 
@@ -33,7 +32,7 @@ class TwitterSearchTweets(TwitterBaseTool):
 
     name: str = NAME
     description: str = PROMPT
-    args_schema: Type[BaseModel] = TwitterSearchTweetsInput
+    args_schema: type[BaseModel] = TwitterSearchTweetsInput
 
     async def _arun(self, query: str, **kwargs):
         context = self.get_context()

intentkit/skills/unrealspeech/base.py

@@ -1,5 +1,3 @@
-from typing import Type
-
 from langchain_core.tools.base import ToolException
 from pydantic import BaseModel, Field
 
@@ -11,7 +9,7 @@ class UnrealSpeechBaseTool(IntentKitSkill):
 
     name: str = Field(description="The name of the tool")
     description: str = Field(description="A description of what the tool does")
-    args_schema: Type[BaseModel]
+    args_schema: type[BaseModel]
 
     def get_api_key(self) -> str:
         context = self.get_context()

intentkit/skills/unrealspeech/text_to_speech.py

@@ -1,6 +1,6 @@
 import logging
 import os
-from typing import Any, Dict, Literal, Optional, Type
+from typing import Any, Literal
 
 import httpx
 from langchain_core.callbacks.manager import CallbackManagerForToolRun
@@ -31,7 +31,7 @@ class TextToSpeechInput(BaseModel):
         default=0.0,
     )
 
-    timestamp_type: Optional[Literal["word", "sentence"]] = Field(
+    timestamp_type: Literal["word", "sentence"] | None = Field(
         description="The type of timestamps to include in the response. 'word' for word-level timestamps, 'sentence' for sentence-level, or None for no timestamps.",
         default="word",
     )
@@ -52,9 +52,9 @@ class TextToSpeech(UnrealSpeechBaseTool):
         "Returns URLs to the generated audio file and word-level timestamps.\n"
         "Provides various voice options and speech customization parameters."
     )
-    args_schema: Type[BaseModel] = TextToSpeechInput
+    args_schema: type[BaseModel] = TextToSpeechInput
 
-    def get_env_var(self, env_var_name: str) -> Optional[str]:
+    def get_env_var(self, env_var_name: str) -> str | None:
         """Helper method to get environment variables."""
         return os.environ.get(env_var_name)
 
@@ -64,11 +64,11 @@ class TextToSpeech(UnrealSpeechBaseTool):
         voice_id: str = "af_sarah",
         bitrate: str = "192k",
         speed: float = 0.0,
-        timestamp_type: Optional[Literal["word", "sentence"]] = "word",
-        config: Optional[Any] = None,
-        run_manager: Optional[CallbackManagerForToolRun] = None,
+        timestamp_type: Literal["word", "sentence"] | None = "word",
+        config: Any | None = None,
+        run_manager: CallbackManagerForToolRun | None = None,
         **kwargs,
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         """Run the tool to convert text to speech."""
 
         # Get the API key from context config if available

intentkit/skills/venice_audio/__init__.py

@@ -1,5 +1,5 @@
 import logging
-from typing import List, Literal, Optional, TypedDict
+from typing import Literal, TypedDict
 
 from intentkit.skills.base import SkillConfig, SkillState
 from intentkit.skills.venice_audio.base import VeniceAudioBaseTool
@@ -7,7 +7,6 @@ from intentkit.skills.venice_audio.venice_audio import VeniceAudioTool
 
 logger = logging.getLogger(__name__)
 
-
 _cache: dict[str, VeniceAudioBaseTool] = {}
 
 _SKILL_NAME_TO_CLASS_MAP = {
@@ -24,15 +23,15 @@ class Config(SkillConfig):
     enabled: bool
     voice_model: Literal["af_heart", "bm_lewis", "custom"]
     states: SkillStates  # type: ignore
-    api_key_provider: Optional[Literal["agent_owner"]]
+    api_key_provider: Literal["agent_owner"] | None
 
     # conditionally required
-    api_key: Optional[str]
-    voice_model_custom: Optional[list[str]]
+    api_key: str | None
+    voice_model_custom: list[str] | None
 
     # optional
-    rate_limit_number: Optional[int]
-    rate_limit_minutes: Optional[int]
+    rate_limit_number: int | None
+    rate_limit_minutes: int | None
 
 
 async def get_skills(
@@ -54,7 +53,7 @@ async def get_skills(
     if not config.get("enabled", False):
         return []
 
-    available_skills: List[VeniceAudioBaseTool] = []
+    available_skills: list[VeniceAudioBaseTool] = []
     skill_states = config.get("states", {})
 
     # Iterate through all known skills defined in the map
@@ -79,7 +78,7 @@ async def get_skills(
 
 def get_venice_audio_skill(
     name: str,
-) -> Optional[VeniceAudioBaseTool]:
+) -> VeniceAudioBaseTool | None:
     """
     Factory function to get a cached Venice Audio skill instance by name.
 

intentkit/skills/venice_audio/base.py

@@ -1,5 +1,4 @@
 import logging
-from typing import Dict, List, Optional, Tuple, Type
 
 from langchain_core.tools.base import ToolException
 from pydantic import BaseModel, Field
@@ -15,7 +14,7 @@ class VeniceAudioBaseTool(IntentKitSkill):
 
     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
+    args_schema: type[BaseModel]  # type: ignore
 
     @property
     def category(self) -> str:
@@ -23,13 +22,13 @@ class VeniceAudioBaseTool(IntentKitSkill):
 
     def validate_voice_model(
         self, context, voice_model: str
-    ) -> Tuple[bool, Optional[Dict[str, object]]]:
+    ) -> 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] = []
+        allowed_voice_models: list[str] = []
 
         if selected_model == "custom":
             allowed_voice_models = custom_models or []

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.
-    """
-
-    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: 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/venice_audio.py

@@ -1,7 +1,7 @@
 import hashlib
 import json
 import logging
-from typing import Any, Dict, Optional, Type
+from typing import Any
 
 import httpx
 from pydantic import BaseModel
@@ -32,16 +32,16 @@ class VeniceAudioTool(VeniceAudioBaseTool):
         "Requires input text. Optional parameters include speed (0.25-4.0, default 1.0) "
         "and audio format (mp3, opus, aac, flac, wav, pcm, default mp3)."
     )
-    args_schema: Type[BaseModel] = VeniceAudioInput
+    args_schema: type[BaseModel] = VeniceAudioInput
 
     async def _arun(
         self,
         voice_input: str,
         voice_model: str,
-        speed: Optional[float] = 1.0,
-        response_format: Optional[AllowedAudioFormat] = "mp3",
+        speed: float | None = 1.0,
+        response_format: AllowedAudioFormat | None = "mp3",
         **kwargs,  # type: ignore
-    ) -> Dict[str, Any]:
+    ) -> dict[str, Any]:
         """
         Generates audio using the configured voice model via Venice AI TTS /audio/speech endpoint.
         Stores the resulting audio using the generic S3 helper.
@@ -92,7 +92,7 @@ class VeniceAudioTool(VeniceAudioBaseTool):
             await self.apply_rate_limit(context)
 
             # --- Prepare API Call ---
-            payload: Dict[str, Any] = {
+            payload: dict[str, Any] = {
                 "model": tts_model_id,
                 "input": voice_input,
                 "voice": voice_model,

intentkit/skills/venice_image/__init__.py

@@ -1,5 +1,5 @@
 import logging
-from typing import NotRequired, Optional, TypedDict
+from typing import NotRequired, TypedDict
 
 from intentkit.skills.base import (
     SkillConfig,
@@ -61,12 +61,12 @@ class Config(SkillConfig):
     enabled: bool  # Keep standard enabled flag
     states: SkillStates
     api_key_provider: str = "agent_owner"
-    api_key: NotRequired[Optional[str]]  # Explicitly Optional
+    api_key: NotRequired[str | None]  # Explicitly Optional
     safe_mode: NotRequired[bool]  # Defaults handled in base or usage
     hide_watermark: NotRequired[bool]  # Defaults handled in base or usage
     negative_prompt: NotRequired[str]  # Defaults handled in base or usage
-    rate_limit_number: NotRequired[Optional[int]]  # Explicitly Optional
-    rate_limit_minutes: NotRequired[Optional[int]]  # Explicitly Optional
+    rate_limit_number: NotRequired[int | None]  # Explicitly Optional
+    rate_limit_minutes: NotRequired[int | None]  # Explicitly Optional
 
 
 _SKILL_NAME_TO_CLASS_MAP: dict[str, type[VeniceImageBaseTool]] = {
@@ -122,7 +122,7 @@ async def get_skills(
 def get_venice_image_skill(
     name: str,
     config: "Config",
-) -> Optional[VeniceImageBaseTool]:
+) -> VeniceImageBaseTool | None:
     """
     Factory function to get a cached Venice Image skill instance by name.