@ainetwork/adk 0.2.1 → 0.2.2

package/dist/esm/modules/index.js

@@ -1,12 +1,9 @@
 import {
   ModelModule
-} from "../chunk-PHKK5NHK.js";
-import {
-  MemoryModule
-} from "../chunk-XBMWWTSX.js";
+} from "../chunk-VNQCUGC3.js";
 import {
   A2AModule
-} from "../chunk-LJVVSUF7.js";
+} from "../chunk-MJEPJ62P.js";
 import "../chunk-DGH3GB5I.js";
 import {
   BaseAuth
@@ -15,11 +12,14 @@ import {
   MCPModule
 } from "../chunk-RNA32QWV.js";
 import "../chunk-3LLHEUY2.js";
+import {
+  MemoryModule
+} from "../chunk-XBMWWTSX.js";
 import {
   BaseModel
 } from "../chunk-CDFLKADH.js";
-import "../chunk-FI2UQLVC.js";
 import "../chunk-MNAWSBYP.js";
+import "../chunk-FI2UQLVC.js";
 import "../chunk-N7VYM2R5.js";
 export {
   A2AModule,

package/dist/esm/modules/models/model.module.d.ts

@@ -6,6 +6,23 @@ import '@a2a-js/sdk';
 import '@a2a-js/sdk/client';
 import '@modelcontextprotocol/sdk/types.js';
 
+/**
+ * Module for managing multiple AI model implementations.
+ *
+ * Allows registration of multiple models and provides a unified interface
+ * for accessing them. Supports setting a default model for convenience.
+ * All models must extend the BaseModel class for consistent API.
+ *
+ * @example
+ * ```typescript
+ * const modelModule = new ModelModule();
+ * modelModule.addModel("gpt-4", new OpenAIModel("gpt-4"), true);
+ * modelModule.addModel("gemini", new GeminiModel("gemini-pro"));
+ *
+ * const defaultModel = modelModule.getModel();
+ * const specificModel = modelModule.getModel("gemini");
+ * ```
+ */
 declare class ModelModule {
     /** Registry of available models indexed by name */
     private models;

package/dist/esm/modules/models/model.module.js

@@ -1,6 +1,6 @@
 import {
   ModelModule
-} from "../../chunk-PHKK5NHK.js";
+} from "../../chunk-VNQCUGC3.js";
 export {
   ModelModule
 };

package/dist/esm/routes/a2a.routes.js

@@ -1,11 +1,11 @@
 import {
   createA2ARouter
-} from "../chunk-2L2KJNAM.js";
-import "../chunk-2RW77MJB.js";
-import "../chunk-CWFIK2EK.js";
-import "../chunk-FI2UQLVC.js";
-import "../chunk-MNAWSBYP.js";
+} from "../chunk-EUUW7J3S.js";
+import "../chunk-7IYLOABV.js";
 import "../chunk-HAE333YO.js";
+import "../chunk-MNAWSBYP.js";
+import "../chunk-FI2UQLVC.js";
+import "../chunk-PUUZRUE6.js";
 import "../chunk-N7VYM2R5.js";
 export {
   createA2ARouter

package/dist/esm/routes/api/threads.routes.js

@@ -1,8 +1,8 @@
 import {
   createThreadApiRouter
-} from "../../chunk-SYCFBEUA.js";
-import "../../chunk-WLT2B2SZ.js";
-import "../../chunk-FL3JRQM4.js";
+} from "../../chunk-QHWDMU43.js";
+import "../../chunk-RL6SRSSK.js";
+import "../../chunk-ZBUAT73P.js";
 export {
   createThreadApiRouter
 };

package/dist/esm/routes/api.routes.js

@@ -1,11 +1,11 @@
 import {
   createApiRouter
-} from "../chunk-VDXWZ4NK.js";
+} from "../chunk-TV2GJAEH.js";
 import "../chunk-AMRKK46V.js";
-import "../chunk-SYCFBEUA.js";
+import "../chunk-QHWDMU43.js";
 import "../chunk-UA44VKFE.js";
-import "../chunk-WLT2B2SZ.js";
-import "../chunk-FL3JRQM4.js";
+import "../chunk-RL6SRSSK.js";
+import "../chunk-ZBUAT73P.js";
 export {
   createApiRouter
 };

package/dist/esm/routes/index.js

@@ -1,30 +1,30 @@
 import "../chunk-XYCWK2ZM.js";
-import {
-  createQueryRouter
-} from "../chunk-EIHZ4S5V.js";
-import "../chunk-ILKEX4ZV.js";
-import "../chunk-7D4SKPIS.js";
 import {
   createA2ARouter
-} from "../chunk-2L2KJNAM.js";
-import "../chunk-2RW77MJB.js";
-import "../chunk-CWFIK2EK.js";
-import "../chunk-FI2UQLVC.js";
-import "../chunk-MNAWSBYP.js";
-import "../chunk-HAE333YO.js";
-import "../chunk-N7VYM2R5.js";
+} from "../chunk-EUUW7J3S.js";
 import {
   createApiRouter
-} from "../chunk-VDXWZ4NK.js";
+} from "../chunk-TV2GJAEH.js";
 import {
   createModelApiRouter
 } from "../chunk-AMRKK46V.js";
 import {
   createThreadApiRouter
-} from "../chunk-SYCFBEUA.js";
+} from "../chunk-QHWDMU43.js";
+import {
+  createQueryRouter
+} from "../chunk-AJRQEL5N.js";
+import "../chunk-GDA6Z5AL.js";
+import "../chunk-7IYLOABV.js";
 import "../chunk-UA44VKFE.js";
-import "../chunk-WLT2B2SZ.js";
-import "../chunk-FL3JRQM4.js";
+import "../chunk-RL6SRSSK.js";
+import "../chunk-HAE333YO.js";
+import "../chunk-4C2TLCWI.js";
+import "../chunk-ZBUAT73P.js";
+import "../chunk-MNAWSBYP.js";
+import "../chunk-FI2UQLVC.js";
+import "../chunk-PUUZRUE6.js";
+import "../chunk-N7VYM2R5.js";
 export {
   createA2ARouter,
   createApiRouter,

package/dist/esm/routes/query.routes.js

@@ -1,13 +1,13 @@
 import {
   createQueryRouter
-} from "../chunk-EIHZ4S5V.js";
-import "../chunk-ILKEX4ZV.js";
-import "../chunk-7D4SKPIS.js";
-import "../chunk-CWFIK2EK.js";
-import "../chunk-FI2UQLVC.js";
+} from "../chunk-AJRQEL5N.js";
+import "../chunk-GDA6Z5AL.js";
+import "../chunk-7IYLOABV.js";
+import "../chunk-4C2TLCWI.js";
+import "../chunk-ZBUAT73P.js";
 import "../chunk-MNAWSBYP.js";
+import "../chunk-FI2UQLVC.js";
 import "../chunk-N7VYM2R5.js";
-import "../chunk-FL3JRQM4.js";
 export {
   createQueryRouter
 };

package/dist/esm/services/a2a.service.d.ts

@@ -1,5 +1,5 @@
 import { AgentExecutor, ExecutionEventBus, RequestContext } from '@a2a-js/sdk/server';
