@microsoft/teams.a2a 2.0.0-preview.10 → 2.0.0-preview.12

package/dist/chat-prompt/plugin2.js

@@ -0,0 +1,215 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.A2AClientPlugin = void 0;
+const client_1 = require("@a2a-js/sdk/client");
+const lodash_camelcase_1 = __importDefault(require("lodash.camelcase"));
+const uuid_1 = require("uuid");
+const pascalCase = (str) => {
+    return (0, lodash_camelcase_1.default)(str).replace(/^(.)/, (match) => match.toUpperCase());
+};
+class A2AClientPlugin {
+    name = 'a2a';
+    _agentConfigs = new Map();
+    _clients = new Map();
+    buildFunctionMetadata;
+    buildPrompt;
+    buildMessageForAgent;
+    buildMessageFromAgentResponse;
+    constructor(options = {}) {
+        this.buildFunctionMetadata = options.buildFunctionMetadata;
+        this.buildPrompt = options.buildPrompt;
+        this.buildMessageForAgent = options.buildMessageForAgent;
+        this.buildMessageFromAgentResponse = options.buildMessageFromAgentResponse;
+    }
+    onUsePlugin(args) {
+        // Just store the config, defer client creation to onBuildFunctions
+        this._agentConfigs.set(args.key, {
+            key: args.key,
+            cardUrl: args.cardUrl,
+            buildFunctionMetadata: args.buildFunctionMetadata,
+            buildMessageForAgent: args.buildMessageForAgent,
+            buildMessageFromAgentResponse: args.buildMessageFromAgentResponse,
+        });
+    }
+    async onBuildFunctions(functions) {
+        const allFunctions = [];
+        for (const [key, config] of this._agentConfigs) {
+            try {
+                // Get agent card (which internally gets or creates client)
+                const agentCard = await this._getAgentCard(key, config);
+                if (!agentCard) {
+                    continue; // Skip if we couldn't get the agent card
+                }
+                // Use custom function metadata builder or default
+                const buildMetadata = config.buildFunctionMetadata ||
+                    this.buildFunctionMetadata ||
+                    this._defaultFunctionMetadata;
+                const { name, description } = buildMetadata(agentCard);
+                allFunctions.push({
+                    name,
+                    description,
+                    parameters: {
+                        type: 'object',
+                        properties: {
+                            message: {
+                                type: 'string',
+                                description: 'Message to send to the agent',
+                            },
+                        },
+                        required: ['message'],
+                    },
+                    handler: (async (args) => {
+                        try {
+                            const agentMessage = args.message;
+                            // Use custom message builder if provided, otherwise use default
+                            const buildMessage = config.buildMessageForAgent ||
+                                this.buildMessageForAgent ||
+                                this._defaultBuildMessage.bind(this);
+                            const messageOrString = buildMessage(agentCard, agentMessage);
+                            // Handle both Message and string returns
+                            const message = typeof messageOrString === 'string'
+                                ? this._createMessage(messageOrString, agentCard)
+                                : messageOrString;
+                            const sendParams = { message };
+                            // Get the client info to send the message
+                            const clientInfo = this._clients.get(key);
+                            if (!clientInfo) {
+                                throw new Error(`Client not found for agent ${key}`);
+                            }
+                            const response = await clientInfo.client.sendMessage(sendParams);
+                            if ('error' in response) {
+                                return response.error.message;
+                            }
+                            const result = response.result;
+                            // Use custom response builder if provided, otherwise use default
+                            const buildResponse = config.buildMessageFromAgentResponse ||
+                                this.buildMessageFromAgentResponse ||
+                                this._defaultBuildMessageFromAgentResponse.bind(this);
+                            return buildResponse(agentCard, result, agentMessage);
+                        }
+                        catch (e) {
+                            console.error(e);
+                            throw e;
+                        }
+                    }).bind(this),
+                });
+            }
+            catch (error) {
+                console.error(`Failed to build function for agent ${key}:`, error);
+                // Continue with other agents even if one fails
+            }
+        }
+        return functions.concat(allFunctions);
+    }
+    async onBuildPrompt(systemPrompt) {
+        // Collect agent details for prompt building
+        const agentPromptParams = [];
+        for (const [key, config] of this._agentConfigs) {
+            try {
+                const agentCard = await this._getAgentCard(key, config);
+                if (agentCard) {
+                    const clientInfo = this._clients.get(key);
+                    if (clientInfo) {
+                        agentPromptParams.push({
+                            card: agentCard,
+                            client: clientInfo.client,
+                        });
+                    }
+                }
+            }
+            catch (error) {
+                console.error(`Failed to get agent card for ${key}:`, error);
+            }
+        }
+        // Use custom buildPrompt if provided, otherwise use default
+        if (this.buildPrompt) {
+            return this.buildPrompt(systemPrompt, agentPromptParams);
+        }
+        // Default prompt building
+        if (agentPromptParams.length === 0) {
+            return systemPrompt;
+        }
+        const agentDetails = agentPromptParams
+            .map(({ card }) => {
+            let details = `<Agent Details>\n<Name>\n${card.name}\n</Name>\n`;
+            if (card.description) {
+                details += `<Description>\n${card.description}\n</Description>\n`;
+            }
+            if (card.skills?.length) {
+                for (const skill of card.skills) {
+                    details += `<SKILL name="${skill.name}" description="${skill.description}" />\n`;
+                    if (skill.examples?.length) {
+                        details += `<EXAMPLES>\n${skill.examples.join('\n')}\n</EXAMPLES>\n`;
+                    }
+                }
+            }
+            // Could add client-specific info here if needed
+            details += '</Agent Details>\n';
+            return details;
+        })
+            .join('');
+        const prompt = (systemPrompt || '') +
+            '\n\nHere are details about available agents that you can message:\n' +
+            agentDetails;
+        return prompt;
+    }
+    _defaultFunctionMetadata(card) {
+        const name = `message${pascalCase(card.name)}`;
+        const description = card.description || `Interact with ${card.name} agent`;
+        return { name, description };
+    }
+    _defaultBuildMessage(card, input, metadata) {
+        return this._createMessage(input, card, metadata);
+    }
+    _createMessage(text, _card, metadata) {
+        return {
+            messageId: (0, uuid_1.v4)(),
+            role: 'user', // Messages TO agents are from user perspective
+            parts: [{ kind: 'text', text }],
+            kind: 'message',
+            // Include metadata if provided
+            ...(metadata && { metadata }),
+        };
+    }
+    _defaultBuildMessageFromAgentResponse(_card, response, _originalInput) {
+        // Extract text from response parts
+        if (response.kind === 'message') {
+            const textParts = response.parts
+                .filter((part) => part.kind === 'text')
+                .map(part => part.text);
+            return textParts.join(' ') || 'Agent responded with no text content.';
+        }
+        else {
+            return 'Agent responded with non-message content.';
+        }
+    }
+    async _getAgentCard(key, config) {
+        // Return cached client info if it exists
+        let clientInfo = this._clients.get(key);
+        if (clientInfo) {
+            return clientInfo.agentCard;
+        }
+        // Create new client and get agent card
+        try {
+            const client = await client_1.A2AClient.fromCardUrl(config.cardUrl);
+            // Get the agent card from the client
+            const agentCard = await client.getAgentCard();
+            clientInfo = {
+                ...config,
+                client,
+                agentCard,
+            };
+            this._clients.set(key, clientInfo);
+            return agentCard;
+        }
+        catch (error) {
+            console.error(`Error creating client or fetching agent card for ${key}:`, error);
+            return null;
+        }
+    }
+}
+exports.A2AClientPlugin = A2AClientPlugin;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoicGx1Z2luMi5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy9jaGF0LXByb21wdC9wbHVnaW4yLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7OztBQUNBLCtDQUErQztBQUMvQyx3RUFBeUM7QUFDekMsK0JBQW9DO0FBMEZwQyxNQUFNLFVBQVUsR0FBRyxDQUFDLEdBQVcsRUFBRSxFQUFFO0lBQ2pDLE9BQU8sSUFBQSwwQkFBUyxFQUFDLEdBQUcsQ0FBQyxDQUFDLE9BQU8sQ0FBQyxNQUFNLEVBQUUsQ0FBQyxLQUFLLEVBQUUsRUFBRSxDQUFDLEtBQUssQ0FBQyxXQUFXLEVBQUUsQ0FBQyxDQUFDO0FBQ3hFLENBQUMsQ0FBQztBQUVGLE1BQWEsZUFBZTtJQUVqQixJQUFJLEdBQUcsS0FBSyxDQUFDO0lBRVosYUFBYSxHQUE4QixJQUFJLEdBQUcsRUFBRSxDQUFDO0lBQ3JELFFBQVEsR0FBa0MsSUFBSSxHQUFHLEVBQUUsQ0FBQztJQUNwRCxxQkFBcUIsQ0FBNEI7SUFDakQsV0FBVyxDQUFrQjtJQUM3QixvQkFBb0IsQ0FBd0I7SUFDNUMsNkJBQTZCLENBQWlDO0lBRXhFLFlBQVksVUFBa0MsRUFBRTtRQUM5QyxJQUFJLENBQUMscUJBQXFCLEdBQUcsT0FBTyxDQUFDLHFCQUFxQixDQUFDO1FBQzNELElBQUksQ0FBQyxXQUFXLEdBQUcsT0FBTyxDQUFDLFdBQVcsQ0FBQztRQUN2QyxJQUFJLENBQUMsb0JBQW9CLEdBQUcsT0FBTyxDQUFDLG9CQUFvQixDQUFDO1FBQ3pELElBQUksQ0FBQyw2QkFBNkIsR0FBRyxPQUFPLENBQUMsNkJBQTZCLENBQUM7SUFDN0UsQ0FBQztJQUVELFdBQVcsQ0FBQyxJQUF3QjtRQUNsQyxtRUFBbUU7UUFDbkUsSUFBSSxDQUFDLGFBQWEsQ0FBQyxHQUFHLENBQUMsSUFBSSxDQUFDLEdBQUcsRUFBRTtZQUMvQixHQUFHLEVBQUUsSUFBSSxDQUFDLEdBQUc7WUFDYixPQUFPLEVBQUUsSUFBSSxDQUFDLE9BQU87WUFDckIscUJBQXFCLEVBQUUsSUFBSSxDQUFDLHFCQUFxQjtZQUNqRCxvQkFBb0IsRUFBRSxJQUFJLENBQUMsb0JBQW9CO1lBQy9DLDZCQUE2QixFQUFFLElBQUksQ0FBQyw2QkFBNkI7U0FDbEUsQ0FBQyxDQUFDO0lBQ0wsQ0FBQztJQUVELEtBQUssQ0FBQyxnQkFBZ0IsQ0FBQyxTQUF5QjtRQUM5QyxNQUFNLFlBQVksR0FBbUIsRUFBRSxDQUFDO1FBRXhDLEtBQUssTUFBTSxDQUFDLEdBQUcsRUFBRSxNQUFNLENBQUMsSUFBSSxJQUFJLENBQUMsYUFBYSxFQUFFLENBQUM7WUFDL0MsSUFBSSxDQUFDO2dCQUNILDJEQUEyRDtnQkFDM0QsTUFBTSxTQUFTLEdBQUcsTUFBTSxJQUFJLENBQUMsYUFBYSxDQUFDLEdBQUcsRUFBRSxNQUFNLENBQUMsQ0FBQztnQkFDeEQsSUFBSSxDQUFDLFNBQVMsRUFBRSxDQUFDO29CQUNmLFNBQVMsQ0FBQyx5Q0FBeUM7Z0JBQ3JELENBQUM7Z0JBRUQsa0RBQWtEO2dCQUNsRCxNQUFNLGFBQWEsR0FDakIsTUFBTSxDQUFDLHFCQUFxQjtvQkFDNUIsSUFBSSxDQUFDLHFCQUFxQjtvQkFDMUIsSUFBSSxDQUFDLHdCQUF3QixDQUFDO2dCQUVoQyxNQUFNLEVBQUUsSUFBSSxFQUFFLFdBQVcsRUFBRSxHQUFHLGFBQWEsQ0FBQyxTQUFTLENBQUMsQ0FBQztnQkFFdkQsWUFBWSxDQUFDLElBQUksQ0FBQztvQkFDaEIsSUFBSTtvQkFDSixXQUFXO29CQUNYLFVBQVUsRUFBRTt3QkFDVixJQUFJLEVBQUUsUUFBUTt3QkFDZCxVQUFVLEVBQUU7NEJBQ1YsT0FBTyxFQUFFO2dDQUNQLElBQUksRUFBRSxRQUFRO2dDQUNkLFdBQVcsRUFBRSw4QkFBOEI7NkJBQzVDO3lCQUNGO3dCQUNELFFBQVEsRUFBRSxDQUFDLFNBQVMsQ0FBQztxQkFDdEI7b0JBQ0QsT0FBTyxFQUFFLENBQUMsS0FBSyxFQUFFLElBQXlCLEVBQUUsRUFBRTt3QkFDNUMsSUFBSSxDQUFDOzRCQUNILE1BQU0sWUFBWSxHQUFHLElBQUksQ0FBQyxPQUFPLENBQUM7NEJBRWxDLGdFQUFnRTs0QkFDaEUsTUFBTSxZQUFZLEdBQ2hCLE1BQU0sQ0FBQyxvQkFBb0I7Z0NBQzNCLElBQUksQ0FBQyxvQkFBb0I7Z0NBQ3pCLElBQUksQ0FBQyxvQkFBb0IsQ0FBQyxJQUFJLENBQUMsSUFBSSxDQUFDLENBQUM7NEJBRXZDLE1BQU0sZUFBZSxHQUFHLFlBQVksQ0FBQyxTQUFTLEVBQUUsWUFBWSxDQUFDLENBQUM7NEJBRTlELHlDQUF5Qzs0QkFDekMsTUFBTSxPQUFPLEdBQVksT0FBTyxlQUFlLEtBQUssUUFBUTtnQ0FDMUQsQ0FBQyxDQUFDLElBQUksQ0FBQyxjQUFjLENBQUMsZUFBZSxFQUFFLFNBQVMsQ0FBQztnQ0FDakQsQ0FBQyxDQUFDLGVBQWUsQ0FBQzs0QkFFcEIsTUFBTSxVQUFVLEdBQXNCLEVBQUUsT0FBTyxFQUFFLENBQUM7NEJBRWxELDBDQUEwQzs0QkFDMUMsTUFBTSxVQUFVLEdBQUcsSUFBSSxDQUFDLFFBQVEsQ0FBQyxHQUFHLENBQUMsR0FBRyxDQUFDLENBQUM7NEJBQzFDLElBQUksQ0FBQyxVQUFVLEVBQUUsQ0FBQztnQ0FDaEIsTUFBTSxJQUFJLEtBQUssQ0FBQyw4QkFBOEIsR0FBRyxFQUFFLENBQUMsQ0FBQzs0QkFDdkQsQ0FBQzs0QkFFRCxNQUFNLFFBQVEsR0FBRyxNQUFNLFVBQVUsQ0FBQyxNQUFNLENBQUMsV0FBVyxDQUFDLFVBQVUsQ0FBQyxDQUFDOzRCQUVqRSxJQUFJLE9BQU8sSUFBSSxRQUFRLEVBQUUsQ0FBQztnQ0FDeEIsT0FBTyxRQUFRLENBQUMsS0FBSyxDQUFDLE9BQU8sQ0FBQzs0QkFDaEMsQ0FBQzs0QkFFRCxNQUFNLE1BQU0sR0FBRyxRQUFRLENBQUMsTUFBTSxDQUFDOzRCQUUvQixpRUFBaUU7NEJBQ2pFLE1BQU0sYUFBYSxHQUNqQixNQUFNLENBQUMsNkJBQTZCO2dDQUNwQyxJQUFJLENBQUMsNkJBQTZCO2dDQUNsQyxJQUFJLENBQUMscUNBQXFDLENBQUMsSUFBSSxDQUFDLElBQUksQ0FBQyxDQUFDOzRCQUV4RCxPQUFPLGFBQWEsQ0FBQyxTQUFTLEVBQUUsTUFBTSxFQUFFLFlBQVksQ0FBQyxDQUFDO3dCQUN4RCxDQUFDO3dCQUFDLE9BQU8sQ0FBQyxFQUFFLENBQUM7NEJBQ1gsT0FBTyxDQUFDLEtBQUssQ0FBQyxDQUFDLENBQUMsQ0FBQzs0QkFDakIsTUFBTSxDQUFDLENBQUM7d0JBQ1YsQ0FBQztvQkFDSCxDQUFDLENBQUMsQ0FBQyxJQUFJLENBQUMsSUFBSSxDQUFDO2lCQUNkLENBQUMsQ0FBQztZQUNMLENBQUM7WUFBQyxPQUFPLEtBQUssRUFBRSxDQUFDO2dCQUNmLE9BQU8sQ0FBQyxLQUFLLENBQUMsc0NBQXNDLEdBQUcsR0FBRyxFQUFFLEtBQUssQ0FBQyxDQUFDO2dCQUNuRSwrQ0FBK0M7WUFDakQsQ0FBQztRQUNILENBQUM7UUFFRCxPQUFPLFNBQVMsQ0FBQyxNQUFNLENBQUMsWUFBWSxDQUFDLENBQUM7SUFDeEMsQ0FBQztJQUVELEtBQUssQ0FBQyxhQUFhLENBQ2pCLFlBQWdDO1FBRWhDLDRDQUE0QztRQUM1QyxNQUFNLGlCQUFpQixHQUEyQixFQUFFLENBQUM7UUFFckQsS0FBSyxNQUFNLENBQUMsR0FBRyxFQUFFLE1BQU0sQ0FBQyxJQUFJLElBQUksQ0FBQyxhQUFhLEVBQUUsQ0FBQztZQUMvQyxJQUFJLENBQUM7Z0JBQ0gsTUFBTSxTQUFTLEdBQUcsTUFBTSxJQUFJLENBQUMsYUFBYSxDQUFDLEdBQUcsRUFBRSxNQUFNLENBQUMsQ0FBQztnQkFDeEQsSUFBSSxTQUFTLEVBQUUsQ0FBQztvQkFDZCxNQUFNLFVBQVUsR0FBRyxJQUFJLENBQUMsUUFBUSxDQUFDLEdBQUcsQ0FBQyxHQUFHLENBQUMsQ0FBQztvQkFDMUMsSUFBSSxVQUFVLEVBQUUsQ0FBQzt3QkFDZixpQkFBaUIsQ0FBQyxJQUFJLENBQUM7NEJBQ3JCLElBQUksRUFBRSxTQUFTOzRCQUNmLE1BQU0sRUFBRSxVQUFVLENBQUMsTUFBTTt5QkFDMUIsQ0FBQyxDQUFDO29CQUNMLENBQUM7Z0JBQ0gsQ0FBQztZQUNILENBQUM7WUFBQyxPQUFPLEtBQUssRUFBRSxDQUFDO2dCQUNmLE9BQU8sQ0FBQyxLQUFLLENBQUMsZ0NBQWdDLEdBQUcsR0FBRyxFQUFFLEtBQUssQ0FBQyxDQUFDO1lBQy9ELENBQUM7UUFDSCxDQUFDO1FBRUQsNERBQTREO1FBQzVELElBQUksSUFBSSxDQUFDLFdBQVcsRUFBRSxDQUFDO1lBQ3JCLE9BQU8sSUFBSSxDQUFDLFdBQVcsQ0FBQyxZQUFZLEVBQUUsaUJBQWlCLENBQUMsQ0FBQztRQUMzRCxDQUFDO1FBRUQsMEJBQTBCO1FBQzFCLElBQUksaUJBQWlCLENBQUMsTUFBTSxLQUFLLENBQUMsRUFBRSxDQUFDO1lBQ25DLE9BQU8sWUFBWSxDQUFDO1FBQ3RCLENBQUM7UUFFRCxNQUFNLFlBQVksR0FBRyxpQkFBaUI7YUFDbkMsR0FBRyxDQUFDLENBQUMsRUFBRSxJQUFJLEVBQUUsRUFBRSxFQUFFO1lBQ2hCLElBQUksT0FBTyxHQUFHLDRCQUE0QixJQUFJLENBQUMsSUFBSSxhQUFhLENBQUM7WUFDakUsSUFBSSxJQUFJLENBQUMsV0FBVyxFQUFFLENBQUM7Z0JBQ3JCLE9BQU8sSUFBSSxrQkFBa0IsSUFBSSxDQUFDLFdBQVcsb0JBQW9CLENBQUM7WUFDcEUsQ0FBQztZQUNELElBQUksSUFBSSxDQUFDLE1BQU0sRUFBRSxNQUFNLEVBQUUsQ0FBQztnQkFDeEIsS0FBSyxNQUFNLEtBQUssSUFBSSxJQUFJLENBQUMsTUFBTSxFQUFFLENBQUM7b0JBQ2hDLE9BQU8sSUFBSSxnQkFBZ0IsS0FBSyxDQUFDLElBQUksa0JBQWtCLEtBQUssQ0FBQyxXQUFXLFFBQVEsQ0FBQztvQkFDakYsSUFBSSxLQUFLLENBQUMsUUFBUSxFQUFFLE1BQU0sRUFBRSxDQUFDO3dCQUMzQixPQUFPLElBQUksZUFBZSxLQUFLLENBQUMsUUFBUSxDQUFDLElBQUksQ0FBQyxJQUFJLENBQUMsaUJBQWlCLENBQUM7b0JBQ3ZFLENBQUM7Z0JBQ0gsQ0FBQztZQUNILENBQUM7WUFDRCxnREFBZ0Q7WUFDaEQsT0FBTyxJQUFJLG9CQUFvQixDQUFDO1lBQ2hDLE9BQU8sT0FBTyxDQUFDO1FBQ2pCLENBQUMsQ0FBQzthQUNELElBQUksQ0FBQyxFQUFFLENBQUMsQ0FBQztRQUVaLE1BQU0sTUFBTSxHQUFHLENBQUMsWUFBWSxJQUFJLEVBQUUsQ0FBQztZQUNqQyxxRUFBcUU7WUFDckUsWUFBWSxDQUFDO1FBRWYsT0FBTyxNQUFNLENBQUM7SUFDaEIsQ0FBQztJQUVPLHdCQUF3QixDQUM5QixJQUFlO1FBRWYsTUFBTSxJQUFJLEdBQUcsVUFBVSxVQUFVLENBQUMsSUFBSSxDQUFDLElBQUksQ0FBQyxFQUFFLENBQUM7UUFDL0MsTUFBTSxXQUFXLEdBQUcsSUFBSSxDQUFDLFdBQVcsSUFBSSxpQkFBaUIsSUFBSSxDQUFDLElBQUksUUFBUSxDQUFDO1FBQzNFLE9BQU8sRUFBRSxJQUFJLEVBQUUsV0FBVyxFQUFFLENBQUM7SUFDL0IsQ0FBQztJQUVPLG9CQUFvQixDQUMxQixJQUFlLEVBQ2YsS0FBYSxFQUNiLFFBQThCO1FBRTlCLE9BQU8sSUFBSSxDQUFDLGNBQWMsQ0FBQyxLQUFLLEVBQUUsSUFBSSxFQUFFLFFBQVEsQ0FBQyxDQUFDO0lBQ3BELENBQUM7SUFFTyxjQUFjLENBQ3BCLElBQVksRUFDWixLQUFpQixFQUNqQixRQUE4QjtRQUU5QixPQUFPO1lBQ0wsU0FBUyxFQUFFLElBQUEsU0FBTSxHQUFFO1lBQ25CLElBQUksRUFBRSxNQUFNLEVBQUUsK0NBQStDO1lBQzdELEtBQUssRUFBRSxDQUFDLEVBQUUsSUFBSSxFQUFFLE1BQU0sRUFBRSxJQUFJLEVBQUUsQ0FBQztZQUMvQixJQUFJLEVBQUUsU0FBUztZQUNmLCtCQUErQjtZQUMvQixHQUFHLENBQUMsUUFBUSxJQUFJLEVBQUUsUUFBUSxFQUFFLENBQUM7U0FDOUIsQ0FBQztJQUNKLENBQUM7SUFFTyxxQ0FBcUMsQ0FDM0MsS0FBZ0IsRUFDaEIsUUFBd0IsRUFDeEIsY0FBc0I7UUFFdEIsbUNBQW1DO1FBQ25DLElBQUksUUFBUSxDQUFDLElBQUksS0FBSyxTQUFTLEVBQUUsQ0FBQztZQUNoQyxNQUFNLFNBQVMsR0FBRyxRQUFRLENBQUMsS0FBSztpQkFDN0IsTUFBTSxDQUFDLENBQUMsSUFBSSxFQUFvQixFQUFFLENBQUMsSUFBSSxDQUFDLElBQUksS0FBSyxNQUFNLENBQUM7aUJBQ3hELEdBQUcsQ0FBQyxJQUFJLENBQUMsRUFBRSxDQUFDLElBQUksQ0FBQyxJQUFJLENBQUMsQ0FBQztZQUMxQixPQUFPLFNBQVMsQ0FBQyxJQUFJLENBQUMsR0FBRyxDQUFDLElBQUksdUNBQXVDLENBQUM7UUFDeEUsQ0FBQzthQUFNLENBQUM7WUFDTixPQUFPLDJDQUEyQyxDQUFDO1FBQ3JELENBQUM7SUFDSCxDQUFDO0lBRU8sS0FBSyxDQUFDLGFBQWEsQ0FBQyxHQUFXLEVBQUUsTUFBb0I7UUFDM0QseUNBQXlDO1FBQ3pDLElBQUksVUFBVSxHQUFHLElBQUksQ0FBQyxRQUFRLENBQUMsR0FBRyxDQUFDLEdBQUcsQ0FBQyxDQUFDO1FBQ3hDLElBQUksVUFBVSxFQUFFLENBQUM7WUFDZixPQUFPLFVBQVUsQ0FBQyxTQUFTLENBQUM7UUFDOUIsQ0FBQztRQUVELHVDQUF1QztRQUN2QyxJQUFJLENBQUM7WUFDSCxNQUFNLE1BQU0sR0FBRyxNQUFNLGtCQUFTLENBQUMsV0FBVyxDQUFDLE1BQU0sQ0FBQyxPQUFPLENBQUMsQ0FBQztZQUUzRCxxQ0FBcUM7WUFDckMsTUFBTSxTQUFTLEdBQUcsTUFBTSxNQUFNLENBQUMsWUFBWSxFQUFFLENBQUM7WUFFOUMsVUFBVSxHQUFHO2dCQUNYLEdBQUcsTUFBTTtnQkFDVCxNQUFNO2dCQUNOLFNBQVM7YUFDVixDQUFDO1lBRUYsSUFBSSxDQUFDLFFBQVEsQ0FBQyxHQUFHLENBQUMsR0FBRyxFQUFFLFVBQVUsQ0FBQyxDQUFDO1lBQ25DLE9BQU8sU0FBUyxDQUFDO1FBQ25CLENBQUM7UUFBQyxPQUFPLEtBQUssRUFBRSxDQUFDO1lBQ2YsT0FBTyxDQUFDLEtBQUssQ0FBQyxvREFBb0QsR0FBRyxHQUFHLEVBQUUsS0FBSyxDQUFDLENBQUM7WUFDakYsT0FBTyxJQUFJLENBQUM7UUFDZCxDQUFDO0lBQ0gsQ0FBQztDQUNGO0FBMVBELDBDQTBQQyJ9

