@ainetwork/adk 0.0.1

package/dist/esm/middlewares/error.middleware.js

@@ -0,0 +1,13 @@
+import { logger } from "../utils/logger.js";
+export const errorMiddleware = (error, req, res, next) => {
+    try {
+        const status = error.status || 500;
+        const message = error.message || "Something went wrong";
+        logger.error(`[${req.method}] ${req.path} >> StatusCode:: ${status}, Message:: ${message}`);
+        res.status(status).json({ message });
+    }
+    catch (error) {
+        next(error);
+    }
+};
+//# sourceMappingURL=error.middleware.js.map

package/dist/esm/middlewares/error.middleware.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"error.middleware.js","sourceRoot":"","sources":["../../../src/middlewares/error.middleware.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAM3C,MAAM,CAAC,MAAM,eAAe,GAAG,CAC9B,KAAoB,EACpB,GAAY,EACZ,GAAa,EACb,IAAkB,EACX,EAAE;IACT,IAAI,CAAC;QACJ,MAAM,MAAM,GAAW,KAAK,CAAC,MAAM,IAAI,GAAG,CAAC;QAC3C,MAAM,OAAO,GAAW,KAAK,CAAC,OAAO,IAAI,sBAAsB,CAAC;QAEhE,MAAM,CAAC,KAAK,CACX,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,IAAI,oBAAoB,MAAM,eAAe,OAAO,EAAE,CAC7E,CAAC;QACF,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;IACtC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,IAAI,CAAC,KAAK,CAAC,CAAC;IACb,CAAC;AACF,CAAC,CAAC"}

package/dist/esm/models/BaseModel.d.ts

@@ -0,0 +1,25 @@
+import type { SessionObject } from "../session/BaseSession.js";
+import type { AgentTool } from "../intent/modules/common/tool.js";
+export interface ToolCall {
+    name: string;
+    arguments?: Record<string, unknown>;
+}
+export interface FetchResponse {
+    content?: string;
+    toolCalls?: ToolCall[];
+}
+export interface IBaseModel {
+    generateMessages<M>(sessionHistory: SessionObject, query: string, systemPrompt?: string): M[];
+    expandMessages<M>(messages: M[], message: string): void;
+    convertToolsToFunctions<F>(tools: AgentTool[]): F[];
+    fetch<M>(messages: M[]): Promise<FetchResponse>;
+    fetchWithContextMessage<M, F>(messages: M[], functions: F[]): Promise<FetchResponse>;
+}
+export declare abstract class BaseModel<MessageType, FunctionType> {
+    abstract generateMessages(sessionHistory: SessionObject, query: string, systemPrompt?: string): MessageType[];
+    abstract expandMessages(messages: MessageType[], message: string): void;
+    abstract convertToolsToFunctions(tools: AgentTool[]): FunctionType[];
+    abstract fetch(messages: MessageType[]): Promise<FetchResponse>;
+    abstract fetchWithContextMessage(messages: MessageType[], functions: FunctionType[]): Promise<FetchResponse>;
+}
+//# sourceMappingURL=BaseModel.d.ts.map

package/dist/esm/models/BaseModel.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"BaseModel.d.ts","sourceRoot":"","sources":["../../../src/models/BaseModel.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kCAAkC,CAAC;AAElE,MAAM,WAAW,QAAQ;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AACD,MAAM,WAAW,aAAa;IAC7B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,QAAQ,EAAE,CAAC;CACvB;AAED,MAAM,WAAW,UAAU;IAC1B,gBAAgB,CAAC,CAAC,EACjB,cAAc,EAAE,aAAa,EAC7B,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GACnB,CAAC,EAAE,CAAC;IACP,cAAc,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACxD,uBAAuB,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,CAAC,EAAE,CAAC;IACpD,KAAK,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC;IAChD,uBAAuB,CAAC,CAAC,EAAE,CAAC,EAC3B,QAAQ,EAAE,CAAC,EAAE,EACb,SAAS,EAAE,CAAC,EAAE,GACZ,OAAO,CAAC,aAAa,CAAC,CAAC;CAC1B;AAED,8BAAsB,SAAS,CAAC,WAAW,EAAE,YAAY;IACxD,QAAQ,CAAC,gBAAgB,CACxB,cAAc,EAAE,aAAa,EAC7B,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GACnB,WAAW,EAAE;IAEhB,QAAQ,CAAC,cAAc,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAEvE,QAAQ,CAAC,uBAAuB,CAAC,KAAK,EAAE,SAAS,EAAE,GAAG,YAAY,EAAE;IAEpE,QAAQ,CAAC,KAAK,CAAC,QAAQ,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,aAAa,CAAC;IAE/D,QAAQ,CAAC,uBAAuB,CAC/B,QAAQ,EAAE,WAAW,EAAE,EACvB,SAAS,EAAE,YAAY,EAAE,GACvB,OAAO,CAAC,aAAa,CAAC;CACzB"}

package/dist/esm/models/BaseModel.js

@@ -0,0 +1,3 @@
+export class BaseModel {
+}
+//# sourceMappingURL=BaseModel.js.map

package/dist/esm/models/BaseModel.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"BaseModel.js","sourceRoot":"","sources":["../../../src/models/BaseModel.ts"],"names":[],"mappings":"AA2BA,MAAM,OAAgB,SAAS;CAiB9B"}

package/dist/esm/models/base.d.ts