-import { QueryService } from './query.service.js';
+import { QueryStreamService } from './query-stream.service.js';
 import '../modules/a2a/a2a.module.js';
 import '@a2a-js/sdk';
 import '../modules/a2a/a2a.tool.js';
@@ -15,9 +15,9 @@ import '../modules/mcp/mcp.tool.js';
 import '../types/memory.js';
 import '../modules/memory/memory.module.js';
 import '../modules/memory/base.memory.js';
+import '../types/stream.js';
 import '../modules/models/model.module.js';
 import '../modules/models/base.model.js';
-import '../types/stream.js';
 import '../types/agent.js';
 import 'http-status-codes';
 
@@ -26,9 +26,9 @@ import 'http-status-codes';
  * This service is responsible for the core business logic of executing an A2A task.
  */
 declare class A2AService implements AgentExecutor {
-    private queryService;
+    private queryStreamService;
     private canceledTasks;
-    constructor(queryService: QueryService);
+    constructor(queryStreamService: QueryStreamService);
     cancelTask: (taskId: string, _eventBus: ExecutionEventBus) => Promise<void>;
     private createTaskStatusUpdateEvent;
     execute(requestContext: RequestContext, eventBus: ExecutionEventBus): Promise<void>;

package/dist/esm/services/a2a.service.js

@@ -1,6 +1,6 @@
 import {
   A2AService
-} from "../chunk-2RW77MJB.js";
+} from "../chunk-PUUZRUE6.js";
 import "../chunk-N7VYM2R5.js";
 export {
   A2AService

package/dist/esm/services/index.js

@@ -1,14 +1,14 @@
+import {
+  QueryService
+} from "../chunk-GDA6Z5AL.js";
 import {
   QueryStreamService
-} from "../chunk-ILKEX4ZV.js";
+} from "../chunk-7IYLOABV.js";
+import "../chunk-MNAWSBYP.js";
+import "../chunk-FI2UQLVC.js";
 import {
   A2AService
-} from "../chunk-2RW77MJB.js";
-import {
-  QueryService
-} from "../chunk-CWFIK2EK.js";
-import "../chunk-FI2UQLVC.js";
-import "../chunk-MNAWSBYP.js";
+} from "../chunk-PUUZRUE6.js";
 import "../chunk-N7VYM2R5.js";
 export {
   A2AService,

package/dist/esm/services/query-stream.service.d.ts

@@ -42,18 +42,20 @@ declare class QueryStreamService {
      */
     private intentTriggering;
     /**
-     * Fulfills the detected intent by generating a response.
+     * Fulfills the detected intent by generating a streaming response.
      *
      * Manages the complete inference loop including:
      * - Loading prompts and conversation history
      * - Collecting available tools from modules
      * - Executing model inference with tool support
      * - Processing tool calls iteratively until completion
+     * - Streaming results as Server-Sent Events
      *
      * @param query - The user's input query
      * @param threadId - Thread identifier for context
      * @param thread - Previous conversation history
-     * @returns Object containing process steps and final response
+     * @param intent - Optional detected intent with custom prompt
+     * @returns AsyncGenerator yielding StreamEvent objects
      */
     intentFulfilling(query: string, threadId: string, thread?: ThreadObject, intent?: Intent): AsyncGenerator<StreamEvent>;
     /**
@@ -64,19 +66,20 @@ declare class QueryStreamService {
      */
     private generateTitle;
     /**
-     * Main entry point for processing user queries.
+     * Main entry point for processing streaming user queries.
      *
      * Handles the complete query lifecycle:
-     * 1. Loads thread from memory
+     * 1. Loads or creates thread from memory
      * 2. Detects intent from the query
-     * 3. Fulfills the intent with AI response
-     * 4. Updates conversation history
+     * 3. Fulfills the intent with streaming AI response
+     * 4. Updates conversation history in real-time
      *
-     * @param type - The type of thread (e.g., chat, workflow)
-     * @param userId - The user's unique identifier
-     * @param threadId - Unique thread identifier
+     * @param threadMetadata - Metadata containing type, userId, and optional threadId
+     * @param threadMetadata.type - The type of thread (e.g., chat, workflow)
+     * @param threadMetadata.userId - The user's unique identifier
+     * @param threadMetadata.threadId - Optional thread identifier
      * @param query - The user's input query
-     * @returns Object containing the AI-generated response
+     * @returns AsyncGenerator yielding StreamEvent objects for SSE
      */
     handleQueryStream(threadMetadata: {
         type: ThreadType;

package/dist/esm/services/query-stream.service.js

@@ -1,8 +1,8 @@
 import {
   QueryStreamService
-} from "../chunk-ILKEX4ZV.js";
-import "../chunk-FI2UQLVC.js";
+} from "../chunk-7IYLOABV.js";
 import "../chunk-MNAWSBYP.js";
+import "../chunk-FI2UQLVC.js";
 import "../chunk-N7VYM2R5.js";
 export {
   QueryStreamService

package/dist/esm/services/query.service.d.ts

@@ -53,6 +53,7 @@ declare class QueryService {
      * @param query - The user's input query
      * @param threadId - Thread identifier for context
      * @param thread - Previous conversation history
+     * @param intent - Optional detected intent with custom prompt
      * @returns Object containing process steps and final response
      */
     private intentFulfilling;
@@ -67,15 +68,17 @@ declare class QueryService {
      * Main entry point for processing user queries.
      *
      * Handles the complete query lifecycle:
-     * 1. Loads thread history from memory
+     * 1. Loads or creates thread from memory
      * 2. Detects intent from the query
      * 3. Fulfills the intent with AI response
      * 4. Updates conversation history
      *
-     * @param type - The type of thread (e.g., chat, workflow)
-     * @param userId - The user's unique identifier
-     * @param threadId - Unique thread identifier
+     * @param threadMetadata - Metadata containing type, userId, and optional threadId
+     * @param threadMetadata.type - The type of thread (e.g., chat, workflow)
+     * @param threadMetadata.userId - The user's unique identifier
+     * @param threadMetadata.threadId - Optional thread identifier
      * @param query - The user's input query
+     * @returns Promise resolving to object with content property containing the response
      */
     handleQuery(threadMetadata: {
         type: ThreadType;

package/dist/esm/services/query.service.js

@@ -1,8 +1,8 @@
 import {
   QueryService
-} from "../chunk-CWFIK2EK.js";
-import "../chunk-FI2UQLVC.js";
+} from "../chunk-GDA6Z5AL.js";
 import "../chunk-MNAWSBYP.js";
+import "../chunk-FI2UQLVC.js";
 import "../chunk-N7VYM2R5.js";
 export {
   QueryService

package/dist/esm/types/agent.d.ts

@@ -36,8 +36,6 @@ type AinAgentManifest = {
     name: string;
     /** Human-readable description of the agent's purpose and capabilities */
     description: string;
-    /** Semantic version of the agent (e.g., "1.0.0") */
-    version: string;
     /** Optional public URL for A2A protocol discovery and communication */
     url?: string;
     /** Optional custom prompts to configure agent behavior */

package/dist/esm/types/agent.js

@@ -1,6 +1,6 @@
 import {
   AinHttpError
-} from "../chunk-FL3JRQM4.js";
+} from "../chunk-ZBUAT73P.js";
 export {
   AinHttpError
 };

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@ainetwork/adk",
-	"version": "0.2.1",
+	"version": "0.2.2",
 	"description": "AI Network Agent Development Kit",
 	"repository": "git@github.com:ainetwork-ai/ain-adk.git",
 	"author": "AI Network (https://ainetwork.ai)",
@@ -46,7 +46,8 @@
 		"biome": "npx @biomejs/biome check",
 		"biome:write": "npx @biomejs/biome check --write",
 		"test": "jest --passWithNoTests",
-		"prepare": "husky"
+		"prepare": "husky",
+		"release": "release-it"
 	},
 	"keywords": [],
 	"license": "MIT",
@@ -61,6 +62,7 @@
 		"eslint": "^9.29.0",
 		"husky": "^9.1.7",
 		"jest": "^30.0.0",
+		"release-it": "^19.0.4",
 		"ts-jest": "^29.4.0",
 		"ts-node": "^10.9.2",
 		"tsup": "^8.5.0",
@@ -69,13 +71,16 @@
 		"typescript-eslint": "^8.34.1"
 	},
 	"dependencies": {
-		"@a2a-js/sdk": "^0.2.4",
-		"@modelcontextprotocol/sdk": "^1.12.3",
+		"@a2a-js/sdk": "^0.3.2",
+		"@modelcontextprotocol/sdk": "^1.17.3",
 		"cors": "^2.8.5",
 		"dotenv": "^16.5.0",
 		"express": "^5.1.0",
 		"http-status-codes": "^2.3.0",
 		"uuid": "^11.1.0",
 		"winston": "^3.17.0"
+	},
+	"publishConfig": {
+		"registry": "https://registry.npmjs.org"
 	}
 }

package/dist/cjs/chunk-2C33PZTV.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-2C33PZTV.cjs","../../src/routes/a2a.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACVA,4CAAkC;AAClC,kCAAuB;AAchB,IAAM,gBAAA,EAAkB,CAAC,KAAA,EAAA,GAA4B;AAC3D,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAGtB,EAAA,MAAM,UAAA,EAAY,IAAI,8BAAA,CAAkB,CAAA;AACxC,EAAA,MAAM,aAAA,EAAe,IAAI,mCAAA;AAAA,IACxB,KAAA,CAAM,WAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,YAAA;AAAA,IACN,KAAA,CAAM,QAAA,CAAS;AAAA,EAChB,CAAA;AACA,EAAA,MAAM,WAAA,EAAa,IAAI,iCAAA,CAAW,YAAY,CAAA;AAC9C,EAAA,MAAM,cAAA,EAAgB,IAAI,oCAAA;AAAA,IACzB,UAAA;AAAA,IACA,SAAA;AAAA,IACA,KAAA,CAAM;AAAA,EACP,CAAA;AAGA,EAAA,MAAA,CAAO,IAAA,CAAK,GAAA,EAAK,aAAA,CAAc,gBAAgB,CAAA;AAE/C,EAAA,OAAO,MAAA;AACR,CAAA;ADNA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-2C33PZTV.cjs","sourcesContent":[null,"import { InMemoryTaskStore } from \"@a2a-js/sdk/server\";\nimport { Router } from \"express\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryService } from \"@/services/query.service.js\";\nimport { A2AController } from \"../controllers/a2a.controller.js\";\nimport { A2AService } from \"../services/a2a.service.js\";\n\n/**\n * Creates and configures the A2A router.\n * This function is a \"composition root\" for the A2A feature,\n * creating and injecting all necessary dependencies.\n * @param intentAnalyzer The core intent analyzer.\n * @param agentCard The agent's card.\n * @returns An Express Router instance.\n */\nexport const createA2ARouter = (agent: AINAgent): Router => {\n\tconst router = Router();\n\n\t// 1. Create dependencies for the A2A feature\n\tconst taskStore = new InMemoryTaskStore();\n\tconst queryService = new QueryService(\n\t\tagent.modelModule,\n\t\tagent.a2aModule,\n\t\tagent.mcpModule,\n\t\tagent.memoryModule,\n\t\tagent.manifest.prompts,\n\t);\n\tconst a2aService = new A2AService(queryService);\n\tconst a2aController = new A2AController(\n\t\ta2aService,\n\t\ttaskStore,\n\t\tagent.generateAgentCard,\n\t);\n\n\t// 2. Define the route\n\trouter.post(\"/\", a2aController.handleA2ARequest);\n\n\treturn router;\n};\n"]}

package/dist/cjs/chunk-344VLDMS.cjs

@@ -1,43 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});
-
-var _chunkZZK4ACTVcjs = require('./chunk-ZZK4ACTV.cjs');
-
-
-var _chunk3J3ORQOGcjs = require('./chunk-3J3ORQOG.cjs');
-
-
-var _chunkKE7SL3H6cjs = require('./chunk-KE7SL3H6.cjs');
-
-// src/routes/query.routes.ts
-var _express = require('express');
-var createQueryRouter = (agent, allowStream = false) => {
-  const router = _express.Router.call(void 0, );
-  const queryService = new (0, _chunkKE7SL3H6cjs.QueryService)(
-    agent.modelModule,
-    agent.a2aModule,
-    agent.mcpModule,
-    agent.memoryModule,
-    agent.manifest.prompts
-  );
-  let queryStreamService;
-  if (allowStream) {
-    queryStreamService = new (0, _chunkZZK4ACTVcjs.QueryStreamService)(
-      agent.modelModule,
-      agent.a2aModule,
-      agent.mcpModule,
-      agent.memoryModule,
-      agent.manifest.prompts
-    );
-  }
-  const queryController = new (0, _chunk3J3ORQOGcjs.QueryController)(queryService, queryStreamService);
-  router.post("/", queryController.handleQueryRequest);
-  if (allowStream) {
-    router.post("/stream", queryController.handleQueryStreamRequest);
-  }
-  return router;
-};
-
-
-
-exports.createQueryRouter = createQueryRouter;
-//# sourceMappingURL=chunk-344VLDMS.cjs.map

package/dist/cjs/chunk-344VLDMS.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-344VLDMS.cjs","../../src/routes/query.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACVA,kCAAuB;AAMhB,IAAM,kBAAA,EAAoB,CAChC,KAAA,EACA,YAAA,EAAc,KAAA,EAAA,GACF;AACZ,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AAEtB,EAAA,MAAM,aAAA,EAAe,IAAI,mCAAA;AAAA,IACxB,KAAA,CAAM,WAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,SAAA;AAAA,IACN,KAAA,CAAM,YAAA;AAAA,IACN,KAAA,CAAM,QAAA,CAAS;AAAA,EAChB,CAAA;AAEA,EAAA,IAAI,kBAAA;AAEJ,EAAA,GAAA,CAAI,WAAA,EAAa;AAChB,IAAA,mBAAA,EAAqB,IAAI,yCAAA;AAAA,MACxB,KAAA,CAAM,WAAA;AAAA,MACN,KAAA,CAAM,SAAA;AAAA,MACN,KAAA,CAAM,SAAA;AAAA,MACN,KAAA,CAAM,YAAA;AAAA,MACN,KAAA,CAAM,QAAA,CAAS;AAAA,IAChB,CAAA;AAAA,EACD;AAEA,EAAA,MAAM,gBAAA,EAAkB,IAAI,sCAAA,CAAgB,YAAA,EAAc,kBAAkB,CAAA;AAC5E,EAAA,MAAA,CAAO,IAAA,CAAK,GAAA,EAAK,eAAA,CAAgB,kBAAkB,CAAA;AAEnD,EAAA,GAAA,CAAI,WAAA,EAAa;AAChB,IAAA,MAAA,CAAO,IAAA,CAAK,SAAA,EAAW,eAAA,CAAgB,wBAAwB,CAAA;AAAA,EAChE;AAEA,EAAA,OAAO,MAAA;AACR,CAAA;ADFA;AACA;AACE;AACF,8CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-344VLDMS.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport { QueryController } from \"@/controllers/query.controller.js\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryService } from \"@/services/query.service.js\";\nimport { QueryStreamService } from \"@/services/query-stream.service.js\";\n\nexport const createQueryRouter = (\n\tagent: AINAgent,\n\tallowStream = false,\n): Router => {\n\tconst router = Router();\n\n\tconst queryService = new QueryService(\n\t\tagent.modelModule,\n\t\tagent.a2aModule,\n\t\tagent.mcpModule,\n\t\tagent.memoryModule,\n\t\tagent.manifest.prompts,\n\t);\n\n\tlet queryStreamService: QueryStreamService | undefined;\n\n\tif (allowStream) {\n\t\tqueryStreamService = new QueryStreamService(\n\t\t\tagent.modelModule,\n\t\t\tagent.a2aModule,\n\t\t\tagent.mcpModule,\n\t\t\tagent.memoryModule,\n\t\t\tagent.manifest.prompts,\n\t\t);\n\t}\n\n\tconst queryController = new QueryController(queryService, queryStreamService);\n\trouter.post(\"/\", queryController.handleQueryRequest);\n\n\tif (allowStream) {\n\t\trouter.post(\"/stream\", queryController.handleQueryStreamRequest);\n\t}\n\n\treturn router;\n};\n"]}

