intentkit 0.8.34.dev3__tar.gz → 0.8.34.dev6__tar.gz

{intentkit-0.8.34.dev3 → intentkit-0.8.34.dev6}/PKG-INFO

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

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

{intentkit-0.8.34.dev3/intentkit/utils → intentkit-0.8.34.dev6/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.dev3 → intentkit-0.8.34.dev6}/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.dev3 → intentkit-0.8.34.dev6}/intentkit/core/chat.py

@@ -6,9 +6,7 @@ including clearing thread history using LangGraph's checkpointer.
 
 import logging
 
-from langgraph.checkpoint.postgres.shallow import AsyncShallowPostgresSaver
-
-from intentkit.models.db import get_connection_pool
+from intentkit.models.db import get_checkpointer
 from intentkit.utils.error import IntentKitAPIError
 
 logger = logging.getLogger(__name__)
@@ -36,8 +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
-        pool = get_connection_pool()
-        checkpointer = AsyncShallowPostgresSaver(pool)
+        checkpointer = get_checkpointer()
 
         # Use the official LangGraph method to delete all thread content
         await checkpointer.adelete_thread(thread_id)

{intentkit-0.8.34.dev3 → intentkit-0.8.34.dev6}/intentkit/core/engine.py

@@ -27,7 +27,6 @@ from langchain_core.messages import (
 )
 from langchain_core.tools import BaseTool
 from langgraph.checkpoint.memory import InMemorySaver
-from langgraph.checkpoint.postgres.shallow import AsyncShallowPostgresSaver
 from langgraph.errors import GraphRecursionError
 from langgraph.graph.state import CompiledStateGraph
 from sqlalchemy import func, update
@@ -55,7 +54,10 @@ from intentkit.models.chat import (
     ChatMessageSkillCall,
 )
 from intentkit.models.credit import CreditAccount, OwnerType
-from intentkit.models.db import get_connection_pool, 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
@@ -124,10 +126,9 @@ async def build_agent(
 
     # ==== Store buffered conversation history in memory.
     try:
-        pool = get_connection_pool()
-        memory = AsyncShallowPostgresSaver(pool)
+        checkpointer = get_checkpointer()
     except RuntimeError:
-        memory = InMemorySaver()
+        checkpointer = InMemorySaver()
 
     # ==== Load skills
     tools: list[BaseTool | dict] = []
@@ -198,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,
     )
@@ -373,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.dev3 → intentkit-0.8.34.dev6}/intentkit/models/db.py

@@ -13,6 +13,7 @@ from intentkit.models.db_mig import safe_migrate
 
 engine: AsyncEngine | None = None
 _connection_pool: AsyncConnectionPool | None = None
+_checkpointer: AsyncShallowPostgresSaver | None = None
 
 
 async def check_connection(conn):
@@ -51,7 +52,7 @@ 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
+    global engine, _connection_pool, _checkpointer
     # Initialize psycopg pool and AsyncShallowPostgresSaver if not already initialized
     if _connection_pool is None:
         if host:
@@ -70,6 +71,7 @@ async def init_db(
                 max_lifetime=3600,  # 1 hour
             )
             _connection_pool = pool