@@ -0,0 +1,24 @@
+import type { AgentTool } from "../intent/modules/common/tool.js";
+export interface ToolCall {
+    name: string;
+    arguments?: Record<string, unknown>;
+}
+export interface FetchResponse {
+    content?: string;
+    toolCalls?: ToolCall[];
+}
+export interface IBaseModel {
+    generateMessages<M>(queries: string[], systemPrompt?: string): M[];
+    expandMessages<M>(messages: M[], message: string): void;
+    convertToolsToFunctions<F>(tools: AgentTool[]): F[];
+    fetch<M>(messages: M[]): Promise<FetchResponse>;
+    fetchWithContextMessage<M, F>(messages: M[], functions: F[]): Promise<FetchResponse>;
+}
+export declare abstract class BaseModel<MessageType, FunctionType> {
+    abstract generateMessages(queries: string[], systemPrompt?: string): MessageType[];
+    abstract expandMessages(messages: MessageType[], message: string): void;
+    abstract convertToolsToFunctions(tools: AgentTool[]): FunctionType[];
+    abstract fetch(messages: MessageType[]): Promise<FetchResponse>;
+    abstract fetchWithContextMessage(messages: MessageType[], functions: FunctionType[]): Promise<FetchResponse>;
+}
+//# sourceMappingURL=base.d.ts.map

package/dist/esm/models/base.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../src/models/base.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kCAAkC,CAAC;AAElE,MAAM,WAAW,QAAQ;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AACD,MAAM,WAAW,aAAa;IAC7B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,QAAQ,EAAE,CAAC;CACvB;AAED,MAAM,WAAW,UAAU;IAC1B,gBAAgB,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,YAAY,CAAC,EAAE,MAAM,GAAG,CAAC,EAAE,CAAC;IACnE,cAAc,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACxD,uBAAuB,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,CAAC,EAAE,CAAC;IACpD,KAAK,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC;IAChD,uBAAuB,CAAC,CAAC,EAAE,CAAC,EAC3B,QAAQ,EAAE,CAAC,EAAE,EACb,SAAS,EAAE,CAAC,EAAE,GACZ,OAAO,CAAC,aAAa,CAAC,CAAC;CAC1B;AAED,8BAAsB,SAAS,CAAC,WAAW,EAAE,YAAY;IACxD,QAAQ,CAAC,gBAAgB,CACxB,OAAO,EAAE,MAAM,EAAE,EACjB,YAAY,CAAC,EAAE,MAAM,GACnB,WAAW,EAAE;IAEhB,QAAQ,CAAC,cAAc,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAEvE,QAAQ,CAAC,uBAAuB,CAAC,KAAK,EAAE,SAAS,EAAE,GAAG,YAAY,EAAE;IAEpE,QAAQ,CAAC,KAAK,CAAC,QAAQ,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,aAAa,CAAC;IAE/D,QAAQ,CAAC,uBAAuB,CAC/B,QAAQ,EAAE,WAAW,EAAE,EACvB,SAAS,EAAE,YAAY,EAAE,GACvB,OAAO,CAAC,aAAa,CAAC;CACzB"}

package/dist/esm/models/base.js

@@ -0,0 +1,3 @@
+export class BaseModel {
+}
+//# sourceMappingURL=base.js.map

package/dist/esm/models/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../src/models/base.ts"],"names":[],"mappings":"AAsBA,MAAM,OAAgB,SAAS;CAgB9B"}

package/dist/esm/models/gemini.d.ts

@@ -0,0 +1,16 @@
+import { type Content, type FunctionDeclaration } from "@google/genai";
+import type { AgentTool } from "../intent/modules/common/tool.js";
+import { type SessionObject } from "../session/BaseSession.js";
+import { BaseModel, type FetchResponse } from "./BaseModel.js";
+export default class GeminiModel extends BaseModel<Content, FunctionDeclaration> {
+    private client;
+    private modelName;
+    constructor(apiKey: string, modelName: string);
+    private getMessageRole;
+    generateMessages(sessionHistory: SessionObject, query: string, systemPrompt?: string): Content[];
+    expandMessages(messages: Content[], message: string): void;
+    fetch(messages: Content[]): Promise<FetchResponse>;
+    fetchWithContextMessage(messages: Content[], functions: FunctionDeclaration[]): Promise<FetchResponse>;
+    convertToolsToFunctions(tools: AgentTool[]): FunctionDeclaration[];
+}
+//# sourceMappingURL=gemini.d.ts.map

package/dist/esm/models/gemini.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../../src/models/gemini.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,KAAK,OAAO,EAEZ,KAAK,mBAAmB,EAExB,MAAM,eAAe,CAAC;AAEvB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iCAAiC,CAAC;AAGjE,OAAO,EAAY,KAAK,aAAa,EAAE,MAAM,0BAA0B,CAAC;AACxE,OAAO,EAAE,SAAS,EAAE,KAAK,aAAa,EAAiB,MAAM,gBAAgB,CAAC;AAE9E,MAAM,CAAC,OAAO,OAAO,WAAY,SAAQ,SAAS,CACjD,OAAO,EACP,mBAAmB,CACnB;IACA,OAAO,CAAC,MAAM,CAAc;IAC5B,OAAO,CAAC,SAAS,CAAS;gBAEd,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM;IAM7C,OAAO,CAAC,cAAc;IAYtB,gBAAgB,CACf,cAAc,EAAE,aAAa,EAC7B,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GACnB,OAAO,EAAE;IAkBZ,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAOpD,KAAK,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,aAAa,CAAC;IASlD,uBAAuB,CAC5B,QAAQ,EAAE,OAAO,EAAE,EACnB,SAAS,EAAE,mBAAmB,EAAE,GAC9B,OAAO,CAAC,aAAa,CAAC;IAiCzB,uBAAuB,CAAC,KAAK,EAAE,SAAS,EAAE,GAAG,mBAAmB,EAAE;CAwBlE"}

package/dist/esm/models/gemini.js

