@ainetwork/adk 0.1.12 → 0.1.13

package/README.md

@@ -10,7 +10,7 @@ A TypeScript library for building AI agents with multi-protocol support includin
 - **Multi-Protocol Support**: Integrate with MCP servers and A2A agents
 - **Modular Architecture**: Flexible module system for models, memory, MCP, and A2A
 - **Multiple AI Models**: Support for OpenAI and Gemini with easy extensibility
-- **Session Management**: Built-in memory module for conversation history
+- **Thread Management**: Built-in memory module for conversation history
 - **Dual Build System**: Supports both ESM and CJS formats for maximum compatibility
 - **Structured Logging**: Winston-based logging system with service-specific loggers
 - **TypeScript First**: Built with strict TypeScript configuration
@@ -45,7 +45,7 @@ To see how to use this package in your project, check out our comprehensive exam
 - **ModelModule** (`src/modules/models/`): Manages AI model integrations (OpenAI, Gemini)
 - **MCPModule** (`src/modules/mcp/`): Handles Model Context Protocol connections
 - **A2AModule** (`src/modules/a2a/`): Manages agent-to-agent communication
-- **MemoryModule** (`src/modules/memory/`): Provides session and conversation history
+- **MemoryModule** (`src/modules/memory/`): Provides threads and conversation history
 
 ### Module System
 
@@ -129,7 +129,7 @@ modelLogger.error('Model API error');
 - `a2a`: Agent-to-agent communication
 - `model`: AI model interactions
 - `server`: HTTP server operations
-- `memory`: Session and memory management
+- `memory`: Threads, Intents, Agent Data management
 
 ### Log Levels
 - `error`: Error conditions
@@ -142,8 +142,13 @@ modelLogger.error('Model API error');
 ### Standard Endpoints
 - `GET /` - Welcome message
 - `POST /query` - Process queries
-  - Request: `{ message: string, sessionId?: string }`
+  - Request: `{ message: string, threadId: string }`
   - Response: `{ content: string }`
+- `POST /query/stream` - Process queries with streaming
+  - Request: `{ message: string, threadId: string }`
+  - Response: `{ content: string }`
+
+### Agent Management
 
 ### A2A Server Endpoints (when `manifest.url` is configured)
 - `GET /agent-card` - Get agent card information

package/dist/cjs/chunk-36ECGO5C.cjs

