@ainetwork/adk 0.1.12 → 0.2.0

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

@@ -1,7 +1,6 @@
-import { Response } from 'express';
 import { A2AModule } from '../modules/a2a/a2a.module.js';
 import { MCPModule } from '../modules/mcp/mcp.module.js';
-import { SessionObject } from '../types/memory.js';
+import { ThreadObject, Intent, ThreadType } from '../types/memory.js';
 import { MemoryModule } from '../modules/memory/memory.module.js';
 import { StreamEvent } from '../types/stream.js';
 import { ModelModule } from '../modules/models/model.module.js';
@@ -33,11 +32,11 @@ declare class QueryStreamService {
     private prompts?;
     constructor(modelModule: ModelModule, a2aModule?: A2AModule, mcpModule?: MCPModule, memoryModule?: MemoryModule, prompts?: AinAgentPrompts);
     /**
-     * Detects the intent from a user query.
+     * Detects the intent from context.
      *
      * @param query - The user's input query
-     * @returns The detected intent (currently returns the query as-is)
-     * @todo Implement actual intent detection logic
+     * @param thread - The thread history
+     * @returns The detected intent
      */
     private intentTriggering;
     /**
@@ -50,11 +49,11 @@ declare class QueryStreamService {
      * - Processing tool calls iteratively until completion
      *
      * @param query - The user's input query
-     * @param sessionId - Session identifier for context
-     * @param sessionHistory - Previous conversation history
+     * @param threadId - Thread identifier for context
+     * @param thread - Previous conversation history
      * @returns Object containing process steps and final response
      */
-    intentFulfilling(query: string, sessionId: string, sessionHistory?: SessionObject): AsyncGenerator<StreamEvent>;
+    intentFulfilling(query: string, threadId: string, thread?: ThreadObject, intent?: Intent): AsyncGenerator<StreamEvent>;
     /**
      * Generates a title for the conversation based on the query.
      *
@@ -66,16 +65,22 @@ declare class QueryStreamService {
      * Main entry point for processing user queries.
      *
      * Handles the complete query lifecycle:
-     * 1. Loads session history from memory
+     * 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
-     * @param sessionId - Unique session identifier
      * @returns Object containing the AI-generated response
      */
-    handleQueryStream(query: string, userId: string, res: Response, _sessionId?: string): Promise<void>;
+    handleQueryStream(threadMetadata: {
+        type: ThreadType;
+        userId: string;
+        threadId?: string;
+    }, query: string): AsyncGenerator<StreamEvent>;
 }
 
 export { QueryStreamService };

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

@@ -1,7 +1,7 @@
 import {
   QueryStreamService
-} from "../chunk-3NEV3O2X.js";
-import "../chunk-VAZWF5IK.js";
+} from "../chunk-M4EJC7LZ.js";
+import "../chunk-MNAWSBYP.js";
 import "../chunk-FI2UQLVC.js";
 import "../chunk-N7VYM2R5.js";
 export {

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

@@ -1,5 +1,6 @@
 import { A2AModule } from '../modules/a2a/a2a.module.js';
 import { MCPModule } from '../modules/mcp/mcp.module.js';
+import { ThreadType } from '../types/memory.js';
 import { MemoryModule } from '../modules/memory/memory.module.js';
 import { ModelModule } from '../modules/models/model.module.js';
 import { AinAgentPrompts } from '../types/agent.js';
@@ -12,7 +13,6 @@ import '../types/mcp.js';
 import '@modelcontextprotocol/sdk/client/stdio.js';
 import '../modules/mcp/mcp.tool.js';
 import '../modules/memory/base.memory.js';
-import '../types/memory.js';
 import '../modules/models/base.model.js';
 import '../types/stream.js';
 import 'http-status-codes';
@@ -32,11 +32,11 @@ declare class QueryService {
     private prompts?;
     constructor(modelModule: ModelModule, a2aModule?: A2AModule, mcpModule?: MCPModule, memoryModule?: MemoryModule, prompts?: AinAgentPrompts);
     /**
-     * Detects the intent from a user query.
+     * Detects the intent from context.
      *
      * @param query - The user's input query
-     * @returns The detected intent (currently returns the query as-is)
-     * @todo Implement actual intent detection logic
+     * @param thread - The thread history
+     * @returns The detected intent
      */
     private intentTriggering;
     /**
@@ -49,26 +49,37 @@ declare class QueryService {
      * - Processing tool calls iteratively until completion
      *
      * @param query - The user's input query
-     * @param sessionId - Session identifier for context
-     * @param sessionHistory - Previous conversation history
+     * @param threadId - Thread identifier for context
+     * @param thread - Previous conversation history
      * @returns Object containing process steps and final response
      */
     private intentFulfilling;
+    /**
+     * Generates a title for the conversation based on the query.
+     *
+     * @param query - The user's input query
+     * @returns Promise resolving to a generated title
+     */
+    private generateTitle;
     /**
      * Main entry point for processing user queries.
      *
      * Handles the complete query lifecycle:
-     * 1. Loads session history from memory
+     * 1. Loads thread history 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
-     * @param sessionId - Unique session identifier
-     * @param userId - Unique user identifier
-     * @returns Object containing the AI-generated response
      */
-    handleQuery(query: string, sessionId: string, userId?: string): Promise<{
+    handleQuery(threadMetadata: {
+        type: ThreadType;
+        userId: string;
+        threadId?: string;
+    }, query: string): Promise<{
         content: string;
     }>;
 }

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

@@ -1,7 +1,7 @@
 import {
   QueryService
-} from "../chunk-DRUCJYPJ.js";
-import "../chunk-VAZWF5IK.js";
+} from "../chunk-A65XRRHY.js";
+import "../chunk-MNAWSBYP.js";
 import "../chunk-FI2UQLVC.js";
 import "../chunk-N7VYM2R5.js";
 export {

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

@@ -1,13 +1,7 @@
-declare enum MemoryType {
-    SESSION = "SESSION",
-    INTENT = "INTENT",
-    AGENT = "AGENT",
-    USER = "USER"
-}
 /**
- * Roles for participants in a chat conversation.
+ * Roles for participants in a message.
  */
-declare enum ChatRole {
+declare enum MessageRole {
     /** User/human participant */
     USER = "USER",
     /** System-generated messages or instructions */
@@ -16,23 +10,23 @@ declare enum ChatRole {
     MODEL = "MODEL"
 }
 /**
- * Content structure for chat messages.
+ * Content structure for message content.
  *
  * Supports multi-part content with different types (text, images, etc.).
  */
-type ChatContentObject = {
+type MessageContentObject = {
     /** Content type (e.g., "text", "image", "tool_use") */
     type: string;
     /** Array of content parts, structure depends on content type */
     parts: any[];
 };
 /**
- * Represents a single message in a chat conversation.
+ * Represents a single message in a thread.
  *
  * @example
  * ```typescript
- * const message: ChatObject = {
- *   role: ChatRole.USER,
+ * const message: MessageObject = {
+ *   role: MessageRole.USER,
  *   content: {
  *     type: "text",
  *     parts: ["Hello, how can you help me?"]
@@ -42,11 +36,11 @@ type ChatContentObject = {
  * };
  * ```
  */
-type ChatObject = {
+type MessageObject = {
     /** Role of the message sender */
-    role: ChatRole;
+    role: MessageRole;
     /** Message content with type and parts */
-    content: ChatContentObject;
+    content: MessageContentObject;
     /** Unix timestamp when the message was created */
     timestamp: number;
     /** Optional metadata for additional context */
@@ -54,32 +48,38 @@ type ChatObject = {
         [key: string]: unknown;
     };
 };
-type SessionMetadata = {
-    title?: string;
-    sessionId: string;
+declare enum ThreadType {
+    WORKFLOW = "WORKFLOW",
+    CHAT = "CHAT"
+}
+type ThreadMetadata = {
+    type: ThreadType;
+    title: string;
+    threadId: string;
     updatedAt: number;
 };
 /**
- * Represents a conversation session containing multiple chat messages.
+ * Represents a conversation thread containing multiple messages.
  *
- * Messages are stored in a key-value structure where keys are unique chat IDs
- * and values are the corresponding chat objects.
+ * Messages are stored in a key-value structure where keys are unique message IDs
+ * and values are the corresponding message objects.
  *
  * @example
  * ```typescript
- * const session: SessionObject = {
- *   chats: {
- *     "<UUID_1>": { role: ChatRole.USER, content: {...}, timestamp: 1234567890 },
- *     "<UUID_2>": { role: ChatRole.MODEL, content: {...}, timestamp: 1234567891 }
+ * const thread: ThreadObject = {
+ * 	 title: "New conversation",
+ *   messages: {
+ *     "<UUID_1>": { role: MessageRole.USER, content: {...}, timestamp: 1234567890 },
+ *     "<UUID_2>": { role: MessageRole.MODEL, content: {...}, timestamp: 1234567891 }
  *   }
  * };
  * ```
  */
-type SessionObject = {
-    title?: string;
-    /** Collection of chat messages indexed by unique chat ID */
-    chats: {
-        [chatId: string]: ChatObject;
+type ThreadObject = {
+    type: ThreadType;
+    title: string;
+    messages: {
+        [messageId: string]: MessageObject;
     };
 };
 interface Intent {
@@ -89,4 +89,4 @@ interface Intent {
     llm?: string;
 }
 
-export { type ChatContentObject, type ChatObject, ChatRole, type Intent, MemoryType, type SessionMetadata, type SessionObject };
+export { type Intent, type MessageContentObject, type MessageObject, MessageRole, type ThreadMetadata, type ThreadObject, ThreadType };

package/dist/esm/types/memory.js

@@ -1,9 +1,9 @@
 import {
-  ChatRole,
-  MemoryType
-} from "../chunk-VAZWF5IK.js";
+  MessageRole,
+  ThreadType
+} from "../chunk-MNAWSBYP.js";
 export {
-  ChatRole,
-  MemoryType
+  MessageRole,
+  ThreadType
 };
 //# sourceMappingURL=memory.js.map

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

@@ -1,6 +1,36 @@
+import { ThreadMetadata } from './memory.js';
+import { TOOL_PROTOCOL_TYPE } from './tool.js';
+import '@a2a-js/sdk';
+import '@a2a-js/sdk/client';
+import '@modelcontextprotocol/sdk/types.js';
+
 type StreamEvent = {
-    event: "tool_start" | "tool_output" | "text_chunk" | "error";
-    data: Record<string, any>;
+    event: "text_chunk";
+    data: {
+        delta: string;
+    };
+} | {
+    event: "tool_start";
+    data: {
+        protocol: TOOL_PROTOCOL_TYPE;
+        toolName: string;
+        toolArgs: unknown;
+    };
+} | {
+    event: "tool_output";
+    data: {
+        protocol: TOOL_PROTOCOL_TYPE;
+        toolName: string;
+        result: unknown;
+    };
+} | {
+    event: "error";
+    data: {
+        message: string;
+    };
+} | {
+    event: "thread_id";
+    data: ThreadMetadata;
 };
 /**
  * Tool call delta for streaming tool invocations

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@ainetwork/adk",
-	"version": "0.1.12",
+	"version": "0.2.0",
 	"description": "AI Network Agent Development Kit",
 	"repository": "git@github.com:ainetwork-ai/ain-adk.git",
 	"author": "AI Network (https://ainetwork.ai)",

package/dist/cjs/chunk-2DKKHZA5.cjs

@@ -1,17 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});
-
-var _chunkE4FRJCLUcjs = require('./chunk-E4FRJCLU.cjs');
-
-// src/routes/api/sessions.routes.ts
-var _express = require('express');
-var createSessionApiRouter = (memoryModule) => {
-  const router = _express.Router.call(void 0, );
-  const sessionApiController = new (0, _chunkE4FRJCLUcjs.SessionApiController)(memoryModule);
-  router.get("/:id", sessionApiController.handleSessionHistory);
-  return router;
-};
-
-
-
-exports.createSessionApiRouter = createSessionApiRouter;
-//# sourceMappingURL=chunk-2DKKHZA5.cjs.map

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

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-2DKKHZA5.cjs","../../src/routes/api/sessions.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA,kCAAuB;AAIhB,IAAM,uBAAA,EAAyB,CAAC,YAAA,EAAA,GAAuC;AAC7E,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AACtB,EAAA,MAAM,qBAAA,EAAuB,IAAI,2CAAA,CAAqB,YAAY,CAAA;AAGlE,EAAA,MAAA,CAAO,GAAA,CAAI,MAAA,EAAQ,oBAAA,CAAqB,oBAAoB,CAAA;AAE5D,EAAA,OAAO,MAAA;AACR,CAAA;ADAA;AACA;AACE;AACF,wDAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-2DKKHZA5.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport { SessionApiController } from \"@/controllers/api/session.api.controller.js\";\nimport type { MemoryModule } from \"@/modules/index.js\";\n\nexport const createSessionApiRouter = (memoryModule: MemoryModule): Router => {\n\tconst router = Router();\n\tconst sessionApiController = new SessionApiController(memoryModule);\n\n\t// APIs (prefix: /api/sessions)\n\trouter.get(\"/:id\", sessionApiController.handleSessionHistory);\n\n\treturn router;\n};\n"]}

package/dist/cjs/chunk-2E6X6IDL.cjs

@@ -1,37 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});
-
-var _chunk3JKRJYJ6cjs = require('./chunk-3JKRJYJ6.cjs');
-
-
-var _chunkDQP6RGIRcjs = require('./chunk-DQP6RGIR.cjs');
-
-
-var _chunkICFTUZP5cjs = require('./chunk-ICFTUZP5.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, _chunkDQP6RGIRcjs.QueryService)(
-    agent.modelModule,
-    agent.a2aModule,
-    agent.mcpModule,
-    agent.memoryModule,
-    agent.manifest.prompts
-  );
-  const a2aService = new (0, _chunk3JKRJYJ6cjs.A2AService)(queryService);
-  const a2aController = new (0, _chunkICFTUZP5cjs.A2AController)(
-    a2aService,
-    taskStore,
-    agent.generateAgentCard
-  );
-  router.post("/a2a", a2aController.handleA2ARequest);
-  return router;
-};
-
-
-
-exports.createA2ARouter = createA2ARouter;
-//# sourceMappingURL=chunk-2E6X6IDL.cjs.map

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

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-2E6X6IDL.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,MAAA,EAAQ,aAAA,CAAc,gBAAgB,CAAA;AAElD,EAAA,OAAO,MAAA;AACR,CAAA;ADNA;AACA;AACE;AACF,0CAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-2E6X6IDL.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(\"/a2a\", a2aController.handleA2ARequest);\n\n\treturn router;\n};\n"]}

package/dist/cjs/chunk-2G7V7SC5.cjs

@@ -1,18 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});
-
-var _chunkYE52UBF3cjs = require('./chunk-YE52UBF3.cjs');
-
-// src/routes/api/sessions.routes.ts
-var _express = require('express');
-var createSessionApiRouter = (memoryModule) => {
-  const router = _express.Router.call(void 0, );
-  const sessionApiController = new (0, _chunkYE52UBF3cjs.SessionApiController)(memoryModule);
-  router.get("/", sessionApiController.handleUserSessionList);
-  router.get("/:id", sessionApiController.handleSessionHistory);
-  return router;
-};
-
-
-
-exports.createSessionApiRouter = createSessionApiRouter;
-//# sourceMappingURL=chunk-2G7V7SC5.cjs.map

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

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-2G7V7SC5.cjs","../../src/routes/api/sessions.routes.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACA;ACJA,kCAAuB;AAIhB,IAAM,uBAAA,EAAyB,CAAC,YAAA,EAAA,GAAuC;AAC7E,EAAA,MAAM,OAAA,EAAS,6BAAA,CAAO;AACtB,EAAA,MAAM,qBAAA,EAAuB,IAAI,2CAAA,CAAqB,YAAY,CAAA;AAGlE,EAAA,MAAA,CAAO,GAAA,CAAI,GAAA,EAAK,oBAAA,CAAqB,qBAAqB,CAAA;AAC1D,EAAA,MAAA,CAAO,GAAA,CAAI,MAAA,EAAQ,oBAAA,CAAqB,oBAAoB,CAAA;AAE5D,EAAA,OAAO,MAAA;AACR,CAAA;ADAA;AACA;AACE;AACF,wDAAC","file":"/Users/shyun/comcom/ain-agent/ain-adk/dist/cjs/chunk-2G7V7SC5.cjs","sourcesContent":[null,"import { Router } from \"express\";\nimport { SessionApiController } from \"@/controllers/api/session.api.controller.js\";\nimport type { MemoryModule } from \"@/modules/index.js\";\n\nexport const createSessionApiRouter = (memoryModule: MemoryModule): Router => {\n\tconst router = Router();\n\tconst sessionApiController = new SessionApiController(memoryModule);\n\n\t// APIs (prefix: /api/sessions)\n\trouter.get(\"/\", sessionApiController.handleUserSessionList);\n\trouter.get(\"/:id\", sessionApiController.handleSessionHistory);\n\n\treturn router;\n};\n"]}