@ainetwork/adk 0.1.6 → 0.1.7

package/dist/esm/services/index.js

@@ -1,14 +1,18 @@
+import {
+  QueryStreamService
+} from "../chunk-EFMGNCWG.js";
 import {
   A2AService
-} from "../chunk-AQ5YZERE.js";
+} from "../chunk-Y6E73I7I.js";
 import {
   QueryService
-} from "../chunk-N6T2JOGC.js";
-import "../chunk-F727XG4N.js";
+} from "../chunk-MC7CPQ6W.js";
+import "../chunk-VAZWF5IK.js";
 import "../chunk-FI2UQLVC.js";
-import "../chunk-64GEXWSE.js";
+import "../chunk-N7VYM2R5.js";
 export {
   A2AService,
-  QueryService
+  QueryService,
+  QueryStreamService
 };
 //# sourceMappingURL=index.js.map

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

@@ -0,0 +1,74 @@
+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 { MemoryModule } from '../modules/memory/memory.module.js';
+import { StreamEvent } from '../types/stream.js';
+import { ModelModule } from '../modules/models/model.module.js';
+import { AinAgentPrompts } from '../types/agent.js';
+import '@a2a-js/sdk';
+import '../modules/a2a/a2a.tool.js';
+import '@a2a-js/sdk/client';
+import '../types/tool.js';
+import '@modelcontextprotocol/sdk/types.js';
+import '../types/mcp.js';
+import '@modelcontextprotocol/sdk/client/stdio.js';
+import '../modules/mcp/mcp.tool.js';
+import '../modules/memory/base.memory.js';
+import '../modules/models/base.model.js';
+import 'http-status-codes';
+
+/**
+ * Service for processing user queries through the agent's AI pipeline.
+ *
+ * Orchestrates the query processing workflow including intent detection,
+ * model inference, tool execution, and response generation. Manages
+ * conversation context and coordinates between different modules.
+ */
+declare class QueryStreamService {
+    private modelModule;
+    private a2aModule?;
+    private mcpModule?;
+    private memoryModule?;
+    private prompts?;
+    constructor(modelModule: ModelModule, a2aModule?: A2AModule, mcpModule?: MCPModule, memoryModule?: MemoryModule, prompts?: AinAgentPrompts);
+    /**
+     * 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
+     */
+    private intentTriggering;
+    /**
+     * 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 sessionId - Session identifier for context
+     * @param sessionHistory - Previous conversation history
+     * @returns Object containing process steps and final response
+     */
+    intentFulfilling(query: string, sessionId: string, sessionHistory: SessionObject): AsyncGenerator<StreamEvent>;
+    /**
+     * Main entry point for processing user queries.
+     *
+     * Handles the complete query lifecycle:
+     * 1. Loads session history from memory
+     * 2. Detects intent from the query
+     * 3. Fulfills the intent with AI response
+     * 4. Updates conversation history
+     *
+     * @param query - The user's input query
+     * @param sessionId - Unique session identifier
+     * @returns Object containing the AI-generated response
+     */
+    handleQueryStream(query: string, sessionId: string, res: Response, userId?: string): Promise<void>;
+}
+
+export { QueryStreamService };

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

@@ -0,0 +1,9 @@
+import {
+  QueryStreamService
+} from "../chunk-EFMGNCWG.js";
+import "../chunk-FI2UQLVC.js";
+import "../chunk-N7VYM2R5.js";
+export {
+  QueryStreamService
+};
+//# sourceMappingURL=query-stream.service.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

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

