intentkit 1.2.0__tar.gz → 1.2.1__tar.gz

{intentkit-1.2.0 → intentkit-1.2.1}/PKG-INFO

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

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

{intentkit-1.2.0 → intentkit-1.2.1}/intentkit/core/agent/__init__.py

@@ -14,6 +14,7 @@ from .management import (
 )
 from .notifications import send_agent_notification
 from .public_info import override_public_info, update_public_info
+from .publish import publish_agent, unpublish_agent
 from .queries import get_agent, get_agent_by_id_or_slug, iterate_agent_id_batches
 from .wallet import process_agent_wallet, set_agent_safe_token_spending_limit
 
@@ -36,4 +37,6 @@ __all__ = [
     "update_agents_statistics",
     "update_public_info",
     "override_public_info",
+    "publish_agent",
+    "unpublish_agent",
 ]

{intentkit-1.2.0 → intentkit-1.2.1}/intentkit/core/agent/public_info.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
 
 from sqlalchemy import func, select
 
@@ -12,6 +12,19 @@ if TYPE_CHECKING:
     from intentkit.models.agent import Agent
 
 
+def apply_public_info_update(db_agent: Any, public_info: AgentPublicInfo) -> None:
+    """Merge the explicitly-set fields of ``public_info`` onto ``db_agent``.
+
+    Shared by ``update_public_info`` and ``publish_agent`` so the field-copy
+    rules and ``public_info_updated_at`` bump live in one place.
+    """
+    update_data = public_info.model_dump(exclude_unset=True)
+    for key, value in update_data.items():
+        if hasattr(db_agent, key):
+            setattr(db_agent, key, value)
+    db_agent.public_info_updated_at = func.now()
+
+
 async def update_public_info(*, agent_id: str, public_info: AgentPublicInfo) -> Agent:
     """Update agent public info with only the fields that are explicitly provided."""
     from intentkit.models.agent import Agent
@@ -25,13 +38,7 @@ async def update_public_info(*, agent_id: str, public_info: AgentPublicInfo) ->
         if not db_agent:
             raise IntentKitAPIError(404, "NotFound", f"Agent {agent_id} not found")
 
-        update_data = public_info.model_dump(exclude_unset=True)
-
-        for key, value in update_data.items():
-            if hasattr(db_agent, key):
-                setattr(db_agent, key, value)
-
-        db_agent.public_info_updated_at = func.now()
+        apply_public_info_update(db_agent, public_info)
 
         await session.commit()
         await session.refresh(db_agent)

intentkit-1.2.1/intentkit/core/agent/publish.py