@@ -0,0 +1,269 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; }
+
+var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
+
+// src/services/query-stream.service.ts
+var _crypto = require('crypto');
+var QueryStreamService = class {
+  
+  
+  
+  
+  
+  constructor(modelModule, a2aModule, mcpModule, memoryModule, prompts) {
+    this.modelModule = modelModule;
+    this.a2aModule = a2aModule;
+    this.mcpModule = mcpModule;
+    this.memoryModule = memoryModule;
+    this.prompts = prompts;
+  }
+  /**
+   * Detects the intent from a user query.
+   *
+   * @param query - The user's input query
+   * @returns The detected intent (currently returns the query as-is)
+   * @todo Implement actual intent detection logic
+   */
+  async intentTriggering(query) {
+    return query;
+  }
+  /**
+   * Fulfills the detected intent by generating a 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
+   *
+   * @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
+   */
+  async *intentFulfilling(query, threadId, thread) {
+    try {
+      const systemPrompt = `
+Today is ${(/* @__PURE__ */ new Date()).toLocaleDateString()}.
+
+${_optionalChain([this, 'access', _ => _.prompts, 'optionalAccess', _2 => _2.agent]) || ""}
+
+${_optionalChain([this, 'access', _3 => _3.prompts, 'optionalAccess', _4 => _4.system]) || ""}
+    `;
+      const modelInstance = this.modelModule.getModel();
+      const messages = modelInstance.generateMessages({
+        query,
+        thread,
+        systemPrompt: systemPrompt.trim()
+      });
+      const tools = [];
+      if (this.mcpModule) {
+        tools.push(...this.mcpModule.getTools());
+      }
+      if (this.a2aModule) {
+        tools.push(...await this.a2aModule.getTools());
+      }
+      const functions = modelInstance.convertToolsToFunctions(tools);
+      const processList = [];
+      let didCallTool = false;
+      while (true) {
+        const responseStream = await modelInstance.fetchStreamWithContextMessage(
+          messages,
+          functions
+        );
+        didCallTool = false;
+        const assembledToolCalls = [];
+        _chunk466ED5WNcjs.loggers.intentStream.debug("messages", { messages });
+        for await (const chunk of responseStream) {
+          const delta = chunk.delta;
+          if (_optionalChain([delta, 'optionalAccess', _5 => _5.tool_calls])) {
+            didCallTool = true;
+            for (const toolCallDelta of delta.tool_calls) {
+              const index = toolCallDelta.index;
+              if (!assembledToolCalls[index]) {
+                assembledToolCalls[index] = {
+                  id: "",
+                  type: "function",
+                  function: { name: "", arguments: "" }
+                };
+              }
+              if (toolCallDelta.id) {
+                assembledToolCalls[index].id = toolCallDelta.id;
+              }
+              if (_optionalChain([toolCallDelta, 'access', _6 => _6.function, 'optionalAccess', _7 => _7.name])) {
+                assembledToolCalls[index].function.name = toolCallDelta.function.name;
+              }
+              if (_optionalChain([toolCallDelta, 'access', _8 => _8.function, 'optionalAccess', _9 => _9.arguments])) {
+                assembledToolCalls[index].function.arguments += toolCallDelta.function.arguments;
+              }
+            }
+          } else if (_optionalChain([chunk, 'access', _10 => _10.delta, 'optionalAccess', _11 => _11.content])) {
+            yield {
+              event: "text_chunk",
+              data: { delta: chunk.delta.content }
+            };
+          }
+        }
+        _chunk466ED5WNcjs.loggers.intentStream.debug("assembledToolCalls", {
+          assembledToolCalls
+        });
+        if (didCallTool && assembledToolCalls.length > 0) {
+          const messagePayload = _optionalChain([this, 'access', _12 => _12.a2aModule, 'optionalAccess', _13 => _13.getMessagePayload, 'call', _14 => _14(
+            query,
+            threadId
+          )]);
+          for (const toolCall of assembledToolCalls) {
+            const toolName = toolCall.function.name;
+            const toolArgs = JSON.parse(toolCall.function.arguments);
+            const selectedTool = tools.filter(
+              (tool) => tool.id === toolName
+            )[0];
+            yield { event: "tool_start", data: { toolName, toolArgs } };
+            let toolResult = "";
+            if (this.mcpModule && selectedTool.protocol === "MCP" /* MCP */) {
+              const toolArgs2 = JSON.parse(toolCall.function.arguments);
+              _chunk466ED5WNcjs.loggers.intent.debug("MCP tool call", { toolName, toolArgs: toolArgs2 });
+              toolResult = await this.mcpModule.useTool(
+                selectedTool,
+                toolArgs2
+              );
+            } else if (this.a2aModule && selectedTool.protocol === "A2A" /* A2A */) {
+              toolResult = await this.a2aModule.useTool(
+                selectedTool,
+                messagePayload,
+                threadId
+              );
+            } else {
+              _chunk466ED5WNcjs.loggers.intent.warn(
+                `Unrecognized tool type: ${selectedTool.protocol}`
+              );
+              continue;
+            }
+            yield {
+              event: "tool_output",
+              data: { toolName, result: toolResult }
+            };
+            _chunk466ED5WNcjs.loggers.intent.debug("toolResult", { toolResult });
+            processList.push(toolResult);
+            modelInstance.appendMessages(messages, toolResult);
+          }
+        }
+        if (!didCallTool) break;
+      }
+    } catch (error) {
+      _chunk466ED5WNcjs.loggers.intent.error("Error in intentFulfilling generator", { error });
+      if (error instanceof Error) {
+        yield {
+          event: "error",
+          data: { message: error.message || "An unknown error occurred." }
+        };
+      } else {
+        yield {
+          event: "error",
+          data: { message: "An unknown error occurred." }
+        };
+      }
+    }
+  }
+  /**
+   * Generates a title for the conversation based on the query.
+   *
+   * @param query - The user's input query
+   * @returns Promise resolving to a generated title
+   */
+  async generateTitle(query) {
+    const DEFAULT_TITLE = "New Chat";
+    try {
+      const modelInstance = this.modelModule.getModel();
+      const messages = modelInstance.generateMessages({
+        query,
+        systemPrompt: `You are a helpful assistant that generates titles for conversations.
+  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.
+  The title must be no more than 5 words long.
+  Respond with only the title. Do not include any punctuation or extra explanations.`
+      });
+      const response = await modelInstance.fetch(messages);
+      return response.content || DEFAULT_TITLE;
+    } catch (error) {
+      _chunk466ED5WNcjs.loggers.intentStream.error("Error generating title", {
+        error,
+        query
+      });
+      return DEFAULT_TITLE;
+    }
+  }
+  /**
+   * Main entry point for processing user queries.
+   *
+   * Handles the complete query lifecycle:
+   * 1. Loads 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 query - The user's input query
+   * @returns Object containing the AI-generated response
+   */
+  async *handleQueryStream(threadMetadata, query) {
+    const { type, userId } = threadMetadata;
+    const queryStartAt = Date.now();
+    const threadMemory = _optionalChain([this, 'access', _15 => _15.memoryModule, 'optionalAccess', _16 => _16.getThreadMemory, 'call', _17 => _17()]);
+    let threadId = threadMetadata.threadId;
+    let thread;
+    console.log("handleQueryStream");
+    if (threadId) {
+      thread = await _optionalChain([threadMemory, 'optionalAccess', _18 => _18.getThread, 'call', _19 => _19(type, userId, threadId)]);
+    } else {
+      threadId = _crypto.randomUUID.call(void 0, );
+      const title = await this.generateTitle(query);
+      console.log(title);
+      const metadata = await _optionalChain([threadMemory, 'optionalAccess', _20 => _20.createThread, 'call', _21 => _21(type, userId, threadId, title)]) || {
+        type,
+        threadId,
+        title,
+        updatedAt: Date.now()
+      };
+      _chunk466ED5WNcjs.loggers.intentStream.info("Create new thread", { metadata });
+      yield { event: "thread_id", data: metadata };
+    }
+    const _intent = await this.intentTriggering(query);
+    try {
+      const stream = this.intentFulfilling(query, threadId, thread);
+      let finalResponseText = "";
+      for await (const event of stream) {
+        if (event.event === "text_chunk" && event.data.delta) {
+          _chunk466ED5WNcjs.loggers.intentStream.debug("text_chunk", { event });
+          finalResponseText += event.data.delta;
+        }
+        yield event;
+      }
+      if (userId) {
+        await _optionalChain([threadMemory, 'optionalAccess', _22 => _22.addMessageToThread, 'call', _23 => _23(userId, threadId, {
+          role: "USER" /* USER */,
+          timestamp: queryStartAt,
+          content: { type: "text", parts: [query] }
+        })]);
+        await _optionalChain([threadMemory, 'optionalAccess', _24 => _24.addMessageToThread, 'call', _25 => _25(userId, threadId, {
+          role: "MODEL" /* MODEL */,
+          timestamp: Date.now(),
+          content: { type: "text", parts: [finalResponseText] }
+        })]);
+      }
+    } catch (error) {
+      const message = error instanceof Error ? error.message : "Stream failed";
+      _chunk466ED5WNcjs.loggers.intentStream.error(message, { error });
+      yield {
+        event: "error",
+        data: { message }
+      };
+    }
+  }
+};
+
+
+
+exports.QueryStreamService = QueryStreamService;
+//# sourceMappingURL=chunk-36ECGO5C.cjs.map