@@ -14,6 +14,7 @@ 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';
 
 /**

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

@@ -1,9 +1,9 @@
 import {
   QueryService
-} from "../chunk-N6T2JOGC.js";
-import "../chunk-F727XG4N.js";
+} from "../chunk-MC7CPQ6W.js";
+import "../chunk-VAZWF5IK.js";
 import "../chunk-FI2UQLVC.js";
-import "../chunk-64GEXWSE.js";
+import "../chunk-N7VYM2R5.js";
 export {
   QueryService
 };

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

@@ -54,6 +54,11 @@ type ChatObject = {
         [key: string]: unknown;
     };
 };
+type SessionMetadata = {
+    title?: string;
+    sessionId: string;
+    updatedAt: number;
+};
 /**
  * Represents a conversation session containing multiple chat messages.
  *
@@ -71,6 +76,7 @@ type ChatObject = {
  * ```
  */
 type SessionObject = {
+    title?: string;
     /** Collection of chat messages indexed by unique chat ID */
     chats: {
         [chatId: string]: ChatObject;
@@ -79,6 +85,8 @@ type SessionObject = {
 interface Intent {
     name: string;
     description: string;
+    prompt?: string;
+    llm?: string;
 }
 
-export { type ChatContentObject, type ChatObject, ChatRole, type Intent, MemoryType, type SessionObject };
+export { type ChatContentObject, type ChatObject, ChatRole, type Intent, MemoryType, type SessionMetadata, type SessionObject };

package/dist/esm/types/memory.js

@@ -1,7 +1,7 @@
 import {
   ChatRole,
   MemoryType
-} from "../chunk-F727XG4N.js";
+} from "../chunk-VAZWF5IK.js";
 export {
   ChatRole,
   MemoryType

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

@@ -0,0 +1,47 @@
+import { ToolCall } 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>;
+};
+/**
+ * Tool call delta for streaming tool invocations
+ */
+interface ToolCallDelta {
+    index: number;
+    id?: string;
+    type?: "function";
+    function?: ToolCall;
+}
+/**
+ * Normalized stream chunk interface for all LLM providers.
+ *
+ * This interface provides a consistent structure for stream responses
+ * across different AI model providers (OpenAI, Gemini, Claude, etc.)
+ */
+interface StreamChunk {
+    /** Text content delta from the model */
+    delta?: {
+        role?: string;
+        content?: string;
+        tool_calls?: ToolCallDelta[];
+    };
+    /** Indicates if the stream has finished and why */
+    finish_reason?: "stop" | "length" | "tool_calls" | "content_filter" | null;
+    /** Provider-specific metadata */
+    metadata?: Record<string, unknown>;
+}
+/**
+ * Async iterable stream interface for LLM responses
+ */
+interface LLMStream extends AsyncIterable<StreamChunk> {
+    /** Cancels the stream */
+    cancel?: () => void;
+    /** Stream metadata */
+    metadata?: Record<string, unknown>;
+}
+
+export type { LLMStream, StreamChunk, StreamEvent, ToolCallDelta };

package/dist/esm/types/stream.js

@@ -0,0 +1 @@
+//# sourceMappingURL=stream.js.map

package/dist/esm/types/stream.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/esm/utils/logger.d.ts

@@ -5,6 +5,7 @@ declare const getLogger: (service: string) => winston.Logger;
 declare const loggers: {
     readonly agent: winston.Logger;
     readonly intent: winston.Logger;
+    readonly intentStream: winston.Logger;
     readonly mcp: winston.Logger;
     readonly a2a: winston.Logger;
     readonly model: winston.Logger;

package/dist/esm/utils/logger.js

@@ -2,7 +2,7 @@ import {
   getLogger,
   logger,
   loggers
-} from "../chunk-64GEXWSE.js";
+} from "../chunk-N7VYM2R5.js";
 export {
   getLogger,
   logger,

package/package.json

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

package/dist/esm/chunk-64GEXWSE.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../../src/utils/logger.ts"],"sourcesContent":["import winston from \"winston\";\n\nconst { combine, timestamp, printf, colorize, errors } = winston.format;\n\nconst createLogger = (service: string) => {\n\tconst logger = winston.createLogger({\n\t\tlevel: (process.env.LOG_LEVEL || \"info\").toLowerCase(),\n\t\tformat: combine(\n\t\t\terrors({ stack: true }),\n\t\t\tcolorize(),\n\t\t\ttimestamp({ format: \"HH:mm:ss\" }),\n\t\t\tprintf(({ level, message, timestamp, service, stack, ...meta }) => {\n\t\t\t\tconst metaStr = Object.keys(meta).length\n\t\t\t\t\t? ` | ${JSON.stringify(meta)}`\n\t\t\t\t\t: \"\";\n\t\t\t\tconst errorStack = stack ? `\\n${stack}` : \"\";\n\t\t\t\treturn `${timestamp} [${service}] ${level}: ${message}${metaStr}${errorStack}`;\n\t\t\t}),\n\t\t),\n\t\tdefaultMeta: { service },\n\t\ttransports: [new winston.transports.Console()],\n\t});\n\n\treturn logger;\n};\n\n// Default logger instance\nexport const logger = createLogger(\"ain-adk\");\n\n// Factory function for creating service-specific loggers\nexport const getLogger = (service: string) => createLogger(service);\n\n// Convenience methods for different components\nexport const loggers = {\n\tagent: getLogger(\"AINAgent\"),\n\tintent: getLogger(\"Intent\"),\n\tmcp: getLogger(\"MCPModule\"),\n\ta2a: getLogger(\"A2AModule\"),\n\tmodel: getLogger(\"Model\"),\n\tserver: getLogger(\"A2AServer\"),\n\tfol: getLogger(\"FOL\"),\n} as const;\n"],"mappings":";AAAA,OAAO,aAAa;AAEpB,IAAM,EAAE,SAAS,WAAW,QAAQ,UAAU,OAAO,IAAI,QAAQ;AAEjE,IAAM,eAAe,CAAC,YAAoB;AACzC,QAAMA,UAAS,QAAQ,aAAa;AAAA,IACnC,QAAQ,QAAQ,IAAI,aAAa,QAAQ,YAAY;AAAA,IACrD,QAAQ;AAAA,MACP,OAAO,EAAE,OAAO,KAAK,CAAC;AAAA,MACtB,SAAS;AAAA,MACT,UAAU,EAAE,QAAQ,WAAW,CAAC;AAAA,MAChC,OAAO,CAAC,EAAE,OAAO,SAAS,WAAAC,YAAW,SAAAC,UAAS,OAAO,GAAG,KAAK,MAAM;AAClE,cAAM,UAAU,OAAO,KAAK,IAAI,EAAE,SAC/B,MAAM,KAAK,UAAU,IAAI,CAAC,KAC1B;AACH,cAAM,aAAa,QAAQ;AAAA,EAAK,KAAK,KAAK;AAC1C,eAAO,GAAGD,UAAS,KAAKC,QAAO,KAAK,KAAK,KAAK,OAAO,GAAG,OAAO,GAAG,UAAU;AAAA,MAC7E,CAAC;AAAA,IACF;AAAA,IACA,aAAa,EAAE,QAAQ;AAAA,IACvB,YAAY,CAAC,IAAI,QAAQ,WAAW,QAAQ,CAAC;AAAA,EAC9C,CAAC;AAED,SAAOF;AACR;AAGO,IAAM,SAAS,aAAa,SAAS;AAGrC,IAAM,YAAY,CAAC,YAAoB,aAAa,OAAO;AAG3D,IAAM,UAAU;AAAA,EACtB,OAAO,UAAU,UAAU;AAAA,EAC3B,QAAQ,UAAU,QAAQ;AAAA,EAC1B,KAAK,UAAU,WAAW;AAAA,EAC1B,KAAK,UAAU,WAAW;AAAA,EAC1B,OAAO,UAAU,OAAO;AAAA,EACxB,QAAQ,UAAU,WAAW;AAAA,EAC7B,KAAK,UAAU,KAAK;AACrB;","names":["logger","timestamp","service"]}

package/dist/esm/chunk-GZFRRFOE.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../../src/modules/models/base.model.ts"],"sourcesContent":["import type { SessionObject } from \"@/types/memory.js\";\nimport type { FetchResponse, IAgentTool } from \"@/types/tool.js\";\n\n/**\n * Abstract base class for AI model implementations.\n *\n * Provides a common interface for different AI model providers (OpenAI, Gemini, etc.)\n * to integrate with the AIN-ADK framework. Each model implementation must handle\n * message formatting, tool conversion, and API communication.\n *\n * @typeParam MessageType - The message format used by the specific model API\n * @typeParam FunctionType - The function/tool format used by the specific model API\n */\nexport abstract class BaseModel<MessageType, FunctionType> {\n\t/**\n\t * Generates an array of messages from session history and current query.\n\t *\n\t * @param sessionHistory - Previous conversation history\n\t * @param query - Current user query\n\t * @param systemPrompt - Optional system prompt to set context\n\t * @returns Array of messages formatted for the specific model API\n\t */\n\tabstract generateMessages(params: {\n\t\tquery: string;\n\t\tsessionHistory?: SessionObject;\n\t\tsystemPrompt?: string;\n\t}): MessageType[];\n\n\t/**\n\t * Appends a new message to the existing message array.\n\t *\n\t * @param messages - Existing message array to expand\n\t * @param message - New message content to append\n\t */\n\tabstract appendMessages(messages: MessageType[], message: string): void;\n\n\t/**\n\t * Converts protocol-agnostic tools to model-specific function format.\n\t *\n\t * @param tools - Array of agent tools from MCP or A2A sources\n\t * @returns Array of functions in the format required by the model API\n\t */\n\tabstract convertToolsToFunctions(tools: IAgentTool[]): FunctionType[];\n\n\t/**\n\t * Fetches a response from the model API without tool support.\n\t *\n\t * @param messages - Array of messages to send to the model\n\t * @returns Promise resolving to the model's response\n\t */\n\tabstract fetch(messages: MessageType[]): Promise<FetchResponse>;\n\n\t/**\n\t * Fetches a response from the model API with tool/function support.\n\t *\n\t * @param messages - Array of messages to send to the model\n\t * @param functions - Array of available functions/tools the model can call\n\t * @returns Promise resolving to the model's response, possibly including tool calls\n\t */\n\tabstract fetchWithContextMessage(\n\t\tmessages: MessageType[],\n\t\tfunctions: FunctionType[],\n\t): Promise<FetchResponse>;\n}\n"],"mappings":";AAaO,IAAe,YAAf,MAAoD;AAkD3D;","names":[]}