@@ -0,0 +1,104 @@
+import { GoogleGenAI, } from "@google/genai";
+import { PROTOCOL_TYPE } from "../intent/modules/common/types.js";
+import { ChatRole } from "../session/BaseSession.js";
+import { BaseModel } from "./BaseModel.js";
+export default class GeminiModel extends BaseModel {
+    constructor(apiKey, modelName) {
+        super();
+        this.client = new GoogleGenAI({ apiKey });
+        this.modelName = modelName;
+    }
+    getMessageRole(role) {
+        switch (role) {
+            case ChatRole.USER:
+                return "user";
+            case ChatRole.MODEL:
+            case ChatRole.SYSTEM:
+                return "model";
+            default:
+                return "model"; /*FIXME*/
+        }
+    }
+    generateMessages(sessionHistory, query, systemPrompt) {
+        const messages = !systemPrompt
+            ? []
+            : [{ role: "model", parts: [{ text: systemPrompt.trim() }] }];
+        const sessionContent = Object.keys(sessionHistory).map((messageId) => {
+            const message = sessionHistory[messageId];
+            // TODO: check message.content.type
+            return {
+                role: this.getMessageRole(message.role),
+                parts: [{ text: message.content.parts[0] }],
+            };
+        });
+        const userContent = { role: "user", parts: [{ text: query }] };
+        return messages.concat(sessionContent).concat(userContent);
+    }
+    expandMessages(messages, message) {
+        messages.push({
+            role: "user",
+            parts: [{ text: message }],
+        });
+    }
+    async fetch(messages) {
+        const response = await this.client.models.generateContent({
+            model: this.modelName,
+            contents: messages,
+        });
+        return { content: response.text };
+    }
+    async fetchWithContextMessage(messages, functions) {
+        if (functions.length > 0) {
+            const response = await this.client.models.generateContent({
+                model: this.modelName,
+                contents: messages,
+                config: {
+                    tools: [{ functionDeclarations: functions }],
+                },
+            });
+            const { text, functionCalls } = response;
+            const hasName = (value) => {
+                return value.name !== undefined;
+            };
+            const toolCalls = functionCalls
+                ?.filter(hasName)
+                .map((value) => {
+                return {
+                    name: value.name,
+                    arguments: value.args,
+                };
+            });
+            return {
+                content: text,
+                toolCalls,
+            };
+        }
+        return await this.fetch(messages);
+    }
+    convertToolsToFunctions(tools) {
+        const functions = [];
+        for (const tool of tools) {
+            if (!tool.enabled) {
+                continue;
+            }
+            if (tool.protocol === PROTOCOL_TYPE.MCP) {
+                const { mcpTool, id } = tool;
+                functions.push({
+                    name: id,
+                    description: mcpTool.description,
+                    parametersJsonSchema: mcpTool.inputSchema,
+                });
+            }
+            else {
+                // PROTOCOL_TYPE.A2A
+                const { id, card } = tool;
+                functions.push({
+                    name: id,
+                    description: card.description,
+                });
+            }
+        }
+        return functions;
+    }
+}
+//# sourceMappingURL=gemini.js.map

package/dist/esm/models/gemini.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../../src/models/gemini.ts"],"names":[],"mappings":"AAAA,OAAO,EAIN,WAAW,GACX,MAAM,eAAe,CAAC;AAGvB,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAC;AAEjE,OAAO,EAAE,QAAQ,EAAsB,MAAM,0BAA0B,CAAC;AACxE,OAAO,EAAE,SAAS,EAAqC,MAAM,gBAAgB,CAAC;AAE9E,MAAM,CAAC,OAAO,OAAO,WAAY,SAAQ,SAGxC;IAIA,YAAY,MAAc,EAAE,SAAiB;QAC5C,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;QAC1C,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC5B,CAAC;IAEO,cAAc,CAAC,IAAc;QACpC,QAAQ,IAAI,EAAE,CAAC;YACd,KAAK,QAAQ,CAAC,IAAI;gBACjB,OAAO,MAAM,CAAC;YACf,KAAK,QAAQ,CAAC,KAAK,CAAC;YACpB,KAAK,QAAQ,CAAC,MAAM;gBACnB,OAAO,OAAO,CAAC;YAChB;gBACC,OAAO,OAAO,CAAC,CAAC,SAAS;QAC3B,CAAC;IACF,CAAC;IAED,gBAAgB,CACf,cAA6B,EAC7B,KAAa,EACb,YAAqB;QAErB,MAAM,QAAQ,GAAc,CAAC,YAAY;YACxC,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;QAC/D,MAAM,cAAc,GAAc,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,GAAG,CAChE,CAAC,SAAiB,EAAE,EAAE;YACrB,MAAM,OAAO,GAAG,cAAc,CAAC,SAAS,CAAC,CAAC;YAC1C,mCAAmC;YACnC,OAAO;gBACN,IAAI,EAAE,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,IAAI,CAAC;gBACvC,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;aAC3C,CAAC;QACH,CAAC,CACD,CAAC;QACF,MAAM,WAAW,GAAY,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;QACxE,OAAO,QAAQ,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;IAC5D,CAAC;IAED,cAAc,CAAC,QAAmB,EAAE,OAAe;QAClD,QAAQ,CAAC,IAAI,CAAC;YACb,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;SAC1B,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,QAAmB;QAC9B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC;YACzD,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,QAAQ,EAAE,QAAQ;SAClB,CAAC,CAAC;QAEH,OAAO,EAAE,OAAO,EAAE,QAAQ,CAAC,IAAI,EAAE,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,uBAAuB,CAC5B,QAAmB,EACnB,SAAgC;QAEhC,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC;gBACzD,KAAK,EAAE,IAAI,CAAC,SAAS;gBACrB,QAAQ,EAAE,QAAQ;gBAClB,MAAM,EAAE;oBACP,KAAK,EAAE,CAAC,EAAE,oBAAoB,EAAE,SAAS,EAAE,CAAC;iBAC5C;aACD,CAAC,CAAC;YAEH,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,QAAQ,CAAC;YACzC,MAAM,OAAO,GAAG,CACf,KAAmB,EACwB,EAAE;gBAC7C,OAAO,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC;YACjC,CAAC,CAAC;YACF,MAAM,SAAS,GAA2B,aAAa;gBACtD,EAAE,MAAM,CAAC,OAAO,CAAC;iBAChB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;gBACd,OAAO;oBACN,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,SAAS,EAAE,KAAK,CAAC,IAAI;iBACrB,CAAC;YACH,CAAC,CAAC,CAAC;YAEJ,OAAO;gBACN,OAAO,EAAE,IAAI;gBACb,SAAS;aACT,CAAC;QACH,CAAC;QACD,OAAO,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;IAED,uBAAuB,CAAC,KAAkB;QACzC,MAAM,SAAS,GAA0B,EAAE,CAAC;QAC5C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YAC1B,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACnB,SAAS;YACV,CAAC;YACD,IAAI,IAAI,CAAC,QAAQ,KAAK,aAAa,CAAC,GAAG,EAAE,CAAC;gBACzC,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,IAAe,CAAC;gBACxC,SAAS,CAAC,IAAI,CAAC;oBACd,IAAI,EAAE,EAAE;oBACR,WAAW,EAAE,OAAO,CAAC,WAAW;oBAChC,oBAAoB,EAAE,OAAO,CAAC,WAAW;iBACzC,CAAC,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACP,oBAAoB;gBACpB,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,GAAG,IAAe,CAAC;gBACrC,SAAS,CAAC,IAAI,CAAC;oBACd,IAAI,EAAE,EAAE;oBACR,WAAW,EAAE,IAAI,CAAC,WAAW;iBAC7B,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;QACD,OAAO,SAAS,CAAC;IAClB,CAAC;CACD"}

