@hashgraphonline/standards-agent-kit 0.2.0 → 0.2.101

package/dist/umd/tools/hcs10/RegisterAgentTool.d.ts

@@ -1,12 +1,13 @@
 import { AIAgentCapability } from '@hashgraphonline/standards-sdk';
 import { z } from 'zod';
 import { BaseServiceBuilder } from 'hedera-agent-kit';
+import { CallbackManagerForToolRun } from '@langchain/core/callbacks/manager';
 import { BaseHCS10TransactionTool } from './base-hcs10-tools';
 import { HCS10TransactionToolParams } from './hcs10-tool-params';
 declare const RegisterAgentZodSchema: z.ZodObject<{
     name: z.ZodString;
     description: z.ZodOptional<z.ZodString>;
-    alias: z.ZodOptional<z.ZodString>;
+    alias: z.ZodEffects<z.ZodOptional<z.ZodString>, string, string | undefined>;
     type: z.ZodOptional<z.ZodEnum<["autonomous", "manual"]>>;
     model: z.ZodOptional<z.ZodString>;
     capabilities: z.ZodOptional<z.ZodArray<z.ZodNativeEnum<typeof AIAgentCapability>, "many">>;
@@ -51,6 +52,7 @@ declare const RegisterAgentZodSchema: z.ZodObject<{
     }>>;
 }, "strip", z.ZodTypeAny, {
     name: string;
+    alias: string;
     tokenFees?: {
         amount: number;
         tokenId: string;
@@ -58,7 +60,6 @@ declare const RegisterAgentZodSchema: z.ZodObject<{
     exemptAccountIds?: string[] | undefined;
     type?: "autonomous" | "manual" | undefined;
     description?: string | undefined;
-    alias?: string | undefined;
     model?: string | undefined;
     capabilities?: AIAgentCapability[] | undefined;
     creator?: string | undefined;
@@ -112,7 +113,7 @@ export declare class RegisterAgentTool extends BaseHCS10TransactionTool<typeof R
     specificInputSchema: z.ZodObject<{
         name: z.ZodString;
         description: z.ZodOptional<z.ZodString>;
-        alias: z.ZodOptional<z.ZodString>;
+        alias: z.ZodEffects<z.ZodOptional<z.ZodString>, string, string | undefined>;
         type: z.ZodOptional<z.ZodEnum<["autonomous", "manual"]>>;
         model: z.ZodOptional<z.ZodString>;
         capabilities: z.ZodOptional<z.ZodArray<z.ZodNativeEnum<typeof AIAgentCapability>, "many">>;
@@ -157,6 +158,7 @@ export declare class RegisterAgentTool extends BaseHCS10TransactionTool<typeof R
         }>>;
     }, "strip", z.ZodTypeAny, {
         name: string;
+        alias: string;
         tokenFees?: {
             amount: number;
             tokenId: string;
@@ -164,7 +166,6 @@ export declare class RegisterAgentTool extends BaseHCS10TransactionTool<typeof R
         exemptAccountIds?: string[] | undefined;
         type?: "autonomous" | "manual" | undefined;
         description?: string | undefined;
-        alias?: string | undefined;
         model?: string | undefined;
         capabilities?: AIAgentCapability[] | undefined;
         creator?: string | undefined;
@@ -212,7 +213,16 @@ export declare class RegisterAgentTool extends BaseHCS10TransactionTool<typeof R
             prefix?: string | undefined;
         } | undefined;
     }>;
+    private specificArgs;
     constructor(params: HCS10TransactionToolParams);
     protected callBuilderMethod(builder: BaseServiceBuilder, specificArgs: z.infer<typeof RegisterAgentZodSchema>): Promise<void>;
+    /**
+     * Override _call to intercept the result and save agent to state if needed
+     */
+    protected _call(args: z.infer<ReturnType<this['schema']>>, runManager?: CallbackManagerForToolRun): Promise<string>;
+    /**
+     * Extract agent data from registration result and save to state
+     */
+    private _handleRegistrationResult;
 }
 export {};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hashgraphonline/standards-agent-kit",
-  "version": "0.2.0",
+  "version": "0.2.101",
   "description": "A modular SDK for building on-chain autonomous agents using Hashgraph Online Standards, including HCS-10 for agent discovery and communication.",
   "type": "module",
   "main": "./dist/cjs/standards-agent-kit.cjs",
@@ -63,7 +63,7 @@
   },
   "dependencies": {
     "@hashgraph/sdk": "^2.67.0",
-    "@hashgraphonline/standards-sdk": "^0.0.148",
+    "@hashgraphonline/standards-sdk": "^0.0.149",
     "@langchain/community": "^0.3.48",
     "@langchain/core": "^0.3.62",
     "@langchain/openai": "^0.5.18",
@@ -79,8 +79,8 @@
     "zod": "^3.25.28"
   },
   "devDependencies": {
-    "@hashgraphonline/standards-agent-kit": "file:.yalc/@hashgraphonline/standards-agent-kit",
-    "@hashgraphonline/standards-agent-plugin": "^0.0.1",
+    "@hashgraphonline/standards-agent-kit": "^0.2",
+    "@hashgraphonline/standards-agent-plugin": "^0.0.3",
     "@rollup/plugin-commonjs": "^28.0.6",
     "@swc/core": "^1.11.21",
     "@swc/jest": "^0.2.36",

package/src/builders/hcs10/hcs10-builder.ts

@@ -202,6 +202,13 @@ export class HCS10Builder extends BaseServiceBuilder {
     return this.network;
   }
 
+  /**
+   * Get state manager instance
+   */
+  public getStateManager(): IStateManager | undefined {
+    return this.stateManager;
+  }
+
   /**
    * Get account and signer information
    */
@@ -392,17 +399,19 @@ export class HCS10Builder extends BaseServiceBuilder {
     try {
       const result = await this.standardClient.getMessages(topicId);
 
-      const mappedMessages: HCSMessageWithTimestamp[] = result.messages.map((sdkMessage) => {
-        const timestamp = sdkMessage?.created?.getTime() || 0;
-
-        return {
-          ...sdkMessage,
-          timestamp: timestamp,
-          data: sdkMessage.data || '',
-          sequence_number: sdkMessage.sequence_number,
-          p: 'hcs-10' as const,
-        } as HCSMessageWithTimestamp;
-      });
+      const mappedMessages: HCSMessageWithTimestamp[] = result.messages.map(
+        (sdkMessage) => {
+          const timestamp = sdkMessage?.created?.getTime() || 0;
+
+          return {
+            ...sdkMessage,
+            timestamp: timestamp,
+            data: sdkMessage.data || '',
+            sequence_number: sdkMessage.sequence_number,
+            p: 'hcs-10' as const,
+          } as HCSMessageWithTimestamp;
+        }
+      );
       mappedMessages.sort(
         (a: { timestamp: number }, b: { timestamp: number }) =>
           a.timestamp - b.timestamp
@@ -432,7 +441,9 @@ export class HCS10Builder extends BaseServiceBuilder {
       );
     }
 
-    return this.standardClient.getMessageStream(topicId) as Promise<{ messages: HCSMessage[] }>;
+    return this.standardClient.getMessageStream(topicId) as Promise<{
+      messages: HCSMessage[];
+    }>;
   }
 
   /**
@@ -630,7 +641,8 @@ export class HCS10Builder extends BaseServiceBuilder {
     }
 
     try {
-      let profilePictureData: { buffer: Buffer; filename: string } | null = null;
+      let profilePictureData: { buffer: Buffer; filename: string } | null =
+        null;
       if (params.profilePicture) {
         profilePictureData = await this.loadProfilePicture(
           params.profilePicture
@@ -1509,7 +1521,10 @@ export class HCS10Builder extends BaseServiceBuilder {
       const messages = await this.getMessages(inboundTopicId);
 
       const unapprovedRequests = messages.messages
-        .filter((msg): msg is HCSMessageWithTimestamp & { op: string } => msg.op === 'connection_request')
+        .filter(
+          (msg): msg is HCSMessageWithTimestamp & { op: string } =>
+            msg.op === 'connection_request'
+        )
         .map((msg) => ({
           requestId: msg.sequence_number,
           fromAccountId: msg.operator_id?.split('@')[1] || 'unknown',
@@ -1517,7 +1532,10 @@ export class HCS10Builder extends BaseServiceBuilder {
           memo: msg.m || '',
           data: msg.data,
         }))
-        .filter((req): req is typeof req & { fromAccountId: string } => req.fromAccountId !== 'unknown');
+        .filter(
+          (req): req is typeof req & { fromAccountId: string } =>
+            req.fromAccountId !== 'unknown'
+        );
 
       this.executeResult = {
         success: true,
@@ -2053,7 +2071,10 @@ export class HCS10Builder extends BaseServiceBuilder {
 
       if (profile.social && Object.keys(profile.social).length > 0) {
         profileDetails += `Social: ${Object.entries(profile.social)
-          .map(([platform, handle]: [string, unknown]): string => `${platform}: ${handle}`)
+          .map(
+            ([platform, handle]: [string, unknown]): string =>
+              `${platform}: ${handle}`
+          )
           .join(', ')}\n`;
       }
 

package/src/tools/hcs10/RegisterAgentTool.ts

@@ -1,9 +1,14 @@
 import { AIAgentCapability } from '@hashgraphonline/standards-sdk';
 import { z } from 'zod';
 import { BaseServiceBuilder } from 'hedera-agent-kit';
-import { HCS10Builder, RegisterAgentParams } from '../../builders/hcs10/hcs10-builder';
+import { CallbackManagerForToolRun } from '@langchain/core/callbacks/manager';
+import {
+  HCS10Builder,
+  RegisterAgentParams,
+} from '../../builders/hcs10/hcs10-builder';
 import { BaseHCS10TransactionTool } from './base-hcs10-tools';
 import { HCS10TransactionToolParams } from './hcs10-tool-params';
+import { RegisteredAgent } from '../../state/state-types';
 
 const RegisterAgentZodSchema = z.object({
   name: z
@@ -19,7 +24,17 @@ const RegisterAgentZodSchema = z.object({
   alias: z
     .string()
     .optional()
-    .describe('Optional custom username/alias for the agent'),
+    .transform((val) => {
+      if (!val || val.toLowerCase().includes('random')) {
+        const timestamp = Date.now().toString(36);
+        const randomChars = Math.random().toString(36);
+        return `bot${timestamp}${randomChars}`;
+      }
+      return val;
+    })
+    .describe(
+      'Optional custom username/alias for the agent. Use "random" to generate a unique alias'
+    ),
   type: z
     .enum(['autonomous', 'manual'])
     .optional()
@@ -124,8 +139,9 @@ export class RegisterAgentTool extends BaseHCS10TransactionTool<
 > {
   name = 'register_agent';
   description =
-    'Creates and registers the AI agent on the Hedera network. Returns JSON string with agent details (accountId, privateKey, topics) on success. Note: This tool requires multiple transactions and cannot be used in returnBytes mode.';
+    'Creates and registers the AI agent on the Hedera network. Returns JSON string with agent details (accountId, privateKey, topics) on success. Note: This tool requires multiple transactions and cannot be used in returnBytes mode. If alias is set to "random" or contains "random", a unique alias will be generated.';
   specificInputSchema = RegisterAgentZodSchema;
+  private specificArgs: z.infer<typeof RegisterAgentZodSchema> | undefined;
 
   constructor(params: HCS10TransactionToolParams) {
     super(params);
@@ -138,7 +154,7 @@ export class RegisterAgentTool extends BaseHCS10TransactionTool<
     specificArgs: z.infer<typeof RegisterAgentZodSchema>
   ): Promise<void> {
     const hcs10Builder = builder as HCS10Builder;
-
+    this.specificArgs = specificArgs;
     const params: RegisterAgentParams = {
       name: specificArgs.name,
     };
@@ -148,6 +164,9 @@ export class RegisterAgentTool extends BaseHCS10TransactionTool<
     }
     if (specificArgs.alias !== undefined) {
       params.alias = specificArgs.alias;
+    } else {
+      const randomSuffix = Date.now().toString(36);
+      params.alias = `${specificArgs.name}${randomSuffix}`;
     }
     if (specificArgs.type !== undefined) {
       params.type = specificArgs.type;
@@ -210,4 +229,91 @@ export class RegisterAgentTool extends BaseHCS10TransactionTool<
 
     await hcs10Builder.registerAgent(params);
   }
-}
+
+  /**
+   * Override _call to intercept the result and save agent to state if needed
+   */
+  protected override async _call(
+    args: z.infer<ReturnType<this['schema']>>,
+    runManager?: CallbackManagerForToolRun
+  ): Promise<string> {
+    const result = await super._call(args, runManager);
+
+    const shouldSetAsCurrent = this.specificArgs?.setAsCurrent !== false;
+
+    if (this.specificArgs && shouldSetAsCurrent) {
+      try {
+        const parsed = JSON.parse(result);
+        if (parsed.rawResult) {
+          this._handleRegistrationResult(parsed.rawResult);
+        } else if (parsed.state || parsed.accountId || parsed.metadata) {
+          this._handleRegistrationResult(parsed);
+        }
+      } catch (e) {}
+    }
+
+    return result;
+  }
+
+  /**
+   * Extract agent data from registration result and save to state
+   */
+  private _handleRegistrationResult(rawResult: any): void {
+    let accountId = rawResult.accountId || rawResult.metadata?.accountId;
+
+    if (!accountId && rawResult.state?.createdResources) {
+      const accountResource = rawResult.state.createdResources.find(
+        (r: string) => r.startsWith('account:')
+      );
+      if (accountResource) {
+        accountId = accountResource.split(':')[1];
+      }
+    }
+
+    const inboundTopicId =
+      rawResult.inboundTopicId ||
+      rawResult.metadata?.inboundTopicId ||
+      rawResult.state?.inboundTopicId;
+
+    const outboundTopicId =
+      rawResult.outboundTopicId ||
+      rawResult.metadata?.outboundTopicId ||
+      rawResult.state?.outboundTopicId;
+
+    const profileTopicId =
+      rawResult.profileTopicId ||
+      rawResult.metadata?.profileTopicId ||
+      rawResult.state?.profileTopicId;
+
+    const privateKey = rawResult.privateKey || rawResult.metadata?.privateKey;
+
+    if (accountId && inboundTopicId && outboundTopicId && this.specificArgs) {
+      const registeredAgent: RegisteredAgent = {
+        name: this.specificArgs.name,
+        accountId,
+        inboundTopicId,
+        outboundTopicId,
+        profileTopicId,
+        privateKey,
+      };
+
+      const hcs10Builder = this.getServiceBuilder() as HCS10Builder;
+      const stateManager = hcs10Builder.getStateManager();
+      if (stateManager) {
+        stateManager.setCurrentAgent(registeredAgent);
+
+        if (stateManager.persistAgentData) {
+          const prefix =
+            this.specificArgs.persistence?.prefix ||
+            this.specificArgs.name.toUpperCase().replace(/[^A-Z0-9]/g, '_');
+          stateManager
+            .persistAgentData(registeredAgent, {
+              type: 'env-file',
+              prefix: prefix,
+            })
+            .catch(() => {});
+        }
+      }
+    }
+  }
+}