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

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.",
+    )

intentkit/skills/venice_image/utils.py

@@ -1,78 +1,77 @@
-import base64
-import io
-import logging
-from typing import Optional
-
-import filetype
-import httpx
-from PIL import Image
-from pydantic import HttpUrl
-
-from intentkit.skills.base import ToolException
-
-logger = logging.getLogger(__name__)
-
-
-async def fetch_image_as_bytes(image_url: HttpUrl) -> bytes:
-    """Fetches image bytes from a given URL. Converts unsupported formats to PNG using Pillow.
-
-    Raises:
-        ToolException: If fetching or converting the image fails.
-    """
-    try:
-        async with httpx.AsyncClient(timeout=90) as client:
-            response = await client.get(str(image_url), follow_redirects=True)
-            response.raise_for_status()
-
-            original_bytes = response.content
-
-            # Guess file type from content
-            kind = filetype.guess(original_bytes)
-            detected_ext = kind.extension if kind else None
-            detected_mime = kind.mime if kind else "unknown"
-
-            if not detected_ext or not detected_mime.startswith("image/"):
-                msg = f"URL {image_url} did not return a recognizable image format. Detected: {detected_mime}"
-                logger.error(msg)
-                raise ToolException(msg)
-
-            if detected_ext in ("jpg", "jpeg", "png"):
-                return original_bytes
-
-            # Convert unsupported image to PNG
-            try:
-                img = Image.open(io.BytesIO(original_bytes)).convert("RGBA")
-                with io.BytesIO() as output:
-                    img.save(output, format="PNG")
-                    logger.info(
-                        f"Converted unsupported image type '{detected_ext}' to PNG."
-                    )
-                    return output.getvalue()
-            except Exception as e:
-                msg = f"Failed to convert image ({detected_ext}) to PNG: {e}"
-                logger.error(msg, exc_info=True)
-                raise ToolException(msg) from e
-
-    except httpx.HTTPStatusError as e:
-        msg = f"HTTP error fetching image {image_url}: Status {e.response.status_code}"
-        logger.error(msg)
-        raise ToolException(msg) from e
-    except httpx.RequestError as e:
-        msg = f"Network error fetching image {image_url}: {e}"
-        logger.error(msg)
-        raise ToolException(msg) from e
-    except Exception as e:
-        msg = f"Unexpected error fetching image {image_url}: {e}"
-        logger.error(msg, exc_info=True)
-        raise ToolException(msg) from e
-
-
-async def fetch_image_as_base64(image_url: HttpUrl) -> Optional[str]:
-    """Fetches an image from the URL and returns the image as a Base64-encoded string."""
-    image_bytes = await fetch_image_as_bytes(image_url)
-
-    if image_bytes is None:
-        return None
-
-    # Convert image bytes to a Base64-encoded string
-    return base64.b64encode(image_bytes).decode("utf-8")
+import base64
+import io
+import logging
+
+import filetype
+import httpx
+from PIL import Image
+from pydantic import HttpUrl
+
+from intentkit.skills.base import ToolException
+
+logger = logging.getLogger(__name__)
+
+
+async def fetch_image_as_bytes(image_url: HttpUrl) -> bytes:
+    """Fetches image bytes from a given URL. Converts unsupported formats to PNG using Pillow.
+
+    Raises:
+        ToolException: If fetching or converting the image fails.
+    """
+    try:
+        async with httpx.AsyncClient(timeout=90) as client:
+            response = await client.get(str(image_url), follow_redirects=True)
+            response.raise_for_status()
+
+            original_bytes = response.content
+
+            # Guess file type from content
+            kind = filetype.guess(original_bytes)
+            detected_ext = kind.extension if kind else None
+            detected_mime = kind.mime if kind else "unknown"
+
+            if not detected_ext or not detected_mime.startswith("image/"):
+                msg = f"URL {image_url} did not return a recognizable image format. Detected: {detected_mime}"
+                logger.error(msg)
+                raise ToolException(msg)
+
+            if detected_ext in ("jpg", "jpeg", "png"):
+                return original_bytes
+
+            # Convert unsupported image to PNG
+            try:
+                img = Image.open(io.BytesIO(original_bytes)).convert("RGBA")
+                with io.BytesIO() as output:
+                    img.save(output, format="PNG")
+                    logger.info(
+                        f"Converted unsupported image type '{detected_ext}' to PNG."
+                    )
+                    return output.getvalue()
+            except Exception as e:
+                msg = f"Failed to convert image ({detected_ext}) to PNG: {e}"
+                logger.error(msg, exc_info=True)
+                raise ToolException(msg) from e
+
+    except httpx.HTTPStatusError as e:
+        msg = f"HTTP error fetching image {image_url}: Status {e.response.status_code}"
+        logger.error(msg)
+        raise ToolException(msg) from e
+    except httpx.RequestError as e:
+        msg = f"Network error fetching image {image_url}: {e}"
+        logger.error(msg)
+        raise ToolException(msg) from e
+    except Exception as e:
+        msg = f"Unexpected error fetching image {image_url}: {e}"
+        logger.error(msg, exc_info=True)
+        raise ToolException(msg) from e
+
+
+async def fetch_image_as_base64(image_url: HttpUrl) -> str | None:
+    """Fetches an image from the URL and returns the image as a Base64-encoded string."""
+    image_bytes = await fetch_image_as_bytes(image_url)
+
+    if image_bytes is None:
+        return None
+
+    # Convert image bytes to a Base64-encoded string
+    return base64.b64encode(image_bytes).decode("utf-8")

