intentkit 0.8.34__tar.gz → 0.8.34.dev2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intentkit
-Version: 0.8.34
+Version: 0.8.34.dev2
 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 → intentkit-0.8.34.dev2}/intentkit/__init__.py

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

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

@@ -7,6 +7,7 @@ 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]
@@ -70,16 +71,9 @@ 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 S3
+        # AWS
         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"
@@ -95,6 +89,7 @@ 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"
@@ -188,6 +183,9 @@ 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 → intentkit-0.8.34.dev2}/intentkit/core/chat.py

@@ -6,7 +6,7 @@ including clearing thread history using LangGraph's checkpointer.
 
 import logging
 
-from intentkit.models.db import get_checkpointer
+from intentkit.models.db import get_langgraph_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_checkpointer()
+        checkpointer = get_langgraph_checkpointer()
 
         # Use the official LangGraph method to delete all thread content
         await checkpointer.adelete_thread(thread_id)

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

@@ -26,7 +26,6 @@ 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
@@ -54,10 +53,7 @@ from intentkit.models.chat import (
     ChatMessageSkillCall,
 )
 from intentkit.models.credit import CreditAccount, OwnerType
-from intentkit.models.db import (
-    get_checkpointer,
-    get_session,
-)
+from intentkit.models.db import get_langgraph_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
@@ -125,10 +121,7 @@ async def build_agent(
     )
 
     # ==== Store buffered conversation history in memory.
-    try:
-        checkpointer = get_checkpointer()
-    except RuntimeError:
-        checkpointer = InMemorySaver()
+    memory = get_langgraph_checkpointer()
 
     # ==== Load skills
     tools: list[BaseTool | dict] = []
@@ -199,7 +192,7 @@ async def build_agent(
         middleware=middleware,
         state_schema=AgentState,
         context_schema=AgentContext,
-        checkpointer=checkpointer,
+        checkpointer=memory,
         debug=config.debug_checkpoint,
         name=agent.id,
     )
@@ -374,7 +367,6 @@ 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 → intentkit-0.8.34.dev2}/intentkit/core/scheduler.py

@@ -13,9 +13,9 @@ from intentkit.core.agent import (
     update_agents_account_snapshot,
     update_agents_statistics,
 )
-from intentkit.core.cleanup import cleanup_checkpoints
 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,15 +141,12 @@ def create_scheduler(
         replace_existing=True,
     )
 
-    # Cleanup old checkpoints daily at UTC 2:20
-    # Cleanup old checkpoints daily at UTC 2:20
-
+    # Clean up old LangGraph checkpoints daily at UTC 2:30
     scheduler.add_job(
         cleanup_checkpoints,
-        trigger=CronTrigger(hour=2, minute=20, timezone="UTC"),
+        trigger=CronTrigger(hour=2, minute=30, timezone="UTC"),
         id="cleanup_checkpoints",
-        name="Cleanup old checkpoints",
-        kwargs={"days": 90, "dry_run": False},
+        name="Clean up old checkpoints",
         replace_existing=True,
     )
 

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

@@ -3,17 +3,19 @@ from contextlib import asynccontextmanager
 from typing import Annotated
 from urllib.parse import quote_plus
 
-from langgraph.checkpoint.postgres.shallow import AsyncShallowPostgresSaver
+from langgraph.checkpoint.memory import InMemorySaver
+from langgraph.checkpoint.postgres.aio import AsyncPostgresSaver
+from langgraph.types import Checkpointer
 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
-_connection_pool: AsyncConnectionPool | None = None
-_checkpointer: AsyncShallowPostgresSaver | None = None
+_langgraph_checkpointer: Checkpointer | None = None
 
 
 async def check_connection(conn):
