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

intentkit/skills/firecrawl/query.py

@@ -1,5 +1,4 @@
 import logging
-from typing import Type
 
 from pydantic import BaseModel, Field
 
@@ -37,7 +36,7 @@ class FirecrawlQueryIndexedContent(FirecrawlBaseTool):
         "Use this tool to search through content that was previously scraped and indexed using Firecrawl tools.\n"
         "This tool can help answer questions based on the indexed web content from Firecrawl scraping/crawling."
     )
-    args_schema: Type[BaseModel] = FirecrawlQueryInput
+    args_schema: type[BaseModel] = FirecrawlQueryInput
 
     async def _arun(
         self,

intentkit/skills/firecrawl/scrape.py

@@ -1,5 +1,4 @@
 import logging
-from typing import List, Optional, Type
 
 import httpx
 from langchain_core.documents import Document
@@ -16,7 +15,7 @@ class FirecrawlScrapeInput(BaseModel):
     url: str = Field(
         description="The URL to scrape. Must be a valid HTTP or HTTPS URL."
     )
-    formats: List[str] = Field(
+    formats: list[str] = Field(
         description="Output formats to include in the response. Options: 'markdown', 'html', 'rawHtml', 'screenshot', 'links', 'json'",
         default=["markdown"],
     )
@@ -24,11 +23,11 @@ class FirecrawlScrapeInput(BaseModel):
         description="Whether to extract only the main content (excluding headers, footers, navigation, etc.)",
         default=True,
     )
-    include_tags: Optional[List[str]] = Field(
+    include_tags: list[str] | None = Field(
         description="HTML tags, classes, or IDs to include in the response (e.g., ['h1', 'p', '.main-content'])",
         default=None,
     )
-    exclude_tags: Optional[List[str]] = Field(
+    exclude_tags: list[str] | None = Field(
         description="HTML tags, classes, or IDs to exclude from the response (e.g., ['#ad', '#footer'])",
         default=None,
     )
@@ -81,15 +80,15 @@ class FirecrawlScrape(FirecrawlBaseTool):
         "This tool can handle JavaScript-rendered content, PDFs, and dynamic websites. "
         "Use this when you want to refresh/update content from a URL that was previously scraped."
     )
-    args_schema: Type[BaseModel] = FirecrawlScrapeInput
+    args_schema: type[BaseModel] = FirecrawlScrapeInput
 
     async def _arun(
         self,
         url: str,
-        formats: List[str] = None,
+        formats: list[str] = None,
         only_main_content: bool = True,
-        include_tags: Optional[List[str]] = None,
-        exclude_tags: Optional[List[str]] = None,
+        include_tags: list[str] | None = None,
+        exclude_tags: list[str] | None = None,
         wait_for: int = 0,
         timeout: int = 30000,
         index_content: bool = True,

intentkit/skills/firecrawl/utils.py

@@ -2,7 +2,7 @@
 
 import logging
 import re
-from typing import Any, Dict, List, Optional, Tuple
+from typing import Any
 
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain_community.vectorstores import FAISS
@@ -36,8 +36,8 @@ class FirecrawlDocumentProcessor:
 
     @staticmethod
     def split_documents(
-        documents: List[Document], chunk_size: int = 1000, chunk_overlap: int = 200
-    ) -> List[Document]:
+        documents: list[Document], chunk_size: int = 1000, chunk_overlap: int = 200
+    ) -> list[Document]:
         """Split documents into smaller chunks for better indexing."""
         text_splitter = RecursiveCharacterTextSplitter(
             chunk_size=chunk_size,
@@ -63,7 +63,7 @@ class FirecrawlDocumentProcessor:
 class FirecrawlVectorStoreManager:
     """Manages vector store operations for Firecrawl content."""
 
-    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:
@@ -81,7 +81,7 @@ class FirecrawlVectorStoreManager:
             openai_api_key=openai_api_key, model="text-embedding-3-small"
         )
 
-    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 (compatible with web_scraper)."""
         import base64
         import os
@@ -106,7 +106,7 @@ class FirecrawlVectorStoreManager:
             raise
 
     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 (compatible with web_scraper)."""
         import base64
@@ -131,7 +131,7 @@ class FirecrawlVectorStoreManager:
             logger.error(f"Error decoding 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 existing vector store for an agent."""
         try:
             vector_store_key = f"vector_store_{agent_id}"
@@ -186,8 +186,8 @@ class FirecrawlMetadataManager:
 
     @staticmethod
     def create_url_metadata(
-        urls: List[str], documents: List[Document], source_type: str
-    ) -> Dict[str, Any]:
+        urls: list[str], documents: list[Document], source_type: str
+    ) -> dict[str, Any]:
         """Create metadata for indexed URLs."""
         return {
             "urls": urls,
@@ -198,7 +198,7 @@ class FirecrawlMetadataManager:
 
     @staticmethod
     @staticmethod
-    async def update_metadata(agent_id: str, new_metadata: Dict[str, Any]) -> None:
+    async def update_metadata(agent_id: str, new_metadata: dict[str, Any]) -> None:
         """Update metadata for an agent."""
         try:
             metadata_key = f"indexed_urls_{agent_id}"
@@ -215,12 +215,12 @@ class FirecrawlMetadataManager:
 
 
 async def index_documents(
-    documents: List[Document],
+    documents: list[Document],
     agent_id: str,
     vector_manager: FirecrawlVectorStoreManager,
     chunk_size: int = 1000,
     chunk_overlap: int = 200,
-) -> Tuple[int, bool]:
+) -> tuple[int, bool]:
     """
     Index documents into the Firecrawl vector store.
 
@@ -281,7 +281,7 @@ async def query_indexed_content(
     agent_id: str,
     vector_manager: FirecrawlVectorStoreManager,
     max_results: int = 4,
-) -> List[Document]:
+) -> list[Document]:
     """
     Query the Firecrawl indexed content.
 

intentkit/skills/github/base.py

@@ -1,5 +1,3 @@
-from typing import Type
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.base import IntentKitSkill
@@ -10,7 +8,7 @@ class GitHubBaseTool(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/github/github_search.py

@@ -1,6 +1,5 @@
 import logging
 from enum import Enum
-from typing import Type
 
 import httpx
 from pydantic import BaseModel, Field
@@ -54,7 +53,7 @@ class GitHubSearch(GitHubBaseTool):
         "- Code snippets across GitHub repositories\n"
         "You must call this tool whenever the user asks about finding something on GitHub."
     )
-    args_schema: Type[BaseModel] = GitHubSearchInput
+    args_schema: type[BaseModel] = GitHubSearchInput
 
     async def _arun(
         self,

intentkit/skills/heurist/base.py

@@ -1,7 +1,5 @@
 """Base class for Heurist AI skills."""
 
-from typing import Type
-
 from langchain_core.tools.base import ToolException
 from pydantic import BaseModel, Field
 
@@ -17,7 +15,7 @@ class HeuristBaseTool(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/heurist/image_generation_animagine_xl.py

@@ -1,5 +1,4 @@
 import logging
-from typing import Optional, Type
 
 import httpx
 from epyxid import XID
@@ -18,16 +17,16 @@ class ImageGenerationAnimagineXLInput(BaseModel):
     prompt: str = Field(
         description="Text prompt describing the image to generate.",
     )
-    neg_prompt: Optional[str] = Field(
+    neg_prompt: str | None = Field(
         default="(worst quality: 1.4), bad quality, nsfw",
         description="Negative prompt describing what to avoid in the generated image.",
     )
-    width: Optional[int] = Field(
+    width: int | None = Field(
         default=1024,
         le=1024,
         description="Width of the generated image.",
     )
-    height: Optional[int] = Field(
+    height: int | None = Field(
         default=680,
         le=1024,
         description="Height of the generated image.",
@@ -53,14 +52,14 @@ class ImageGenerationAnimagineXL(HeuristBaseTool):
         "AnimagineXL specializes in creating high-quality Japanese anime-style illustrations.\n"
         "If you have height and width, remember to specify them.\n"
     )
-    args_schema: Type[BaseModel] = ImageGenerationAnimagineXLInput
+    args_schema: type[BaseModel] = ImageGenerationAnimagineXLInput
 
     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,
+        neg_prompt: str | None = "(worst quality: 1.4), bad quality, nsfw",
+        width: int | None = 1024,
+        height: int | None = 680,
         **kwargs,
     ) -> str:
         """Implementation of the tool to generate Japanese anime-style images using Heurist AI's AnimagineXL model.

intentkit/skills/heurist/image_generation_arthemy_comics.py

@@ -1,5 +1,4 @@
 import logging
-from typing import Optional, Type
 
 import httpx
 from epyxid import XID
@@ -18,16 +17,16 @@ class ImageGenerationArthemyComicsInput(BaseModel):
     prompt: str = Field(
         description="Text prompt describing the image to generate.",
     )
-    neg_prompt: Optional[str] = Field(
+    neg_prompt: str | None = Field(
         default="(worst quality: 1.4), bad quality, nsfw",
         description="Negative prompt describing what to avoid in the generated image.",
     )
-    width: Optional[int] = Field(
+    width: int | None = Field(
         default=1024,
         le=1024,
         description="Width of the generated image.",
     )
-    height: Optional[int] = Field(
+    height: int | None = Field(
         default=1024,
         le=1024,
         description="Height of the generated image.",
@@ -53,14 +52,14 @@ class ImageGenerationArthemyComics(HeuristBaseTool):
         "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
+    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,
+        neg_prompt: str | None = "(worst quality: 1.4), bad quality, nsfw",
+        width: int | None = 1024,
+        height: int | None = 680,
         **kwargs,
     ) -> str:
         """Implementation of the tool to generate comic-style images using Heurist AI's ArthemyComics model.

intentkit/skills/heurist/image_generation_arthemy_real.py

@@ -1,5 +1,4 @@
 import logging
-from typing import Optional, Type
 
 import httpx
 from epyxid import XID
@@ -18,16 +17,16 @@ class ImageGenerationArthemyRealInput(BaseModel):
     prompt: str = Field(
         description="Text prompt describing the image to generate.",
     )
-    neg_prompt: Optional[str] = Field(
+    neg_prompt: str | None = Field(
         default="(worst quality: 1.4), bad quality, nsfw",
         description="Negative prompt describing what to avoid in the generated image.",
     )
-    width: Optional[int] = Field(
+    width: int | None = Field(
         default=1024,
         le=1024,
         description="Width of the generated image.",
     )
-    height: Optional[int] = Field(
+    height: int | None = Field(
         default=1024,
         le=1024,
         description="Height of the generated image.",
@@ -53,14 +52,14 @@ class ImageGenerationArthemyReal(HeuristBaseTool):
         "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
+    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,
+        neg_prompt: str | None = "(worst quality: 1.4), bad quality, nsfw",
+        width: int | None = 1024,
+        height: int | None = 680,
         **kwargs,
     ) -> str:
         """Implementation of the tool to generate realistic images using Heurist AI's ArthemyReal model.

intentkit/skills/heurist/image_generation_braindance.py

@@ -1,5 +1,4 @@
 import logging
-from typing import Optional, Type
 
 import httpx
 from epyxid import XID
@@ -18,16 +17,16 @@ class ImageGenerationBrainDanceInput(BaseModel):
     prompt: str = Field(
         description="Text prompt describing the image to generate.",
     )
-    neg_prompt: Optional[str] = Field(
+    neg_prompt: str | None = Field(
         default="(worst quality: 1.4), bad quality, nsfw",
         description="Negative prompt describing what to avoid in the generated image.",
     )
-    width: Optional[int] = Field(
+    width: int | None = Field(
         default=1024,
         le=1024,
         description="Width of the generated image.",
     )
-    height: Optional[int] = Field(
+    height: int | None = Field(
         default=1024,
         le=1024,
         description="Height of the generated image.",
@@ -53,14 +52,14 @@ class ImageGenerationBrainDance(HeuristBaseTool):
         "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
+    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,
+        neg_prompt: str | None = "(worst quality: 1.4), bad quality, nsfw",
+        width: int | None = 1024,
+        height: int | None = 680,
         **kwargs,
     ) -> str:
         """Implementation of the tool to generate artistic images using Heurist AI's BrainDance model.

intentkit/skills/heurist/image_generation_cyber_realistic_xl.py

@@ -1,5 +1,4 @@
 import logging
-from typing import Optional, Type
 
 import httpx
 from epyxid import XID
@@ -18,16 +17,16 @@ class ImageGenerationCyberRealisticXLInput(BaseModel):
     prompt: str = Field(
         description="Text prompt describing the image to generate.",
     )
-    neg_prompt: Optional[str] = Field(
+    neg_prompt: str | None = Field(
         default="(worst quality: 1.4), bad quality, nsfw",
         description="Negative prompt describing what to avoid in the generated image.",
     )
-    width: Optional[int] = Field(
+    width: int | None = Field(
         default=1024,
         le=1024,
         description="Width of the generated image.",
     )
-    height: Optional[int] = Field(
+    height: int | None = Field(
         default=680,
         le=1024,
         description="Height of the generated image.",
@@ -53,14 +52,14 @@ class ImageGenerationCyberRealisticXL(HeuristBaseTool):
         "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
+    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,
+        neg_prompt: str | None = "(worst quality: 1.4), bad quality, nsfw",
+        width: int | None = 1024,
+        height: int | None = 680,
         **kwargs,
     ) -> str:
         """Implementation of the tool to generate hyperrealistic cyberpunk images using Heurist AI's CyberRealisticXL model.

intentkit/skills/heurist/image_generation_flux_1_dev.py

@@ -1,5 +1,4 @@
 import logging
-from typing import Optional, Type
 
 import httpx
 from epyxid import XID
@@ -18,16 +17,16 @@ class ImageGenerationFlux1DevInput(BaseModel):
     prompt: str = Field(
         description="Text prompt describing the image to generate.",
     )
-    neg_prompt: Optional[str] = Field(
+    neg_prompt: str | None = Field(
         default="",
         description="Negative prompt describing what to avoid in the generated image.",
     )
-    width: Optional[int] = Field(
+    width: int | None = Field(
         default=1024,
         le=2048,
         description="Width of the generated image.",
     )
-    height: Optional[int] = Field(
+    height: int | None = Field(
         default=1024,
         le=2048,
         description="Height of the generated image.",
@@ -53,14 +52,14 @@ class ImageGenerationFlux1Dev(HeuristBaseTool):
         "Flux.1-dev is a versatile, general-purpose model capable of generating images in any style.\n"
         "If you have height and width, remember to specify them.\n"
     )
-    args_schema: Type[BaseModel] = ImageGenerationFlux1DevInput
+    args_schema: type[BaseModel] = ImageGenerationFlux1DevInput
 
     async def _arun(
         self,
         prompt: str,
-        neg_prompt: Optional[str] = "",
-        width: Optional[int] = 1024,
-        height: Optional[int] = 680,
+        neg_prompt: str | None = "",
+        width: int | None = 1024,
+        height: int | None = 680,
         **kwargs,
     ) -> str:
         """Implementation of the tool to generate images using Heurist AI's Flux.1-dev model.

intentkit/skills/heurist/image_generation_sdxl.py

@@ -1,5 +1,4 @@
 import logging
-from typing import Optional, Type
 
 import httpx
 from epyxid import XID
@@ -18,16 +17,16 @@ class ImageGenerationSDXLInput(BaseModel):
     prompt: str = Field(
         description="Text prompt describing the image to generate.",
     )
-    neg_prompt: Optional[str] = Field(
+    neg_prompt: str | None = Field(
         default="(worst quality: 1.4), bad quality, nsfw",
         description="Negative prompt describing what to avoid in the generated image.",
     )
-    width: Optional[int] = Field(
+    width: int | None = Field(
         default=1024,
         le=1024,
         description="Width of the generated image.",
     )
-    height: Optional[int] = Field(
+    height: int | None = Field(
         default=1024,
         le=1024,
         description="Height of the generated image.",
@@ -53,14 +52,14 @@ class ImageGenerationSDXL(HeuristBaseTool):
         "SDXL is a versatile, general-purpose model capable of generating high-quality images in any style.\n"
         "If you have height and width, remember to specify them.\n"
     )
-    args_schema: Type[BaseModel] = ImageGenerationSDXLInput
+    args_schema: type[BaseModel] = ImageGenerationSDXLInput
 
     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,
+        neg_prompt: str | None = "(worst quality: 1.4), bad quality, nsfw",
+        width: int | None = 1024,
+        height: int | None = 680,
         **kwargs,
     ) -> str:
         """Implementation of the tool to generate images using Heurist AI's SDXL model.

intentkit/skills/http/base.py

@@ -1,5 +1,3 @@
-from typing import Type
-
 from pydantic import BaseModel, Field
 
 from intentkit.skills.base import IntentKitSkill
@@ -10,7 +8,7 @@ class HttpBaseTool(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/http/get.py

@@ -1,5 +1,5 @@
 import logging
-from typing import Any, Dict, Optional, Type
+from typing import Any
 
 import httpx
 from langchain_core.tools import ToolException
@@ -14,15 +14,15 @@ class HttpGetInput(BaseModel):
     """Input for HTTP GET request."""
 
     url: str = Field(description="The URL to send the GET request to")
-    headers: Optional[Dict[str, str]] = Field(
+    headers: dict[str, str] | None = Field(
         description="Optional headers to include in the request",
         default=None,
     )
-    params: Optional[Dict[str, Any]] = Field(
+    params: dict[str, Any] | None = Field(
         description="Optional query parameters to include in the request",
         default=None,
     )
-    timeout: Optional[float] = Field(
+    timeout: float | None = Field(
         description="Request timeout in seconds (default: 30)",
         default=30.0,
     )
@@ -47,13 +47,13 @@ class HttpGet(HttpBaseTool):
         "Returns the response content as text. "
         "Use this when you need to fetch data from web APIs or websites."
     )
-    args_schema: Type[BaseModel] = HttpGetInput
+    args_schema: type[BaseModel] = HttpGetInput
 
     async def _arun(
         self,
         url: str,
-        headers: Optional[Dict[str, str]] = None,
-        params: Optional[Dict[str, Any]] = None,
+        headers: dict[str, str] | None = None,
+        params: dict[str, Any] | None = None,
         timeout: float = 30.0,
         **kwargs,
     ) -> str:

intentkit/skills/http/post.py

@@ -1,5 +1,5 @@
 import logging
-from typing import Any, Dict, Optional, Type, Union
+from typing import Any
 
 import httpx
 from langchain_core.tools import ToolException
@@ -14,19 +14,19 @@ class HttpPostInput(BaseModel):
     """Input for HTTP POST request."""
 
     url: str = Field(description="The URL to send the POST request to")
-    data: Optional[Union[Dict[str, Any], str]] = Field(
+    data: dict[str, Any] | str | None = Field(
         description="The data to send in the request body. Can be a dictionary (will be sent as JSON) or a string",
         default=None,
     )
-    headers: Optional[Dict[str, str]] = Field(
+    headers: dict[str, str] | None = Field(
         description="Optional headers to include in the request",
         default=None,
     )
-    params: Optional[Dict[str, Any]] = Field(
+    params: dict[str, Any] | None = Field(
         description="Optional query parameters to include in the request",
         default=None,
     )
-    timeout: Optional[float] = Field(
+    timeout: float | None = Field(
         description="Request timeout in seconds (default: 30)",
         default=30.0,
     )
@@ -52,14 +52,14 @@ class HttpPost(HttpBaseTool):
         "Returns the response content as text. "
         "Use this when you need to send data to web APIs or submit forms."
     )
-    args_schema: Type[BaseModel] = HttpPostInput
+    args_schema: type[BaseModel] = HttpPostInput
 
     async def _arun(
         self,
         url: str,
-        data: Optional[Union[Dict[str, Any], str]] = None,
-        headers: Optional[Dict[str, str]] = None,
-        params: Optional[Dict[str, Any]] = None,
+        data: dict[str, Any] | str | None = None,
+        headers: dict[str, str] | None = None,
+        params: dict[str, Any] | None = None,
         timeout: float = 30.0,
         **kwargs,
     ) -> str: