@bpinhosilva/agent-orchestrator 1.0.0-alpha.2 → 1.0.0-alpha.21

package/dist/agents/dto/create-agent.dto.js

@@ -0,0 +1,63 @@
+"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.CreateAgentDto = void 0;
+const openapi = require("@nestjs/swagger");
+const class_validator_1 = require("class-validator");
+class CreateAgentDto {
+    name;
+    description;
+    role;
+    systemInstructions;
+    modelId;
+    status;
+    providerId;
+    static _OPENAPI_METADATA_FACTORY() {
+        return { name: { required: true, type: () => String }, description: { required: false, type: () => String }, role: { required: false, type: () => String }, systemInstructions: { required: false, type: () => String }, modelId: { required: true, type: () => String }, status: { required: false, type: () => String }, providerId: { required: true, type: () => String, format: "uuid" } };
+    }
+}
+exports.CreateAgentDto = CreateAgentDto;
+__decorate([
+    (0, class_validator_1.IsString)(),
+    (0, class_validator_1.IsNotEmpty)(),
+    __metadata("design:type", String)
+], CreateAgentDto.prototype, "name", void 0);
+__decorate([
+    (0, class_validator_1.IsString)(),
+    (0, class_validator_1.IsOptional)(),
+    __metadata("design:type", String)
+], CreateAgentDto.prototype, "description", void 0);
+__decorate([
+    (0, class_validator_1.IsString)(),
+    (0, class_validator_1.IsOptional)(),
+    __metadata("design:type", String)
+], CreateAgentDto.prototype, "role", void 0);
+__decorate([
+    (0, class_validator_1.IsString)(),
+    (0, class_validator_1.IsOptional)(),
+    __metadata("design:type", String)
+], CreateAgentDto.prototype, "systemInstructions", void 0);
+__decorate([
+    (0, class_validator_1.IsString)(),
+    (0, class_validator_1.IsNotEmpty)(),
+    __metadata("design:type", String)
+], CreateAgentDto.prototype, "modelId", void 0);
+__decorate([
+    (0, class_validator_1.IsString)(),
+    (0, class_validator_1.IsOptional)(),
+    __metadata("design:type", String)
+], CreateAgentDto.prototype, "status", void 0);
+__decorate([
+    (0, class_validator_1.IsUUID)(),
+    (0, class_validator_1.IsNotEmpty)(),
+    __metadata("design:type", String)
+], CreateAgentDto.prototype, "providerId", void 0);
+//# sourceMappingURL=create-agent.dto.js.map

package/dist/agents/dto/create-agent.dto.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"create-agent.dto.js","sourceRoot":"","sources":["../../../src/agents/dto/create-agent.dto.ts"],"names":[],"mappings":";;;;;;;;;;;;;AAAA,qDAA2E;AAE3E,MAAa,cAAc;IAGzB,IAAI,CAAS;IAIb,WAAW,CAAU;IAIrB,IAAI,CAAU;IAId,kBAAkB,CAAU;IAI5B,OAAO,CAAS;IAIhB,MAAM,CAAU;IAIhB,UAAU,CAAS;;;;CACpB;AA5BD,wCA4BC;AAzBC;IAFC,IAAA,0BAAQ,GAAE;IACV,IAAA,4BAAU,GAAE;;4CACA;AAIb;IAFC,IAAA,0BAAQ,GAAE;IACV,IAAA,4BAAU,GAAE;;mDACQ;AAIrB;IAFC,IAAA,0BAAQ,GAAE;IACV,IAAA,4BAAU,GAAE;;4CACC;AAId;IAFC,IAAA,0BAAQ,GAAE;IACV,IAAA,4BAAU,GAAE;;0DACe;AAI5B;IAFC,IAAA,0BAAQ,GAAE;IACV,IAAA,4BAAU,GAAE;;+CACG;AAIhB;IAFC,IAAA,0BAAQ,GAAE;IACV,IAAA,4BAAU,GAAE;;8CACG;AAIhB;IAFC,IAAA,wBAAM,GAAE;IACR,IAAA,4BAAU,GAAE;;kDACM"}

package/dist/agents/dto/update-agent.dto.d.ts

@@ -0,0 +1,5 @@
+import { CreateAgentDto } from './create-agent.dto';
+declare const UpdateAgentDto_base: import("@nestjs/mapped-types").MappedType<Partial<CreateAgentDto>>;
+export declare class UpdateAgentDto extends UpdateAgentDto_base {
+}
+export {};

package/dist/agents/dto/update-agent.dto.js

@@ -0,0 +1,13 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.UpdateAgentDto = void 0;
+const openapi = require("@nestjs/swagger");
+const mapped_types_1 = require("@nestjs/mapped-types");
+const create_agent_dto_1 = require("./create-agent.dto");
+class UpdateAgentDto extends (0, mapped_types_1.PartialType)(create_agent_dto_1.CreateAgentDto) {
+    static _OPENAPI_METADATA_FACTORY() {
+        return {};
+    }
+}
+exports.UpdateAgentDto = UpdateAgentDto;
+//# sourceMappingURL=update-agent.dto.js.map

package/dist/agents/dto/update-agent.dto.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"update-agent.dto.js","sourceRoot":"","sources":["../../../src/agents/dto/update-agent.dto.ts"],"names":[],"mappings":";;;;AAAA,uDAAmD;AACnD,yDAAoD;AAEpD,MAAa,cAAe,SAAQ,IAAA,0BAAW,EAAC,iCAAc,CAAC;;;;CAAG;AAAlE,wCAAkE"}

package/dist/agents/entities/agent.entity.d.ts

@@ -0,0 +1,14 @@
+import { Model } from '../../models/entities/model.entity';
+import { Provider } from '../../providers/entities/provider.entity';
+export declare class AgentEntity {
+    id: string;
+    name: string;
+    description: string;
+    systemInstructions: string;
+    role: string | null;
+    status: string | null;
+    provider: Provider | null;
+    model: Model | null;
+    createdAt: Date;
+    updatedAt: Date;
+}

package/dist/agents/entities/agent.entity.js

@@ -0,0 +1,87 @@
+"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.AgentEntity = void 0;
+const openapi = require("@nestjs/swagger");
+const typeorm_1 = require("typeorm");
+const model_entity_1 = require("../../models/entities/model.entity");
+const provider_entity_1 = require("../../providers/entities/provider.entity");
+let AgentEntity = class AgentEntity {
+    id;
+    name;
+    description;
+    systemInstructions;
+    role;
+    status;
+    provider;
+    model;
+    createdAt;
+    updatedAt;
+    static _OPENAPI_METADATA_FACTORY() {
+        return { id: { required: true, type: () => String }, name: { required: true, type: () => String }, description: { required: true, type: () => String }, systemInstructions: { required: true, type: () => String }, role: { required: true, type: () => String, nullable: true }, status: { required: true, type: () => String, nullable: true }, provider: { required: true, type: () => require("../../providers/entities/provider.entity").Provider, nullable: true }, model: { required: true, type: () => require("../../models/entities/model.entity").Model, nullable: true }, createdAt: { required: true, type: () => Date }, updatedAt: { required: true, type: () => Date } };
+    }
+};
+exports.AgentEntity = AgentEntity;
+__decorate([
+    (0, typeorm_1.PrimaryGeneratedColumn)('uuid'),
+    __metadata("design:type", String)
+], AgentEntity.prototype, "id", void 0);
+__decorate([
+    (0, typeorm_1.Column)({ unique: true }),
+    __metadata("design:type", String)
+], AgentEntity.prototype, "name", void 0);
+__decorate([
+    (0, typeorm_1.Column)('text', { nullable: true }),
+    __metadata("design:type", String)
+], AgentEntity.prototype, "description", void 0);
+__decorate([
+    (0, typeorm_1.Column)('text', { nullable: true }),
+    __metadata("design:type", String)
+], AgentEntity.prototype, "systemInstructions", void 0);
+__decorate([
+    (0, typeorm_1.Column)({ type: 'text', nullable: true }),
+    __metadata("design:type", Object)
+], AgentEntity.prototype, "role", void 0);
+__decorate([
+    (0, typeorm_1.Column)({ type: 'text', nullable: true }),
+    __metadata("design:type", Object)
+], AgentEntity.prototype, "status", void 0);
+__decorate([
+    (0, typeorm_1.Index)(),
+    (0, typeorm_1.ManyToOne)(() => provider_entity_1.Provider, {
+        nullable: true,
+        eager: true,
+        onDelete: 'SET NULL',
+    }),
+    (0, typeorm_1.JoinColumn)({ name: 'providerId' }),
+    __metadata("design:type", Object)
+], AgentEntity.prototype, "provider", void 0);
+__decorate([
+    (0, typeorm_1.Index)(),
+    (0, typeorm_1.ManyToOne)(() => model_entity_1.Model, (model) => model.agents, {
+        nullable: true,
+        eager: true,
+        onDelete: 'SET NULL',
+    }),
+    __metadata("design:type", Object)
+], AgentEntity.prototype, "model", void 0);
+__decorate([
+    (0, typeorm_1.CreateDateColumn)(),
+    __metadata("design:type", Date)
+], AgentEntity.prototype, "createdAt", void 0);
+__decorate([
+    (0, typeorm_1.UpdateDateColumn)(),
+    __metadata("design:type", Date)
+], AgentEntity.prototype, "updatedAt", void 0);
+exports.AgentEntity = AgentEntity = __decorate([
+    (0, typeorm_1.Entity)('agents')
+], AgentEntity);
+//# sourceMappingURL=agent.entity.js.map