package/dist/cjs/chunk-36ECGO5C.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-36ECGO5C.cjs","../../src/services/query-stream.service.ts"],"names":["toolArgs"],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA,gCAA2B;AA8BpB,IAAM,mBAAA,EAAN,MAAyB;AAAA,EACvB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,WAAA,CACC,WAAA,EACA,SAAA,EACA,SAAA,EACA,YAAA,EACA,OAAA,EACC;AACD,IAAA,IAAA,CAAK,YAAA,EAAc,WAAA;AACnB,IAAA,IAAA,CAAK,UAAA,EAAY,SAAA;AACjB,IAAA,IAAA,CAAK,UAAA,EAAY,SAAA;AACjB,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AACpB,IAAA,IAAA,CAAK,QAAA,EAAU,OAAA;AAAA,EAChB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAc,gBAAA,CAAiB,KAAA,EAAe;AAE7C,IAAA,OAAO,KAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAgBA,MAAA,CAAc,gBAAA,CACb,KAAA,EACA,QAAA,EACA,MAAA,EAC8B;AAC9B,IAAA,IAAI;AACH,MAAA,MAAM,aAAA,EAAe,CAAA;AAAA,SAAA,EAAA,iBACb,IAAI,IAAA,CAAK,CAAA,CAAA,CAAE,kBAAA,CAAmB,CAAC,CAAA;AAAA;AAAA,kBAExC,IAAA,mBAAK,OAAA,6BAAS,QAAA,GAAS,EAAE,CAAA;AAAA;AAAA,kBAEzB,IAAA,qBAAK,OAAA,6BAAS,SAAA,GAAU,EAAE,CAAA;AAAA,IAAA,CAAA;AAGzB,MAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,WAAA,CAAY,QAAA,CAAS,CAAA;AAChD,MAAA,MAAM,SAAA,EAAW,aAAA,CAAc,gBAAA,CAAiB;AAAA,QAC/C,KAAA;AAAA,QACA,MAAA;AAAA,QACA,YAAA,EAAc,YAAA,CAAa,IAAA,CAAK;AAAA,MACjC,CAAC,CAAA;AAED,MAAA,MAAM,MAAA,EAAsB,CAAC,CAAA;AAC7B,MAAA,GAAA,CAAI,IAAA,CAAK,SAAA,EAAW;AACnB,QAAA,KAAA,CAAM,IAAA,CAAK,GAAG,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAC,CAAA;AAAA,MACxC;AACA,MAAA,GAAA,CAAI,IAAA,CAAK,SAAA,EAAW;AACnB,QAAA,KAAA,CAAM,IAAA,CAAK,GAAI,MAAM,IAAA,CAAK,SAAA,CAAU,QAAA,CAAS,CAAE,CAAA;AAAA,MAChD;AACA,MAAA,MAAM,UAAA,EAAY,aAAA,CAAc,uBAAA,CAAwB,KAAK,CAAA;AAE7D,MAAA,MAAM,YAAA,EAAwB,CAAC,CAAA;AAC/B,MAAA,IAAI,YAAA,EAAc,KAAA;AAElB,MAAA,MAAA,CAAO,IAAA,EAAM;AACZ,QAAA,MAAM,eAAA,EACL,MAAM,aAAA,CAAc,6BAAA;AAAA,UACnB,QAAA;AAAA,UACA;AAAA,QACD,CAAA;AACD,QAAA,YAAA,EAAc,KAAA;AAEd,QAAA,MAAM,mBAAA,EAIA,CAAC,CAAA;AAEP,QAAA,yBAAA,CAAQ,YAAA,CAAa,KAAA,CAAM,UAAA,EAAY,EAAE,SAAS,CAAC,CAAA;AAEnD,QAAA,IAAA,MAAA,CAAA,MAAiB,MAAA,GAAS,cAAA,EAAgB;AACzC,UAAA,MAAM,MAAA,EAAQ,KAAA,CAAM,KAAA;AACpB,UAAA,GAAA,iBAAI,KAAA,6BAAO,YAAA,EAAY;AACtB,YAAA,YAAA,EAAc,IAAA;AACd,YAAA,IAAA,CAAA,MAAW,cAAA,GAAiB,KAAA,CAAM,UAAA,EAAY;AAC7C,cAAA,MAAM,MAAA,EAAQ,aAAA,CAAc,KAAA;AAE5B,cAAA,GAAA,CAAI,CAAC,kBAAA,CAAmB,KAAK,CAAA,EAAG;AAC/B,gBAAA,kBAAA,CAAmB,KAAK,EAAA,EAAI;AAAA,kBAC3B,EAAA,EAAI,EAAA;AAAA,kBACJ,IAAA,EAAM,UAAA;AAAA,kBACN,QAAA,EAAU,EAAE,IAAA,EAAM,EAAA,EAAI,SAAA,EAAW,GAAG;AAAA,gBACrC,CAAA;AAAA,cACD;AAEA,cAAA,GAAA,CAAI,aAAA,CAAc,EAAA,EAAI;AACrB,gBAAA,kBAAA,CAAmB,KAAK,CAAA,CAAE,GAAA,EAAK,aAAA,CAAc,EAAA;AAAA,cAC9C;AAEA,cAAA,GAAA,iBAAI,aAAA,qBAAc,QAAA,6BAAU,MAAA,EAAM;AACjC,gBAAA,kBAAA,CAAmB,KAAK,CAAA,CAAE,QAAA,CAAS,KAAA,EAClC,aAAA,CAAc,QAAA,CAAS,IAAA;AAAA,cACzB;AAEA,cAAA,GAAA,iBAAI,aAAA,qBAAc,QAAA,6BAAU,WAAA,EAAW;AACtC,gBAAA,kBAAA,CAAmB,KAAK,CAAA,CAAE,QAAA,CAAS,UAAA,GAClC,aAAA,CAAc,QAAA,CAAS,SAAA;AAAA,cACzB;AAAA,YACD;AAAA,UACD,EAAA,KAAA,GAAA,iBAAW,KAAA,uBAAM,KAAA,+BAAO,SAAA,EAAS;AAChC,YAAA,MAAM;AAAA,cACL,KAAA,EAAO,YAAA;AAAA,cACP,IAAA,EAAM,EAAE,KAAA,EAAO,KAAA,CAAM,KAAA,CAAM,QAAQ;AAAA,YACpC,CAAA;AAAA,UACD;AAAA,QACD;AAEA,QAAA,yBAAA,CAAQ,YAAA,CAAa,KAAA,CAAM,oBAAA,EAAsB;AAAA,UAChD;AAAA,QACD,CAAC,CAAA;AAED,QAAA,GAAA,CAAI,YAAA,GAAe,kBAAA,CAAmB,OAAA,EAAS,CAAA,EAAG;AACjD,UAAA,MAAM,eAAA,kBAAiB,IAAA,uBAAK,SAAA,+BAAW,iBAAA;AAAA,YACtC,KAAA;AAAA,YACA;AAAA,UACD,GAAA;AACA,UAAA,IAAA,CAAA,MAAW,SAAA,GAAY,kBAAA,EAAoB;AAC1C,YAAA,MAAM,SAAA,EAAW,QAAA,CAAS,QAAA,CAAS,IAAA;AACnC,YAAA,MAAM,SAAA,EAAW,IAAA,CAAK,KAAA,CAAM,QAAA,CAAS,QAAA,CAAS,SAAS,CAAA;AACvD,YAAA,MAAM,aAAA,EAAe,KAAA,CAAM,MAAA;AAAA,cAC1B,CAAC,IAAA,EAAA,GAAS,IAAA,CAAK,GAAA,IAAO;AAAA,YACvB,CAAA,CAAE,CAAC,CAAA;AAEH,YAAA,MAAM,EAAE,KAAA,EAAO,YAAA,EAAc,IAAA,EAAM,EAAE,QAAA,EAAU,SAAS,EAAE,CAAA;AAE1D,YAAA,IAAI,WAAA,EAAa,EAAA;AAEjB,YAAA,GAAA,CACC,IAAA,CAAK,UAAA,GACL,YAAA,CAAa,SAAA,IAAA,eAAA,EACZ;AACD,cAAA,MAAMA,UAAAA,EAAW,IAAA,CAAK,KAAA,CAAM,QAAA,CAAS,QAAA,CAAS,SAAS,CAAA;AAGvD,cAAA,yBAAA,CAAQ,MAAA,CAAO,KAAA,CAAM,eAAA,EAAiB,EAAE,QAAA,EAAU,QAAA,EAAAA,UAAS,CAAC,CAAA;AAC5D,cAAA,WAAA,EAAa,MAAM,IAAA,CAAK,SAAA,CAAU,OAAA;AAAA,gBACjC,YAAA;AAAA,gBACAA;AAAA,cACD,CAAA;AAAA,YACD,EAAA,KAAA,GAAA,CACC,IAAA,CAAK,UAAA,GACL,YAAA,CAAa,SAAA,IAAA,eAAA,EACZ;AACD,cAAA,WAAA,EAAa,MAAM,IAAA,CAAK,SAAA,CAAU,OAAA;AAAA,gBACjC,YAAA;AAAA,gBACA,cAAA;AAAA,gBACA;AAAA,cACD,CAAA;AAAA,YACD,EAAA,KAAO;AAEN,cAAA,yBAAA,CAAQ,MAAA,CAAO,IAAA;AAAA,gBACd,CAAA,wBAAA,EAA2B,YAAA,CAAa,QAAQ,CAAA;AAAA,cAAA;AAEjD,cAAA;AAAA,YAAA;AAED,YAAA;AAAM,cAAA;AACE,cAAA;AAC8B,YAAA;AAEtC,YAAA;AAEA,YAAA;AACA,YAAA;AAAiD,UAAA;AAClD,QAAA;AAGD,QAAA;AAAkB,MAAA;AACnB,IAAA;AAEA,MAAA;AACA,MAAA;AACC,QAAA;AAAM,UAAA;AACE,UAAA;AACwD,QAAA;AAChE,MAAA;AAEA,QAAA;AAAM,UAAA;AACE,UAAA;AACuC,QAAA;AAC/C,MAAA;AACD,IAAA;AACD,EAAA;AACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAAA;AAUC,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AAAgD,QAAA;AAC/C,QAAA;AACc;AAAA;AAAA,oFAAA;AAAA,MAAA;AAKf,MAAA;AACA,MAAA;AAA2B,IAAA;AAE3B,MAAA;AAAqD,QAAA;AACpD,QAAA;AACA,MAAA;AAED,MAAA;AAAO,IAAA;AACR,EAAA;AACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAAA;AAyBC,IAAA;AACA,IAAA;AACA,IAAA;AAEA,IAAA;AACA,IAAA;AAEA,IAAA;AAEA,IAAA;AACC,MAAA;AAA6D,IAAA;AAE7D,MAAA;AACA,MAAA;AAEA,MAAA;AAEA,MAAA;AAEE,QAAA;AACA,QAAA;AACA,QAAA;AACA,QAAA;AACoB,MAAA;AAEtB,MAAA;AACA,MAAA;AAA2C,IAAA;AAI5C,IAAA;AAEA,IAAA;AAEC,MAAA;AAEA,MAAA;AACA,MAAA;AACC,QAAA;AACC,UAAA;AACA,UAAA;AAAgC,QAAA;AAEjC,QAAA;AAAM,MAAA;AAGP,MAAA;AACC,QAAA;AAAyD,UAAA;AACxD,UAAA;AACW,UAAA;AAC6B,QAAA;AAEzC,QAAA;AAAyD,UAAA;AACxD,UAAA;AACoB,UAAA;AACgC,QAAA;AACpD,MAAA;AACF,IAAA;AAEA,MAAA;AACA,MAAA;AACA,MAAA;AAAM,QAAA;AACE,QAAA;AACS,MAAA;AACjB,IAAA;AACD,EAAA;AAEF;AD9FA;AACA;AACA;AACA","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-36ECGO5C.cjs","sourcesContent":[null,"import { randomUUID } from \"node:crypto\";\nimport type {\n\tA2AModule,\n\tMCPModule,\n\tMemoryModule,\n\tModelModule,\n} from \"@/modules/index.js\";\nimport type { AinAgentPrompts } from \"@/types/agent.js\";\nimport {\n\tMessageRole,\n\ttype ThreadMetadata,\n\ttype ThreadObject,\n\ttype ThreadType,\n} from \"@/types/memory.js\";\nimport type { StreamEvent } from \"@/types/stream\";\nimport {\n\ttype IA2ATool,\n\ttype IAgentTool,\n\ttype IMCPTool,\n\tTOOL_PROTOCOL_TYPE,\n} from \"@/types/tool.js\";\nimport { loggers } from \"@/utils/logger.js\";\n\n/**\n * Service for processing user queries through the agent's AI pipeline.\n *\n * Orchestrates the query processing workflow including intent detection,\n * model inference, tool execution, and response generation. Manages\n * conversation context and coordinates between different modules.\n */\nexport class QueryStreamService {\n\tprivate modelModule: ModelModule;\n\tprivate a2aModule?: A2AModule;\n\tprivate mcpModule?: MCPModule;\n\tprivate memoryModule?: MemoryModule;\n\tprivate prompts?: AinAgentPrompts;\n\n\tconstructor(\n\t\tmodelModule: ModelModule,\n\t\ta2aModule?: A2AModule,\n\t\tmcpModule?: MCPModule,\n\t\tmemoryModule?: MemoryModule,\n\t\tprompts?: AinAgentPrompts,\n\t) {\n\t\tthis.modelModule = modelModule;\n\t\tthis.a2aModule = a2aModule;\n\t\tthis.mcpModule = mcpModule;\n\t\tthis.memoryModule = memoryModule;\n\t\tthis.prompts = prompts;\n\t}\n\n\t/**\n\t * Detects the intent from a user query.\n\t *\n\t * @param query - The user's input query\n\t * @returns The detected intent (currently returns the query as-is)\n\t * @todo Implement actual intent detection logic\n\t */\n\tprivate async intentTriggering(query: string) {\n\t\t/* TODO */\n\t\treturn query;\n\t}\n\n\t/**\n\t * Fulfills the detected intent by generating a response.\n\t *\n\t * Manages the complete inference loop including:\n\t * - Loading prompts and conversation history\n\t * - Collecting available tools from modules\n\t * - Executing model inference with tool support\n\t * - Processing tool calls iteratively until completion\n\t *\n\t * @param query - The user's input query\n\t * @param threadId - Thread identifier for context\n\t * @param thread - Previous conversation history\n\t * @returns Object containing process steps and final response\n\t */\n\tpublic async *intentFulfilling(\n\t\tquery: string,\n\t\tthreadId: string,\n\t\tthread?: ThreadObject,\n\t): AsyncGenerator<StreamEvent> {\n\t\ttry {\n\t\t\tconst systemPrompt = `\nToday is ${new Date().toLocaleDateString()}.\n\n${this.prompts?.agent || \"\"}\n\n${this.prompts?.system || \"\"}\n    `;\n\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tthread,\n\t\t\t\tsystemPrompt: systemPrompt.trim(),\n\t\t\t});\n\n\t\t\tconst tools: IAgentTool[] = [];\n\t\t\tif (this.mcpModule) {\n\t\t\t\ttools.push(...this.mcpModule.getTools());\n\t\t\t}\n\t\t\tif (this.a2aModule) {\n\t\t\t\ttools.push(...(await this.a2aModule.getTools()));\n\t\t\t}\n\t\t\tconst functions = modelInstance.convertToolsToFunctions(tools);\n\n\t\t\tconst processList: string[] = [];\n\t\t\tlet didCallTool = false;\n\n\t\t\twhile (true) {\n\t\t\t\tconst responseStream =\n\t\t\t\t\tawait modelInstance.fetchStreamWithContextMessage(\n\t\t\t\t\t\tmessages,\n\t\t\t\t\t\tfunctions,\n\t\t\t\t\t);\n\t\t\t\tdidCallTool = false;\n\n\t\t\t\tconst assembledToolCalls: {\n\t\t\t\t\tid: string;\n\t\t\t\t\ttype: \"function\";\n\t\t\t\t\tfunction: { name: string; arguments: string };\n\t\t\t\t}[] = [];\n\n\t\t\t\tloggers.intentStream.debug(\"messages\", { messages });\n\n\t\t\t\tfor await (const chunk of responseStream) {\n\t\t\t\t\tconst delta = chunk.delta;\n\t\t\t\t\tif (delta?.tool_calls) {\n\t\t\t\t\t\tdidCallTool = true;\n\t\t\t\t\t\tfor (const toolCallDelta of delta.tool_calls) {\n\t\t\t\t\t\t\tconst index = toolCallDelta.index;\n\n\t\t\t\t\t\t\tif (!assembledToolCalls[index]) {\n\t\t\t\t\t\t\t\tassembledToolCalls[index] = {\n\t\t\t\t\t\t\t\t\tid: \"\",\n\t\t\t\t\t\t\t\t\ttype: \"function\",\n\t\t\t\t\t\t\t\t\tfunction: { name: \"\", arguments: \"\" },\n\t\t\t\t\t\t\t\t};\n\t\t\t\t\t\t\t}\n\n\t\t\t\t\t\t\tif (toolCallDelta.id) {\n\t\t\t\t\t\t\t\tassembledToolCalls[index].id = toolCallDelta.id;\n\t\t\t\t\t\t\t}\n\n\t\t\t\t\t\t\tif (toolCallDelta.function?.name) {\n\t\t\t\t\t\t\t\tassembledToolCalls[index].function.name =\n\t\t\t\t\t\t\t\t\ttoolCallDelta.function.name;\n\t\t\t\t\t\t\t}\n\n\t\t\t\t\t\t\tif (toolCallDelta.function?.arguments) {\n\t\t\t\t\t\t\t\tassembledToolCalls[index].function.arguments +=\n\t\t\t\t\t\t\t\t\ttoolCallDelta.function.arguments;\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\t\t\t\t\t} else if (chunk.delta?.content) {\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\tevent: \"text_chunk\",\n\t\t\t\t\t\t\tdata: { delta: chunk.delta.content },\n\t\t\t\t\t\t};\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\tloggers.intentStream.debug(\"assembledToolCalls\", {\n\t\t\t\t\tassembledToolCalls,\n\t\t\t\t});\n\n\t\t\t\tif (didCallTool && assembledToolCalls.length > 0) {\n\t\t\t\t\tconst messagePayload = this.a2aModule?.getMessagePayload(\n\t\t\t\t\t\tquery,\n\t\t\t\t\t\tthreadId,\n\t\t\t\t\t);\n\t\t\t\t\tfor (const toolCall of assembledToolCalls) {\n\t\t\t\t\t\tconst toolName = toolCall.function.name;\n\t\t\t\t\t\tconst toolArgs = JSON.parse(toolCall.function.arguments);\n\t\t\t\t\t\tconst selectedTool = tools.filter(\n\t\t\t\t\t\t\t(tool) => tool.id === toolName,\n\t\t\t\t\t\t)[0];\n\n\t\t\t\t\t\tyield { event: \"tool_start\", data: { toolName, toolArgs } };\n\n\t\t\t\t\t\tlet toolResult = \"\";\n\n\t\t\t\t\t\tif (\n\t\t\t\t\t\t\tthis.mcpModule &&\n\t\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.MCP\n\t\t\t\t\t\t) {\n\t\t\t\t\t\t\tconst toolArgs = JSON.parse(toolCall.function.arguments) as\n\t\t\t\t\t\t\t\t| { [x: string]: unknown }\n\t\t\t\t\t\t\t\t| undefined;\n\t\t\t\t\t\t\tloggers.intent.debug(\"MCP tool call\", { toolName, toolArgs });\n\t\t\t\t\t\t\ttoolResult = await this.mcpModule.useTool(\n\t\t\t\t\t\t\t\tselectedTool as IMCPTool,\n\t\t\t\t\t\t\t\ttoolArgs,\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t} else if (\n\t\t\t\t\t\t\tthis.a2aModule &&\n\t\t\t\t\t\t\tselectedTool.protocol === TOOL_PROTOCOL_TYPE.A2A\n\t\t\t\t\t\t) {\n\t\t\t\t\t\t\ttoolResult = await this.a2aModule.useTool(\n\t\t\t\t\t\t\t\tselectedTool as IA2ATool,\n\t\t\t\t\t\t\t\tmessagePayload!,\n\t\t\t\t\t\t\t\tthreadId,\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\t// Unrecognized tool type. It cannot be happened...\n\t\t\t\t\t\t\tloggers.intent.warn(\n\t\t\t\t\t\t\t\t`Unrecognized tool type: ${selectedTool.protocol}`,\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\tcontinue;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tyield {\n\t\t\t\t\t\t\tevent: \"tool_output\",\n\t\t\t\t\t\t\tdata: { toolName, result: toolResult },\n\t\t\t\t\t\t};\n\t\t\t\t\t\tloggers.intent.debug(\"toolResult\", { toolResult });\n\n\t\t\t\t\t\tprocessList.push(toolResult);\n\t\t\t\t\t\tmodelInstance.appendMessages(messages, toolResult);\n\t\t\t\t\t}\n\t\t\t\t}\n\n\t\t\t\tif (!didCallTool) break;\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tloggers.intent.error(\"Error in intentFulfilling generator\", { error });\n\t\t\tif (error instanceof Error) {\n\t\t\t\tyield {\n\t\t\t\t\tevent: \"error\",\n\t\t\t\t\tdata: { message: error.message || \"An unknown error occurred.\" },\n\t\t\t\t};\n\t\t\t} else {\n\t\t\t\tyield {\n\t\t\t\t\tevent: \"error\",\n\t\t\t\t\tdata: { message: \"An unknown error occurred.\" },\n\t\t\t\t};\n\t\t\t}\n\t\t}\n\t}\n\n\t/**\n\t * Generates a title for the conversation based on the query.\n\t *\n\t * @param query - The user's input query\n\t * @returns Promise resolving to a generated title\n\t */\n\n\tprivate async generateTitle(query: string): Promise<string> {\n\t\tconst DEFAULT_TITLE = \"New Chat\";\n\t\ttry {\n\t\t\tconst modelInstance = this.modelModule.getModel();\n\t\t\tconst messages = modelInstance.generateMessages({\n\t\t\t\tquery,\n\t\t\t\tsystemPrompt: `You are a helpful assistant that generates titles for conversations.\n  Please analyze the user's query and create a concise title that accurately reflects the conversation's core topic.\n  The title must be no more than 5 words long.\n  Respond with only the title. Do not include any punctuation or extra explanations.`,\n\t\t\t});\n\t\t\tconst response = await modelInstance.fetch(messages);\n\t\t\treturn response.content || DEFAULT_TITLE;\n\t\t} catch (error) {\n\t\t\tloggers.intentStream.error(\"Error generating title\", {\n\t\t\t\terror,\n\t\t\t\tquery,\n\t\t\t});\n\t\t\treturn DEFAULT_TITLE;\n\t\t}\n\t}\n\n\t/**\n\t * Main entry point for processing user queries.\n\t *\n\t * Handles the complete query lifecycle:\n\t * 1. Loads thread from memory\n\t * 2. Detects intent from the query\n\t * 3. Fulfills the intent with AI response\n\t * 4. Updates conversation history\n\t *\n\t * @param type - The type of thread (e.g., chat, workflow)\n\t * @param userId - The user's unique identifier\n\t * @param threadId - Unique thread identifier\n\t * @param query - The user's input query\n\t * @returns Object containing the AI-generated response\n\t */\n\tpublic async *handleQueryStream(\n\t\tthreadMetadata: {\n\t\t\ttype: ThreadType;\n\t\t\tuserId: string;\n\t\t\tthreadId?: string;\n\t\t},\n\t\tquery: string,\n\t): AsyncGenerator<StreamEvent> {\n\t\tconst { type, userId } = threadMetadata;\n\t\tconst queryStartAt = Date.now();\n\t\tconst threadMemory = this.memoryModule?.getThreadMemory();\n\n\t\tlet threadId = threadMetadata.threadId;\n\t\tlet thread: ThreadObject | undefined;\n\n\t\tconsole.log('handleQueryStream');\n\n\t\tif (threadId) {\n\t\t\tthread = await threadMemory?.getThread(type, userId, threadId);\n\t\t} else {\n\t\t\tthreadId = randomUUID();\n\t\t\tconst title = await this.generateTitle(query);\n\n\t\t\tconsole.log(title);\n\n\t\t\tconst metadata =\n\t\t\t\t(await threadMemory?.createThread(type, userId, threadId, title)) ||\n\t\t\t\t({\n\t\t\t\t\ttype,\n\t\t\t\t\tthreadId,\n\t\t\t\t\ttitle,\n\t\t\t\t\tupdatedAt: Date.now(),\n\t\t\t\t} as ThreadMetadata);\n\t\t\tloggers.intentStream.info(\"Create new thread\", { metadata });\n\t\t\tyield { event: \"thread_id\", data: metadata };\n\t\t}\n\n\t\t// 2. intent triggering\n\t\tconst _intent = await this.intentTriggering(query);\n\n\t\ttry {\n\t\t\t// 3. intent fulfillment\n\t\t\tconst stream = this.intentFulfilling(query, threadId, thread);\n\n\t\t\tlet finalResponseText = \"\";\n\t\t\tfor await (const event of stream) {\n\t\t\t\tif (event.event === \"text_chunk\" && event.data.delta) {\n\t\t\t\t\tloggers.intentStream.debug(\"text_chunk\", { event });\n\t\t\t\t\tfinalResponseText += event.data.delta;\n\t\t\t\t}\n\t\t\t\tyield event;\n\t\t\t}\n\n\t\t\tif (userId) {\n\t\t\t\tawait threadMemory?.addMessageToThread(userId, threadId, {\n\t\t\t\t\trole: MessageRole.USER,\n\t\t\t\t\ttimestamp: queryStartAt,\n\t\t\t\t\tcontent: { type: \"text\", parts: [query] },\n\t\t\t\t});\n\t\t\t\tawait threadMemory?.addMessageToThread(userId, threadId, {\n\t\t\t\t\trole: MessageRole.MODEL,\n\t\t\t\t\ttimestamp: Date.now(),\n\t\t\t\t\tcontent: { type: \"text\", parts: [finalResponseText] },\n\t\t\t\t});\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tconst message = error instanceof Error ? error.message : \"Stream failed\";\n\t\t\tloggers.intentStream.error(message, { error });\n\t\t\tyield {\n\t\t\t\tevent: \"error\",\n\t\t\t\tdata: { message },\n\t\t\t};\n\t\t}\n\t}\n}\n"]}