@@ -52,9 +54,9 @@ async def init_db(
         auto_migrate: Whether to run migrations automatically (default: True)
         pool_size: Database connection pool size (default: 3)
     """
-    global engine, _connection_pool, _checkpointer
-    # Initialize psycopg pool and AsyncShallowPostgresSaver if not already initialized
-    if _connection_pool is None:
+    global engine, _langgraph_checkpointer
+    # Initialize psycopg pool and AsyncPostgresSaver if not already initialized
+    if _langgraph_checkpointer is None:
         if host:
             conn_string = (
                 f"postgresql://{username}:{quote_plus(password)}@{host}:{port}/{dbname}"
@@ -70,17 +72,13 @@ async def init_db(
                 # Set connection max lifetime to prevent stale connections
                 max_lifetime=3600,  # 1 hour
             )
-            _connection_pool = pool
-            _checkpointer = AsyncShallowPostgresSaver(pool)
+            _langgraph_checkpointer = AsyncPostgresSaver(pool)
             if auto_migrate:
                 # Migrate can not use pool, so we start from scratch
-                async with AsyncShallowPostgresSaver.from_conn_string(
-                    conn_string
-                ) as saver:
+                async with AsyncPostgresSaver.from_conn_string(conn_string) as saver:
                     await saver.setup()
         else:
-            # For in-memory, we don't need a pool, but we need to handle it if requested
-            pass
+            _langgraph_checkpointer = InMemorySaver()
     # Initialize SQLAlchemy engine with pool settings
     if engine is None:
         if host:
@@ -143,23 +141,75 @@ def get_engine() -> AsyncEngine:
     return engine
 
 
-def get_connection_pool() -> AsyncConnectionPool:
-    """Get the AsyncConnectionPool instance.
+def get_langgraph_checkpointer() -> Checkpointer:
+    """Get the AsyncPostgresSaver instance for langgraph.
 
     Returns:
-        AsyncConnectionPool: The AsyncConnectionPool instance
+        AsyncPostgresSaver: The AsyncPostgresSaver instance
     """
-    if _connection_pool is None:
+    if _langgraph_checkpointer is None:
         raise RuntimeError("Database pool not initialized. Call init_db first.")
-    return _connection_pool
+    return _langgraph_checkpointer
 
 
-def get_checkpointer() -> AsyncShallowPostgresSaver:
-    """Get the AsyncShallowPostgresSaver instance.
-
-    Returns:
-        AsyncShallowPostgresSaver: The AsyncShallowPostgresSaver instance
+async def cleanup_checkpoints() -> None:
     """
-    if _checkpointer is None:
-        raise RuntimeError("Database checkpointer not initialized. Call init_db first.")
-    return _checkpointer
+    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
+        );
+    """)
+
+    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

intentkit-0.8.34.dev2/intentkit/models/db_mig.py

@@ -0,0 +1,103 @@
+"""Database migration utilities."""
+
+import logging
+from collections.abc import Callable
+
+from sqlalchemy import Column, MetaData, inspect, text
+
+from intentkit.models.base import Base
+
+logger = logging.getLogger(__name__)
+
+
+async def add_column_if_not_exists(
+    conn, dialect, table_name: str, column: Column
+) -> None:
+    """Add a column to a table if it doesn't exist.
+
+    Args:
+        conn: SQLAlchemy conn
+        table_name: Name of the table
+        column: Column to add
+    """
+
+    # Use run_sync to perform inspection on the connection
+    def _get_columns(connection):
+        inspector = inspect(connection)
+        return [c["name"] for c in inspector.get_columns(table_name)]
+
+    columns = await conn.run_sync(_get_columns)
+
+    if column.name not in columns:
+        # Build column definition
+        column_def = f"{column.name} {column.type.compile(dialect)}"
+
+        # Add DEFAULT if specified
+        if column.default is not None:
+            if hasattr(column.default, "arg"):
+                default_value = column.default.arg
+                if not isinstance(default_value, Callable):
+                    if isinstance(default_value, bool):
+                        default_value = str(default_value).lower()
+                    elif isinstance(default_value, str):
+                        default_value = f"'{default_value}'"
+                    elif isinstance(default_value, list | dict):
+                        default_value = "'{}'"
+                    column_def += f" DEFAULT {default_value}"
+
+        # Execute ALTER TABLE
+        await conn.execute(text(f"ALTER TABLE {table_name} ADD COLUMN {column_def}"))
+        logger.info(f"Added column {column.name} to table {table_name}")
+
+
+async def update_table_schema(conn, dialect, model_cls) -> None:
+    """Update table schema by adding missing columns from the model.
+
+    Args:
+        conn: SQLAlchemy conn
+        dialect: SQLAlchemy dialect
+        model_cls: SQLAlchemy model class to check for new columns
+    """
+    if not hasattr(model_cls, "__table__"):
+        return
+
+    table_name = model_cls.__tablename__
+    for name, column in model_cls.__table__.columns.items():
+        if name != "id":  # Skip primary key
+            await add_column_if_not_exists(conn, dialect, table_name, column)
+
+
+async def safe_migrate(engine) -> None:
+    """Safely migrate all SQLAlchemy models by adding new columns.
+
+    Args:
+        engine: SQLAlchemy engine
+    """
+    logger.info("Starting database schema migration")
+    dialect = engine.dialect
+
+    async with engine.begin() as conn:
+        try:
+            # Create tables if they don't exist
+            await conn.run_sync(Base.metadata.create_all)
+
+            # Get existing table metadata
+            metadata = MetaData()
+            await conn.run_sync(metadata.reflect)
+
+            # Update schema for all model classes
+            for mapper in Base.registry.mappers:
+                model_cls = mapper.class_
+                if hasattr(model_cls, "__tablename__"):
+                    table_name = model_cls.__tablename__
+                    if table_name in metadata.tables:
+                        # We need a sync wrapper for the async update_table_schema
+                        async def update_table_wrapper():
+                            await update_table_schema(conn, dialect, model_cls)
+
+                        await update_table_wrapper()
+        except Exception as e:
+            logger.error(f"Error updating database schema: {str(e)}")
+            raise
+
+    logger.info("Database schema updated successfully")

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

@@ -167,6 +167,7 @@ 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 → intentkit-0.8.34.dev2}/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__)
 
@@ -77,6 +77,7 @@ class ImageGenerationAnimagineXL(HeuristBaseTool):
         """
         context = self.get_context()
         skill_config = context.agent.skill_config(self.category)