package/dist/agents/entities/agent.entity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.entity.js","sourceRoot":"","sources":["../../../src/agents/entities/agent.entity.ts"],"names":[],"mappings":";;;;;;;;;;;;;AAAA,qCASiB;AACjB,qEAA2D;AAC3D,8EAAoE;AAG7D,IAAM,WAAW,GAAjB,MAAM,WAAW;IAEtB,EAAE,CAAS;IAGX,IAAI,CAAS;IAGb,WAAW,CAAS;IAGpB,kBAAkB,CAAS;IAG3B,IAAI,CAAgB;IAGpB,MAAM,CAAgB;IAStB,QAAQ,CAAkB;IAQ1B,KAAK,CAAe;IAGpB,SAAS,CAAO;IAGhB,SAAS,CAAO;;;;CACjB,CAAA;AAzCY,kCAAW;AAEtB;IADC,IAAA,gCAAsB,EAAC,MAAM,CAAC;;uCACpB;AAGX;IADC,IAAA,gBAAM,EAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;;yCACZ;AAGb;IADC,IAAA,gBAAM,EAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;gDACf;AAGpB;IADC,IAAA,gBAAM,EAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;uDACR;AAG3B;IADC,IAAA,gBAAM,EAAC,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;yCACrB;AAGpB;IADC,IAAA,gBAAM,EAAC,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;2CACnB;AAStB;IAPC,IAAA,eAAK,GAAE;IACP,IAAA,mBAAS,EAAC,GAAG,EAAE,CAAC,0BAAQ,EAAE;QACzB,QAAQ,EAAE,IAAI;QACd,KAAK,EAAE,IAAI;QACX,QAAQ,EAAE,UAAU;KACrB,CAAC;IACD,IAAA,oBAAU,EAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;;6CACT;AAQ1B;IANC,IAAA,eAAK,GAAE;IACP,IAAA,mBAAS,EAAC,GAAG,EAAE,CAAC,oBAAK,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE;QAC/C,QAAQ,EAAE,IAAI;QACd,KAAK,EAAE,IAAI;QACX,QAAQ,EAAE,UAAU;KACrB,CAAC;;0CACkB;AAGpB;IADC,IAAA,0BAAgB,GAAE;8BACR,IAAI;8CAAC;AAGhB;IADC,IAAA,0BAAgB,GAAE;8BACR,IAAI;8CAAC;sBAxCL,WAAW;IADvB,IAAA,gBAAM,EAAC,QAAQ,CAAC;GACJ,WAAW,CAyCvB"}