package/dist/chat-prompt/types.d.ts

@@ -1,60 +1,68 @@
-import { AgentManager, AgentManagerOptions } from '../client/agent-manager';
-import * as schema from '../common/schema';
-import { TaskAndHistory } from '../server/types/a2a-types';
+import { AgentCard, Message, Task } from '@a2a-js/sdk';
+import { A2AClient } from '@a2a-js/sdk/client';
+import { ILogger } from '@microsoft/teams.common';
 /**
- * Parameters for registering an agent with the A2A plugin.
- * usage: new ChatPrompt(..., [new A2APlugin(...)]).use(A2APluginParams)
+ * New types that use SDK's AgentCard and A2AClient instead of internal schema types
  */
-export type A2APluginUseParams = {
+export type BuildFunctionMetadata = (card: AgentCard) => {
+    name: string;
+    description: string;
+};
+export type AgentPromptParams = {
+    card: AgentCard;
+    client: A2AClient;
+};
+export type BuildPrompt = (systemPrompt: string | undefined, agentDetails: AgentPromptParams[]) => string | undefined;
+export type BuildMessageForAgent = (card: AgentCard, input: string, metadata?: Record<string, any>) => Message | string;
+export type BuildMessageFromAgentResponse = (card: AgentCard, response: Task | Message, originalInput: string) => string;
+/**
+ * Options for constructing an A2AClientPlugin using the official SDK.
+ */
+export type A2AClientPluginOptions = {
     /**
-     * Unique key for this agent (used for config and lookup)
+     * Optional function to customize the function name and description for each agent card.
      */
-    key: string;
+    buildFunctionMetadata?: BuildFunctionMetadata;
     /**
-     * The agent's base URL
+     * Optional function to customize the prompt given all agent cards.
      */
-    url: string;
+    buildPrompt?: BuildPrompt;
     /**
-     * Optional agent card for the agent
+     * Optional function to customize the message format sent to each agent.
      */
-    agentCard?: schema.AgentCard;
+    buildMessageForAgent?: BuildMessageForAgent;
     /**
-     * Optional function to customize function metadata for this agent
+     * Optional function to customize how agent responses are processed into strings.
      */
-    buildFunctionMetadata?: BuildFunctionMetadata;
+    buildMessageFromAgentResponse?: BuildMessageFromAgentResponse;
     /**
-     * Optional function to customize TaskSendParams for this agent
+     * Logger
      */
-    buildTaskSendParams?: BuildTaskSendParams;
-};
-export type AgentPromptParams = {
-    card: schema.AgentCard;
-    latestTask?: TaskAndHistory | null;
+    logger?: ILogger;
 };
-export type BuildFunctionMetadata = (card: schema.AgentCard) => {
-    name: string;
-    description: string;
-};
-export type BuildTaskSendParams = (card: schema.AgentCard, input: string, continueTaskId?: string | null, metadata?: Record<string, any>) => schema.TaskSendParams;
-export type BuildPrompt = (incomingSystemPrompt: string | undefined, agentDetails: AgentPromptParams[]) => string | undefined;
 /**
- * Options for constructing an A2APlugin.
+ * Parameters for registering an agent with the A2AClientPlugin.
+ * Usage: new ChatPrompt(..., [new A2AClientPlugin(...)]).usePlugin('a2a', A2APluginUseParams)
  */
-export type A2APluginOptions = {
+export type A2APluginUseParams = {
     /**
-     * Optional A2AAgentManager instance to use for agent management.
+     * Unique key to identify this agent
      */
-    manager?: AgentManager | AgentManagerOptions;
+    key: string;
     /**
-     * Optional function to customize the function name and description for each agent card.
+     * URL to the agent's card endpoint
+     */
+    cardUrl: string;
+    /**
+     * Custom function metadata builder for this specific agent
      */
     buildFunctionMetadata?: BuildFunctionMetadata;
     /**
-     * Optional function to customize the prompt given all agent cards.
+     * Custom message builder for this specific agent
      */
-    buildPrompt?: BuildPrompt;
+    buildMessageForAgent?: BuildMessageForAgent;
     /**
-     * Optional function to customize TaskSendParams given the input and context.
+     * Custom response processor for this specific agent
      */
-    buildTaskSendParams?: BuildTaskSendParams;
+    buildMessageFromAgentResponse?: BuildMessageFromAgentResponse;
 };

package/dist/index.d.ts

@@ -1,10 +1,3 @@
 export * from './chat-prompt/plugin';
 export * from './chat-prompt/types';
-export * from './client/agent-client';
-export * from './client/agent-manager';
-export * from './client/agent-task-store';
-export * from './common/schema';
-export * as schema from './common/schema';
 export * from './server/plugin';
-export * from './server/types/a2a-types';
-export * from './server/types/event-types';

package/dist/index.js

@@ -10,41 +10,11 @@ var __createBinding = (this && this.__createBinding) || (Object.create ? (functi
     if (k2 === undefined) k2 = k;
     o[k2] = m[k];
 }));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
 var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
-var __importStar = (this && this.__importStar) || (function () {
-    var ownKeys = function(o) {
-        ownKeys = Object.getOwnPropertyNames || function (o) {
-            var ar = [];
-            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
-            return ar;
-        };
-        return ownKeys(o);
-    };
-    return function (mod) {
-        if (mod && mod.__esModule) return mod;
-        var result = {};
-        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
-        __setModuleDefault(result, mod);
-        return result;
-    };
-})();
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.schema = void 0;
 __exportStar(require("./chat-prompt/plugin"), exports);
 __exportStar(require("./chat-prompt/types"), exports);
-__exportStar(require("./client/agent-client"), exports);
-__exportStar(require("./client/agent-manager"), exports);
-__exportStar(require("./client/agent-task-store"), exports);
-__exportStar(require("./common/schema"), exports);
-exports.schema = __importStar(require("./common/schema"));
 __exportStar(require("./server/plugin"), exports);
-__exportStar(require("./server/types/a2a-types"), exports);
-__exportStar(require("./server/types/event-types"), exports);
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi9zcmMvaW5kZXgudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6Ijs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7O0FBQUEsdURBQXFDO0FBQ3JDLHNEQUFvQztBQUNwQyx3REFBc0M7QUFDdEMseURBQXVDO0FBQ3ZDLDREQUEwQztBQUMxQyxrREFBZ0M7QUFDaEMsMERBQTBDO0FBQzFDLGtEQUFnQztBQUNoQywyREFBeUM7QUFDekMsNkRBQTJDIn0=
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi9zcmMvaW5kZXgudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6Ijs7Ozs7Ozs7Ozs7Ozs7OztBQUFBLHVEQUFxQztBQUNyQyxzREFBb0M7QUFDcEMsa0RBQWdDIn0=

package/dist/server/plugin.d.ts

@@ -1,15 +1,13 @@
+import { AgentCard, Message, Task } from '@a2a-js/sdk';
+import { A2ARequestHandler, AgentExecutor, ExecutionEventBus, RequestContext, TaskStore } from '@a2a-js/sdk/server';
+import { RequestHandler } from 'express';
 import { EmitPluginEvent, HttpPlugin, IPlugin } from '@microsoft/teams.apps';
-import { ILogger, IStorage } from '@microsoft/teams.common';
-import { AgentManager, AgentManagerOptions } from '../client/agent-manager';
-import * as schema from '../common/schema';
-import { TaskManager } from './tasks/task-manager';
-import { TaskStore } from './tasks/task-store';
-import { A2AEvents } from './types/event-types';
+import { ILogger } from '@microsoft/teams.common';
 interface IA2APluginOptions {
     /**
      * The agent card to be used for the A2A plugin.
      */
-    agentCard: schema.AgentCard;
+    agentCard: AgentCard;
     /**
      * Path to the A2A server
      * @default '/a2a'
@@ -21,29 +19,34 @@ interface IA2APluginOptions {
      */
     taskStore?: TaskStore;
     /**
-     * The agent manager options that can be used to configure the agent manager.
-     * @default {}
-     */
-    managerOptions?: AgentManagerOptions;
+    * For a completely custom executor, you may provide your own executor that will
+    * get executed whenever the a2a agent is InMemoryTaskStore
+    */
+    agentExecutor?: AgentExecutor;
 }
+export type Respond = (message: string | Message | Task) => Promise<void>;
+export type A2AEvents = {
+    'a2a:message': {
+        requestContext: RequestContext;
+        respond: Respond;
+        publishUpdate: ExecutionEventBus['publish'];
+    };
+};
 export declare class A2APlugin implements IPlugin {
+    readonly log: ILogger;
     protected readonly emit: EmitPluginEvent<A2AEvents>;
     protected readonly _httpPlugin: HttpPlugin;
-    protected readonly _storage: IStorage;
-    protected readonly _logger: ILogger;
     __eventType: A2AEvents;
-    protected _card: schema.AgentCard;
-    protected _path: string;
-    protected _taskManager: TaskManager;
-    protected _taskStore: TaskStore;
-    protected _clientManager: AgentManager;
-    get clientManager(): AgentManager;
+    readonly card: AgentCard;
+    readonly path: string;
+    readonly taskStore: TaskStore;
+    readonly customExecutor?: AgentExecutor;
+    private readonly middlewares;
     constructor(options: IA2APluginOptions);
+    use(middleware: RequestHandler): void;
     onInit(): void;
-    private onTaskRequest;
-    private onSendRequest;
-    private onGetTaskRequest;
-    private onTaskGetRequest;
-    private onTaskSendRequest;
+    _createLoggingMiddleware(): RequestHandler;
+    _setupExecutor(): AgentExecutor;
+    _setupRequestHandler(): A2ARequestHandler;
 }
 export {};