refactor: public info

Author: hyacinthus

intentkit/core/agent/__init__.py

@@ -7,6 +7,7 @@
 )
 from .management import create_agent, deploy_agent, override_agent, patch_agent
 from .notifications import send_agent_notification
+from .public_info import override_public_info, update_public_info
 from .queries import get_agent, get_agent_by_id_or_slug, iterate_agent_id_batches
 from .wallet import process_agent_wallet
 
@@ -25,4 +26,6 @@
     "update_agents_account_snapshot",
     "update_agents_assets",
     "update_agents_statistics",
+    "update_public_info",
+    "override_public_info",
 ]

intentkit/core/agent/public_info.py

@@ -0,0 +1,65 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from sqlalchemy import func, select
+
+from intentkit.config.db import get_session
+from intentkit.models.agent import AgentPublicInfo, AgentTable
+from intentkit.utils.error import IntentKitAPIError
+
+if TYPE_CHECKING:
+    from intentkit.models.agent import Agent
+
+
+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
+
+    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")
+
+        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()
+
+        await session.commit()
+        await session.refresh(db_agent)
+
+        return Agent.model_validate(db_agent)
+
+
+async def override_public_info(*, agent_id: str, public_info: AgentPublicInfo) -> Agent:
+    """Override agent public info with all fields from this instance."""
+    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")
+
+        update_data = public_info.model_dump()
+        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()
+
+        await session.commit()
+        await session.refresh(db_agent)
+
+        return Agent.model_validate(db_agent)

intentkit/core/manager/skills/public_info.py

@@ -8,6 +8,7 @@
 from langchain_core.tools import ArgsSchema
 from pydantic import BaseModel
 
+from intentkit.core.agent.public_info import update_public_info
 from intentkit.core.manager.service import get_latest_public_info
 from intentkit.core.manager.skills.base import ManagerSkill
 from intentkit.models.agent import AgentPublicInfo
@@ -88,7 +89,10 @@ async def _arun(self, **kwargs: Any) -> str:
         )
 
         public_info = AgentPublicInfo.model_validate(kwargs["public_info_update"])
-        updated_agent = await public_info.update(context.agent_id)
+        updated_agent = await update_public_info(
+            agent_id=context.agent_id,
+            public_info=public_info,
+        )
         updated_public_info = AgentPublicInfo.model_validate(updated_agent)
 
         return json.dumps(updated_public_info.model_dump(mode="json"), indent=2)

intentkit/models/agent/__init__.py

@@ -3,8 +3,7 @@
 from .autonomous import AgentAutonomous, AgentAutonomousStatus
 from .core import AgentCore, AgentVisibility
 from .db import AgentTable, AgentUserInputColumns
-from .example import AgentExample
-from .public_info import AgentPublicInfo
+from .public_info import AgentExample, AgentPublicInfo
 from .response import AgentResponse
 from .user_input import AgentCreate, AgentUpdate, AgentUserInput
 

intentkit/models/agent/example.py

@@ -1,19 +1,45 @@
 from __future__ import annotations
 
 from decimal import Decimal
-from typing import TYPE_CHECKING, Annotated, Any
+from typing import Annotated, Any
 
 from pydantic import BaseModel, ConfigDict
 from pydantic import Field as PydanticField
-from sqlalchemy import func, select
 
-from intentkit.config.db import get_session
-from intentkit.models.agent.db import AgentTable
-from intentkit.models.agent.example import AgentExample
-from intentkit.utils.error import IntentKitAPIError
 
-if TYPE_CHECKING:
-    from intentkit.models.agent.agent import Agent
+class AgentExample(BaseModel):
+    """Agent example configuration."""
+
+    name: Annotated[
+        str,
+        PydanticField(
+            description="Name of the example",
+            max_length=50,
+            json_schema_extra={
+                "x-placeholder": "Add a name for the example",
+            },
+        ),
+    ]
+    description: Annotated[
+        str,
+        PydanticField(
+            description="Description of the example",
+            max_length=200,
+            json_schema_extra={
+                "x-placeholder": "Add a short description for the example",
+            },
+        ),
+    ]
+    prompt: Annotated[
+        str,
+        PydanticField(
+            description="Example prompt",
+            max_length=2000,
+            json_schema_extra={
+                "x-placeholder": "The prompt will be sent to the agent",
+            },
+        ),
+    ]
 
 
 class AgentPublicInfo(BaseModel):
@@ -132,72 +158,3 @@ class AgentPublicInfo(BaseModel):
             description="Public extra data of the agent",
         ),
     ]
-
-    async def update(self, agent_id: str) -> "Agent":
-        """Update agent public info with only the fields that are explicitly provided.
-
-        This method only updates fields that are explicitly set in this instance,
-        leaving other fields unchanged. This is more efficient than override as it
-        reduces context usage and minimizes the risk of accidentally changing fields.
-
-        Args:
-            agent_id: The ID of the agent to update
-
-        Returns:
-            The updated Agent instance
-        """
-        from intentkit.models.agent.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")
-
-            update_data = self.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()
-
-            await session.commit()
-            await session.refresh(db_agent)
-
-            return Agent.model_validate(db_agent)
-
-    async def override(self, agent_id: str) -> "Agent":
-        """Override agent public info with all fields from this instance.
-
-        Args:
-            agent_id: The ID of the agent to override
-
-        Returns:
-            The updated Agent instance
-        """
-        from intentkit.models.agent.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")
-
-            update_data = self.model_dump()
-            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()
-
-            await session.commit()
-            await session.refresh(db_agent)
-
-            return Agent.model_validate(db_agent)

intentkit/models/agent/public_info.py

@@ -12,7 +12,7 @@
 from pydantic.main import IncEx
 
 from intentkit.models.agent.agent import Agent
-from intentkit.models.agent.example import AgentExample
+from intentkit.models.agent.public_info import AgentExample
 from intentkit.models.agent_data import AgentData