package/dist/esm/models/openai.d.ts

@@ -0,0 +1,16 @@
+import type { ChatCompletionMessageParam as CCMessageParam, ChatCompletionTool } from "openai/resources";
+import { type SessionObject } from "../session/BaseSession.js";
+import type { AgentTool } from "../intent/modules/common/tool.js";
+import { BaseModel, type FetchResponse } from "./BaseModel.js";
+export default class AzureOpenAI extends BaseModel<CCMessageParam, ChatCompletionTool> {
+    private client;
+    private modelName;
+    constructor(baseUrl: string, apiKey: string, apiVersion: string, modelName: string);
+    private getMessageRole;
+    generateMessages(sessionHistory: SessionObject, query: string, systemPrompt?: string): CCMessageParam[];
+    expandMessages(messages: CCMessageParam[], message: string): void;
+    fetch(messages: CCMessageParam[]): Promise<FetchResponse>;
+    fetchWithContextMessage(messages: CCMessageParam[], functions: ChatCompletionTool[]): Promise<FetchResponse>;
+    convertToolsToFunctions(tools: AgentTool[]): ChatCompletionTool[];
+}
+//# sourceMappingURL=openai.d.ts.map

package/dist/esm/models/openai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../src/models/openai.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACX,0BAA0B,IAAI,cAAc,EAE5C,kBAAkB,EAClB,MAAM,kBAAkB,CAAC;AAI1B,OAAO,EAAY,KAAK,aAAa,EAAE,MAAM,0BAA0B,CAAC;AACxE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kCAAkC,CAAC;AAClE,OAAO,EAAE,SAAS,EAAE,KAAK,aAAa,EAAiB,MAAM,gBAAgB,CAAC;AAE9E,MAAM,CAAC,OAAO,OAAO,WAAY,SAAQ,SAAS,CACjD,cAAc,EACd,kBAAkB,CAClB;IACA,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,SAAS,CAAS;gBAGzB,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,MAAM;IAWlB,OAAO,CAAC,cAAc;IAYtB,gBAAgB,CACf,cAAc,EAAE,aAAa,EAC7B,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GACnB,cAAc,EAAE;IAiBnB,cAAc,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAO3D,KAAK,CAAC,QAAQ,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC,aAAa,CAAC;IAWzD,uBAAuB,CAC5B,QAAQ,EAAE,cAAc,EAAE,EAC1B,SAAS,EAAE,kBAAkB,EAAE,GAC7B,OAAO,CAAC,aAAa,CAAC;IA6BzB,uBAAuB,CAAC,KAAK,EAAE,SAAS,EAAE,GAAG,kBAAkB,EAAE;CA8BjE"}

package/dist/esm/models/openai.js