package/dist/cjs/chunk-4DXS33VP.cjs

@@ -1,43 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});
-
-var _chunkZZK4ACTVcjs = require('./chunk-ZZK4ACTV.cjs');
-
-
-var _chunk3J3ORQOGcjs = require('./chunk-3J3ORQOG.cjs');
-
-
-var _chunk7GWYMGWWcjs = require('./chunk-7GWYMGWW.cjs');
-
-// src/routes/query.routes.ts
-var _express = require('express');
-var createQueryRouter = (agent, allowStream = false) => {
-  const router = _express.Router.call(void 0, );
-  const queryService = new (0, _chunk7GWYMGWWcjs.QueryService)(
-    agent.modelModule,
-    agent.a2aModule,
-    agent.mcpModule,
-    agent.memoryModule,
-    agent.manifest.prompts
-  );
-  let queryStreamService;
-  if (allowStream) {
-    queryStreamService = new (0, _chunkZZK4ACTVcjs.QueryStreamService)(
-      agent.modelModule,
-      agent.a2aModule,
-      agent.mcpModule,
-      agent.memoryModule,
-      agent.manifest.prompts
-    );
-  }
-  const queryController = new (0, _chunk3J3ORQOGcjs.QueryController)(queryService, queryStreamService);
-  router.post("/", queryController.handleQueryRequest);
-  if (allowStream) {
-    router.post("/stream", queryController.handleQueryStreamRequest);
-  }
-  return router;
-};
-
-
-
-exports.createQueryRouter = createQueryRouter;
-//# sourceMappingURL=chunk-4DXS33VP.cjs.map

