intentkit 0.7.5.dev3__py3-none-any.whl → 0.8.34.dev7__py3-none-any.whl

intentkit/models/db.py

@@ -1,18 +1,19 @@
+from collections.abc import AsyncGenerator
 from contextlib import asynccontextmanager
-from typing import Annotated, AsyncGenerator, Optional
+from typing import Annotated
 from urllib.parse import quote_plus
 
-from intentkit.models.db_mig import safe_migrate
-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.ext.asyncio import AsyncEngine, AsyncSession, create_async_engine
 
-engine = None
-_langgraph_checkpointer: Optional[Checkpointer] = None
+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):
@@ -28,11 +29,11 @@ async def check_connection(conn):
 
 
 async def init_db(
-    host: Optional[str],
-    username: Optional[str],
-    password: Optional[str],
-    dbname: Optional[str],
-    port: Annotated[Optional[str], Field(default="5432", description="Database port")],
+    host: str | None,
+    username: str | None,
+    password: str | None,
+    dbname: str | None,
+    port: Annotated[str | None, Field(default="5432", description="Database port")],
     auto_migrate: Annotated[
         bool, Field(default=True, description="Whether to run migrations automatically")
     ],
@@ -51,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}"
@@ -69,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:
@@ -97,6 +102,7 @@ async def init_db(
 
 
 async def get_db() -> AsyncGenerator[AsyncSession, None]:
+    assert engine is not None, "Database engine not initialized. Call init_db first."
     async with AsyncSession(engine) as session:
         yield session
 
@@ -119,6 +125,7 @@ async def get_session() -> AsyncGenerator[AsyncSession, None]:
         # session is automatically closed
         ```
     """
+    assert engine is not None, "Database engine not initialized. Call init_db first."
     session = AsyncSession(engine)
     try:
         yield session
@@ -132,15 +139,27 @@ def get_engine() -> AsyncEngine:
     Returns:
         AsyncEngine: The SQLAlchemy async engine
     """
+    assert engine is not None, "Database engine not initialized. Call init_db first."
     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
+
+
+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/models/db_mig.py

@@ -1,11 +1,12 @@
 """Database migration utilities."""
 
 import logging
-from typing import Callable
+from collections.abc import Callable
 
-from intentkit.models.base import Base
 from sqlalchemy import Column, MetaData, inspect, text
 
+from intentkit.models.base import Base
+
 logger = logging.getLogger(__name__)
 
 
@@ -40,7 +41,7 @@ async def add_column_if_not_exists(
                         default_value = str(default_value).lower()
                     elif isinstance(default_value, str):
                         default_value = f"'{default_value}'"
-                    elif isinstance(default_value, (list, dict)):
+                    elif isinstance(default_value, list | dict):
                         default_value = "'{}'"
                     column_def += f" DEFAULT {default_value}"
 
@@ -95,8 +96,124 @@ async def safe_migrate(engine) -> None:
                             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
+
+        # Step 1: Add checkpoint_ns column
+        await conn.execute(
+            text(
+                f"ALTER TABLE {table} ADD COLUMN IF NOT EXISTS checkpoint_ns TEXT DEFAULT ''"
+            )
+        )
+
+        # Step 2: Check current and expected 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"
+
+        # Step 3: Handle duplicates BEFORE dropping columns (so we can use checkpoint_id, version, etc.)
+        if current_cols != expected_cols:
+            logger.info(f"Migrating {table} PK from {current_cols} to {expected_cols}")
+
+            if table == "checkpoints" and expected_cols == {
+                "thread_id",
+                "checkpoint_ns",
+            }:
+                # Keep only the latest checkpoint based on MAX(checkpoint_id)
+                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
+                    )
+                """)
+                )
+
+            elif table == "checkpoint_blobs" and expected_cols == {
+                "thread_id",
+                "checkpoint_ns",
+                "channel",
+            }:
+                # Keep only blobs referenced by checkpoints via channel_versions
+                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
+                    )
+                """)
+                )
+
+        # Step 4: Update Primary Key FIRST (before dropping columns)
+        if current_cols != expected_cols:
+            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})"))
+
+        # Step 5: Drop columns that ShallowPostgresSaver doesn't use
+        # (Can only drop AFTER removing from PK)
+        if table == "checkpoints":
+            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":
+            await conn.execute(
+                text("ALTER TABLE checkpoint_blobs DROP COLUMN IF EXISTS version")
+            )

intentkit/models/draft.py

@@ -0,0 +1,222 @@
+from datetime import datetime, timezone
+from enum import Enum
+from typing import Annotated
+
+from epyxid import XID
+from fastapi import status
+from pydantic import BaseModel, Field
+from sqlalchemy import Column, DateTime, Index, String, func, select
+
+from intentkit.models.agent import (
+    AgentUserInput,
+    AgentUserInputColumns,
+)
+from intentkit.models.base import Base
+from intentkit.models.db import get_session
+from intentkit.utils.error import IntentKitAPIError
+
+
+class AgentState(str, Enum):
+    """Agent state."""
+
+    PRIVATE = "private"
+    PUBLIC = "public"
+    CITIZEN = "citizen"
+
+
+class AgentExtra(BaseModel):
+    """Agent extra data in AgentUpdate."""
+
+    state: AgentState = Field(default=AgentState.PRIVATE, description="Agent state")
+    draft_id: str = Field(description="Draft ID")
+    project_id: str | None = Field(
+        default=None, description="Project ID, forward compatible"
+    )
+    request_id: str | None = Field(
+        default=None, description="Request ID, forward compatible"
+    )
+    create_tx_id: str | None = Field(
+        default=None, description="Transaction hash used when the agent was created"
+    )
+
+
+class AgentDraftTable(Base, AgentUserInputColumns):
+    """Agent table db model."""
+
+    __tablename__ = "agent_drafts"
+
+    id = Column(
+        String,
+        primary_key=True,
+        comment="Unique identifier for the agent. Must be URL-safe, containing only lowercase letters, numbers, and hyphens",
+    )
+    agent_id = Column(
+        String,
+        nullable=False,
+        comment="Agent id",
+    )
+    owner = Column(
+        String,
+        nullable=True,
+        comment="Owner identifier of the agent, used for access control",
+    )
+    team_id = Column(
+        String,
+        nullable=True,
+        comment="Team identifier of the agent, used for access control",
+    )
+    version = Column(
+        String,
+        nullable=True,
+        comment="Version hash of the agent",
+    )
+    project_id = Column(
+        String,
+        nullable=True,
+        comment="Project ID, forward compatible",
+    )
+    last_draft_id = Column(
+        String,
+        nullable=True,
+        comment="ID of the last draft that was deployed",
+    )
+    deployed_at = Column(
+        DateTime(timezone=True),
+        nullable=True,
+        comment="Timestamp when the agent was deployed",
+    )
+    # auto timestamp
+    created_at = Column(
+        DateTime(timezone=True),
+        nullable=False,
+        server_default=func.now(),
+        comment="Timestamp when the agent was created",
+    )
+    updated_at = Column(
+        DateTime(timezone=True),
+        nullable=False,
+        server_default=func.now(),
+        onupdate=lambda: datetime.now(timezone.utc),
+        comment="Timestamp when the agent was last updated",
+    )
+
+    # Indexes for optimal query performance
+    __table_args__ = (
+        # Index for queries filtering by agent_id and owner (most common pattern)
+        Index("idx_agent_drafts_agent_owner", "agent_id", "owner"),
+        # Index for queries ordering by created_at (for latest draft queries)
+        Index("idx_agent_drafts_created_at", "created_at"),
+        # Composite index for agent_id, owner, and created_at (covers all common query patterns)
+        Index(
+            "idx_agent_drafts_agent_owner_created", "agent_id", "owner", "created_at"
+        ),
+    )
+
+
+class AgentDraft(AgentUserInput):
+    """Agent draft model."""
+
+    id: Annotated[
+        str,
+        Field(
+            default_factory=lambda: str(XID()),
+            description="Unique identifier for the draft",
+        ),
+    ]
+    agent_id: Annotated[
+        str,
+        Field(
+            description="Agent id",
+        ),
+    ]
+    owner: Annotated[
+        str | None,
+        Field(
+            default=None,
+            description="Owner identifier of the agent, used for access control",
+            max_length=50,
+        ),
+    ]
+    team_id: Annotated[
+        str | None,
+        Field(
+            default=None,
+            description="Team identifier of the agent, used for access control",
+            max_length=50,
+        ),
+    ]
+    version: Annotated[
+        str | None,
+        Field(
+            default=None,
+            description="Version hash of the agent",
+        ),
+    ]
+    project_id: Annotated[
+        str | None,
+        Field(
+            default=None,
+            description="Project ID, forward compatible",
+        ),
+    ]
+    last_draft_id: Annotated[
+        str | None,
+        Field(
+            default=None,
+            description="ID of the last draft that was deployed",
+        ),
+    ]
+    deployed_at: Annotated[
+        datetime | None,
+        Field(
+            default=None,
+            description="Timestamp when the agent was deployed",
+        ),
+    ]
+    # auto timestamp
+    created_at: Annotated[
+        datetime,
+        Field(
+            description="Timestamp when the agent was created, will ignore when importing"
+        ),
+    ]
+    updated_at: Annotated[
+        datetime,
+        Field(
+            description="Timestamp when the agent was last updated, will ignore when importing"
+        ),
+    ]
+
+    @staticmethod
+    async def exist(agent_id: str, user_id: str | None = None) -> None:
+        """Check if an agent exists in the draft table.
+
+        Args:
+            agent_id: The agent ID to check
+            user_id: Optional user ID to check ownership
+
+        Raises:
+            IntentKitAPIError: 404 if agent not found, 403 if user doesn't own the agent
+        """
+        async with get_session() as session:
+            query = (
+                select(AgentDraftTable)
+                .where(AgentDraftTable.agent_id == agent_id)
+                .limit(1)
+            )
+            result = await session.execute(query)
+            draft = result.scalar_one_or_none()
+
+            if not draft:
+                raise IntentKitAPIError(
+                    status.HTTP_404_NOT_FOUND,
+                    "AgentNotFound",
+                    f"Agent {agent_id} not found",
+                )
+
+            if user_id is not None and draft.owner != user_id:
+                raise IntentKitAPIError(
+                    status.HTTP_403_FORBIDDEN,
+                    "AgentForbidden",
+                    "Agent does not belong to user",
+                )

intentkit/models/llm.csv

@@ -0,0 +1,31 @@
+id,name,provider,enabled,input_price,output_price,price_level,context_length,output_length,intelligence,speed,supports_image_input,supports_skill_calls,supports_structured_output,has_reasoning,supports_search,supports_temperature,supports_frequency_penalty,supports_presence_penalty,api_base,timeout
+minimax/minimax-m2:free,MiniMax M2,gatewayz,FALSE,0,0,1,204800,131000,4,2,FALSE,TRUE,FALSE,FALSE,FALSE,TRUE,FALSE,FALSE,https://api.gatewayz.ai/v1,300
+qwen/qwen3-coder:exacto,Qwen3 Coder,gatewayz,FALSE,0.38,1.53,2,262000,262000,3,3,FALSE,TRUE,FALSE,FALSE,FALSE,TRUE,FALSE,FALSE,https://api.gatewayz.ai/v1,300
+google/gemini-2.5-flash,Gemini 2.5 Flash,gatewayz,FALSE,0.3,2.5,2,1050000,65000,3,4,TRUE,TRUE,FALSE,TRUE,FALSE,TRUE,FALSE,FALSE,https://api.gatewayz.ai/v1,300
+google/gemini-2.5-pro,Gemini 2.5 Pro,gatewayz,FALSE,1.25,10,4,1050000,65000,4,2,TRUE,TRUE,FALSE,TRUE,FALSE,TRUE,FALSE,FALSE,https://api.gatewayz.ai/v1,300
+anthropic/claude-sonnet-4.5,Anthropic Claude Sonnet 4.5,gatewayz,FALSE,3,15,5,200000,64000,5,1,TRUE,TRUE,FALSE,TRUE,FALSE,TRUE,FALSE,FALSE,https://api.gatewayz.ai/v1,300
+gpt-5-nano,GPT-5 Nano,openai,TRUE,0.05,0.4,1,400000,128000,3,5,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,FALSE,FALSE,,180
+gpt-5-mini,GPT-5 Mini,openai,TRUE,0.25,2,2,400000,128000,4,4,TRUE,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,FALSE,,180
+gpt-5,GPT-5,openai,TRUE,1.25,10,4,400000,128000,5,3,TRUE,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,FALSE,,180
+gpt-5.1,GPT-5.1,openai,TRUE,1.25,10,4,400000,128000,5,3,TRUE,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,FALSE,,180
+gpt-5.1-codex,GPT-5.1 Codex,openai,TRUE,1.25,10,4,400000,128000,5,3,TRUE,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,FALSE,,180
+gpt-5.1-codex-mini,GPT-5.1 Codex Mini,openai,TRUE,0.25,2,2,400000,128000,4,4,TRUE,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,FALSE,,180
+gpt-4o,GPT-4o,openai,FALSE,2.5,10,4,128000,4096,4,3,TRUE,TRUE,TRUE,FALSE,TRUE,TRUE,FALSE,FALSE,,180
+gpt-4o-mini,GPT-4o Mini,openai,FALSE,0.15,0.6,1,128000,4096,3,4,FALSE,TRUE,TRUE,FALSE,TRUE,TRUE,FALSE,FALSE,,180
+gpt-4.1-nano,GPT-4.1 Nano,openai,TRUE,0.1,0.4,1,128000,4096,3,5,FALSE,TRUE,TRUE,FALSE,FALSE,TRUE,FALSE,FALSE,,180
+gpt-4.1-mini,GPT-4.1 Mini,openai,TRUE,0.4,1.6,2,128000,4096,4,4,FALSE,TRUE,TRUE,FALSE,TRUE,TRUE,FALSE,FALSE,,180
+gpt-4.1,GPT-4.1,openai,TRUE,2,8,4,128000,4096,5,3,TRUE,TRUE,TRUE,FALSE,TRUE,TRUE,FALSE,FALSE,,180
+o3,OpenAI o3,openai,TRUE,2,8,4,200000,100000,5,1,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,FALSE,FALSE,,180
+o4-mini,OpenAI o4-mini,openai,TRUE,1.1,4.4,3,128000,4096,4,3,FALSE,TRUE,TRUE,TRUE,FALSE,FALSE,FALSE,FALSE,,180
+deepseek-chat,Deepseek V3.2 Exp,deepseek,TRUE,0.28,0.42,2,128000,4096,4,3,FALSE,TRUE,TRUE,FALSE,FALSE,TRUE,TRUE,TRUE,https://api.deepseek.com,300
+deepseek-reasoner,DeepSeek V3.2 Exp (Thinking Mode),deepseek,TRUE,0.28,0.42,2,128000,32000,5,1,FALSE,TRUE,TRUE,TRUE,FALSE,TRUE,TRUE,TRUE,https://api.deepseek.com,300
+grok-4,Grok 4,xai,TRUE,3,15,4,256000,4096,5,3,TRUE,TRUE,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,,180
+grok-4-fast-non-reasoning,Grok 4 Fast,xai,TRUE,0.2,0.5,2,2000000,4096,3,4,TRUE,TRUE,TRUE,FALSE,TRUE,TRUE,FALSE,FALSE,,180
+grok-4-fast-reasoning,Grok 4 Fast Reasoning,xai,TRUE,0.2,0.5,2,2000000,4096,3,4,TRUE,TRUE,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,,180
+grok-code-fast-1,Grok Code Fast 1,xai,TRUE,0.2,1.5,3,256000,4096,4,4,FALSE,TRUE,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,,180
+grok-3,Grok 3,xai,TRUE,3,15,4,131072,4096,5,3,FALSE,TRUE,TRUE,FALSE,TRUE,TRUE,FALSE,FALSE,,180
+grok-3-mini,Grok 3 Mini,xai,TRUE,0.3,0.5,2,131072,4096,5,3,FALSE,TRUE,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,,180
+eternalai,Eternal AI (Llama-3.3-70B),eternal,FALSE,0.25,0.75,2,64000,4096,4,3,FALSE,TRUE,TRUE,FALSE,FALSE,TRUE,TRUE,TRUE,https://api.eternalai.org/v1,300
+reigent,REI Network,reigent,FALSE,0.5,1.5,3,32000,4096,4,3,FALSE,TRUE,TRUE,FALSE,FALSE,FALSE,FALSE,FALSE,https://api.reisearch.box/v1,300
+venice-uncensored,Venice Uncensored,venice,TRUE,0.5,2,3,32000,4096,4,3,FALSE,TRUE,TRUE,FALSE,FALSE,FALSE,FALSE,FALSE,https://api.venice.ai/api/v1,300
+venice-llama-4-maverick-17b,Venice Llama 4 Maverick 17B,venice,TRUE,1.5,6,4,32000,4096,4,3,FALSE,TRUE,TRUE,FALSE,FALSE,FALSE,FALSE,FALSE,https://api.venice.ai/api/v1,300