intentkit 0.8.34.dev2__tar.gz → 0.8.34.dev4__tar.gz

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intentkit
-Version: 0.8.34.dev2
+Version: 0.8.34.dev4
 Summary: Intent-based AI Agent Platform - Core Package
 Project-URL: Homepage, https://github.com/crestalnetwork/intentkit
 Project-URL: Repository, https://github.com/crestalnetwork/intentkit

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/__init__.py

@@ -3,7 +3,7 @@
 A powerful platform for building AI agents with blockchain and cryptocurrency capabilities.
 """
 
-__version__ = "0.8.34-dev2"
+__version__ = "0.8.34-dev4"
 __author__ = "Ruihua"
 __email__ = "ruihua@crestal.network"
 

{intentkit-0.8.34.dev2/intentkit/utils → intentkit-0.8.34.dev4/intentkit/clients}/s3.py

@@ -13,6 +13,8 @@ import httpx
 from botocore.exceptions import ClientError
 from mypy_boto3_s3.client import S3Client
 
+from intentkit.config.config import config
+
 logger = logging.getLogger(__name__)
 
 # Global variables for S3 configuration
@@ -22,31 +24,47 @@ _prefix: str | None = None
 _cdn_url: str | None = None
 
 
-def init_s3(bucket: str, cdn_url: str, env: str) -> None:
+def get_s3_client() -> S3Client | None:
     """
-    Initialize S3 configuration.
-
-    Args:
-        bucket: S3 bucket name
-        cdn_url: CDN URL for the S3 bucket
-        env: Environment name for the prefix
-
-    Raises:
-        ValueError: If bucket or cdn_url is empty
+    Get or initialize S3 client and configuration.
+    Returns None if configuration is missing.
     """
     global _bucket, _client, _prefix, _cdn_url
 
-    if not bucket:
-        raise ValueError("S3 bucket name cannot be empty")
-    if not cdn_url:
-        raise ValueError("S3 CDN URL cannot be empty")
+    if _client is not None:
+        return _client
 
-    _bucket = bucket
-    _cdn_url = cdn_url
-    _prefix = f"{env}/intentkit/"
-    _client = cast(S3Client, boto3.client("s3"))
+    if not config.aws_s3_bucket or not config.aws_s3_cdn_url:
+        # Only log once or if needed, but here we just return None
+        # The calling functions usually log "S3 not initialized"
+        return None
 
-    logger.info(f"S3 initialized with bucket: {bucket}, prefix: {_prefix}")
+    _bucket = config.aws_s3_bucket
+    _cdn_url = config.aws_s3_cdn_url
+    _prefix = f"{config.env}/intentkit/"
+
+    try:
+        if config.aws_s3_endpoint_url:
+            _client = cast(
+                S3Client,
+                boto3.client(
+                    "s3",
+                    endpoint_url=config.aws_s3_endpoint_url,
+                    region_name=config.aws_s3_region_name,
+                    aws_access_key_id=config.aws_s3_access_key_id,
+                    aws_secret_access_key=config.aws_s3_secret_access_key,
+                ),
+            )
+            logger.info(
+                f"S3 initialized with custom endpoint: {config.aws_s3_endpoint_url}, bucket: {_bucket}, prefix: {_prefix}"
+            )
+        else:
+            _client = cast(S3Client, boto3.client("s3"))
+            logger.info(f"S3 initialized with bucket: {_bucket}, prefix: {_prefix}")
+        return _client
+    except Exception as e:
+        logger.error(f"Failed to initialize S3 client: {e}")
+        return None
 
 
 async def store_image(url: str, key: str) -> str:
@@ -64,15 +82,16 @@ async def store_image(url: str, key: str) -> str:
         ClientError: If the upload fails
         httpx.HTTPError: If the download fails
     """
-    if not _client or not _bucket or not _prefix or not _cdn_url:
+    client = get_s3_client()
+    if not client or not _bucket or not _prefix or not _cdn_url:
         # If S3 is not initialized, log and return the original URL
         logger.info("S3 not initialized. Returning original URL.")
         return url
 
     try:
         # Download the image from the URL asynchronously
-        async with httpx.AsyncClient() as client:
-            response = await client.get(url, follow_redirects=True)
+        async with httpx.AsyncClient() as http_client:
+            response = await http_client.get(url, follow_redirects=True)
             response.raise_for_status()
 
             # Prepare the S3 key with prefix