+        skill_config = skill_config
 
         # Get the Heurist API key from configuration
         if "api_key" in skill_config and skill_config["api_key"]:

{intentkit-0.8.34 → intentkit-0.8.34.dev2}/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__)
 
@@ -76,6 +76,7 @@ class ImageGenerationArthemyComics(HeuristBaseTool):
         """
         context = self.get_context()
         skill_config = context.agent.skill_config(self.category)
+        skill_config = skill_config
 
         # Get the Heurist API key from configuration
         if "api_key" in skill_config and skill_config["api_key"]:

{intentkit-0.8.34 → intentkit-0.8.34.dev2}/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__)
 
@@ -76,6 +76,7 @@ class ImageGenerationArthemyReal(HeuristBaseTool):
         """
         context = self.get_context()
         skill_config = context.agent.skill_config(self.category)
+        skill_config = skill_config
 
         # Get the Heurist API key from configuration
         if "api_key" in skill_config and skill_config["api_key"]:

{intentkit-0.8.34 → intentkit-0.8.34.dev2}/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__)
 
@@ -76,6 +76,7 @@ class ImageGenerationBrainDance(HeuristBaseTool):
         """
         context = self.get_context()
         skill_config = context.agent.skill_config(self.category)
+        skill_config = skill_config
 
         # Get the Heurist API key from configuration
         if "api_key" in skill_config and skill_config["api_key"]:

{intentkit-0.8.34 → intentkit-0.8.34.dev2}/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__)
 
@@ -76,6 +76,7 @@ class ImageGenerationCyberRealisticXL(HeuristBaseTool):
         """
         context = self.get_context()
         skill_config = context.agent.skill_config(self.category)
+        skill_config = skill_config
 
         # Get the Heurist API key from configuration
         if "api_key" in skill_config and skill_config["api_key"]:

{intentkit-0.8.34 → intentkit-0.8.34.dev2}/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__)
 
@@ -76,6 +76,7 @@ class ImageGenerationFlux1Dev(HeuristBaseTool):
         """
         context = self.get_context()
         skill_config = context.agent.skill_config(self.category)
+        skill_config = skill_config
 
         # Get the Heurist API key from configuration
         if "api_key" in skill_config and skill_config["api_key"]:

{intentkit-0.8.34 → intentkit-0.8.34.dev2}/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__)
 
@@ -76,6 +76,7 @@ class ImageGenerationSDXL(HeuristBaseTool):
         """
         context = self.get_context()
         skill_config = context.agent.skill_config(self.category)
+        skill_config = skill_config
 
         # Get the Heurist API key from configuration
         if "api_key" in skill_config and skill_config["api_key"]:

{intentkit-0.8.34 → intentkit-0.8.34.dev2}/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 → intentkit-0.8.34.dev2}/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 → intentkit-0.8.34.dev2}/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 → intentkit-0.8.34.dev2}/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 → intentkit-0.8.34.dev2}/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 → intentkit-0.8.34.dev2}/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 → intentkit-0.8.34.dev2}/intentkit/skills/venice_image/api.py

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

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

@@ -5,14 +5,13 @@ 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__)