package/dist/esm/chunk-NH7BYWQV.js

@@ -1,27 +0,0 @@
-import {
-  QueryController
-} from "./chunk-WIJDRP2A.js";
-import {
-  QueryService
-} from "./chunk-N6T2JOGC.js";
-
-// src/routes/query.routes.ts
-import { Router } from "express";
-var createQueryRouter = (agent) => {
-  const router = Router();
-  const queryService = new QueryService(
-    agent.modelModule,
-    agent.a2aModule,
-    agent.mcpModule,
-    agent.memoryModule,
-    agent.manifest.prompts
-  );
-  const queryController = new QueryController(queryService);
-  router.post("/query", queryController.handleQueryRequest);
-  return router;
-};
-
-export {
-  createQueryRouter
-};
-//# sourceMappingURL=chunk-NH7BYWQV.js.map

package/dist/esm/chunk-NH7BYWQV.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../../src/routes/query.routes.ts"],"sourcesContent":["import { Router } from \"express\";\nimport { QueryController } from \"@/controllers/query.controller.js\";\nimport type { AINAgent } from \"@/index.js\";\nimport { QueryService } from \"@/services/query.service.js\";\n\nexport const createQueryRouter = (agent: AINAgent): 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\tconst queryController = new QueryController(queryService);\n\trouter.post(\"/query\", queryController.handleQueryRequest);\n\n\treturn router;\n};\n"],"mappings":";;;;;;;;AAAA,SAAS,cAAc;AAKhB,IAAM,oBAAoB,CAAC,UAA4B;AAC7D,QAAM,SAAS,OAAO;AAEtB,QAAM,eAAe,IAAI;AAAA,IACxB,MAAM;AAAA,IACN,MAAM;AAAA,IACN,MAAM;AAAA,IACN,MAAM;AAAA,IACN,MAAM,SAAS;AAAA,EAChB;AACA,QAAM,kBAAkB,IAAI,gBAAgB,YAAY;AACxD,SAAO,KAAK,UAAU,gBAAgB,kBAAkB;AAExD,SAAO;AACR;","names":[]}