@@ -0,0 +1,126 @@
+"""Publish / unpublish helpers for team-owned agents."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from sqlalchemy import delete, func, select
+
+from intentkit.config.db import get_session
+from intentkit.core.agent.public_info import apply_public_info_update
+from intentkit.models.agent import AgentPublicInfo, AgentTable
+from intentkit.models.agent.core import AgentVisibility
+from intentkit.models.team import TeamTable
+from intentkit.models.team_feed import TeamSubscriptionTable
+from intentkit.utils.error import IntentKitAPIError
+
+if TYPE_CHECKING:
+    from intentkit.models.agent import Agent
+
+
+async def publish_agent(*, agent_id: str, public_info: AgentPublicInfo) -> "Agent":
+    """Mark an agent as public after merging in the supplied public info.
+
+    Enforces the owning team's ``public_agent_limit`` before flipping
+    visibility. Only fields explicitly provided in ``public_info`` are written
+    so callers can update a single field (matches ``update_public_info``).
+
+    Raises:
+        IntentKitAPIError 404: agent missing.
+        IntentKitAPIError 400: agent has no team_id (cannot enforce limit).
+        IntentKitAPIError 403: team has reached its public_agent_limit.
+    """
+    from intentkit.models.agent import Agent
+
+    async with get_session() as session:
+        result = await session.execute(
+            select(AgentTable).where(AgentTable.id == agent_id)
+        )
+        db_agent = result.scalar_one_or_none()
+        if not db_agent:
+            raise IntentKitAPIError(404, "NotFound", f"Agent {agent_id} not found")
+
+        if not db_agent.team_id:
+            raise IntentKitAPIError(
+                400,
+                "AgentHasNoTeam",
+                "Only team-owned agents can be published",
+            )
+
+        # Re-publishing an already public agent is allowed and bypasses the
+        # quota check so operators can update public_info without losing
+        # access to their own existing slot.
+        is_already_public = (
+            db_agent.visibility is not None
+            and db_agent.visibility >= AgentVisibility.PUBLIC
+        )
+
+        if not is_already_public:
+            # SELECT FOR UPDATE on the team row serializes concurrent publishes
+            # against the same team so the limit can't be exceeded by a race
+            # between two simultaneous quota checks.
+            team = await session.get(TeamTable, db_agent.team_id, with_for_update=True)
+            if team is None:
+                raise IntentKitAPIError(
+                    404, "TeamNotFound", f"Team {db_agent.team_id} not found"
+                )
+
+            current_public_count = (
+                await session.scalar(
+                    select(func.count(AgentTable.id)).where(
+                        AgentTable.team_id == db_agent.team_id,
+                        AgentTable.visibility >= AgentVisibility.PUBLIC,
+                        AgentTable.archived_at.is_(None),
+                    )
+                )
+                or 0
+            )
+
+            if current_public_count >= team.public_agent_limit:
+                raise IntentKitAPIError(
+                    403,
+                    "PublicAgentLimitReached",
+                    f"Team has reached its public agent limit ({team.public_agent_limit})",
+                )
+
+        apply_public_info_update(db_agent, public_info)
+        db_agent.visibility = AgentVisibility.PUBLIC
+
+        await session.commit()
+        await session.refresh(db_agent)
+
+        return Agent.model_validate(db_agent)
+
+
+async def unpublish_agent(*, agent_id: str) -> "Agent":
+    """Flip an agent back to TEAM visibility and clear its subscriptions.
+
+    Activity / post feed rows are intentionally retained. Only the
+    ``team_subscriptions`` rows are cleared so the agent stops appearing on
+    subscriber timelines going forward.
+
+    Raises:
+        IntentKitAPIError 404: agent missing.
+    """
+    from intentkit.models.agent import Agent
+
+    async with get_session() as session:
+        result = await session.execute(
+            select(AgentTable).where(AgentTable.id == agent_id)
+        )
+        db_agent = result.scalar_one_or_none()
+        if not db_agent:
+            raise IntentKitAPIError(404, "NotFound", f"Agent {agent_id} not found")
+
+        db_agent.visibility = AgentVisibility.TEAM
+
+        await session.execute(
+            delete(TeamSubscriptionTable).where(
+                TeamSubscriptionTable.agent_id == agent_id
+            )
+        )
+
+        await session.commit()
+        await session.refresh(db_agent)
+
+        return Agent.model_validate(db_agent)

{intentkit-1.2.0 → intentkit-1.2.1}/intentkit/models/team.py

@@ -171,6 +171,13 @@ class TeamTable(Base):
         default=TeamPlan.NONE.value,
         server_default=TeamPlan.NONE.value,
     )
+    public_agent_limit: Mapped[int] = mapped_column(
+        Integer,
+        nullable=False,
+        default=1,
+        server_default="1",
+        comment="Maximum number of agents this team is allowed to publish",
+    )
     plan_expires_at: Mapped[datetime | None] = mapped_column(
         DateTime(timezone=True),
         nullable=True,
@@ -363,6 +370,14 @@ class Team(TeamCreate):
         TeamPlan,
         Field(default=TeamPlan.NONE, description="Pricing plan tier"),
     ] = TeamPlan.NONE
+    public_agent_limit: Annotated[
+        int,
+        Field(
+            default=1,
+            ge=0,
+            description="Maximum number of agents this team is allowed to publish",
+        ),
+    ] = 1
 
     @field_validator("plan", mode="before")
     @classmethod

{intentkit-1.2.0 → intentkit-1.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "intentkit"
-version = "1.2.0"
+version = "1.2.1"
 description = "Intent-based AI Agent Platform - Core Package"
 authors = [
     { name = "Ruihua", email = "ruihua@crestal.network" },