package/dist/agents/enums/provider.enum.d.ts

@@ -0,0 +1,4 @@
+export declare enum Provider {
+    GOOGLE = "google",
+    ANTHROPIC = "anthropic"
+}

package/dist/agents/enums/provider.enum.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Provider = void 0;
+var Provider;
+(function (Provider) {
+    Provider["GOOGLE"] = "google";
+    Provider["ANTHROPIC"] = "anthropic";
+})(Provider || (exports.Provider = Provider = {}));
+//# sourceMappingURL=provider.enum.js.map

package/dist/agents/enums/provider.enum.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.enum.js","sourceRoot":"","sources":["../../../src/agents/enums/provider.enum.ts"],"names":[],"mappings":";;;AAAA,IAAY,QAGX;AAHD,WAAY,QAAQ;IAClB,6BAAiB,CAAA;IACjB,mCAAuB,CAAA;AACzB,CAAC,EAHW,QAAQ,wBAAR,QAAQ,QAGnB"}

package/dist/agents/implementations/claude.agent.d.ts

@@ -0,0 +1,27 @@
+import { Agent, AgentResponse } from '../interfaces/agent.interface';
+import { Task } from '../../tasks/entities/task.entity';
+import { Project } from '../../projects/entities/project.entity';
+import { ConfigService } from '@nestjs/config';
+export declare class ClaudeAgent implements Agent {
+    private readonly configService;
+    private readonly logger;
+    private name;
+    private description?;
+    private systemInstructions?;
+    private role?;
+    private provider;
+    private model;
+    constructor(configService: ConfigService, model?: string);
+    private validateApiKey;
+    getName(): string;
+    getDescription(): string;
+    getSystemInstructions(): string;
+    getRole(): string;
+    getProvider(): string;
+    getModel(): string;
+    updateConfig(config: Record<string, any>): void;
+    private buildSystemPrompt;
+    processText(input: string): Promise<AgentResponse>;
+    isFeatureSupported(feature: string): boolean;
+    performTask(task: Task, project: Project): Promise<AgentResponse>;
+}

package/dist/agents/implementations/claude.agent.js