package/dist/cjs/chunk-3J3ORQOG.cjs

@@ -0,0 +1,65 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class;
+
+var _chunkDX2JH3FJcjs = require('./chunk-DX2JH3FJ.cjs');
+
+// src/controllers/query.controller.ts
+var _httpstatuscodes = require('http-status-codes');
+var QueryController = (_class = class {
+  
+  
+  constructor(queryService, queryStreamService) {;_class.prototype.__init.call(this);_class.prototype.__init2.call(this);
+    this.queryService = queryService;
+    this.queryStreamService = queryStreamService;
+  }
+  __init() {this.handleQueryRequest = async (req, res, next) => {
+    const { type, message, threadId } = req.body;
+    const userId = res.locals.userId;
+    try {
+      const result = await this.queryService.handleQuery(
+        { type, userId, threadId },
+        message
+      );
+      res.status(200).json(result);
+    } catch (error) {
+      next(error);
+    }
+  }}
+  __init2() {this.handleQueryStreamRequest = async (req, res, next) => {
+    const { type, threadId, message } = req.body;
+    const userId = res.locals.userId;
+    if (!this.queryStreamService) {
+      const error = new (0, _chunkDX2JH3FJcjs.AinHttpError)(
+        _httpstatuscodes.StatusCodes.NOT_IMPLEMENTED,
+        "Stream query not supported"
+      );
+      return next(error);
+    }
+    const stream = this.queryStreamService.handleQueryStream(
+      { type, userId, threadId },
+      message
+    );
+    try {
+      for await (const event of stream) {
+        res.write(
+          `event: ${event.event}
+data: ${JSON.stringify(event.data)}
+
+`
+        );
+      }
+    } catch (error) {
+      const errMsg = _optionalChain([error, 'optionalAccess', _2 => _2.message]) || "Failed to handle query stream";
+      res.write(`event: error
+data: ${errMsg}
+
+`);
+    } finally {
+      res.end();
+    }
+  }}
+}, _class);
+
+
+
+exports.QueryController = QueryController;
+//# sourceMappingURL=chunk-3J3ORQOG.cjs.map

