nestjs-a2a/dist/services/a2a.registry.js

NestJS module for creating Google Agent to Agent Server

"use strict";
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
    return c > 3 && r && Object.defineProperty(target, key, r), r;
};
var __metadata = (this && this.__metadata) || function (k, v) {
    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.A2ARegistry = void 0;
const common_1 = require("@nestjs/common");
const core_1 = require("@nestjs/core");
const constant_1 = require("../constant");
let A2ARegistry = class A2ARegistry {
    constructor(discovery, metadataScanner) {
        this.discovery = discovery;
        this.metadataScanner = metadataScanner;
        this.discoveredSkills = [];
    }
    onApplicationBootstrap() {
        this.discoverSkills();
    }
    discoverSkills() {
        const providers = this.discovery.getProviders();
        const controllers = this.discovery.getControllers();
        let isFallbackExists = false;
        [...providers, ...controllers]
            .filter((wrapper) => Boolean(wrapper.instance) && typeof wrapper.instance === 'object')
            .forEach(({ instance, token }) => {
            this.metadataScanner
                .getAllMethodNames(instance)
                .forEach((methodName) => {
                const methodRef = instance[methodName];
                const methodMetaKeys = Reflect.getOwnMetadataKeys(methodRef);
                if (methodMetaKeys.includes(constant_1.A2A_AGENT_SKILL_TOKEN)) {
                    const skill = Reflect.getMetadata(constant_1.A2A_AGENT_SKILL_TOKEN, methodRef);
                    /**
                     * Check if there are more than one fallback skill.
                     * If there are, throw an error.
                     */
                    if (skill.fallback) {
                        if (isFallbackExists) {
                            throw new Error('More than one fallback skill found');
                        }
                        isFallbackExists = true;
                    }
                    this.discoveredSkills.push({
                        providerClass: token,
                        methodName,
                        skill: Reflect.getMetadata(constant_1.A2A_AGENT_SKILL_TOKEN, methodRef),
                    });
                }
            });
        });
    }
    getAvailableSkills() {
        return this.discoveredSkills;
    }
    getSkill(id) {
        return this.discoveredSkills.find((skill) => skill.skill.id === id);
    }
    getFallbackSkill() {
        var _a;
        return ((_a = this.discoveredSkills.find((skill) => skill.skill.fallback)) !== null && _a !== void 0 ? _a : this.discoveredSkills[0]);
    }
};
exports.A2ARegistry = A2ARegistry;
exports.A2ARegistry = A2ARegistry = __decorate([
    (0, common_1.Injectable)(),
    __metadata("design:paramtypes", [core_1.DiscoveryService,
        core_1.MetadataScanner])
], A2ARegistry);