package/dist/esm/chunk-WIJDRP2A.js

@@ -1,26 +0,0 @@
-// src/controllers/query.controller.ts
-var QueryController = class {
-  queryService;
-  constructor(queryService) {
-    this.queryService = queryService;
-  }
-  handleQueryRequest = async (req, res, next) => {
-    try {
-      const { message, sessionId } = req.body;
-      const userId = res.locals.userId;
-      const result = await this.queryService.handleQuery(
-        message,
-        sessionId,
-        userId
-      );
-      res.status(200).json(result);
-    } catch (error) {
-      next(error);
-    }
-  };
-};
-
-export {
-  QueryController
-};
-//# sourceMappingURL=chunk-WIJDRP2A.js.map

package/dist/esm/chunk-WIJDRP2A.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../../src/controllers/query.controller.ts"],"sourcesContent":["import type { NextFunction, Request, Response } from \"express\";\nimport type { QueryService } from \"@/services\";\n\nexport class QueryController {\n\tprivate queryService;\n\n\tconstructor(queryService: QueryService) {\n\t\tthis.queryService = queryService;\n\t}\n\n\tpublic handleQueryRequest = async (\n\t\treq: Request,\n\t\tres: Response,\n\t\tnext: NextFunction,\n\t) => {\n\t\ttry {\n\t\t\tconst { message, sessionId } = req.body;\n\t\t\tconst userId = res.locals.userId;\n\n\t\t\tconst result = await this.queryService.handleQuery(\n\t\t\t\tmessage,\n\t\t\t\tsessionId,\n\t\t\t\tuserId,\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"],"mappings":";AAGO,IAAM,kBAAN,MAAsB;AAAA,EACpB;AAAA,EAER,YAAY,cAA4B;AACvC,SAAK,eAAe;AAAA,EACrB;AAAA,EAEO,qBAAqB,OAC3B,KACA,KACA,SACI;AACJ,QAAI;AACH,YAAM,EAAE,SAAS,UAAU,IAAI,IAAI;AACnC,YAAM,SAAS,IAAI,OAAO;AAE1B,YAAM,SAAS,MAAM,KAAK,aAAa;AAAA,QACtC;AAAA,QACA;AAAA,QACA;AAAA,MACD;AAEA,UAAI,OAAO,GAAG,EAAE,KAAK,MAAM;AAAA,IAC5B,SAAS,OAAO;AACf,WAAK,KAAK;AAAA,IACX;AAAA,EACD;AACD;","names":[]}