package/dist/cjs/chunk-3J3ORQOG.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-3J3ORQOG.cjs","../../src/controllers/query.controller.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACHA,oDAA4B;AAIrB,IAAM,gBAAA,YAAN,MAAsB;AAAA,EACpB;AAAA,EACA;AAAA,EAER,WAAA,CACC,YAAA,EACA,kBAAA,EACC;AACD,IAAA,IAAA,CAAK,aAAA,EAAe,YAAA;AACpB,IAAA,IAAA,CAAK,mBAAA,EAAqB,kBAAA;AAAA,EAC3B;AAAA,iBAEO,mBAAA,EAAqB,MAAA,CAC3B,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,MAAM,EAAE,IAAA,EAAM,OAAA,EAAS,SAAS,EAAA,EAAI,GAAA,CAAI,IAAA;AACxC,IAAA,MAAM,OAAA,EAAS,GAAA,CAAI,MAAA,CAAO,MAAA;AAE1B,IAAA,IAAI;AACH,MAAA,MAAM,OAAA,EAAS,MAAM,IAAA,CAAK,YAAA,CAAa,WAAA;AAAA,QACtC,EAAE,IAAA,EAAM,MAAA,EAAQ,SAAS,CAAA;AAAA,QACzB;AAAA,MACD,CAAA;AAEA,MAAA,GAAA,CAAI,MAAA,CAAO,GAAG,CAAA,CAAE,IAAA,CAAK,MAAM,CAAA;AAAA,IAC5B,EAAA,MAAA,CAAS,KAAA,EAAO;AACf,MAAA,IAAA,CAAK,KAAK,CAAA;AAAA,IACX;AAAA,EACD,EAAA;AAAA,kBAEO,yBAAA,EAA2B,MAAA,CACjC,GAAA,EACA,GAAA,EACA,IAAA,EAAA,GACI;AACJ,IAAA,MAAM,EAAE,IAAA,EAAM,QAAA,EAAU,QAAQ,EAAA,EAAI,GAAA,CAAI,IAAA;AACxC,IAAA,MAAM,OAAA,EAAS,GAAA,CAAI,MAAA,CAAO,MAAA;AAE1B,IAAA,GAAA,CAAI,CAAC,IAAA,CAAK,kBAAA,EAAoB;AAC7B,MAAA,MAAM,MAAA,EAAQ,IAAI,mCAAA;AAAA,QACjB,4BAAA,CAAY,eAAA;AAAA,QACZ;AAAA,MACD,CAAA;AACA,MAAA,OAAO,IAAA,CAAK,KAAK,CAAA;AAAA,IAClB;AAEA,IAAA,MAAM,OAAA,EAAS,IAAA,CAAK,kBAAA,CAAmB,iBAAA;AAAA,MACtC,EAAE,IAAA,EAAM,MAAA,EAAQ,SAAS,CAAA;AAAA,MACzB;AAAA,IACD,CAAA;AAEA,IAAA,IAAI;AACH,MAAA,IAAA,MAAA,CAAA,MAAiB,MAAA,GAAS,MAAA,EAAQ;AACjC,QAAA,GAAA,CAAI,KAAA;AAAA,UACH,CAAA,OAAA,EAAU,KAAA,CAAM,KAAK,CAAA;AAAA,MAAA,EAAW,IAAA,CAAK,SAAA,CAAU,KAAA,CAAM,IAAI,CAAC,CAAA;AAAA;AAAA;AAAA,QAC3D,CAAA;AAAA,MACD;AAAA,IACD,EAAA,MAAA,CAAS,KAAA,EAAgB;AACxB,MAAA,MAAM,OAAA,kBACJ,KAAA,6BAAiB,UAAA,GAAW,+BAAA;AAC9B,MAAA,GAAA,CAAI,KAAA,CAAM,CAAA;AAAA,MAAA,EAAuB,MAAM,CAAA;AAAA;AAAA,CAAM,CAAA;AAAA,IAC9C,EAAA,QAAE;AACD,MAAA,GAAA,CAAI,GAAA,CAAI,CAAA;AAAA,IACT;AAAA,EACD,EAAA;AACD,UAAA;ADZA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-3J3ORQOG.cjs","sourcesContent":[null,"import type { NextFunction, Request, Response } from \"express\";\nimport { StatusCodes } from \"http-status-codes\";\nimport type { QueryService, QueryStreamService } from \"@/services\";\nimport { AinHttpError } from \"@/types/agent\";\n\nexport class QueryController {\n\tprivate queryService;\n\tprivate queryStreamService;\n\n\tconstructor(\n\t\tqueryService: QueryService,\n\t\tqueryStreamService?: QueryStreamService,\n\t) {\n\t\tthis.queryService = queryService;\n\t\tthis.queryStreamService = queryStreamService;\n\t}\n\n\tpublic handleQueryRequest = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\tconst { type, message, threadId } = req.body;\n\t\tconst userId = res.locals.userId;\n\n\t\ttry {\n\t\t\tconst result = await this.queryService.handleQuery(\n\t\t\t\t{ type, userId, threadId },\n\t\t\t\tmessage,\n\t\t\t);\n\n\t\t\tres.status(200).json(result);\n\t\t} catch (error) {\n\t\t\tnext(error);\n\t\t}\n\t};\n\n\tpublic handleQueryStreamRequest = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\tconst { type, threadId, message } = req.body;\n\t\tconst userId = res.locals.userId;\n\n\t\tif (!this.queryStreamService) {\n\t\t\tconst error = new AinHttpError(\n\t\t\t\tStatusCodes.NOT_IMPLEMENTED,\n\t\t\t\t\"Stream query not supported\",\n\t\t\t);\n\t\t\treturn next(error);\n\t\t}\n\n\t\tconst stream = this.queryStreamService.handleQueryStream(\n\t\t\t{ type, userId, threadId },\n\t\t\tmessage,\n\t\t);\n\n\t\ttry {\n\t\t\tfor await (const event of stream) {\n\t\t\t\tres.write(\n\t\t\t\t\t`event: ${event.event}\\ndata: ${JSON.stringify(event.data)}\\n\\n`,\n\t\t\t\t);\n\t\t\t}\n\t\t} catch (error: unknown) {\n\t\t\tconst errMsg =\n\t\t\t\t(error as Error)?.message || \"Failed to handle query stream\";\n\t\t\tres.write(`event: error\\ndata: ${errMsg}\\n\\n`);\n\t\t} finally {\n\t\t\tres.end();\n\t\t}\n\t};\n}\n"]}