package/dist/cjs/chunk-55IME4BG.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-55IME4BG.cjs","../../src/services/a2a.service.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA,gCAA2B;AAepB,IAAM,WAAA,YAAN,MAA0C;AAAA,EACxC;AAAA,iBACA,cAAA,kBAA6B,IAAI,GAAA,CAAY,EAAA;AAAA,EAErD,WAAA,CAAY,YAAA,EAA4B;AACvC,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AAAA,EACrB;AAAA,kBAEO,WAAA,EAAa,MAAA,CACnB,MAAA,EACA,SAAA,EAAA,GACmB;AACnB,IAAA,IAAA,CAAK,aAAA,CAAc,GAAA,CAAI,MAAM,CAAA;AAAA,EAC9B,EAAA;AAAA,kBAEQ,4BAAA,EAA8B,CACrC,MAAA,EACA,SAAA,EACA,KAAA,EACA,OAAA,EAAA,GAC2B;AAC3B,IAAA,OAAO;AAAA,MACN,IAAA,EAAM,eAAA;AAAA,MACN,MAAA;AAAA,MACA,SAAA;AAAA,MACA,MAAA,EAAQ;AAAA,QACP,KAAA;AAAA,QACA,OAAA,EAAS,QAAA,EACN;AAAA,UACA,IAAA,EAAM,SAAA;AAAA,UACN,IAAA,EAAM,OAAA;AAAA,UACN,SAAA,EAAW,gCAAA,CAAW;AAAA,UACtB,KAAA,EAAO,CAAC,EAAE,IAAA,EAAM,MAAA,EAAQ,IAAA,EAAM,QAAQ,CAAC,CAAA;AAAA,UACvC,MAAA;AAAA,UACA;AAAA,QACD,EAAA,EACC,KAAA,CAAA;AAAA,QACH,SAAA,EAAA,iBAAW,IAAI,IAAA,CAAK,CAAA,CAAA,CAAE,WAAA,CAAY;AAAA,MACnC,CAAA;AAAA,MACA,KAAA,EAAO,MAAA,IAAU;AAAA,IAClB,CAAA;AAAA,EACD,EAAA;AAAA,EAEA,MAAM,OAAA,CACL,cAAA,EACA,QAAA,EACgB;AAChB,IAAA,MAAM,YAAA,EAAc,cAAA,CAAe,WAAA;AACnC,IAAA,MAAM,EAAE,OAAA,EAAS,IAAA,EAAM,SAAS,EAAA,EAAI,WAAA,CAAY,QAAA;AAKhD,IAAA,MAAM,aAAA,EAAe,cAAA,CAAe,IAAA;AAEpC,IAAA,MAAM,OAAA,kBAAS,YAAA,2BAAc,KAAA,GAAM,gCAAA,CAAW;AAC9C,IAAA,MAAM,UAAA,EACL,WAAA,CAAY,UAAA,mBAAa,YAAA,6BAAc,YAAA,GAAa,gCAAA,CAAW;AAEhE,IAAA,GAAA,CAAI,CAAC,YAAA,EAAc;AAClB,MAAA,MAAM,YAAA,EAAoB;AAAA,QACzB,IAAA,EAAM,MAAA;AAAA,QACN,EAAA,EAAI,MAAA;AAAA,QACJ,SAAA;AAAA,QACA,MAAA,EAAQ;AAAA,UACP,KAAA,EAAO,WAAA;AAAA,UACP,SAAA,EAAA,iBAAW,IAAI,IAAA,CAAK,CAAA,CAAA,CAAE,WAAA,CAAY;AAAA,QACnC,CAAA;AAAA,QACA,OAAA,EAAS,CAAC,WAAW,CAAA;AAAA,QACrB,QAAA,EAAU,WAAA,CAAY,QAAA;AAAA,QACtB,SAAA,EAAW,CAAC;AAAA,MACb,CAAA;AACA,MAAA,QAAA,CAAS,OAAA,CAAQ,WAAW,CAAA;AAAA,IAC7B;AAEA,IAAA,MAAM,oBAAA,EAAsB,IAAA,CAAK,2BAAA;AAAA,MAChC,MAAA;AAAA,MACA,SAAA;AAAA,MACA;AAAA,IACD,CAAA;AACA,IAAA,QAAA,CAAS,OAAA,CAAQ,mBAAmB,CAAA;AAEpC,IAAA,MAAM,QAAA,EAAkB,WAAA,CAAY,KAAA,CAClC,MAAA,CAAO,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,KAAA,IAAS,MAAM,CAAA,CACrC,GAAA,CAAI,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,IAAI,CAAA,CACvB,IAAA,CAAK,IAAI,CAAA;AACX,IAAA,GAAA,CAAI,OAAA,CAAQ,OAAA,IAAW,CAAA,EAAG;AACzB,MAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA,CAAK,CAAA,gCAAA,EAAmC,MAAM,CAAA,CAAA,CAAG,CAAA;AAChE,MAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,2BAAA;AAAA,QAC1B,MAAA;AAAA,QACA,SAAA;AAAA,QACA,QAAA;AAAA,QACA;AAAA,MACD,CAAA;AACA,MAAA,QAAA,CAAS,OAAA,CAAQ,aAAa,CAAA;AAC9B,MAAA,MAAA;AAAA,IACD;AAEA,IAAA,IAAI;AACH,MAAA,MAAM,SAAA,EAAW,MAAM,IAAA,CAAK,YAAA,CAAa,WAAA;AAAA,QACxC,EAAE,MAAA,EAAQ,OAAA,EAAS,IAAA,EAAM,SAAS,CAAA;AAAA,QAClC;AAAA,MACD,CAAA;AAEA,MAAA,GAAA,CAAI,IAAA,CAAK,aAAA,CAAc,GAAA,CAAI,MAAM,CAAA,EAAG;AACnC,QAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA,CAAK,CAAA,KAAA,EAAQ,MAAM,CAAA,cAAA,CAAgB,CAAA;AAClD,QAAA,MAAM,eAAA,EAAiB,IAAA,CAAK,2BAAA;AAAA,UAC3B,MAAA;AAAA,UACA,SAAA;AAAA,UACA;AAAA,QACD,CAAA;AACA,QAAA,QAAA,CAAS,OAAA,CAAQ,cAAc,CAAA;AAC/B,QAAA,MAAA;AAAA,MACD;AAEA,MAAA,MAAM,YAAA,EAAc,IAAA,CAAK,2BAAA;AAAA,QACxB,MAAA;AAAA,QACA,SAAA;AAAA,QACA,WAAA;AAAA,QACA,QAAA,CAAS;AAAA,MACV,CAAA;AACA,MAAA,QAAA,CAAS,OAAA,CAAQ,WAAW,CAAA;AAC5B,MAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA,CAAK,CAAA,KAAA,EAAQ,MAAM,CAAA,wBAAA,CAA0B,CAAA;AAAA,IAC7D,EAAA,MAAA,CAAS,KAAA,EAAY;AACpB,MAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,CAAA,sBAAA,EAAyB,MAAM,CAAA,CAAA,CAAA,EAAK,KAAK,CAAA;AAC9D,MAAA,MAAM,YAAA,EAAc,IAAA,CAAK,2BAAA;AAAA,QACxB,MAAA;AAAA,QACA,SAAA;AAAA,QACA,QAAA;AAAA,QACA,CAAA,aAAA,EAAgB,KAAA,CAAM,OAAO,CAAA;AAAA,MAAA;AAE9B,MAAA;AAA4B,IAAA;AAC7B,EAAA;AAEF;ADxCA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-55IME4BG.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport type { Task, TaskStatusUpdateEvent } from \"@a2a-js/sdk\";\nimport type {\n\tAgentExecutor,\n\tExecutionEventBus,\n\tRequestContext,\n} from \"@a2a-js/sdk/server\";\nimport type { ThreadType } from \"@/types/memory.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport type { QueryService } from \"./query.service.js\";\n\n/**\n * Implements the AgentExecutor interface from the a2a-js-sdk.\n * This service is responsible for the core business logic of executing an A2A task.\n */\nexport class A2AService implements AgentExecutor {\n\tprivate queryService: QueryService;\n\tprivate canceledTasks: Set<string> = new Set<string>();\n\n\tconstructor(queryService: QueryService) {\n\t\tthis.queryService = queryService;\n\t}\n\n\tpublic cancelTask = async (\n\t\ttaskId: string,\n\t\t_eventBus: ExecutionEventBus,\n\t): Promise<void> => {\n\t\tthis.canceledTasks.add(taskId);\n\t};\n\n\tprivate createTaskStatusUpdateEvent = (\n\t\ttaskId: string,\n\t\tcontextId: string,\n\t\tstate: \"working\" | \"failed\" | \"canceled\" | \"completed\",\n\t\tmessage?: string,\n\t): TaskStatusUpdateEvent => {\n\t\treturn {\n\t\t\tkind: \"status-update\",\n\t\t\ttaskId: taskId,\n\t\t\tcontextId: contextId,\n\t\t\tstatus: {\n\t\t\t\tstate: state,\n\t\t\t\tmessage: message\n\t\t\t\t\t? {\n\t\t\t\t\t\t\tkind: \"message\",\n\t\t\t\t\t\t\trole: \"agent\",\n\t\t\t\t\t\t\tmessageId: randomUUID(),\n\t\t\t\t\t\t\tparts: [{ kind: \"text\", text: message }],\n\t\t\t\t\t\t\ttaskId: taskId,\n\t\t\t\t\t\t\tcontextId: contextId,\n\t\t\t\t\t\t}\n\t\t\t\t\t: undefined,\n\t\t\t\ttimestamp: new Date().toISOString(),\n\t\t\t},\n\t\t\tfinal: state !== \"working\",\n\t\t};\n\t};\n\n\tasync execute(\n\t\trequestContext: RequestContext,\n\t\teventBus: ExecutionEventBus,\n\t): Promise<void> {\n\t\tconst userMessage = requestContext.userMessage;\n\t\tconst { agentId, type, threadId } = userMessage.metadata as {\n\t\t\tagentId: string;\n\t\t\ttype: ThreadType;\n\t\t\tthreadId: string;\n\t\t};\n\t\tconst existingTask = requestContext.task;\n\n\t\tconst taskId = existingTask?.id || randomUUID();\n\t\tconst contextId =\n\t\t\tuserMessage.contextId || existingTask?.contextId || randomUUID();\n\n\t\tif (!existingTask) {\n\t\t\tconst initialTask: Task = {\n\t\t\t\tkind: \"task\",\n\t\t\t\tid: taskId,\n\t\t\t\tcontextId: contextId,\n\t\t\t\tstatus: {\n\t\t\t\t\tstate: \"submitted\",\n\t\t\t\t\ttimestamp: new Date().toISOString(),\n\t\t\t\t},\n\t\t\t\thistory: [userMessage],\n\t\t\t\tmetadata: userMessage.metadata,\n\t\t\t\tartifacts: [],\n\t\t\t};\n\t\t\teventBus.publish(initialTask);\n\t\t}\n\n\t\tconst workingStatusUpdate = this.createTaskStatusUpdateEvent(\n\t\t\ttaskId,\n\t\t\tcontextId,\n\t\t\t\"working\",\n\t\t);\n\t\teventBus.publish(workingStatusUpdate);\n\n\t\tconst message: string = userMessage.parts\n\t\t\t.filter((part) => part.kind === \"text\")\n\t\t\t.map((part) => part.text)\n\t\t\t.join(\"\\n\");\n\t\tif (message.length === 0) {\n\t\t\tloggers.server.warn(`Empty message received for task ${taskId}.`);\n\t\t\tconst failureUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\ttaskId,\n\t\t\t\tcontextId,\n\t\t\t\t\"failed\",\n\t\t\t\t\"No message found to process.\",\n\t\t\t);\n\t\t\teventBus.publish(failureUpdate);\n\t\t\treturn;\n\t\t}\n\n\t\ttry {\n\t\t\tconst response = await this.queryService.handleQuery(\n\t\t\t\t{ userId: agentId, type, threadId },\n\t\t\t\tmessage,\n\t\t\t);\n\n\t\t\tif (this.canceledTasks.has(taskId)) {\n\t\t\t\tloggers.server.info(`Task ${taskId} was canceled.`);\n\t\t\t\tconst canceledUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\t\ttaskId,\n\t\t\t\t\tcontextId,\n\t\t\t\t\t\"canceled\",\n\t\t\t\t);\n\t\t\t\teventBus.publish(canceledUpdate);\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tconst finalUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\ttaskId,\n\t\t\t\tcontextId,\n\t\t\t\t\"completed\",\n\t\t\t\tresponse.content,\n\t\t\t);\n\t\t\teventBus.publish(finalUpdate);\n\t\t\tloggers.server.info(`Task ${taskId} completed successfully.`);\n\t\t} catch (error: any) {\n\t\t\tloggers.server.error(`Error processing task ${taskId}:`, error);\n\t\t\tconst errorUpdate = this.createTaskStatusUpdateEvent(\n\t\t\t\ttaskId,\n\t\t\t\tcontextId,\n\t\t\t\t\"failed\",\n\t\t\t\t`Agent error: ${error.message}`,\n\t\t\t);\n\t\t\teventBus.publish(errorUpdate);\n\t\t}\n\t}\n}\n"]}