@@ -0,0 +1,148 @@
+"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);
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+var ClaudeAgent_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ClaudeAgent = void 0;
+const claude_agent_sdk_1 = require("@anthropic-ai/claude-agent-sdk");
+const common_1 = require("@nestjs/common");
+const config_1 = require("@nestjs/config");
+const agent_registry_1 = require("../registry/agent.registry");
+let ClaudeAgent = ClaudeAgent_1 = class ClaudeAgent {
+    configService;
+    logger = new common_1.Logger(ClaudeAgent_1.name);
+    name = 'ClaudeAgent';
+    description;
+    systemInstructions;
+    role;
+    provider = 'anthropic';
+    model;
+    constructor(configService, model = 'claude-opus-4-6') {
+        this.configService = configService;
+        this.model = model;
+    }
+    validateApiKey() {
+        const apiKey = this.configService.get('ANTHROPIC_API_KEY');
+        if (!apiKey) {
+            this.logger.error('ANTHROPIC_API_KEY environment variable is not set');
+            throw new Error('ANTHROPIC_API_KEY is required to initialize ClaudeAgent');
+        }
+        return apiKey;
+    }
+    getName() {
+        return this.name;
+    }
+    getDescription() {
+        return this.description || '';
+    }
+    getSystemInstructions() {
+        return this.systemInstructions || '';
+    }
+    getRole() {
+        return this.role || '';
+    }
+    getProvider() {
+        return this.provider;
+    }
+    getModel() {
+        return this.model;
+    }
+    updateConfig(config) {
+        if (config['name'])
+            this.name = config['name'];
+        if (config['description'])
+            this.description = config['description'];
+        if (config['systemInstructions'])
+            this.systemInstructions = config['systemInstructions'];
+        if (config['role'])
+            this.role = config['role'];
+        if (config['provider'])
+            this.provider = config['provider'];
+        if (config['model'])
+            this.model = config['model'];
+    }
+    buildSystemPrompt() {
+        return [
+            `Your name is ${this.name}.`,
+            this.role ? `Your role is: ${this.role}.` : '',
+            this.description ? `Description of yourself: ${this.description}.` : '',
+            this.systemInstructions
+                ? `Instructions:\n${this.systemInstructions}`
+                : '',
+        ]
+            .filter(Boolean)
+            .join('\n');
+    }
+    async processText(input) {
+        this.validateApiKey();
+        this.logger.debug(`Processing input with ClaudeAgent. Model: ${this.model}`);
+        try {
+            const systemPrompt = this.buildSystemPrompt();
+            let resultContent = '';
+            let resultMetadata = {};
+            const q = (0, claude_agent_sdk_1.query)({
+                prompt: input,
+                options: {
+                    model: this.model,
+                    systemPrompt,
+                    permissionMode: 'bypassPermissions',
+                    persistSession: false,
+                    tools: [],
+                },
+            });
+            for await (const message of q) {
+                if (message.type === 'result' && message.subtype === 'success') {
+                    resultContent = message.result;
+                    resultMetadata = {
+                        model: this.model,
+                        usage: message.usage,
+                        totalCostUsd: message.total_cost_usd,
+                        numTurns: message.num_turns,
+                    };
+                }
+            }
+            return {
+                content: resultContent,
+                metadata: resultMetadata,
+            };
+        }
+        catch (error) {
+            this.logger.error(`Claude Error: ${error}`);
+            throw error;
+        }
+    }
+    isFeatureSupported(feature) {
+        void feature;
+        return false;
+    }
+    async performTask(task, project) {
+        this.logger.debug(`Performing task: ${task.title} for project: ${project.title}`);
+        const prompt = `
+Task: ${task.title}
+Description: ${task.description}
+Project: ${project.title}
+Project Description: ${project.description}
+
+Please perform the task and provide the output.
+`;
+        return this.processText(prompt);
+    }
+};
+exports.ClaudeAgent = ClaudeAgent;
+exports.ClaudeAgent = ClaudeAgent = ClaudeAgent_1 = __decorate([
+    (0, common_1.Injectable)({ scope: common_1.Scope.TRANSIENT }),
+    (0, agent_registry_1.RegisterAgent)('anthropic'),
+    __param(1, (0, common_1.Optional)()),
+    __metadata("design:paramtypes", [config_1.ConfigService, String])
+], ClaudeAgent);
+//# sourceMappingURL=claude.agent.js.map