package/dist/cjs/chunk-3QNRWPMM.cjs

@@ -0,0 +1,37 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+var _chunkICFTUZP5cjs = require('./chunk-ICFTUZP5.cjs');
+
+
+var _chunkJAPFZA2Kcjs = require('./chunk-JAPFZA2K.cjs');
+
+
+var _chunk55IME4BGcjs = require('./chunk-55IME4BG.cjs');
+
+// src/routes/a2a.routes.ts
+var _server = require('@a2a-js/sdk/server');
+var _express = require('express');
+var createA2ARouter = (agent) => {
+  const router = _express.Router.call(void 0, );
+  const taskStore = new (0, _server.InMemoryTaskStore)();
+  const queryService = new (0, _chunkJAPFZA2Kcjs.QueryService)(
+    agent.modelModule,
+    agent.a2aModule,
+    agent.mcpModule,
+    agent.memoryModule,
+    agent.manifest.prompts
+  );
+  const a2aService = new (0, _chunk55IME4BGcjs.A2AService)(queryService);
+  const a2aController = new (0, _chunkICFTUZP5cjs.A2AController)(
+    a2aService,
+    taskStore,
+    agent.generateAgentCard
+  );
+  router.post("/", a2aController.handleA2ARequest);
+  return router;
+};
+
+
+
+exports.createA2ARouter = createA2ARouter;
+//# sourceMappingURL=chunk-3QNRWPMM.cjs.map