@@ -93,7 +112,7 @@ async def store_image(url: str, key: str) -> str:
                     content_type = "image/jpeg"
 
             # Upload to S3
-            _client.upload_fileobj(
+            client.upload_fileobj(
                 file_obj,
                 _bucket,
                 prefixed_key,
@@ -131,7 +150,8 @@ async def store_image_bytes(
         ClientError: If the upload fails
         ValueError: If S3 is not initialized or image_bytes is empty
     """
-    if not _client or not _bucket or not _prefix or not _cdn_url:
+    client = get_s3_client()
+    if not client or not _bucket or not _prefix or not _cdn_url:
         # If S3 is not initialized, log and return empty string
         logger.info("S3 not initialized. Cannot store image bytes.")
         return ""
@@ -158,7 +178,7 @@ async def store_image_bytes(
 
         logger.info("uploading image to s3")
         # Upload to S3
-        _client.upload_fileobj(
+        client.upload_fileobj(
             file_obj,
             _bucket,
             prefixed_key,
@@ -189,7 +209,8 @@ async def store_file(
     size: int | None = None,
 ) -> str:
     """Store raw file bytes with automatic content type detection."""
-    if not _client or not _bucket or not _prefix or not _cdn_url:
+    client = get_s3_client()
+    if not client or not _bucket or not _prefix or not _cdn_url:
         logger.info("S3 not initialized. Cannot store file bytes.")
         return ""
 
@@ -220,7 +241,7 @@ async def store_file(
         effective_size,
     )
 
-    _client.upload_fileobj(
+    client.upload_fileobj(
         file_obj,
         _bucket,
         prefixed_key,
@@ -257,7 +278,8 @@ async def store_file_bytes(
         ClientError: If the upload fails
         ValueError: If S3 is not initialized, file_bytes is empty, or file exceeds size limit
     """
-    if not _client or not _bucket or not _prefix or not _cdn_url:
+    client = get_s3_client()
+    if not client or not _bucket or not _prefix or not _cdn_url:
         logger.info("S3 not initialized. Cannot store file bytes.")
         return ""
     if not file_bytes:
@@ -303,7 +325,7 @@ async def store_file_bytes(
         logger.info(f"Uploading {file_type} to S3 with content type {content_type}")
 
         # Upload to S3
-        _client.upload_fileobj(
+        client.upload_fileobj(
             file_obj,
             _bucket,
             prefixed_key,

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/config/config.py

@@ -7,7 +7,6 @@ from dotenv import load_dotenv
 
 from intentkit.utils.chain import ChainProvider, QuicknodeChainProvider
 from intentkit.utils.logging import setup_logging
-from intentkit.utils.s3 import init_s3
 from intentkit.utils.slack_alert import init_slack
 
 SecretsMap = dict[str, str | int]
@@ -71,9 +70,16 @@ class Config:
         self.redis_host: str | None = self.load("REDIS_HOST")
         self.redis_port: int = self.load_int("REDIS_PORT", 6379)
         self.redis_db: int = self.load_int("REDIS_DB", 0)
-        # AWS
+        # AWS S3
         self.aws_s3_bucket: str | None = self.load("AWS_S3_BUCKET")
         self.aws_s3_cdn_url: str | None = self.load("AWS_S3_CDN_URL")
+        # If using custom S3 endpoint
+        self.aws_s3_endpoint_url: str | None = self.load("AWS_S3_ENDPOINT_URL")
+        self.aws_s3_region_name: str | None = self.load("AWS_S3_REGION_NAME")
+        self.aws_s3_access_key_id: str | None = self.load("AWS_S3_ACCESS_KEY_ID")
+        self.aws_s3_secret_access_key: str | None = self.load(
+            "AWS_S3_SECRET_ACCESS_KEY"
+        )
         # Internal API
         self.internal_base_url: str = self.load(
             "INTERNAL_BASE_URL", "http://intent-api"
@@ -89,7 +95,6 @@ class Config:
         self.debug_password: str | None = self.load("DEBUG_PASSWORD")
         # Payment
         self.payment_enabled: bool = self.load("PAYMENT_ENABLED", "false") == "true"
-        self.x402_fee_address: str | None = self.load("X402_FEE_ADDRESS")
         # Open API for agent
         self.open_api_base_url: str = self.load(
             "OPEN_API_BASE_URL", "http://localhost:8000"
@@ -183,9 +188,6 @@ class Config:
         # If the slack alert token exists, init it
         if self.slack_alert_token and self.slack_alert_channel:
             init_slack(self.slack_alert_token, self.slack_alert_channel)
-        # If the AWS S3 bucket and CDN URL exist, init it
-        if self.aws_s3_bucket and self.aws_s3_cdn_url:
-            init_s3(self.aws_s3_bucket, self.aws_s3_cdn_url, self.env)
 
     @overload
     def load(self, key: str) -> str | None: ...  # noqa: F811

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/core/chat.py

@@ -6,7 +6,7 @@ including clearing thread history using LangGraph's checkpointer.
 
 import logging
 
-from intentkit.models.db import get_langgraph_checkpointer
+from intentkit.models.db import get_checkpointer
 from intentkit.utils.error import IntentKitAPIError
 
 logger = logging.getLogger(__name__)
@@ -34,7 +34,7 @@ async def clear_thread_memory(agent_id: str, chat_id: str) -> bool:
         thread_id = f"{agent_id}-{chat_id}"
 
         # Get the LangGraph checkpointer instance
-        checkpointer = get_langgraph_checkpointer()
+        checkpointer = get_checkpointer()
 
         # Use the official LangGraph method to delete all thread content
         await checkpointer.adelete_thread(thread_id)

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/core/engine.py

@@ -26,6 +26,7 @@ from langchain_core.messages import (
     HumanMessage,
 )
 from langchain_core.tools import BaseTool
+from langgraph.checkpoint.memory import InMemorySaver
 from langgraph.errors import GraphRecursionError
 from langgraph.graph.state import CompiledStateGraph
 from sqlalchemy import func, update
@@ -53,7 +54,10 @@ from intentkit.models.chat import (
     ChatMessageSkillCall,
 )
 from intentkit.models.credit import CreditAccount, OwnerType
-from intentkit.models.db import get_langgraph_checkpointer, get_session
+from intentkit.models.db import (
+    get_checkpointer,
+    get_session,
+)
 from intentkit.models.llm import LLMModelInfo, create_llm_model
 from intentkit.models.skill import AgentSkillData, ChatSkillData, Skill
 from intentkit.models.user import User
@@ -121,7 +125,10 @@ async def build_agent(
     )
 
     # ==== Store buffered conversation history in memory.
-    memory = get_langgraph_checkpointer()
+    try:
+        checkpointer = get_checkpointer()
+    except RuntimeError:
+        checkpointer = InMemorySaver()
 
     # ==== Load skills
     tools: list[BaseTool | dict] = []
@@ -192,7 +199,7 @@ async def build_agent(
         middleware=middleware,
         state_schema=AgentState,
         context_schema=AgentContext,
-        checkpointer=memory,
+        checkpointer=checkpointer,
         debug=config.debug_checkpoint,
         name=agent.id,
     )
@@ -367,6 +374,7 @@ async def stream_agent_raw(
         payer = user_message.user_id
         if user_message.author_type in [
             AuthorType.TELEGRAM,
+            AuthorType.DISCORD,
             AuthorType.TWITTER,
             AuthorType.API,
             AuthorType.X402,

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/core/scheduler.py

@@ -15,7 +15,6 @@ from intentkit.core.agent import (
 )
 from intentkit.core.credit import refill_all_free_credits
 from intentkit.models.agent_data import AgentQuota
-from intentkit.models.db import cleanup_checkpoints
 
 
 def create_scheduler(
@@ -141,13 +140,4 @@ def create_scheduler(
         replace_existing=True,
     )
 
-    # Clean up old LangGraph checkpoints daily at UTC 2:30
-    scheduler.add_job(
-        cleanup_checkpoints,
-        trigger=CronTrigger(hour=2, minute=30, timezone="UTC"),
-        id="cleanup_checkpoints",
-        name="Clean up old checkpoints",
-        replace_existing=True,
-    )
-
     return scheduler

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/models/db.py

@@ -3,19 +3,17 @@ from contextlib import asynccontextmanager
 from typing import Annotated
 from urllib.parse import quote_plus
 
-from langgraph.checkpoint.memory import InMemorySaver
-from langgraph.checkpoint.postgres.aio import AsyncPostgresSaver
-from langgraph.types import Checkpointer
+from langgraph.checkpoint.postgres.shallow import AsyncShallowPostgresSaver
 from psycopg import OperationalError
 from psycopg_pool import AsyncConnectionPool
 from pydantic import Field
-from sqlalchemy import text
 from sqlalchemy.ext.asyncio import AsyncEngine, AsyncSession, create_async_engine
 
 from intentkit.models.db_mig import safe_migrate
 
 engine: AsyncEngine | None = None
-_langgraph_checkpointer: Checkpointer | None = None
+_connection_pool: AsyncConnectionPool | None = None
+_checkpointer: AsyncShallowPostgresSaver | None = None
 
 
 async def check_connection(conn):
@@ -54,9 +52,9 @@ async def init_db(
         auto_migrate: Whether to run migrations automatically (default: True)
         pool_size: Database connection pool size (default: 3)
     """
-    global engine, _langgraph_checkpointer
-    # Initialize psycopg pool and AsyncPostgresSaver if not already initialized
-    if _langgraph_checkpointer is None:
+    global engine, _connection_pool, _checkpointer
+    # Initialize psycopg pool and AsyncShallowPostgresSaver if not already initialized
+    if _connection_pool is None:
         if host:
             conn_string = (
                 f"postgresql://{username}:{quote_plus(password)}@{host}:{port}/{dbname}"
@@ -72,13 +70,17 @@ async def init_db(
                 # Set connection max lifetime to prevent stale connections
                 max_lifetime=3600,  # 1 hour
             )
-            _langgraph_checkpointer = AsyncPostgresSaver(pool)
+            _connection_pool = pool
+            _checkpointer = AsyncShallowPostgresSaver(pool)
             if auto_migrate:
                 # Migrate can not use pool, so we start from scratch
-                async with AsyncPostgresSaver.from_conn_string(conn_string) as saver:
+                async with AsyncShallowPostgresSaver.from_conn_string(
+                    conn_string
+                ) as saver:
                     await saver.setup()
         else:
-            _langgraph_checkpointer = InMemorySaver()
+            # For in-memory, we don't need a pool, but we need to handle it if requested
+            pass
     # Initialize SQLAlchemy engine with pool settings
     if engine is None:
         if host:
@@ -141,75 +143,23 @@ def get_engine() -> AsyncEngine:
     return engine
 
 
-def get_langgraph_checkpointer() -> Checkpointer:
-    """Get the AsyncPostgresSaver instance for langgraph.
+def get_connection_pool() -> AsyncConnectionPool:
+    """Get the AsyncConnectionPool instance.
 
     Returns:
-        AsyncPostgresSaver: The AsyncPostgresSaver instance
+        AsyncConnectionPool: The AsyncConnectionPool instance
     """
-    if _langgraph_checkpointer is None:
+    if _connection_pool is None:
         raise RuntimeError("Database pool not initialized. Call init_db first.")
-    return _langgraph_checkpointer
+    return _connection_pool
 
 
-async def cleanup_checkpoints() -> None:
-    """
-    Clean up old checkpoints, writes, and blobs, keeping only the latest one for each thread.
-    This helps reduce database size while maintaining the current state of agents.
-    """
-    assert engine is not None, "Database engine not initialized. Call init_db first."
-
-    # SQL to keep only the latest checkpoint for each thread
-    # We assume checkpoint_id is sortable (ULID/UUIDv7 style which LangGraph uses)
-    cleanup_checkpoints_sql = text("""
-        DELETE FROM checkpoints
-        WHERE (thread_id, checkpoint_ns, checkpoint_id) NOT IN (
-            SELECT DISTINCT ON (thread_id, checkpoint_ns)
-                thread_id, checkpoint_ns, checkpoint_id
-            FROM checkpoints
-            ORDER BY thread_id, checkpoint_ns, checkpoint_id DESC
-        );
-    """)
-
-    # SQL to clean up writes that reference deleted checkpoints
-    cleanup_writes_sql = text("""
-        DELETE FROM checkpoint_writes
-        WHERE (thread_id, checkpoint_ns, checkpoint_id) NOT IN (
-            SELECT thread_id, checkpoint_ns, checkpoint_id
-            FROM checkpoints
-        );
-    """)
-
-    # SQL to clean up blobs that are not referenced by any remaining checkpoint
-    # Optimized using NOT EXISTS instead of NOT IN for better performance
-    # Using CTE to materialize active channel versions first
-    cleanup_blobs_sql = text("""
-        WITH active_versions AS (
-            SELECT DISTINCT 
-                thread_id, 
-                checkpoint_ns, 
-                key as channel, 
-                value as version
-            FROM checkpoints, 
-                 jsonb_each_text(checkpoint -> 'channel_versions')
-        )
-        DELETE FROM checkpoint_blobs cb
-        WHERE NOT EXISTS (
-            SELECT 1 
-            FROM active_versions av
-            WHERE av.thread_id = cb.thread_id
-              AND av.checkpoint_ns = cb.checkpoint_ns
-              AND av.channel = cb.channel
-              AND av.version = cb.version
-        );
-    """)
+def get_checkpointer() -> AsyncShallowPostgresSaver:
+    """Get the AsyncShallowPostgresSaver instance.
 
-    async with AsyncSession(engine) as session:
-        try:
-            await session.execute(cleanup_checkpoints_sql)
-            await session.execute(cleanup_writes_sql)
-            await session.execute(cleanup_blobs_sql)
-            await session.commit()
-        except Exception as e:
-            await session.rollback()
-            raise e
+    Returns:
+        AsyncShallowPostgresSaver: The AsyncShallowPostgresSaver instance
+    """
+    if _checkpointer is None:
+        raise RuntimeError("Database pool not initialized. Call init_db first.")
+    return _checkpointer

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/models/skills.csv

@@ -167,7 +167,6 @@ firecrawl,firecrawl_clear_indexed_content,firecrawl_clear_indexed_content,TRUE,1
 xmtp,xmtp_transfer,xmtp_transfer,TRUE,1,5,5,,,0x445750026A4a1906b61302442E085f9cbAfe206a
 xmtp,xmtp_swap,xmtp_swap,TRUE,1,5,5,,,0x445750026A4a1906b61302442E085f9cbAfe206a
 xmtp,xmtp_get_swap_price,xmtp_get_swap_price,TRUE,1,5,5,,,0x445750026A4a1906b61302442E085f9cbAfe206a
-x402,x402_ask_agent,x402_ask_agent,TRUE,1,1,1,,,0x445750026A4a1906b61302442E085f9cbAfe206a
 x402,x402_http_request,x402_http_request,TRUE,1,1,1,,,0x445750026A4a1906b61302442E085f9cbAfe206a
 casino,deck_shuffle,casino_deck_shuffle,true,1,5,5,,,0x3cdd051eec909f94965f9c1c657f5b70a172b2c0
 casino,deck_draw,casino_deck_draw,true,1,5,5,,,0x3cdd051eec909f94965f9c1c657f5b70a172b2c0

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/skills/heurist/image_generation_animagine_xl.py

@@ -4,9 +4,9 @@ import httpx
 from epyxid import XID
 from pydantic import BaseModel, Field
 
+from intentkit.clients.s3 import store_image
 from intentkit.config.config import config
 from intentkit.skills.heurist.base import HeuristBaseTool
-from intentkit.utils.s3 import store_image
 
 logger = logging.getLogger(__name__)
 

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/skills/heurist/image_generation_arthemy_comics.py

@@ -4,9 +4,9 @@ import httpx
 from epyxid import XID
 from pydantic import BaseModel, Field
 
+from intentkit.clients.s3 import store_image
 from intentkit.config.config import config
 from intentkit.skills.heurist.base import HeuristBaseTool
-from intentkit.utils.s3 import store_image
 
 logger = logging.getLogger(__name__)
 

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/skills/heurist/image_generation_arthemy_real.py

@@ -4,9 +4,9 @@ import httpx
 from epyxid import XID
 from pydantic import BaseModel, Field
 
+from intentkit.clients.s3 import store_image
 from intentkit.config.config import config
 from intentkit.skills.heurist.base import HeuristBaseTool
-from intentkit.utils.s3 import store_image
 
 logger = logging.getLogger(__name__)
 

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/skills/heurist/image_generation_braindance.py

@@ -4,9 +4,9 @@ import httpx
 from epyxid import XID
 from pydantic import BaseModel, Field
 
+from intentkit.clients.s3 import store_image
 from intentkit.config.config import config
 from intentkit.skills.heurist.base import HeuristBaseTool
-from intentkit.utils.s3 import store_image
 
 logger = logging.getLogger(__name__)
 

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/skills/heurist/image_generation_cyber_realistic_xl.py

@@ -4,9 +4,9 @@ import httpx
 from epyxid import XID
 from pydantic import BaseModel, Field
 
+from intentkit.clients.s3 import store_image
 from intentkit.config.config import config
 from intentkit.skills.heurist.base import HeuristBaseTool
-from intentkit.utils.s3 import store_image
 
 logger = logging.getLogger(__name__)
 

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/skills/heurist/image_generation_flux_1_dev.py

@@ -4,9 +4,9 @@ import httpx
 from epyxid import XID
 from pydantic import BaseModel, Field
 
+from intentkit.clients.s3 import store_image
 from intentkit.config.config import config
 from intentkit.skills.heurist.base import HeuristBaseTool
-from intentkit.utils.s3 import store_image
 
 logger = logging.getLogger(__name__)
 

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/skills/heurist/image_generation_sdxl.py

@@ -4,9 +4,9 @@ import httpx
 from epyxid import XID
 from pydantic import BaseModel, Field
 
+from intentkit.clients.s3 import store_image
 from intentkit.config.config import config
 from intentkit.skills.heurist.base import HeuristBaseTool
-from intentkit.utils.s3 import store_image
 
 logger = logging.getLogger(__name__)
 

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/skills/openai/dalle_image_generation.py

@@ -6,8 +6,8 @@ import openai
 from epyxid import XID
 from pydantic import BaseModel, Field
 
+from intentkit.clients.s3 import store_image
 from intentkit.skills.openai.base import OpenAIBaseTool
-from intentkit.utils.s3 import store_image
 
 logger = logging.getLogger(__name__)
 

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/skills/openai/gpt_avatar_generator.py

@@ -7,8 +7,8 @@ import openai
 from epyxid import XID
 from pydantic import BaseModel, Field
 
+from intentkit.clients.s3 import store_image_bytes
 from intentkit.skills.openai.base import OpenAIBaseTool
-from intentkit.utils.s3 import store_image_bytes
 
 logger = logging.getLogger(__name__)
 

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/skills/openai/gpt_image_generation.py

@@ -8,8 +8,8 @@ import openai
 from epyxid import XID
 from pydantic import BaseModel, Field
 
+from intentkit.clients.s3 import store_image_bytes
 from intentkit.skills.openai.base import OpenAIBaseTool
-from intentkit.utils.s3 import store_image_bytes
 
 logger = logging.getLogger(__name__)
 

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/skills/openai/gpt_image_mini_generator.py

@@ -8,9 +8,9 @@ import openai
 from epyxid import XID
 from pydantic import BaseModel
 
+from intentkit.clients.s3 import store_image_bytes
 from intentkit.skills.openai.base import OpenAIBaseTool
 from intentkit.skills.openai.gpt_image_generation import GPTImageGenerationInput
-from intentkit.utils.s3 import store_image_bytes
 
 logger = logging.getLogger(__name__)
 

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/skills/openai/gpt_image_to_image.py

@@ -10,8 +10,8 @@ import openai
 from epyxid import XID
 from pydantic import BaseModel, Field
 
+from intentkit.clients.s3 import store_image_bytes
 from intentkit.skills.openai.base import OpenAIBaseTool
-from intentkit.utils.s3 import store_image_bytes
 
 logger = logging.getLogger(__name__)
 

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/skills/venice_audio/venice_audio.py

@@ -6,9 +6,9 @@ from typing import Any
 import httpx
 from pydantic import BaseModel
 
+from intentkit.clients.s3 import store_file
 from intentkit.skills.venice_audio.base import VeniceAudioBaseTool
 from intentkit.skills.venice_audio.input import AllowedAudioFormat, VeniceAudioInput
-from intentkit.utils.s3 import store_file
 
 logger = logging.getLogger(__name__)
 

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/skills/venice_image/api.py

@@ -12,7 +12,7 @@ from typing import Any
 
 import httpx
 
-from intentkit.utils.s3 import store_image_bytes
+from intentkit.clients.s3 import store_image_bytes
 
 logger = logging.getLogger(__name__)
 

{intentkit-0.8.34.dev2 → intentkit-0.8.34.dev4}/intentkit/skills/venice_image/image_generation/image_generation_base.py

@@ -5,13 +5,14 @@ from typing import Any, Literal
 
 from pydantic import BaseModel, Field
 
+from intentkit.clients.s3 import store_image_bytes
+
 # Import the generic base
 from intentkit.skills.base import ToolException
 from intentkit.skills.venice_image.base import VeniceImageBaseTool
 from intentkit.skills.venice_image.image_generation.image_generation_input import (
     VeniceImageGenerationInput,
 )
-from intentkit.utils.s3 import store_image_bytes
 
 logger = logging.getLogger(__name__)