package/dist/cjs/chunk-6YVTJEMQ.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6YVTJEMQ.cjs","../../src/modules/a2a/a2a.module.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACPA,gCAA2B;AAS3B,4CAA0B;AAsBnB,IAAM,UAAA,YAAN,MAAgB;AAAA;AAAA,iBAEd,eAAA,kBAA8C,IAAI,GAAA,CAAI,EAAA;AAAA;AAAA,kBAEtD,YAAA,kBAAuC,IAAI,GAAA,CAAI,EAAA;AAAA,kBAC/C,QAAA,EAAkB,gCAAA,EAAW;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOrC,MAAa,gBAAA,CAAiB,GAAA,EAA4B;AACzD,IAAA,IAAA,CAAK,cAAA,CAAe,GAAA,CAAI,GAAA,EAAK,IAAI,CAAA;AAAA,EAClC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAa,QAAA,CAAA,EAA+B;AAC3C,IAAA,MAAM,MAAA,EAAmB,CAAC,CAAA;AAC1B,IAAA,IAAA,CAAA,MAAW,IAAA,GAAO,CAAC,GAAG,IAAA,CAAK,cAAA,CAAe,IAAA,CAAK,CAAC,CAAA,EAAG;AAClD,MAAA,MAAM,KAAA,EAAO,IAAA,CAAK,cAAA,CAAe,GAAA,CAAI,GAAG,CAAA;AACxC,MAAA,GAAA,CAAI,CAAC,KAAA,GAAQ,CAAC,IAAA,CAAK,OAAA,EAAS;AAC3B,QAAA,IAAI;AACH,UAAA,MAAM,OAAA,EAAS,IAAI,sBAAA,CAAU,GAAG,CAAA;AAChC,UAAA,MAAM,KAAA,EAAkB,MAAM,MAAA,CAAO,YAAA,CAAa,CAAA;AAClD,UAAA,MAAM,SAAA,EAAW,IAAA,CAAK,IAAA,CAAK,UAAA,CAAW,GAAA,EAAK,GAAG,CAAA;AAC9C,UAAA,MAAM,QAAA,EAAU,IAAI,8BAAA,CAAQ,QAAA,EAAU,MAAA,EAAQ,IAAI,CAAA;AAElD,UAAA,KAAA,CAAM,IAAA,CAAK,OAAO,CAAA;AAAA,QACnB,EAAA,MAAA,CAAS,MAAA,EAAa;AAErB,UAAA,GAAA,CAAI,IAAA,EAAM;AACT,YAAA,IAAA,CAAK,OAAA,CAAQ,CAAA;AAAA,UACd;AAAA,QACD;AAAA,MACD,EAAA,KAAO;AACN,QAAA,KAAA,CAAM,IAAA,CAAK,IAAI,CAAA;AAAA,MAChB;AAAA,IACD;AACA,IAAA,OAAO,KAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,kBAQQ,oBAAA,EAAsB,CAAC,QAAA,EAAA,GAAiC;AAC/D,IAAA,MAAM,WAAA,mBAAa,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,QAAQ,CAAA,UAAK;AAAA,MACpD,MAAA,EAAQ,KAAA,CAAA;AAAA,MACR,SAAA,EAAW,KAAA;AAAA,IACZ,GAAA;AACA,IAAA,GAAA,CAAI,CAAC,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,QAAQ,CAAA,EAAG;AACpC,MAAA,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,QAAA,EAAU,UAAU,CAAA;AAAA,IAC1C;AAEA,IAAA,OAAO,UAAA;AAAA,EACR,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYO,iBAAA,CAAkB,KAAA,EAAe,QAAA,EAA2B;AAClE,IAAA,MAAM,eAAA,EAA0B;AAAA,MAC/B,SAAA,EAAW,gCAAA,CAAW;AAAA,MACtB,IAAA,EAAM,SAAA;AAAA,MACN,IAAA,EAAM,MAAA;AAAA;AAAA,MACN,QAAA,EAAU;AAAA,QACT,OAAA,EAAS,IAAA,CAAK,OAAA;AAAA,QACd,IAAA,EAAA,iBAAA;AAAA,QACA;AAAA,MACD,CAAA;AAAA,MACA,KAAA,EAAO;AAAA,QACN;AAAA,UACC,IAAA,EAAM,MAAA;AAAA,UACN,IAAA,EAAM;AAAA,QACP;AAAA,MACD;AAAA,IACD,CAAA;AAEA,IAAA,MAAM,WAAA,EAAa,IAAA,CAAK,mBAAA,CAAoB,QAAQ,CAAA;AACpD,IAAA,GAAA,CAAI,UAAA,CAAW,MAAA,EAAQ;AACtB,MAAA,cAAA,CAAe,OAAA,EAAS,UAAA,CAAW,MAAA;AAAA,IACpC;AACA,IAAA,GAAA,CAAI,UAAA,CAAW,SAAA,EAAW;AACzB,MAAA,cAAA,CAAe,UAAA,EAAY,UAAA,CAAW,SAAA;AAAA,IACvC;AAEA,IAAA,OAAO,cAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAaA,MAAa,OAAA,CACZ,IAAA,EACA,cAAA,EACA,QAAA,EACkB;AAClB,IAAA,MAAM,UAAA,EAAsB,CAAC,CAAA;AAC7B,IAAA,MAAM,OAAA,EAAS,IAAA,CAAK,MAAA;AACpB,IAAA,MAAM,OAAA,EAA4B;AAAA,MACjC,OAAA,EAAS;AAAA,IACV,CAAA;AACA,IAAA,MAAM,WAAA,EAAa,IAAA,CAAK,mBAAA,CAAoB,QAAQ,CAAA;AAEpD,IAAA,IAAI;AACH,MAAA,MAAM,OAAA,EAAS,MAAA,CAAO,iBAAA,CAAkB,MAAM,CAAA;AAC9C,MAAA,IAAA,MAAA,CAAA,MAAiB,MAAA,GAAS,MAAA,EAAQ;AACjC,QAAA,GAAA,CAAI,KAAA,CAAM,KAAA,IAAS,eAAA,EAAiB;AACnC,UAAA,MAAM,WAAA,EAAa,KAAA;AACnB,UAAA,GAAA,CACC,UAAA,CAAW,MAAA,GACX,UAAA,CAAW,MAAA,CAAO,MAAA,IAAU,gBAAA,EAC3B;AACD,YAAA,UAAA,CAAW,OAAA,EAAS,KAAA,CAAA;AAAA,UACrB;AAEA,UAAA,MAAM,MAAA,kBAAQ,UAAA,mBAAW,MAAA,qBAAO,OAAA,6BAAS,KAAA,qBACvC,MAAA,mBAAO,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,KAAA,IAAS,MAAM,CAAA,qBACrC,GAAA,mBAAI,CAAC,IAAA,EAAA,GAAmB,IAAA,CAAK,IAAI,CAAA,qBACjC,IAAA,mBAAK,IAAI,GAAA;AACX,UAAA,GAAA,CAAI,KAAA,EAAO;AACV,YAAA,SAAA,CAAU,IAAA,CAAK,KAAK,CAAA;AAAA,UACrB;AAAA,QACD,EAAA,KAAA,GAAA,CAAW,KAAA,CAAM,KAAA,IAAS,SAAA,EAAW;AAEpC,UAAA,MAAM,IAAA,EAAM,KAAA;AACZ,UAAA,GAAA,CAAI,GAAA,CAAI,OAAA,GAAU,GAAA,CAAI,OAAA,IAAW,UAAA,CAAW,MAAA,EAAQ;AACnD,YAAA,UAAA,CAAW,OAAA,EAAS,GAAA,CAAI,MAAA;AAAA,UACzB;AACA,UAAA,GAAA,CAAI,GAAA,CAAI,UAAA,GAAa,GAAA,CAAI,UAAA,IAAc,UAAA,CAAW,SAAA,EAAW;AAC5D,YAAA,UAAA,CAAW,UAAA,EAAY,GAAA,CAAI,SAAA;AAAA,UAC5B;AAAA,QACD,EAAA,KAAA,GAAA,CAAW,KAAA,CAAM,KAAA,IAAS,MAAA,EAAQ;AAEjC,UAAA,MAAM,KAAA,EAAO,KAAA;AACb,UAAA,GAAA,CAAI,IAAA,CAAK,GAAA,IAAO,UAAA,CAAW,MAAA,EAAQ;AAClC,YAAA,UAAA,CAAW,OAAA,EAAS,IAAA,CAAK,EAAA;AAAA,UAC1B;AACA,UAAA,GAAA,CAAI,IAAA,CAAK,UAAA,GAAa,IAAA,CAAK,UAAA,IAAc,UAAA,CAAW,SAAA,EAAW;AAC9D,YAAA,UAAA,CAAW,UAAA,EAAY,IAAA,CAAK,SAAA;AAAA,UAC7B;AAAA,QACD,EAAA,KAAO;AACN,UAAA,yBAAA,CAAQ,GAAA,CAAI,IAAA,CAAK,+CAAA,EAAiD;AAAA,YACjE;AAAA,UACD,CAAC,CAAA;AAAA,QACF;AAAA,MACD;AAAA,IACD,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,yBAAA,CAAQ,GAAA,CAAI,KAAA,CAAM,iCAAA,EAAmC,EAAE,MAAM,CAAC,CAAA;AAC9D,MAAA,IAAA,CAAK,OAAA,CAAQ,CAAA;AACb,MAAA,MAAM,WAAA,EAAa,CAAA,qBAAA,EAAwB,IAAA,CAAK,IAAA,CAAK,IAAI,CAAA;AAAA,EAAM,OAAO,MAAA,IAAU,SAAA,EAAW,MAAA,EAAQ,IAAA,CAAK,SAAA,CAAU,KAAA,EAAO,IAAA,EAAM,CAAC,CAAC,CAAA,CAAA;AAC1H,MAAA;AACR,IAAA;AAE6C,IAAA;AAA0B;AACxE,EAAA;AACD;ADzCqE;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-6YVTJEMQ.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport type {\n\tAgentCard,\n\tMessage,\n\tMessageSendParams,\n\tTask,\n\tTaskStatusUpdateEvent,\n\tTextPart,\n} from \"@a2a-js/sdk\";\nimport { A2AClient } from \"@a2a-js/sdk/client\";\nimport { ThreadType } from \"@/types/memory.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { A2ATool } from \"./a2a.tool.js\";\n\n/**\n * Represents an active A2A communication session.\n */\ninterface A2ASession {\n\t/** Current task ID for multi-turn conversations */\n\ttaskId: string | undefined;\n\t/** Context ID for maintaining conversation state */\n\tcontextId: string | undefined;\n}\n\n/**\n * Module for managing Agent-to-Agent (A2A) protocol connections.\n *\n * This module handles connections to other A2A-compatible agents, manages\n * conversation sessions, and provides an interface for inter-agent communication.\n * Supports multi-turn conversations with task and context tracking.\n */\nexport class A2AModule {\n\t/** Map of A2A server URLs to their corresponding tool instances */\n\tprivate a2aPeerServers: Map<string, A2ATool | null> = new Map();\n\t/** Map of session IDs to their A2A session state */\n\tprivate a2aSessions: Map<string, A2ASession> = new Map();\n\tprivate agentId: string = randomUUID(); /* FIXME */\n\n\t/**\n\t * Registers a new A2A peer server URL for connection.\n\t *\n\t * @param url - The URL of the A2A-compatible agent to connect to\n\t */\n\tpublic async addA2APeerServer(url: string): Promise<void> {\n\t\tthis.a2aPeerServers.set(url, null);\n\t}\n\n\t/**\n\t * Retrieves tools from all registered A2A peer servers.\n\t *\n\t * Attempts to connect to each registered server, fetch their agent cards,\n\t * and create tool instances. Disables tools for unreachable servers.\n\t *\n\t * @returns Promise resolving to array of available A2A tools\n\t */\n\tpublic async getTools(): Promise<A2ATool[]> {\n\t\tconst tools: A2ATool[] = [];\n\t\tfor (const url of [...this.a2aPeerServers.keys()]) {\n\t\t\tconst tool = this.a2aPeerServers.get(url);\n\t\t\tif (!tool || !tool.enabled) {\n\t\t\t\ttry {\n\t\t\t\t\tconst client = new A2AClient(url);\n\t\t\t\t\tconst card: AgentCard = await client.getAgentCard();\n\t\t\t\t\tconst toolName = card.name.replaceAll(\" \", \"-\");\n\t\t\t\t\tconst a2aTool = new A2ATool(toolName, client, card);\n\n\t\t\t\t\ttools.push(a2aTool);\n\t\t\t\t} catch (_error: any) {\n\t\t\t\t\t// Agent not responded\n\t\t\t\t\tif (tool) {\n\t\t\t\t\t\ttool.disable();\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t} else {\n\t\t\t\ttools.push(tool);\n\t\t\t}\n\t\t}\n\t\treturn tools;\n\t}\n\n\t/**\n\t * Gets or creates an A2A session for the given session ID.\n\t *\n\t * @param threadId - The session identifier\n\t * @returns A2ASession object with task and context IDs\n\t */\n\tprivate getA2ASessionWithId = (threadId: string): A2ASession => {\n\t\tconst a2aSession = this.a2aSessions.get(threadId) ?? {\n\t\t\ttaskId: undefined,\n\t\t\tcontextId: undefined,\n\t\t};\n\t\tif (!this.a2aSessions.has(threadId)) {\n\t\t\tthis.a2aSessions.set(threadId, a2aSession);\n\t\t}\n\n\t\treturn a2aSession;\n\t};\n\n\t/**\n\t * Constructs a message payload for A2A communication.\n\t *\n\t * Includes session context (task ID and context ID) if available\n\t * for maintaining conversation continuity.\n\t *\n\t * @param query - The message content to send\n\t * @param threadId - The session identifier\n\t * @returns Formatted Message object for A2A protocol\n\t */\n\tpublic getMessagePayload(query: string, threadId: string): Message {\n\t\tconst messagePayload: Message = {\n\t\t\tmessageId: randomUUID(),\n\t\t\tkind: \"message\",\n\t\t\trole: \"user\", // FIXME: it could be 'agent'\n\t\t\tmetadata: {\n\t\t\t\tagentId: this.agentId,\n\t\t\t\ttype: ThreadType.CHAT,\n\t\t\t\tthreadId,\n\t\t\t},\n\t\t\tparts: [\n\t\t\t\t{\n\t\t\t\t\tkind: \"text\",\n\t\t\t\t\ttext: query,\n\t\t\t\t},\n\t\t\t],\n\t\t};\n\n\t\tconst a2aSession = this.getA2ASessionWithId(threadId);\n\t\tif (a2aSession.taskId) {\n\t\t\tmessagePayload.taskId = a2aSession.taskId;\n\t\t}\n\t\tif (a2aSession.contextId) {\n\t\t\tmessagePayload.contextId = a2aSession.contextId;\n\t\t}\n\n\t\treturn messagePayload;\n\t}\n\n\t/**\n\t * Executes an A2A tool by sending a message to the remote agent.\n\t *\n\t * Handles streaming responses, maintains session state, and extracts\n\t * text content from various event types in the response stream.\n\t *\n\t * @param tool - The A2ATool instance to use\n\t * @param messagePayload - The message to send to the agent\n\t * @param threadId - The session identifier for context tracking\n\t * @returns Promise resolving to array of text responses from the agent\n\t */\n\tpublic async useTool(\n\t\ttool: A2ATool,\n\t\tmessagePayload: Message,\n\t\tthreadId: string,\n\t): Promise<string> {\n\t\tconst finalText: string[] = [];\n\t\tconst client = tool.client;\n\t\tconst params: MessageSendParams = {\n\t\t\tmessage: messagePayload,\n\t\t};\n\t\tconst a2aSession = this.getA2ASessionWithId(threadId);\n\n\t\ttry {\n\t\t\tconst stream = client.sendMessageStream(params);\n\t\t\tfor await (const event of stream) {\n\t\t\t\tif (event.kind === \"status-update\") {\n\t\t\t\t\tconst typedEvent = event as TaskStatusUpdateEvent;\n\t\t\t\t\tif (\n\t\t\t\t\t\ttypedEvent.final &&\n\t\t\t\t\t\ttypedEvent.status.state !== \"input-required\"\n\t\t\t\t\t) {\n\t\t\t\t\t\ta2aSession.taskId = undefined;\n\t\t\t\t\t}\n\t\t\t\t\t// TODO: handle 'file', 'data' parts\n\t\t\t\t\tconst texts = typedEvent.status.message?.parts\n\t\t\t\t\t\t.filter((part) => part.kind === \"text\")\n\t\t\t\t\t\t.map((part: TextPart) => part.text)\n\t\t\t\t\t\t.join(\"\\n\");\n\t\t\t\t\tif (texts) {\n\t\t\t\t\t\tfinalText.push(texts);\n\t\t\t\t\t}\n\t\t\t\t} else if (event.kind === \"message\") {\n\t\t\t\t\t// FIXME: handling text in 'message'?\n\t\t\t\t\tconst msg = event as Message;\n\t\t\t\t\tif (msg.taskId && msg.taskId !== a2aSession.taskId) {\n\t\t\t\t\t\ta2aSession.taskId = msg.taskId;\n\t\t\t\t\t}\n\t\t\t\t\tif (msg.contextId && msg.contextId !== a2aSession.contextId) {\n\t\t\t\t\t\ta2aSession.contextId = msg.contextId;\n\t\t\t\t\t}\n\t\t\t\t} else if (event.kind === \"task\") {\n\t\t\t\t\t// FIXME: handling text in 'task'?\n\t\t\t\t\tconst task = event as Task;\n\t\t\t\t\tif (task.id !== a2aSession.taskId) {\n\t\t\t\t\t\ta2aSession.taskId = task.id;\n\t\t\t\t\t}\n\t\t\t\t\tif (task.contextId && task.contextId !== a2aSession.contextId) {\n\t\t\t\t\t\ta2aSession.contextId = task.contextId;\n\t\t\t\t\t}\n\t\t\t\t} else {\n\t\t\t\t\tloggers.a2a.warn(\"Received unknown event structure from stream:\", {\n\t\t\t\t\t\tevent,\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tloggers.a2a.error(\"Error communicating with agent:\", { error });\n\t\t\ttool.disable();\n\t\t\tconst toolResult = `[Bot Called A2A Tool ${tool.card.name}]\\n${typeof error === \"string\" ? error : JSON.stringify(error, null, 2)}`;\n\t\t\treturn toolResult;\n\t\t}\n\n\t\treturn `[Bot Called A2A Tool ${tool.card.name}]\\n${finalText.join(\"\\n\")}`;\n\t}\n}\n"]}