+            _checkpointer = AsyncShallowPostgresSaver(pool)
             if auto_migrate:
                 # Migrate can not use pool, so we start from scratch
                 async with AsyncShallowPostgresSaver.from_conn_string(
@@ -150,3 +152,14 @@ def get_connection_pool() -> AsyncConnectionPool:
     if _connection_pool is None:
         raise RuntimeError("Database pool not initialized. Call init_db first.")
     return _connection_pool
+
+
+def get_checkpointer() -> AsyncShallowPostgresSaver:
+    """Get the AsyncShallowPostgresSaver instance.
+
+    Returns:
+        AsyncShallowPostgresSaver: The AsyncShallowPostgresSaver instance
+    """
+    if _checkpointer is None:
+        raise RuntimeError("Database checkpointer not initialized. Call init_db first.")
+    return _checkpointer

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

@@ -0,0 +1,220 @@
+"""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()
+
+            # Migrate checkpoints tables
+            await migrate_checkpoints_table(conn)
+        except Exception as e:
+            logger.error(f"Error updating database schema: {str(e)}")
+            raise
+
+    logger.info("Database schema updated successfully")
+
+
+async def migrate_checkpoints_table(conn) -> None:
+    """Migrate checkpoints tables to support langgraph 2.0."""
+    tables = ["checkpoints", "checkpoint_blobs", "checkpoint_writes"]
+
+    def _get_tables(connection):
+        insp = inspect(connection)
+        return insp.get_table_names()
+
+    existing_tables = await conn.run_sync(_get_tables)
+
+    for table in tables:
+        if table not in existing_tables:
+            continue
+
+        # 1. Add checkpoint_ns column
+        await conn.execute(
+            text(
+                f"ALTER TABLE {table} ADD COLUMN IF NOT EXISTS checkpoint_ns TEXT DEFAULT ''"
+            )
+        )
+
+        # 2. Drop columns that ShallowPostgresSaver doesn't use
+        if table == "checkpoints":
+            # ShallowPostgresSaver doesn't use checkpoint_id or parent_checkpoint_id
+            await conn.execute(
+                text("ALTER TABLE checkpoints DROP COLUMN IF EXISTS checkpoint_id")
+            )
+            await conn.execute(
+                text(
+                    "ALTER TABLE checkpoints DROP COLUMN IF EXISTS parent_checkpoint_id"
+                )
+            )
+        elif table == "checkpoint_blobs":
+            # ShallowPostgresSaver doesn't use version column
+            await conn.execute(
+                text("ALTER TABLE checkpoint_blobs DROP COLUMN IF EXISTS version")
+            )
+
+        # 3. Update Primary Key
+        def _check_pk(connection, table_name=table):
+            insp = inspect(connection)
+            return insp.get_pk_constraint(table_name)
+
+        pk = await conn.run_sync(_check_pk)
+        current_cols = set(pk.get("constrained_columns", []))
+
+        # Expected columns depend on table
+        expected_cols = set()
+        pk_cols = ""
+        if table == "checkpoints":
+            expected_cols = {"thread_id", "checkpoint_ns"}
+            pk_cols = "thread_id, checkpoint_ns"
+        elif table == "checkpoint_blobs":
+            expected_cols = {"thread_id", "checkpoint_ns", "channel"}
+            pk_cols = "thread_id, checkpoint_ns, channel"
+        elif table == "checkpoint_writes":
+            expected_cols = {
+                "thread_id",
+                "checkpoint_ns",
+                "checkpoint_id",
+                "task_id",
+                "idx",
+            }
+            pk_cols = "thread_id, checkpoint_ns, checkpoint_id, task_id, idx"
+
+        if current_cols != expected_cols:
+            logger.info(f"Migrating {table} PK from {current_cols} to {expected_cols}")
+
+            # If migrating checkpoints to (thread_id, checkpoint_ns), we need to handle duplicates
+            if table == "checkpoints" and expected_cols == {
+                "thread_id",
+                "checkpoint_ns",
+            }:
+                # Keep only the latest checkpoint for each (thread_id, checkpoint_ns) based on checkpoint_id (time-ordered)
+                await conn.execute(
+                    text("""
+                    DELETE FROM checkpoints
+                    WHERE (thread_id, checkpoint_ns, checkpoint_id) NOT IN (
+                        SELECT thread_id, checkpoint_ns, MAX(checkpoint_id)
+                        FROM checkpoints
+                        GROUP BY thread_id, checkpoint_ns
+                    )
+                """)
+                )
+
+            # If migrating checkpoint_blobs to (thread_id, checkpoint_ns, channel), we need to handle duplicates
+            elif table == "checkpoint_blobs" and expected_cols == {
+                "thread_id",
+                "checkpoint_ns",
+                "channel",
+            }:
+                # Keep only blobs that are referenced by the remaining checkpoints
+                # The relationship is: checkpoints.checkpoint -> 'channel_versions' ->> blob.channel = blob.version
+                await conn.execute(
+                    text("""
+                    DELETE FROM checkpoint_blobs cb
+                    WHERE NOT EXISTS (
+                        SELECT 1
+                        FROM checkpoints cp
+                        WHERE cp.thread_id = cb.thread_id
+                          AND cp.checkpoint_ns = cb.checkpoint_ns
+                          AND (cp.checkpoint -> 'channel_versions' ->> cb.channel) = cb.version
+                    )
+                """)
+                )
+
+            if pk.get("name"):
+                await conn.execute(
+                    text(f"ALTER TABLE {table} DROP CONSTRAINT {pk['name']}")
+                )
+
+            await conn.execute(text(f"ALTER TABLE {table} ADD PRIMARY KEY ({pk_cols})"))

{intentkit-0.8.34.dev3 → intentkit-0.8.34.dev6}/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.dev3 → intentkit-0.8.34.dev6}/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,7 +77,6 @@ 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.dev3 → intentkit-0.8.34.dev6}/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,7 +76,6 @@ 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.dev3 → intentkit-0.8.34.dev6}/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,7 +76,6 @@ 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.dev3 → intentkit-0.8.34.dev6}/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,7 +76,6 @@ 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.dev3 → intentkit-0.8.34.dev6}/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,7 +76,6 @@ 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.dev3 → intentkit-0.8.34.dev6}/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,7 +76,6 @@ 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"]: