@ton-agent-kit/plugin-identity 1.1.0 → 1.2.0

package/src/actions/register-agent.ts

@@ -1,125 +1,125 @@
-import { z } from "zod";
-import { Address, internal, toNano } from "@ton/core";
-import { defineAction, toFriendlyAddress, sendTransaction } from "@ton-agent-kit/core";
-import { loadAgentRegistry, saveAgentRegistry } from "../utils";
-import { resolveContractAddress } from "../reputation-config";
-import { buildRegisterBody } from "../reputation-helpers";
-
-export function createRegisterAgentAction(contractAddress?: string) {
-  return defineAction({
-    name: "register_agent",
-    description:
-      "Register an AI agent with its capabilities, name, and description. Other agents can discover it via discover_agent. Supports both on-chain (Tact contract) and local JSON modes.",
-    schema: z.object({
-      name: z
-        .string()
-        .describe("Agent name (e.g., 'market-data', 'trading-bot')"),
-      capabilities: z
-        .union([z.array(z.string()), z.string()])
-        .describe("List of capabilities (e.g., ['price_feed', 'analytics', 'trading'])"),
-      description: z.string().optional().describe("Human-readable description of the agent"),
-      endpoint: z.string().optional().describe("API endpoint where the agent can be reached"),
-      available: z
-        .boolean()
-        .optional()
-        .describe("Whether the agent is currently available. Defaults to true."),
-    }),
-    handler: async (agent, params) => {
-      const agentId = `agent_${params.name.toLowerCase().replace(/[^a-z0-9]/g, "-")}`;
-
-      // Coerce capabilities
-      let capabilities: string[];
-      if (typeof params.capabilities === "string") {
-        try {
-          capabilities = JSON.parse(params.capabilities);
-        } catch {
-          capabilities = params.capabilities.split(",").map((c: string) => c.trim());
-        }
-      } else {
-        capabilities = params.capabilities;
-      }
-
-      const available = params.available !== false;
-
-      // Resolve contract address (factory → config → default → null)
-      const addr = resolveContractAddress(contractAddress, agent.network);
-
-      // Always save to JSON registry (needed for discovery — contract doesn't store names/capabilities)
-      const agentRecord = {
-        id: agentId,
-        name: params.name,
-        address: agent.wallet.address.toRawString(),
-        capabilities,
-        description: params.description || "",
-        endpoint: params.endpoint || null,
-        available,
-        network: agent.network,
-        registeredAt: new Date().toISOString(),
-        reputation: { score: 0, totalTasks: 0, successfulTasks: 0 },
-      };
-
-      const registry = loadAgentRegistry();
-      registry[agentId] = agentRecord;
-      saveAgentRegistry(registry);
-
-      // ── On-chain mode ──
-      if (addr) {
-        try {
-          const capStr = capabilities.join(",");
-          const body = buildRegisterBody(params.name, capStr, available);
-
-          await sendTransaction(agent, [
-            internal({
-              to: Address.parse(addr),
-              value: toNano("0.03"),
-              bounce: true,
-              body,
-            }),
-          ]);
-
-          return {
-            agentId,
-            name: params.name,
-            address: agent.wallet.address.toRawString(),
-            friendlyAddress: toFriendlyAddress(agent.wallet.address, agent.network),
-            capabilities,
-            available,
-            onChain: true,
-            contractAddress: addr,
-            status: "registered (on-chain + local)",
-            message: `Agent "${params.name}" registered on-chain at ${addr.slice(0, 16)}... and locally`,
-          };
-        } catch (err: any) {
-          // On-chain failed but JSON saved — return partial success
-          return {
-            agentId,
-            name: params.name,
-            address: agent.wallet.address.toRawString(),
-            friendlyAddress: toFriendlyAddress(agent.wallet.address, agent.network),
-            capabilities,
-            available,
-            onChain: false,
-            status: "registered (local only — on-chain failed)",
-            message: `Agent "${params.name}" saved locally. On-chain registration failed: ${err.message?.slice(0, 80)}`,
-          };
-        }
-      }
-
-      // ── JSON-only mode ──
-      return {
-        agentId,
-        name: params.name,
-        address: agent.wallet.address.toRawString(),
-        friendlyAddress: toFriendlyAddress(agent.wallet.address, agent.network),
-        capabilities,
-        available,
-        onChain: false,
-        description: params.description || "",
-        status: "registered",
-        dnsHint: `${params.name}.agents.ton (requires TON DNS domain)`,
-      };
-    },
-  });
-}
-
-export const registerAgentAction = createRegisterAgentAction();
+import { z } from "zod";
+import { Address, internal, toNano } from "@ton/core";
+import { defineAction, toFriendlyAddress, sendTransaction } from "@ton-agent-kit/core";
+import { loadAgentRegistry, saveAgentRegistry } from "../utils";
+import { resolveContractAddress } from "../reputation-config";
+import { buildRegisterBody } from "../reputation-helpers";
+
+export function createRegisterAgentAction(contractAddress?: string) {
+  return defineAction({
+    name: "register_agent",
+    description:
+      "Register an AI agent with its capabilities, name, and description. Other agents can discover it via discover_agent. Supports both on-chain (Tact contract) and local JSON modes.",
+    schema: z.object({
+      name: z
+        .string()
+        .describe("Agent name (e.g., 'market-data', 'trading-bot')"),
+      capabilities: z
+        .union([z.array(z.string()), z.string()])
+        .describe("List of capabilities (e.g., ['price_feed', 'analytics', 'trading'])"),
+      description: z.string().optional().describe("Human-readable description of the agent"),
+      endpoint: z.string().optional().describe("API endpoint where the agent can be reached"),
+      available: z
+        .boolean()
+        .optional()
+        .describe("Whether the agent is currently available. Defaults to true."),
+    }),
+    handler: async (agent, params) => {
+      const agentId = `agent_${params.name.toLowerCase().replace(/[^a-z0-9]/g, "-")}`;
+
+      // Coerce capabilities
+      let capabilities: string[];
+      if (typeof params.capabilities === "string") {
+        try {
+          capabilities = JSON.parse(params.capabilities);
+        } catch {
+          capabilities = params.capabilities.split(",").map((c: string) => c.trim());
+        }
+      } else {
+        capabilities = params.capabilities;
+      }
+
+      const available = params.available !== false;
+
+      // Resolve contract address (factory → config → default → null)
+      const addr = resolveContractAddress(contractAddress, agent.network);
+
+      // Always save to JSON registry (needed for discovery — contract doesn't store names/capabilities)
+      const agentRecord = {
+        id: agentId,
+        name: params.name,
+        address: agent.wallet.address.toRawString(),
+        capabilities,
+        description: params.description || "",
+        endpoint: params.endpoint || null,
+        available,
+        network: agent.network,
+        registeredAt: new Date().toISOString(),
+        reputation: { score: 0, totalTasks: 0, successfulTasks: 0 },
+      };
+
+      const registry = loadAgentRegistry();
+      registry[agentId] = agentRecord;
+      saveAgentRegistry(registry);
+
+      // ── On-chain mode ──
+      if (addr) {
+        try {
+          const capStr = capabilities.join(",");
+          const body = buildRegisterBody(params.name, capStr, available);
+
+          await sendTransaction(agent, [
+            internal({
+              to: Address.parse(addr),
+              value: toNano("0.05"),
+              bounce: true,
+              body,
+            }),
+          ]);
+
+          return {
+            agentId,
+            name: params.name,
+            address: agent.wallet.address.toRawString(),
+            friendlyAddress: toFriendlyAddress(agent.wallet.address, agent.network),
+            capabilities,
+            available,
+            onChain: true,
+            contractAddress: addr,
+            status: "registered (on-chain + local)",
+            message: `Agent "${params.name}" registered on-chain at ${addr.slice(0, 16)}... and locally`,
+          };
+        } catch (err: any) {
+          // On-chain failed but JSON saved — return partial success
+          return {
+            agentId,
+            name: params.name,
+            address: agent.wallet.address.toRawString(),
+            friendlyAddress: toFriendlyAddress(agent.wallet.address, agent.network),
+            capabilities,
+            available,
+            onChain: false,
+            status: "registered (local only — on-chain failed)",
+            message: `Agent "${params.name}" saved locally. On-chain registration failed: ${err.message?.slice(0, 80)}`,
+          };
+        }
+      }
+
+      // ── JSON-only mode ──
+      return {
+        agentId,
+        name: params.name,
+        address: agent.wallet.address.toRawString(),
+        friendlyAddress: toFriendlyAddress(agent.wallet.address, agent.network),
+        capabilities,
+        available,
+        onChain: false,
+        description: params.description || "",
+        status: "registered",
+        dnsHint: `${params.name}.agents.ton (requires TON DNS domain)`,
+      };
+    },
+  });
+}
+
+export const registerAgentAction = createRegisterAgentAction();

package/src/actions/trigger-cleanup.ts

@@ -1,43 +1,43 @@
-import { z } from "zod";
-import { Address, internal, toNano, beginCell } from "@ton/core";
-import { defineAction, sendTransaction } from "@ton-agent-kit/core";
-import { resolveContractAddress } from "../reputation-config";
-import { storeTriggerCleanup } from "../contracts/Reputation_Reputation";
-
-export const triggerCleanupAction = defineAction({
-  name: "trigger_cleanup",
-  description:
-    "Manually trigger cleanup of inactive, ghost, or low-score agents from the reputation contract. " +
-    "Cleanup also runs automatically on every registration and rating. " +
-    "Conditions: score<20 with 100+ ratings, inactive 30+ days, ghost (0 ratings after 7 days).",
-  schema: z.object({
-    maxClean: z.coerce.number().optional().describe("Max agents to clean. Default 10, max 50."),
-  }),
-  handler: async (agent, params) => {
-    const addr = resolveContractAddress(undefined, agent.network);
-    if (!addr) {
-      return { triggered: false, message: "No reputation contract configured" };
-    }
-
-    const maxClean = Math.min(params.maxClean || 10, 50);
-    const body = beginCell()
-      .store(storeTriggerCleanup({ $$type: "TriggerCleanup", maxClean: BigInt(maxClean) }))
-      .endCell();
-
-    await sendTransaction(agent, [
-      internal({
-        to: Address.parse(addr),
-        value: toNano("0.03"),
-        bounce: true,
-        body,
-      }),
-    ]);
-
-    return {
-      triggered: true,
-      maxClean,
-      contractAddress: addr,
-      message: `Cleanup triggered for up to ${maxClean} agents.`,
-    };
-  },
-});
+import { z } from "zod";
+import { Address, internal, toNano, beginCell } from "@ton/core";
+import { defineAction, sendTransaction } from "@ton-agent-kit/core";
+import { resolveContractAddress } from "../reputation-config";
+import { storeTriggerCleanup } from "../contracts/Reputation_Reputation";
+
+export const triggerCleanupAction = defineAction({
+  name: "trigger_cleanup",
+  description:
+    "Manually trigger cleanup of inactive, ghost, or low-score agents from the reputation contract. " +
+    "Cleanup also runs automatically on every registration and rating. " +
+    "Conditions: score<20 with 100+ ratings, inactive 30+ days, ghost (0 ratings after 7 days).",
+  schema: z.object({
+    maxClean: z.coerce.number().optional().describe("Max agents to clean. Default 10, max 50."),
+  }),
+  handler: async (agent, params) => {
+    const addr = resolveContractAddress(undefined, agent.network);
+    if (!addr) {
+      return { triggered: false, message: "No reputation contract configured" };
+    }
+
+    const maxClean = Math.min(params.maxClean || 10, 50);
+    const body = beginCell()
+      .store(storeTriggerCleanup({ $$type: "TriggerCleanup", maxClean: BigInt(maxClean) }))
+      .endCell();
+
+    await sendTransaction(agent, [
+      internal({
+        to: Address.parse(addr),
+        value: toNano("0.05"),
+        bounce: true,
+        body,
+      }),
+    ]);
+
+    return {
+      triggered: true,
+      maxClean,
+      contractAddress: addr,
+      message: `Cleanup triggered for up to ${maxClean} agents.`,
+    };
+  },
+});

package/src/actions/withdraw-reputation-fees.ts

@@ -1,52 +1,52 @@
-import { z } from "zod";
-import { Address, internal, toNano } from "@ton/core";
-import { defineAction, sendTransaction } from "@ton-agent-kit/core";
-import { loadReputationConfig } from "../reputation-config";
-import { buildWithdrawBody } from "../reputation-helpers";
-
-export function createWithdrawReputationFeesAction(contractAddress?: string) {
-  return defineAction({
-    name: "withdraw_reputation_fees",
-    description:
-      "Withdraw accumulated fees from the on-chain Reputation contract. Only the contract owner (deployer) can call this.",
-    schema: z.object({
-      confirm: z.boolean().optional().describe("Set to true to confirm the withdrawal. Defaults to true."),
-    }),
-    handler: async (agent, _params) => {
-      const addr = contractAddress || loadReputationConfig()?.contractAddress;
-      if (!addr) {
-        return {
-          withdrawn: false,
-          message: "No reputation contract deployed. Run deploy_reputation_contract first.",
-        };
-      }
-
-      try {
-        const body = buildWithdrawBody();
-
-        await sendTransaction(agent, [
-          internal({
-            to: Address.parse(addr),
-            value: toNano("0.02"),
-            bounce: true,
-            body,
-          }),
-        ]);
-
-        return {
-          withdrawn: true,
-          contractAddress: addr,
-          message: `Withdrawal sent to reputation contract ${addr.slice(0, 16)}...`,
-        };
-      } catch (err: any) {
-        return {
-          withdrawn: false,
-          error: err.message,
-          message: `Failed to withdraw: ${err.message}`,
-        };
-      }
-    },
-  });
-}
-
-export const withdrawReputationFeesAction = createWithdrawReputationFeesAction();
+import { z } from "zod";
+import { Address, internal, toNano } from "@ton/core";
+import { defineAction, sendTransaction } from "@ton-agent-kit/core";
+import { loadReputationConfig } from "../reputation-config";
+import { buildWithdrawBody } from "../reputation-helpers";
+
+export function createWithdrawReputationFeesAction(contractAddress?: string) {
+  return defineAction({
+    name: "withdraw_reputation_fees",
+    description:
+      "Withdraw accumulated fees from the on-chain Reputation contract. Only the contract owner (deployer) can call this.",
+    schema: z.object({
+      confirm: z.boolean().optional().describe("Set to true to confirm the withdrawal. Defaults to true."),
+    }),
+    handler: async (agent, _params) => {
+      const addr = contractAddress || loadReputationConfig()?.contractAddress;
+      if (!addr) {
+        return {
+          withdrawn: false,
+          message: "No reputation contract deployed. Run deploy_reputation_contract first.",
+        };
+      }
+
+      try {
+        const body = buildWithdrawBody();
+
+        await sendTransaction(agent, [
+          internal({
+            to: Address.parse(addr),
+            value: toNano("0.05"),
+            bounce: true,
+            body,
+          }),
+        ]);
+
+        return {
+          withdrawn: true,
+          contractAddress: addr,
+          message: `Withdrawal sent to reputation contract ${addr.slice(0, 16)}...`,
+        };
+      } catch (err: any) {
+        return {
+          withdrawn: false,
+          error: err.message,
+          message: `Failed to withdraw: ${err.message}`,
+        };
+      }
+    },
+  });
+}
+
+export const withdrawReputationFeesAction = createWithdrawReputationFeesAction();

package/src/reputation-config.ts

@@ -19,7 +19,7 @@ export interface ReputationConfig {
  * These are used when no factory parameter or local config is found.
  */
 export const DEFAULT_REPUTATION_CONTRACTS: Record<string, string> = {
-  testnet: "0:5352445990487167d19102e3d1ed2715d69263972ef014bdc1a7561230e2087c",
+  testnet: "0:f141dbcefae9e3f6ee4b08748fc6a0cf1e279ed8ff9e8a20ca44df82be49f839",
 };
 
 /**