@@ -0,0 +1,110 @@
+import { AzureOpenAI as AzureOpenAIClient } from "openai";
+import { PROTOCOL_TYPE } from "../intent/modules/common/types.js";
+import { ChatRole } from "../session/BaseSession.js";
+import { BaseModel } from "./BaseModel.js";
+export default class AzureOpenAI extends BaseModel {
+    constructor(baseUrl, apiKey, apiVersion, modelName) {
+        super();
+        this.client = new AzureOpenAIClient({
+            baseURL: baseUrl,
+            apiKey: apiKey,
+            apiVersion: apiVersion,
+        });
+        this.modelName = modelName;
+    }
+    getMessageRole(role) {
+        switch (role) {
+            case ChatRole.USER:
+                return "user";
+            case ChatRole.MODEL:
+            case ChatRole.SYSTEM:
+                return "system";
+            default:
+                return "system"; /*FIXME*/
+        }
+    }
+    generateMessages(sessionHistory, query, systemPrompt) {
+        const messages = !systemPrompt
+            ? []
+            : [{ role: "system", content: systemPrompt.trim() }];
+        const sessionContent = Object.keys(sessionHistory).map((messageId) => {
+            const message = sessionHistory[messageId];
+            return {
+                role: this.getMessageRole(message.role),
+                content: message.content.parts[0],
+            };
+        });
+        const userContent = { role: "user", content: query };
+        return messages.concat(sessionContent).concat(userContent);
+    }
+    expandMessages(messages, message) {
+        messages.push({
+            role: "user",
+            content: message,
+        });
+    }
+    async fetch(messages) {
+        const response = await this.client.chat.completions.create({
+            model: this.modelName,
+            messages,
+        });
+        return {
+            content: response.choices[0].message.content || undefined,
+        };
+    }
+    async fetchWithContextMessage(messages, functions) {
+        if (functions.length > 0) {
+            const response = await this.client.chat.completions.create({
+                model: this.modelName,
+                messages,
+                tools: functions,
+                tool_choice: "auto",
+            });
+            const { content, tool_calls } = response.choices[0].message;
+            const toolCalls = tool_calls?.map((value) => {
+                return {
+                    name: value.function.name,
+                    // FIXME: value.function.arguments could not be a valid JSON
+                    arguments: JSON.parse(value.function.arguments),
+                };
+            });
+            return {
+                content: content || undefined,
+                toolCalls,
+            };
+        }
+        return await this.fetch(messages);
+    }
+    convertToolsToFunctions(tools) {
+        const functions = [];
+        for (const tool of tools) {
+            if (!tool.enabled) {
+                continue;
+            }
+            if (tool.protocol === PROTOCOL_TYPE.MCP) {
+                const { mcpTool, id } = tool;
+                functions.push({
+                    type: "function",
+                    function: {
+                        name: id,
+                        description: mcpTool.description,
+                        parameters: mcpTool.inputSchema,
+                    },
+                });
+            }
+            else {
+                // PROTOCOL_TYPE.A2A
+                const { id, card } = tool;
+                functions.push({
+                    type: "function",
+                    function: {
+                        name: id,
+                        description: card.description,
+                    },
+                });
+            }
+        }
+        return functions;
+    }
+}
+//# sourceMappingURL=openai.js.map

package/dist/esm/models/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../src/models/openai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,IAAI,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAO1D,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAC;AAEjE,OAAO,EAAE,QAAQ,EAAsB,MAAM,0BAA0B,CAAC;AAExE,OAAO,EAAE,SAAS,EAAqC,MAAM,gBAAgB,CAAC;AAE9E,MAAM,CAAC,OAAO,OAAO,WAAY,SAAQ,SAGxC;IAIA,YACC,OAAe,EACf,MAAc,EACd,UAAkB,EAClB,SAAiB;QAEjB,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,IAAI,iBAAiB,CAAC;YACnC,OAAO,EAAE,OAAO;YAChB,MAAM,EAAE,MAAM;YACd,UAAU,EAAE,UAAU;SACtB,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC5B,CAAC;IAEO,cAAc,CAAC,IAAc;QACpC,QAAQ,IAAI,EAAE,CAAC;YACd,KAAK,QAAQ,CAAC,IAAI;gBACjB,OAAO,MAAM,CAAC;YACf,KAAK,QAAQ,CAAC,KAAK,CAAC;YACpB,KAAK,QAAQ,CAAC,MAAM;gBACnB,OAAO,QAAQ,CAAC;YACjB;gBACC,OAAO,QAAQ,CAAC,CAAC,SAAS;QAC5B,CAAC;IACF,CAAC;IAED,gBAAgB,CACf,cAA6B,EAC7B,KAAa,EACb,YAAqB;QAErB,MAAM,QAAQ,GAAqB,CAAC,YAAY;YAC/C,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACtD,MAAM,cAAc,GAAqB,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,GAAG,CACvE,CAAC,SAAiB,EAAE,EAAE;YACrB,MAAM,OAAO,GAAG,cAAc,CAAC,SAAS,CAAC,CAAC;YAC1C,OAAO;gBACN,IAAI,EAAE,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,IAAI,CAAC;gBACvC,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;aACjC,CAAC;QACH,CAAC,CACD,CAAC;QACF,MAAM,WAAW,GAAmB,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QACrE,OAAO,QAAQ,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;IAC5D,CAAC;IAED,cAAc,CAAC,QAA0B,EAAE,OAAe;QACzD,QAAQ,CAAC,IAAI,CAAC;YACb,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,OAAO;SAChB,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,QAA0B;QACrC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;YAC1D,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,QAAQ;SACR,CAAC,CAAC;QAEH,OAAO;YACN,OAAO,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,IAAI,SAAS;SACzD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,uBAAuB,CAC5B,QAA0B,EAC1B,SAA+B;QAE/B,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBAC1D,KAAK,EAAE,IAAI,CAAC,SAAS;gBACrB,QAAQ;gBACR,KAAK,EAAE,SAAS;gBAChB,WAAW,EAAE,MAAM;aACnB,CAAC,CAAC;YAEH,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAE5D,MAAM,SAAS,GAA2B,UAAU,EAAE,GAAG,CACxD,CAAC,KAAoC,EAAE,EAAE;gBACxC,OAAO;oBACN,IAAI,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI;oBACzB,4DAA4D;oBAC5D,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC;iBAC/C,CAAC;YACH,CAAC,CACD,CAAC;YAEF,OAAO;gBACN,OAAO,EAAE,OAAO,IAAI,SAAS;gBAC7B,SAAS;aACT,CAAC;QACH,CAAC;QACD,OAAO,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;IAED,uBAAuB,CAAC,KAAkB;QACzC,MAAM,SAAS,GAAyB,EAAE,CAAC;QAC3C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YAC1B,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACnB,SAAS;YACV,CAAC;YACD,IAAI,IAAI,CAAC,QAAQ,KAAK,aAAa,CAAC,GAAG,EAAE,CAAC;gBACzC,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,IAAe,CAAC;gBACxC,SAAS,CAAC,IAAI,CAAC;oBACd,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE;wBACT,IAAI,EAAE,EAAE;wBACR,WAAW,EAAE,OAAO,CAAC,WAAW;wBAChC,UAAU,EAAE,OAAO,CAAC,WAAW;qBAC/B;iBACD,CAAC,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACP,oBAAoB;gBACpB,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,GAAG,IAAe,CAAC;gBACrC,SAAS,CAAC,IAAI,CAAC;oBACd,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE;wBACT,IAAI,EAAE,EAAE;wBACR,WAAW,EAAE,IAAI,CAAC,WAAW;qBAC7B;iBACD,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;QACD,OAAO,SAAS,CAAC;IAClB,CAAC;CACD"}