package/dist/agents/implementations/claude.agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"claude.agent.js","sourceRoot":"","sources":["../../../src/agents/implementations/claude.agent.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,qEAAmE;AAEnE,2CAAqE;AAGrE,2CAA+C;AAE/C,+DAA2D;AAIpD,IAAM,WAAW,mBAAjB,MAAM,WAAW;IAUH;IATF,MAAM,GAAG,IAAI,eAAM,CAAC,aAAW,CAAC,IAAI,CAAC,CAAC;IAC/C,IAAI,GAAW,aAAa,CAAC;IAC7B,WAAW,CAAU;IACrB,kBAAkB,CAAU;IAC5B,IAAI,CAAU;IACd,QAAQ,GAAW,WAAW,CAAC;IAC/B,KAAK,CAAS;IAEtB,YACmB,aAA4B,EACjC,QAAgB,iBAAiB;QAD5B,kBAAa,GAAb,aAAa,CAAe;QAG7C,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAEO,cAAc;QACpB,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAS,mBAAmB,CAAC,CAAC;QACnE,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,mDAAmD,CAAC,CAAC;YACvE,MAAM,IAAI,KAAK,CACb,yDAAyD,CAC1D,CAAC;QACJ,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAED,cAAc;QACZ,OAAO,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC;IAChC,CAAC;IAED,qBAAqB;QACnB,OAAO,IAAI,CAAC,kBAAkB,IAAI,EAAE,CAAC;IACvC,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;IACzB,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,YAAY,CAAC,MAA2B;QACtC,IAAI,MAAM,CAAC,MAAM,CAAC;YAAE,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,MAAM,CAAW,CAAC;QACzD,IAAI,MAAM,CAAC,aAAa,CAAC;YACvB,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,aAAa,CAAW,CAAC;QACrD,IAAI,MAAM,CAAC,oBAAoB,CAAC;YAC9B,IAAI,CAAC,kBAAkB,GAAG,MAAM,CAAC,oBAAoB,CAAW,CAAC;QACnE,IAAI,MAAM,CAAC,MAAM,CAAC;YAAE,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,MAAM,CAAW,CAAC;QACzD,IAAI,MAAM,CAAC,UAAU,CAAC;YAAE,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,UAAU,CAAW,CAAC;QACrE,IAAI,MAAM,CAAC,OAAO,CAAC;YAAE,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,OAAO,CAAW,CAAC;IAC9D,CAAC;IAEO,iBAAiB;QACvB,OAAO;YACL,gBAAgB,IAAI,CAAC,IAAI,GAAG;YAC5B,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,iBAAiB,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,EAAE;YAC9C,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,4BAA4B,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC,EAAE;YACvE,IAAI,CAAC,kBAAkB;gBACrB,CAAC,CAAC,kBAAkB,IAAI,CAAC,kBAAkB,EAAE;gBAC7C,CAAC,CAAC,EAAE;SACP;aACE,MAAM,CAAC,OAAO,CAAC;aACf,IAAI,CAAC,IAAI,CAAC,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAa;QAC7B,IAAI,CAAC,cAAc,EAAE,CAAC;QACtB,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,6CAA6C,IAAI,CAAC,KAAK,EAAE,CAC1D,CAAC;QACF,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC9C,IAAI,aAAa,GAAG,EAAE,CAAC;YACvB,IAAI,cAAc,GAAwB,EAAE,CAAC;YAE7C,MAAM,CAAC,GAAG,IAAA,wBAAK,EAAC;gBACd,MAAM,EAAE,KAAK;gBACb,OAAO,EAAE;oBACP,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,YAAY;oBACZ,cAAc,EAAE,mBAAmB;oBACnC,cAAc,EAAE,KAAK;oBACrB,KAAK,EAAE,EAAE;iBACV;aACF,CAAC,CAAC;YAEH,IAAI,KAAK,EAAE,MAAM,OAAO,IAAI,CAA8B,EAAE,CAAC;gBAC3D,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;oBAC/D,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC;oBAC/B,cAAc,GAAG;wBACf,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,KAAK,EAAE,OAAO,CAAC,KAAK;wBACpB,YAAY,EAAE,OAAO,CAAC,cAAc;wBACpC,QAAQ,EAAE,OAAO,CAAC,SAAS;qBAC5B,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,aAAa;gBACtB,QAAQ,EAAE,cAAc;aACzB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iBAAiB,KAAK,EAAE,CAAC,CAAC;YAC5C,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,kBAAkB,CAAC,OAAe;QAChC,KAAK,OAAO,CAAC;QACb,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,IAAU,EAAE,OAAgB;QAC5C,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,oBAAoB,IAAI,CAAC,KAAK,iBAAiB,OAAO,CAAC,KAAK,EAAE,CAC/D,CAAC;QACF,MAAM,MAAM,GAAG;QACX,IAAI,CAAC,KAAK;eACH,IAAI,CAAC,WAAW;WACpB,OAAO,CAAC,KAAK;uBACD,OAAO,CAAC,WAAW;;;CAGzC,CAAC;QACE,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;CACF,CAAA;AAzIY,kCAAW;sBAAX,WAAW;IAFvB,IAAA,mBAAU,EAAC,EAAE,KAAK,EAAE,cAAK,CAAC,SAAS,EAAE,CAAC;IACtC,IAAA,8BAAa,EAAC,WAAW,CAAC;IAYtB,WAAA,IAAA,iBAAQ,GAAE,CAAA;qCADqB,sBAAa;GAVpC,WAAW,CAyIvB"}

package/dist/agents/implementations/gemini.agent.d.ts

@@ -1,8 +1,30 @@
 import { Agent, AgentResponse } from '../interfaces/agent.interface';