package/dist/cjs/chunk-3QNRWPMM.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-3QNRWPMM.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-3QNRWPMM.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-4MCEFDVX.cjs

@@ -0,0 +1,43 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});
+
+var _chunkXPD3ANDGcjs = require('./chunk-XPD3ANDG.cjs');
+
+
+var _chunkXVMXTCJIcjs = require('./chunk-XVMXTCJI.cjs');
+
+
+var _chunkJAPFZA2Kcjs = require('./chunk-JAPFZA2K.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, _chunkJAPFZA2Kcjs.QueryService)(
+    agent.modelModule,
+    agent.a2aModule,
+    agent.mcpModule,
+    agent.memoryModule,
+    agent.manifest.prompts
+  );
+  let queryStreamService;
+  if (allowStream) {
+    queryStreamService = new (0, _chunkXPD3ANDGcjs.QueryStreamService)(
+      agent.modelModule,
+      agent.a2aModule,
+      agent.mcpModule,
+      agent.memoryModule,
+      agent.manifest.prompts
+    );
+  }
+  const queryController = new (0, _chunkXVMXTCJIcjs.QueryController)(queryService, queryStreamService);
+  router.post("/", queryController.handleQueryRequest);
+  if (allowStream) {
+    router.post("/stream", queryController.handleQueryStreamRequest);
+  }
+  return router;
+};
+
+
+
+exports.createQueryRouter = createQueryRouter;
+//# sourceMappingURL=chunk-4MCEFDVX.cjs.map