package/dist/esm/modules/a2a/a2a.module.d.ts

@@ -0,0 +1,71 @@
+import { type Message } from "@a2a-js/sdk";
+import { A2ATool } from "./a2a.tool.js";
+/**
+ * Module for managing Agent-to-Agent (A2A) protocol connections.
+ *
+ * This module handles connections to other A2A-compatible agents, manages
+ * conversation sessions, and provides an interface for inter-agent communication.
+ * Supports multi-turn conversations with task and context tracking.
+ *
+ * @example
+ * ```typescript
+ * const a2aModule = new A2AModule();
+ * await a2aModule.addA2APeerServer("https://api.example.com/agent");
+ *
+ * const tools = await a2aModule.getTools();
+ * const message = a2aModule.getMessagePayload("Hello", "session-123");
+ * const response = await a2aModule.useTool(tools[0], message, "session-123");
+ * ```
+ */
+export declare class A2AModule {
+    /** Map of A2A server URLs to their corresponding tool instances */
+    private a2aPeerServers;
+    /** Map of session IDs to their A2A session state */
+    private a2aSessions;
+    /**
+     * Registers a new A2A peer server URL for connection.
+     *
+     * @param url - The URL of the A2A-compatible agent to connect to
+     */
+    addA2APeerServer(url: string): Promise<void>;
+    /**
+     * Retrieves tools from all registered A2A peer servers.
+     *
+     * Attempts to connect to each registered server, fetch their agent cards,
+     * and create tool instances. Disables tools for unreachable servers.
+     *
+     * @returns Promise resolving to array of available A2A tools
+     */
+    getTools(): Promise<A2ATool[]>;
+    /**
+     * Gets or creates an A2A session for the given session ID.
+     *
+     * @param sessionId - The session identifier
+     * @returns A2ASession object with task and context IDs
+     */
+    private getA2ASessionWithId;
+    /**
+     * Constructs a message payload for A2A communication.
+     *
+     * Includes session context (task ID and context ID) if available
+     * for maintaining conversation continuity.
+     *
+     * @param query - The message content to send
+     * @param sessionId - The session identifier
+     * @returns Formatted Message object for A2A protocol
+     */
+    getMessagePayload(query: string, sessionId: string): Message;
+    /**
+     * Executes an A2A tool by sending a message to the remote agent.
+     *
+     * Handles streaming responses, maintains session state, and extracts
+     * text content from various event types in the response stream.
+     *
+     * @param tool - The A2ATool instance to use
+     * @param messagePayload - The message to send to the agent
+     * @param sessionId - The session identifier for context tracking
+     * @returns Promise resolving to array of text responses from the agent
+     */
+    useTool(tool: A2ATool, messagePayload: Message, sessionId: string): Promise<string[]>;
+}
+//# sourceMappingURL=a2a.module.d.ts.map

package/dist/esm/modules/a2a/a2a.module.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"a2a.module.d.ts","sourceRoot":"","sources":["../../../../src/modules/a2a/a2a.module.ts"],"names":[],"mappings":"AACA,OAAO,EAGN,KAAK,OAAO,EAKZ,MAAM,aAAa,CAAC;AAErB,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAYxC;;;;;;;;;;;;;;;;GAgBG;AACH,qBAAa,SAAS;IACrB,mEAAmE;IACnE,OAAO,CAAC,cAAc,CAA0C;IAChE,oDAAoD;IACpD,OAAO,CAAC,WAAW,CAAsC;IAEzD;;;;OAIG;IACU,gBAAgB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIzD;;;;;;;OAOG;IACU,QAAQ,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IAyB3C;;;;;OAKG;IACH,OAAO,CAAC,mBAAmB,CAUzB;IAEF;;;;;;;;;OASG;IACI,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO;IA2BnE;;;;;;;;;;OAUG;IACU,OAAO,CACnB,IAAI,EAAE,OAAO,EACb,cAAc,EAAE,OAAO,EACvB,SAAS,EAAE,MAAM,GACf,OAAO,CAAC,MAAM,EAAE,CAAC;CA2DpB"}

package/dist/esm/modules/a2a/a2a.module.js