+import { Task } from '../../tasks/entities/task.entity';
+import { Project } from '../../projects/entities/project.entity';
+import { ConfigService } from '@nestjs/config';
 export declare class GeminiAgent implements Agent {
+    private readonly configService;
     private readonly logger;
     private genAI;
-    constructor();
+    private name;
+    private description?;
+    private systemInstructions?;
+    private role?;
+    private provider;
+    private model;
+    private enableGrounding;
+    constructor(configService: ConfigService, model?: string);
+    private getGenAI;
     getName(): string;
+    getDescription(): string;
+    getSystemInstructions(): string;
+    getRole(): string;
+    getProvider(): string;
+    getModel(): string;
+    updateConfig(config: Record<string, any>): void;
     processText(input: string): Promise<AgentResponse>;
+    isFeatureSupported(feature: string): boolean;
+    private getTools;
+    private isGroundingSupported;
+    performTask(task: Task, project: Project): Promise<AgentResponse>;
 }

package/dist/agents/implementations/gemini.agent.js

@@ -8,52 +8,187 @@ var __decorate = (this && this.__decorate) || function (decorators, target, key,
 var __metadata = (this && this.__metadata) || function (k, v) {
     if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
 };
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
 var GeminiAgent_1;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.GeminiAgent = void 0;
-const generative_ai_1 = require("@google/generative-ai");
+const genai_1 = require("@google/genai");
 const common_1 = require("@nestjs/common");
+const config_1 = require("@nestjs/config");
+const agent_registry_1 = require("../registry/agent.registry");
 let GeminiAgent = GeminiAgent_1 = class GeminiAgent {
-    constructor() {
-        this.logger = new common_1.Logger(GeminiAgent_1.name);
-        const apiKey = process.env.GEMINI_API_KEY;
+    configService;
+    logger = new common_1.Logger(GeminiAgent_1.name);
+    genAI = null;
+    name = 'GeminiAgent';
+    description;
+    systemInstructions;
+    role;
+    provider = 'google';
+    model;
+    enableGrounding = true;
+    constructor(configService, model = 'gemini-2.5-flash-lite') {
+        this.configService = configService;
+        this.model = model;
+    }
+    getGenAI() {
+        if (this.genAI)
+            return this.genAI;
+        const apiKey = this.configService.get('GEMINI_API_KEY');
         if (!apiKey) {
             this.logger.error('GEMINI_API_KEY environment variable is not set');
             throw new Error('GEMINI_API_KEY is required to initialize GeminiAgent');
         }
-        this.genAI = new generative_ai_1.GoogleGenerativeAI(apiKey);
+        this.genAI = new genai_1.GoogleGenAI({ apiKey });
+        return this.genAI;
     }
     getName() {
-        return 'GeminiAgent';
+        return this.name;
+    }
+    getDescription() {
+        return this.description || '';
+    }
+    getSystemInstructions() {
+        return this.systemInstructions || '';
+    }
+    getRole() {
+        return this.role || '';
+    }
+    getProvider() {
+        return this.provider;
+    }
+    getModel() {
+        return this.model;
+    }
+    updateConfig(config) {
+        if (config['name'])
+            this.name = config['name'];
+        if (config['description'])
+            this.description = config['description'];
+        if (config['systemInstructions'])
+            this.systemInstructions = config['systemInstructions'];
+        if (config['role'])
+            this.role = config['role'];
+        if (config['provider'])
+            this.provider = config['provider'];
+        if (config['model'])
+            this.model = config['model'];
+        if (config['enableGrounding'] !== undefined)
+            this.enableGrounding = !!config['enableGrounding'];
     }
     async processText(input) {
-        this.logger.debug(`Processing input with GeminiAgent`);
+        const isGrounded = this.isGroundingSupported();
+        this.logger.debug(`Processing input with GeminiAgent. Model: ${this.model}, Grounding Supported: ${isGrounded}, Grounding Enabled: ${this.enableGrounding}`);
         try {
-            const model = this.genAI.getGenerativeModel({
-                model: 'gemini-2.5-flash-lite',
+            const contents = input;
+            const tools = this.getTools();
+            if (tools.length > 0) {
+                this.logger.debug(`Adding tools to request: ${JSON.stringify(tools)}`);
+            }
+            const genAI = this.getGenAI();
+            const response = await genAI.models.generateContent({
+                model: this.model,
+                contents,
+                config: {
+                    systemInstruction: [
+                        `Your name is ${this.name}.`,
+                        this.role ? `Your role is: ${this.role}.` : '',
+                        this.description
+                            ? `Description of yourself: ${this.description}.`
+                            : '',
+                        this.systemInstructions
+                            ? `Instructions:\n${this.systemInstructions}`
+                            : '',
+                    ]
+                        .filter(Boolean)
+                        .join('\n'),
+                    tools,
+                },
             });
-            const result = await model.generateContent(input);
-            const output = result.response.text();
+            const candidate = response.candidates?.[0];
+            const groundingMetadata = candidate?.groundingMetadata;
+            if (groundingMetadata) {
+                this.logger.debug(`Grounding Metadata found: ${JSON.stringify(groundingMetadata, null, 2)}`);
+            }
             return {
-                content: output,
+                content: response.text || '',
+                image: response.data,
                 metadata: {
-                    model: 'gemini-2.5-flash-lite',
+                    model: this.model,
+                    groundingMetadata: groundingMetadata || undefined,
+                    usage: response.usageMetadata,
                 },
             };
         }
         catch (error) {
-            if (error instanceof Error) {
-                this.logger.error(`Error processing text: ${error.message}`, error.stack);
-            }
-            else {
-                this.logger.error(`Error processing text: ${String(error)}`);
-            }
+            this.logger.error(`Gemini Error: ${error}`);
             throw error;
         }
     }
+    isFeatureSupported(feature) {
+        if (feature === 'grounding' || feature === 'googleSearch') {
+            return this.isGroundingSupported();
+        }
+        return false;
+    }
+    getTools() {
+        const tools = [];
+        if (this.enableGrounding && this.isGroundingSupported()) {
+            tools.push({
+                googleSearch: {
+                    searchTypes: {
+                        webSearch: {},
+                    },
+                },
+            });
+        }
+        return tools;
+    }
+    isGroundingSupported() {
+        const supportedPrefixes = [
+            'gemini-1.5-flash',
+            'gemini-1.5-pro',
+            'gemini-2.0-flash',
+            'gemini-2.0-pro',
+            'gemini-2.5-flash',
+            'gemini-2.5-pro',
+            'gemini-3.1-flash',
+            'gemini-3.1-pro',
+            'gemini-3-flash',
+        ];
+        const excludedSuffixes = [
+            '-image',
+            '-vision',
+            '-thinking',
+            '-experimental',
+        ];
+        const modelName = this.model.startsWith('models/')
+            ? this.model.slice(7)
+            : this.model;
+        const hasSupportedPrefix = supportedPrefixes.some((p) => modelName.startsWith(p));
+        const hasExcludedSuffix = excludedSuffixes.some((s) => modelName.includes(s));
+        return hasSupportedPrefix && !hasExcludedSuffix;
+    }
+    async performTask(task, project) {
+        this.logger.debug(`Performing task: ${task.title} for project: ${project.title}`);
+        const prompt = `
+Task: ${task.title}
+Description: ${task.description}
+Project: ${project.title}
+Project Description: ${project.description}
+
+Please perform the task and provide the output.
+`;
+        return this.processText(prompt);
+    }
 };
 exports.GeminiAgent = GeminiAgent;
 exports.GeminiAgent = GeminiAgent = GeminiAgent_1 = __decorate([
-    (0, common_1.Injectable)(),
-    __metadata("design:paramtypes", [])
+    (0, common_1.Injectable)({ scope: common_1.Scope.TRANSIENT }),
+    (0, agent_registry_1.RegisterAgent)('google'),
+    __param(1, (0, common_1.Optional)()),
+    __metadata("design:paramtypes", [config_1.ConfigService, String])
 ], GeminiAgent);
+//# sourceMappingURL=gemini.agent.js.map