package/dist/cjs/chunk-4MCEFDVX.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-4MCEFDVX.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-4MCEFDVX.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-55IME4BG.cjs

@@ -0,0 +1,114 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class;
+
+var _chunk466ED5WNcjs = require('./chunk-466ED5WN.cjs');
+
+// src/services/a2a.service.ts
+var _crypto = require('crypto');
+var A2AService = (_class = class {
+  
+  __init() {this.canceledTasks = /* @__PURE__ */ new Set()}
+  constructor(queryService) {;_class.prototype.__init.call(this);_class.prototype.__init2.call(this);_class.prototype.__init3.call(this);
+    this.queryService = queryService;
+  }
+  __init2() {this.cancelTask = async (taskId, _eventBus) => {
+    this.canceledTasks.add(taskId);
+  }}
+  __init3() {this.createTaskStatusUpdateEvent = (taskId, contextId, state, message) => {
+    return {
+      kind: "status-update",
+      taskId,
+      contextId,
+      status: {
+        state,
+        message: message ? {
+          kind: "message",
+          role: "agent",
+          messageId: _crypto.randomUUID.call(void 0, ),
+          parts: [{ kind: "text", text: message }],
+          taskId,
+          contextId
+        } : void 0,
+        timestamp: (/* @__PURE__ */ new Date()).toISOString()
+      },
+      final: state !== "working"
+    };
+  }}
+  async execute(requestContext, eventBus) {
+    const userMessage = requestContext.userMessage;
+    const { agentId, type, threadId } = userMessage.metadata;
+    const existingTask = requestContext.task;
+    const taskId = _optionalChain([existingTask, 'optionalAccess', _ => _.id]) || _crypto.randomUUID.call(void 0, );
+    const contextId = userMessage.contextId || _optionalChain([existingTask, 'optionalAccess', _2 => _2.contextId]) || _crypto.randomUUID.call(void 0, );
+    if (!existingTask) {
+      const initialTask = {
+        kind: "task",
+        id: taskId,
+        contextId,
+        status: {
+          state: "submitted",
+          timestamp: (/* @__PURE__ */ new Date()).toISOString()
+        },
+        history: [userMessage],
+        metadata: userMessage.metadata,
+        artifacts: []
+      };
+      eventBus.publish(initialTask);
+    }
+    const workingStatusUpdate = this.createTaskStatusUpdateEvent(
+      taskId,
+      contextId,
+      "working"
+    );
+    eventBus.publish(workingStatusUpdate);
+    const message = userMessage.parts.filter((part) => part.kind === "text").map((part) => part.text).join("\n");
+    if (message.length === 0) {
+      _chunk466ED5WNcjs.loggers.server.warn(`Empty message received for task ${taskId}.`);
+      const failureUpdate = this.createTaskStatusUpdateEvent(
+        taskId,
+        contextId,
+        "failed",
+        "No message found to process."
+      );
+      eventBus.publish(failureUpdate);
+      return;
+    }
+    try {
+      const response = await this.queryService.handleQuery(
+        { userId: agentId, type, threadId },
+        message
+      );
+      if (this.canceledTasks.has(taskId)) {
+        _chunk466ED5WNcjs.loggers.server.info(`Task ${taskId} was canceled.`);
+        const canceledUpdate = this.createTaskStatusUpdateEvent(
+          taskId,
+          contextId,
+          "canceled"
+        );
+        eventBus.publish(canceledUpdate);
+        return;
+      }
+      const finalUpdate = this.createTaskStatusUpdateEvent(
+        taskId,
+        contextId,
+        "completed",
+        response.content
+      );
+      eventBus.publish(finalUpdate);
+      _chunk466ED5WNcjs.loggers.server.info(`Task ${taskId} completed successfully.`);
+    } catch (error) {
+      _chunk466ED5WNcjs.loggers.server.error(`Error processing task ${taskId}:`, error);
+      const errorUpdate = this.createTaskStatusUpdateEvent(
+        taskId,
+        contextId,
+        "failed",
+        `Agent error: ${error.message}`
+      );
+      eventBus.publish(errorUpdate);
+    }
+  }
+}, _class);
+
+
+
+exports.A2AService = A2AService;
+//# sourceMappingURL=chunk-55IME4BG.cjs.map

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

@@ -0,0 +1 @@
+{"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"]}