@@ -0,0 +1,191 @@
+import { randomUUID } from "node:crypto";
+import { A2AClient, } from "@a2a-js/sdk";
+import { loggers } from "../../utils/logger.js";
+import { A2ATool } from "./a2a.tool.js";
+/**
+ * Module for managing Agent-to-Agent (A2A) protocol connections.
+ *
+ * This module handles connections to other A2A-compatible agents, manages
+ * conversation sessions, and provides an interface for inter-agent communication.
+ * Supports multi-turn conversations with task and context tracking.
+ *
+ * @example
+ * ```typescript
+ * const a2aModule = new A2AModule();
+ * await a2aModule.addA2APeerServer("https://api.example.com/agent");
+ *
+ * const tools = await a2aModule.getTools();
+ * const message = a2aModule.getMessagePayload("Hello", "session-123");
+ * const response = await a2aModule.useTool(tools[0], message, "session-123");
+ * ```
+ */
+export class A2AModule {
+    constructor() {
+        /** Map of A2A server URLs to their corresponding tool instances */
+        this.a2aPeerServers = new Map();
+        /** Map of session IDs to their A2A session state */
+        this.a2aSessions = new Map();
+        /**
+         * Gets or creates an A2A session for the given session ID.
+         *
+         * @param sessionId - The session identifier
+         * @returns A2ASession object with task and context IDs
+         */
+        this.getA2ASessionWithId = (sessionId) => {
+            const a2aSession = this.a2aSessions.get(sessionId) ?? {
+                taskId: undefined,
+                contextId: undefined,
+            };
+            if (!this.a2aSessions.has(sessionId)) {
+                this.a2aSessions.set(sessionId, a2aSession);
+            }
+            return a2aSession;
+        };
+    }
+    /**
+     * Registers a new A2A peer server URL for connection.
+     *
+     * @param url - The URL of the A2A-compatible agent to connect to
+     */
+    async addA2APeerServer(url) {
+        this.a2aPeerServers.set(url, null);
+    }
+    /**
+     * Retrieves tools from all registered A2A peer servers.
+     *
+     * Attempts to connect to each registered server, fetch their agent cards,
+     * and create tool instances. Disables tools for unreachable servers.
+     *
+     * @returns Promise resolving to array of available A2A tools
+     */
+    async getTools() {
+        const tools = [];
+        for (const url of [...this.a2aPeerServers.keys()]) {
+            const tool = this.a2aPeerServers.get(url);
+            if (!tool || !tool.enabled) {
+                try {
+                    const client = new A2AClient(url);
+                    const card = await client.getAgentCard();
+                    const toolName = card.name.replaceAll(" ", "-");
+                    const a2aTool = new A2ATool(toolName, client, card);
+                    tools.push(a2aTool);
+                }
+                catch (_error) {
+                    // Agent not responded
+                    if (tool) {
+                        tool.disable();
+                    }
+                }
+            }
+            else {
+                tools.push(tool);
+            }
+        }
+        return tools;
+    }
+    /**
+     * Constructs a message payload for A2A communication.
+     *
+     * Includes session context (task ID and context ID) if available
+     * for maintaining conversation continuity.
+     *
+     * @param query - The message content to send
+     * @param sessionId - The session identifier
+     * @returns Formatted Message object for A2A protocol
+     */
+    getMessagePayload(query, sessionId) {
+        const messagePayload = {
+            messageId: randomUUID(),
+            kind: "message",
+            role: "user", // FIXME: it could be 'agent'
+            metadata: {
+                sessionId,
+            },
+            parts: [
+                {
+                    kind: "text",
+                    text: query,
+                },
+            ],
+        };
+        const a2aSession = this.getA2ASessionWithId(sessionId);
+        if (a2aSession.taskId) {
+            messagePayload.taskId = a2aSession.taskId;
+        }
+        if (a2aSession.contextId) {
+            messagePayload.contextId = a2aSession.contextId;
+        }
+        return messagePayload;
+    }
+    /**
+     * Executes an A2A tool by sending a message to the remote agent.
+     *
+     * Handles streaming responses, maintains session state, and extracts
+     * text content from various event types in the response stream.
+     *
+     * @param tool - The A2ATool instance to use
+     * @param messagePayload - The message to send to the agent
+     * @param sessionId - The session identifier for context tracking
+     * @returns Promise resolving to array of text responses from the agent
+     */
+    async useTool(tool, messagePayload, sessionId) {
+        const finalText = [];
+        const client = tool.client;
+        const params = {
+            message: messagePayload,
+        };
+        const a2aSession = this.getA2ASessionWithId(sessionId);
+        try {
+            const stream = client.sendMessageStream(params);
+            for await (const event of stream) {
+                if (event.kind === "status-update") {
+                    const typedEvent = event;
+                    if (typedEvent.final &&
+                        typedEvent.status.state !== "input-required") {
+                        a2aSession.taskId = undefined;
+                    }
+                    // TODO: handle 'file', 'data' parts
+                    const texts = typedEvent.status.message?.parts
+                        .filter((part) => part.kind === "text")
+                        .map((part) => part.text)
+                        .join("\n");
+                    if (texts) {
+                        finalText.push(texts);
+                    }
+                }
+                else if (event.kind === "message") {
+                    // FIXME: handling text in 'message'?
+                    const msg = event;
+                    if (msg.taskId && msg.taskId !== a2aSession.taskId) {
+                        a2aSession.taskId = msg.taskId;
+                    }
+                    if (msg.contextId && msg.contextId !== a2aSession.contextId) {
+                        a2aSession.contextId = msg.contextId;
+                    }
+                }
+                else if (event.kind === "task") {
+                    // FIXME: handling text in 'task'?
+                    const task = event;
+                    if (task.id !== a2aSession.taskId) {
+                        a2aSession.taskId = task.id;
+                    }
+                    if (task.contextId && task.contextId !== a2aSession.contextId) {
+                        a2aSession.contextId = task.contextId;
+                    }
+                }
+                else {
+                    loggers.a2a.warn("Received unknown event structure from stream:", {
+                        event,
+                    });
+                }
+            }
+        }
+        catch (error) {
+            loggers.a2a.error("Error communicating with agent:", { error });
+            tool.disable();
+            // TODO: add failed & disabled text for next inference?
+        }
+        return finalText;
+    }
+}
+//# sourceMappingURL=a2a.module.js.map