intentkit/skills/web_scraper/base.py

@@ -1,5 +1,3 @@
-from typing import Type
-
 from langchain_core.tools.base import ToolException
 from pydantic import BaseModel, Field
 
@@ -12,7 +10,7 @@ class WebScraperBaseTool(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]
 
     @property
     def category(self) -> str:

intentkit/skills/web_scraper/document_indexer.py

@@ -1,5 +1,4 @@
 import logging
-from typing import Type
 
 from pydantic import BaseModel, Field
 
@@ -65,7 +64,7 @@ class DocumentIndexer(WebScraperBaseTool):
         "Perfect for adding content from Google Docs, Notion pages, PDFs, or any other document sources. "
         "The indexed content can then be queried using the query_indexed_content tool."
     )
-    args_schema: Type[BaseModel] = DocumentIndexerInput
+    args_schema: type[BaseModel] = DocumentIndexerInput
 
     async def _arun(
         self,

intentkit/skills/web_scraper/scrape_and_index.py

@@ -1,5 +1,4 @@
 import logging
-from typing import List, Type
 
 from pydantic import BaseModel, Field
 
@@ -19,7 +18,7 @@ logger = logging.getLogger(__name__)
 class ScrapeAndIndexInput(BaseModel):
     """Input for ScrapeAndIndex tool."""
 
-    urls: List[str] = Field(
+    urls: list[str] = Field(
         description="List of URLs to scrape and index. Each URL should be a valid web address starting with http:// or https://",
         min_items=1,
         max_items=25,
@@ -67,11 +66,11 @@ class ScrapeAndIndex(WebScraperBaseTool):
         "Use this tool to collect and index web content that you want to reference later.\n"
         "The indexed content can then be queried using the query_indexed_content tool."
     )
-    args_schema: Type[BaseModel] = ScrapeAndIndexInput
+    args_schema: type[BaseModel] = ScrapeAndIndexInput
 
     async def _arun(
         self,
-        urls: List[str],
+        urls: list[str],
         chunk_size: int = DEFAULT_CHUNK_SIZE,
         chunk_overlap: int = DEFAULT_CHUNK_OVERLAP,
         **kwargs,
@@ -171,7 +170,7 @@ class QueryIndexedContent(WebScraperBaseTool):
         "Use this tool to search through content that was previously scraped and indexed.\n"
         "This tool can help answer questions based on the indexed web content."
     )
-    args_schema: Type[BaseModel] = QueryIndexInput
+    args_schema: type[BaseModel] = QueryIndexInput
 
     async def _arun(
         self,

intentkit/skills/web_scraper/utils.py

@@ -10,7 +10,6 @@ import base64
 import logging
 import os
 import tempfile
-from typing import Dict, List, Optional, Tuple
 
 from langchain_community.vectorstores import FAISS
 from langchain_core.documents import Document
@@ -63,7 +62,7 @@ METADATA_KEY_PREFIX = "indexed_urls"
 class VectorStoreManager:
     """Manages vector store operations including creation, saving, loading, and merging."""
 
-    def __init__(self, embedding_api_key: Optional[str] = None):
+    def __init__(self, embedding_api_key: str | None = None):
         self._embedding_api_key = embedding_api_key
 
     def _resolve_api_key(self) -> str:
@@ -79,13 +78,13 @@ class VectorStoreManager:
         api_key = self._resolve_api_key()
         return OpenAIEmbeddings(api_key=api_key)
 
-    def get_storage_keys(self, agent_id: str) -> Tuple[str, str]:
+    def get_storage_keys(self, agent_id: str) -> tuple[str, str]:
         """Get storage keys for vector store and metadata."""
         vector_store_key = f"{VECTOR_STORE_KEY_PREFIX}_{agent_id}"
         metadata_key = f"{METADATA_KEY_PREFIX}_{agent_id}"
         return vector_store_key, metadata_key
 
-    def encode_vector_store(self, vector_store: FAISS) -> Dict[str, str]:
+    def encode_vector_store(self, vector_store: FAISS) -> dict[str, str]:
         """Encode FAISS vector store to base64 for storage."""
         with tempfile.TemporaryDirectory() as temp_dir:
             vector_store.save_local(temp_dir)
@@ -102,7 +101,7 @@ class VectorStoreManager:
             return encoded_files
 
     def decode_vector_store(
-        self, encoded_files: Dict[str, str], embeddings: OpenAIEmbeddings
+        self, encoded_files: dict[str, str], embeddings: OpenAIEmbeddings
     ) -> FAISS:
         """Decode base64 files back to FAISS vector store."""
         with tempfile.TemporaryDirectory() as temp_dir:
@@ -119,18 +118,18 @@ class VectorStoreManager:
                 allow_dangerous_deserialization=True,
             )
 
-    async def get_existing_vector_store(self, agent_id: str) -> Optional[Dict]:
+    async def get_existing_vector_store(self, agent_id: str) -> dict | None:
         """Get existing vector store data if it exists."""
         vector_store_key, _ = self.get_storage_keys(agent_id)
         return await AgentSkillData.get(agent_id, "web_scraper", vector_store_key)
 
     async def merge_with_existing(
         self,
-        new_documents: List[Document],
+        new_documents: list[Document],
         agent_id: str,
         chunk_size: int = DEFAULT_CHUNK_SIZE,
         chunk_overlap: int = DEFAULT_CHUNK_OVERLAP,
-    ) -> Tuple[FAISS, bool]:
+    ) -> tuple[FAISS, bool]:
         """
         Merge new documents with existing vector store or create new one.
 
@@ -205,7 +204,7 @@ class VectorStoreManager:
             logger.error(f"[{agent_id}] Failed to save vector store: {e}")
             raise
 
-    async def load_vector_store(self, agent_id: str) -> Optional[FAISS]:
+    async def load_vector_store(self, agent_id: str) -> FAISS | None:
         """Load vector store for an agent."""
         stored_data = await self.get_existing_vector_store(agent_id)
 
@@ -250,10 +249,10 @@ class DocumentProcessor:
 
     @staticmethod
     def create_chunks(
-        documents: List[Document],
+        documents: list[Document],
         chunk_size: int = DEFAULT_CHUNK_SIZE,
         chunk_overlap: int = DEFAULT_CHUNK_OVERLAP,
-    ) -> List[Document]:
+    ) -> list[Document]:
         """Split documents into chunks."""
         text_splitter = RecursiveCharacterTextSplitter(
             chunk_size=chunk_size,
@@ -292,7 +291,7 @@ class DocumentProcessor:
         title: str,
         source: str,
         tags: str = "",
-        extra_metadata: Optional[Dict] = None,
+        extra_metadata: dict | None = None,
     ) -> Document:
         """Create a Document with standardized metadata."""
         cleaned_content = DocumentProcessor.clean_text(content)
@@ -324,18 +323,18 @@ class MetadataManager:
     def __init__(self, vector_manager: VectorStoreManager):
         self._vector_manager = vector_manager
 
-    async def get_existing_metadata(self, agent_id: str) -> Dict:
+    async def get_existing_metadata(self, agent_id: str) -> dict:
         """Get existing metadata for an agent."""
         _, metadata_key = self._vector_manager.get_storage_keys(agent_id)
         return await AgentSkillData.get(agent_id, "web_scraper", metadata_key) or {}
 
     def create_url_metadata(
         self,
-        urls: List[str],
-        split_docs: List[Document],
+        urls: list[str],
+        split_docs: list[Document],
         source_type: str = "web_scraper",
-        extra_fields: Optional[Dict] = None,
-    ) -> Dict:
+        extra_fields: dict | None = None,
+    ) -> dict:
         """Create metadata for a list of URLs."""
         metadata = {}
         current_time = str(asyncio.get_event_loop().time())
@@ -361,9 +360,9 @@ class MetadataManager:
         title: str,
         source: str,
         tags: str,
-        split_docs: List[Document],
+        split_docs: list[Document],
         document_length: int,
-    ) -> Dict:
+    ) -> dict:
         """Create metadata for a document."""
         # Generate unique key
         key = f"document_{title.lower().replace(' ', '_')}"
@@ -382,7 +381,7 @@ class MetadataManager:
             }
         }
 
-    async def update_metadata(self, agent_id: str, new_metadata: Dict) -> None:
+    async def update_metadata(self, agent_id: str, new_metadata: dict) -> None:
         """Update metadata for an agent."""
         _, metadata_key = self._vector_manager.get_storage_keys(agent_id)
 
@@ -408,12 +407,12 @@ class ResponseFormatter:
     @staticmethod
     def format_indexing_response(
         operation_type: str,
-        urls_or_content: List[str] | str,
+        urls_or_content: list[str] | str,
         total_chunks: int,
         chunk_size: int,
         chunk_overlap: int,
         was_merged: bool,
-        extra_info: Optional[Dict] = None,
+        extra_info: dict | None = None,
         current_size_bytes: int = 0,
         size_limit_reached: bool = False,
         total_requested_urls: int = 0,
@@ -481,13 +480,13 @@ class ResponseFormatter:
 
 
 async def scrape_and_index_urls(
-    urls: List[str],
+    urls: list[str],
     agent_id: str,
     vector_manager: VectorStoreManager,
     chunk_size: int = DEFAULT_CHUNK_SIZE,
     chunk_overlap: int = DEFAULT_CHUNK_OVERLAP,
     requests_per_second: int = DEFAULT_REQUESTS_PER_SECOND,
-) -> Tuple[int, bool, List[str]]:
+) -> tuple[int, bool, list[str]]:
     """
     Scrape URLs and index their content into vector store with size limits.
 
@@ -636,12 +635,12 @@ async def scrape_and_index_urls(
 
 # Convenience function that combines all operations
 async def index_documents(
-    documents: List[Document],
+    documents: list[Document],
     agent_id: str,
     vector_manager: VectorStoreManager,
     chunk_size: int = DEFAULT_CHUNK_SIZE,
     chunk_overlap: int = DEFAULT_CHUNK_OVERLAP,
-) -> Tuple[int, bool]:
+) -> tuple[int, bool]:
     """
     Complete document indexing workflow.