package/dist/esm/modules/a2a/a2a.module.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"a2a.module.js","sourceRoot":"","sources":["../../../../src/modules/a2a/a2a.module.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EACN,SAAS,GAOT,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAC5C,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAYxC;;;;;;;;;;;;;;;;GAgBG;AACH,MAAM,OAAO,SAAS;IAAtB;QACC,mEAAmE;QAC3D,mBAAc,GAAgC,IAAI,GAAG,EAAE,CAAC;QAChE,oDAAoD;QAC5C,gBAAW,GAA4B,IAAI,GAAG,EAAE,CAAC;QA4CzD;;;;;WAKG;QACK,wBAAmB,GAAG,CAAC,SAAiB,EAAc,EAAE;YAC/D,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI;gBACrD,MAAM,EAAE,SAAS;gBACjB,SAAS,EAAE,SAAS;aACpB,CAAC;YACF,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;gBACtC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;YAC7C,CAAC;YAED,OAAO,UAAU,CAAC;QACnB,CAAC,CAAC;IAiHH,CAAC;IA3KA;;;;OAIG;IACI,KAAK,CAAC,gBAAgB,CAAC,GAAW;QACxC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IACpC,CAAC;IAED;;;;;;;OAOG;IACI,KAAK,CAAC,QAAQ;QACpB,MAAM,KAAK,GAAc,EAAE,CAAC;QAC5B,KAAK,MAAM,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;YACnD,MAAM,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC1C,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;gBAC5B,IAAI,CAAC;oBACJ,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,GAAG,CAAC,CAAC;oBAClC,MAAM,IAAI,GAAc,MAAM,MAAM,CAAC,YAAY,EAAE,CAAC;oBACpD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;oBAChD,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;oBAEpD,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACrB,CAAC;gBAAC,OAAO,MAAW,EAAE,CAAC;oBACtB,sBAAsB;oBACtB,IAAI,IAAI,EAAE,CAAC;wBACV,IAAI,CAAC,OAAO,EAAE,CAAC;oBAChB,CAAC;gBACF,CAAC;YACF,CAAC;iBAAM,CAAC;gBACP,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClB,CAAC;QACF,CAAC;QACD,OAAO,KAAK,CAAC;IACd,CAAC;IAoBD;;;;;;;;;OASG;IACI,iBAAiB,CAAC,KAAa,EAAE,SAAiB;QACxD,MAAM,cAAc,GAAY;YAC/B,SAAS,EAAE,UAAU,EAAE;YACvB,IAAI,EAAE,SAAS;YACf,IAAI,EAAE,MAAM,EAAE,6BAA6B;YAC3C,QAAQ,EAAE;gBACT,SAAS;aACT;YACD,KAAK,EAAE;gBACN;oBACC,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,KAAK;iBACX;aACD;SACD,CAAC;QAEF,MAAM,UAAU,GAAG,IAAI,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;QACvD,IAAI,UAAU,CAAC,MAAM,EAAE,CAAC;YACvB,cAAc,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;QAC3C,CAAC;QACD,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;YAC1B,cAAc,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;QACjD,CAAC;QAED,OAAO,cAAc,CAAC;IACvB,CAAC;IAED;;;;;;;;;;OAUG;IACI,KAAK,CAAC,OAAO,CACnB,IAAa,EACb,cAAuB,EACvB,SAAiB;QAEjB,MAAM,SAAS,GAAa,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,MAAM,MAAM,GAAsB;YACjC,OAAO,EAAE,cAAc;SACvB,CAAC;QACF,MAAM,UAAU,GAAG,IAAI,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAEvD,IAAI,CAAC;YACJ,MAAM,MAAM,GAAG,MAAM,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAChD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBAClC,IAAI,KAAK,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;oBACpC,MAAM,UAAU,GAAG,KAA8B,CAAC;oBAClD,IACC,UAAU,CAAC,KAAK;wBAChB,UAAU,CAAC,MAAM,CAAC,KAAK,KAAK,gBAAgB,EAC3C,CAAC;wBACF,UAAU,CAAC,MAAM,GAAG,SAAS,CAAC;oBAC/B,CAAC;oBACD,oCAAoC;oBACpC,MAAM,KAAK,GAAG,UAAU,CAAC,MAAM,CAAC,OAAO,EAAE,KAAK;yBAC5C,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;yBACtC,GAAG,CAAC,CAAC,IAAc,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;yBAClC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACb,IAAI,KAAK,EAAE,CAAC;wBACX,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBACvB,CAAC;gBACF,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;oBACrC,qCAAqC;oBACrC,MAAM,GAAG,GAAG,KAAgB,CAAC;oBAC7B,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,KAAK,UAAU,CAAC,MAAM,EAAE,CAAC;wBACpD,UAAU,CAAC,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC;oBAChC,CAAC;oBACD,IAAI,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,SAAS,KAAK,UAAU,CAAC,SAAS,EAAE,CAAC;wBAC7D,UAAU,CAAC,SAAS,GAAG,GAAG,CAAC,SAAS,CAAC;oBACtC,CAAC;gBACF,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBAClC,kCAAkC;oBAClC,MAAM,IAAI,GAAG,KAAa,CAAC;oBAC3B,IAAI,IAAI,CAAC,EAAE,KAAK,UAAU,CAAC,MAAM,EAAE,CAAC;wBACnC,UAAU,CAAC,MAAM,GAAG,IAAI,CAAC,EAAE,CAAC;oBAC7B,CAAC;oBACD,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,KAAK,UAAU,CAAC,SAAS,EAAE,CAAC;wBAC/D,UAAU,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;oBACvC,CAAC;gBACF,CAAC;qBAAM,CAAC;oBACP,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,+CAA+C,EAAE;wBACjE,KAAK;qBACL,CAAC,CAAC;gBACJ,CAAC;YACF,CAAC;QACF,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACzB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,iCAAiC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAChE,IAAI,CAAC,OAAO,EAAE,CAAC;YACf,uDAAuD;QACxD,CAAC;QAED,OAAO,SAAS,CAAC;IAClB,CAAC;CACD"}