@bpinhosilva/agent-orchestrator 1.0.0-alpha.12 → 1.0.0-alpha.13

package/CHANGELOG.md

@@ -1,3 +1,10 @@
+# [1.0.0-alpha.13](https://github.com/bpinhosilva/agent-orchestrator/compare/v1.0.0-alpha.12...v1.0.0-alpha.13) (2026-03-26)
+
+
+### Features
+
+* add Claude agent implementation with Anthropic SDK and refactor agent-provider ([1400a0d](https://github.com/bpinhosilva/agent-orchestrator/commit/1400a0dec6beb8393ddc4c5a23b9b860532079e8))
+
 # [1.0.0-alpha.12](https://github.com/bpinhosilva/agent-orchestrator/compare/v1.0.0-alpha.11...v1.0.0-alpha.12) (2026-03-24)
 
 

package/dist/agents/agents.module.js

@@ -12,7 +12,10 @@ const typeorm_1 = require("@nestjs/typeorm");
 const agents_controller_1 = require("./agents.controller");
 const agents_service_1 = require("./agents.service");
 const agent_entity_1 = require("./entities/agent.entity");
-const gemini_agent_1 = require("./implementations/gemini.agent");
+// Side-effect imports: @RegisterAgent decorators fire at class-definition time
+// and self-register each implementation in AGENT_REGISTRY. No DI instantiation needed.
+require("./implementations/gemini.agent");
+require("./implementations/claude.agent");
 let AgentsModule = class AgentsModule {
 };
 exports.AgentsModule = AgentsModule;
@@ -20,7 +23,7 @@ exports.AgentsModule = AgentsModule = __decorate([
     (0, common_1.Module)({
         imports: [typeorm_1.TypeOrmModule.forFeature([agent_entity_1.AgentEntity])],
         controllers: [agents_controller_1.AgentsController],
-        providers: [agents_service_1.AgentsService, gemini_agent_1.GeminiAgent],
+        providers: [agents_service_1.AgentsService],
         exports: [agents_service_1.AgentsService],
     })
 ], AgentsModule);

package/dist/agents/agents.service.js

@@ -19,6 +19,7 @@ const typeorm_1 = require("@nestjs/typeorm");
 const typeorm_2 = require("typeorm");
 const agent_entity_1 = require("./entities/agent.entity");
 const agent_registry_1 = require("./registry/agent.registry");
+const AGENT_RELATIONS = ['model', 'provider'];
 let AgentsService = AgentsService_1 = class AgentsService {
     constructor(agentRepository) {
         this.agentRepository = agentRepository;
@@ -27,7 +28,9 @@ let AgentsService = AgentsService_1 = class AgentsService {
     }
     async onModuleInit() {
         this.logger.log('Initializing Agent instances from database...');
-        const agents = await this.agentRepository.find({ relations: ['model'] });
+        const agents = await this.agentRepository.find({
+            relations: AGENT_RELATIONS,
+        });
         for (const agentEntity of agents) {
             if (agentEntity.status !== 'inactive') {
                 try {
@@ -46,14 +49,14 @@ let AgentsService = AgentsService_1 = class AgentsService {
             this.agentInstances.delete(agentEntity.id);
             return;
         }
-        const provider = agentEntity.provider;
-        if (!provider) {
+        const providerName = agentEntity.provider?.name?.toLowerCase();
+        if (!providerName) {
             this.logger.warn(`No provider found for agent #${agentEntity.id}. Skipping instance creation.`);
             return;
         }
-        const AgentClass = (0, agent_registry_1.getAgentImplementation)(provider);
+        const AgentClass = (0, agent_registry_1.getAgentImplementation)(providerName);
         if (!AgentClass) {
-            this.logger.warn(`No implementation found for provider: ${provider}. Skipping instance creation.`);
+            this.logger.warn(`No implementation found for provider: ${providerName}. Skipping instance creation.`);
             return;
         }
         try {
@@ -63,7 +66,7 @@ let AgentsService = AgentsService_1 = class AgentsService {
                 description: agentEntity.description,
                 systemInstructions: agentEntity.systemInstructions,
                 role: agentEntity.role,
-                provider: agentEntity.provider,
+                provider: providerName,
             });
             this.agentInstances.set(agentEntity.id, instance);
             this.logger.debug(`Synchronized agent instance #${agentEntity.id}`);
@@ -71,24 +74,41 @@ let AgentsService = AgentsService_1 = class AgentsService {
         catch (error) {
             const errorMessage = error instanceof Error ? error.message : String(error);
             this.logger.error(`Error synchronizing agent #${agentEntity.id}: ${errorMessage}`);
-            throw error;
+            throw new common_1.BadRequestException(`Failed to initialize agent instance: ${errorMessage}`);
         }
     }
     async create(createAgentDto) {
-        const agent = this.agentRepository.create(createAgentDto);
-        const savedAgent = await this.agentRepository.save(agent);
-        // Reload with relations for proper synchronization
-        const fullyLoadedAgent = await this.findOne(savedAgent.id);
-        this.syncAgentInstance(fullyLoadedAgent);
-        return fullyLoadedAgent;
+        const { modelId, providerId, ...rest } = createAgentDto;
+        const agentData = { ...rest };
+        if (modelId) {
+            agentData.model = { id: modelId };
+        }
+        if (providerId) {
+            agentData.provider = { id: providerId };
+        }
+        return await this.agentRepository.manager.transaction(async (manager) => {
+            const agent = manager.create(agent_entity_1.AgentEntity, agentData);
+            const savedAgent = await manager.save(agent);
+            // Reload with full relations for proper synchronization
+            const fullyLoadedAgent = await manager.findOne(agent_entity_1.AgentEntity, {
+                where: { id: savedAgent.id },
+                relations: AGENT_RELATIONS,
+            });
+            if (!fullyLoadedAgent) {
+                throw new Error(`Agent #${savedAgent.id} not found immediately after save`);
+            }
+            // This will throw BadRequestException if instantiation fails, rolling back the transaction
+            this.syncAgentInstance(fullyLoadedAgent);
+            return fullyLoadedAgent;
+        });
     }
     async findAll() {
-        return this.agentRepository.find({ relations: ['model'] });
+        return this.agentRepository.find({ relations: AGENT_RELATIONS });
     }
     async findOne(id) {
         const agent = await this.agentRepository.findOne({
             where: { id },
-            relations: ['model'],
+            relations: AGENT_RELATIONS,
         });
         if (!agent) {
             throw new common_1.NotFoundException(`Agent #${id} not found`);
@@ -96,21 +116,27 @@ let AgentsService = AgentsService_1 = class AgentsService {
         return agent;
     }
     async update(id, updateAgentDto) {
-        const { modelId, ...rest } = updateAgentDto;
+        const { modelId, providerId, ...rest } = updateAgentDto;
         const updateData = { ...rest };
         if (modelId) {
             updateData.model = { id: modelId };
         }
-        await this.agentRepository.update(id, updateData);
-        const updatedAgent = await this.agentRepository.findOne({
-            where: { id },
-            relations: ['model'],
-        });
-        if (!updatedAgent) {
-            throw new common_1.NotFoundException(`Agent #${id} not found`);
+        if (providerId) {
+            updateData.provider = { id: providerId };
         }
-        this.syncAgentInstance(updatedAgent);
-        return updatedAgent;
+        return await this.agentRepository.manager.transaction(async (manager) => {
+            await manager.update(agent_entity_1.AgentEntity, id, updateData);
+            const updatedAgent = await manager.findOne(agent_entity_1.AgentEntity, {
+                where: { id },
+                relations: AGENT_RELATIONS,
+            });
+            if (!updatedAgent) {
+                throw new common_1.NotFoundException(`Agent #${id} not found`);
+            }
+            // This will throw BadRequestException if instantiation fails, rolling back the transaction
+            this.syncAgentInstance(updatedAgent);
+            return updatedAgent;
+        });
     }
     async remove(id) {
         const agent = await this.findOne(id);
@@ -138,7 +164,7 @@ let AgentsService = AgentsService_1 = class AgentsService {
             this.logger.debug(`Agent instance #${agentId} not found in memory. Checking database...`);
             const agentEntity = await this.agentRepository.findOne({
                 where: { id: agentId },
-                relations: ['model'],
+                relations: AGENT_RELATIONS,
             });
             if (!agentEntity) {
                 throw new common_1.NotFoundException(`Agent #${agentId} not found`);

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

@@ -45,7 +45,7 @@ __decorate([
     __metadata("design:type", String)
 ], CreateAgentDto.prototype, "status", void 0);
 __decorate([
-    (0, class_validator_1.IsString)(),
-    (0, class_validator_1.IsOptional)(),
+    (0, class_validator_1.IsUUID)(),
+    (0, class_validator_1.IsNotEmpty)(),
     __metadata("design:type", String)
-], CreateAgentDto.prototype, "provider", void 0);
+], CreateAgentDto.prototype, "providerId", void 0);

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

@@ -12,6 +12,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.AgentEntity = void 0;
 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 {
 };
 exports.AgentEntity = AgentEntity;
@@ -40,8 +41,13 @@ __decorate([
     __metadata("design:type", String)
 ], AgentEntity.prototype, "status", void 0);
 __decorate([
-    (0, typeorm_1.Column)({ type: 'text', nullable: true }),
-    __metadata("design:type", String)
+    (0, typeorm_1.ManyToOne)(() => provider_entity_1.Provider, {
+        nullable: true,
+        eager: true,
+        onDelete: 'SET NULL',
+    }),
+    (0, typeorm_1.JoinColumn)({ name: 'providerId' }),
+    __metadata("design:type", provider_entity_1.Provider)
 ], AgentEntity.prototype, "provider", void 0);
 __decorate([
     (0, typeorm_1.ManyToOne)(() => model_entity_1.Model, (model) => model.agents, {

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

@@ -0,0 +1,136 @@
+"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 agent_registry_1 = require("../registry/agent.registry");
+let ClaudeAgent = ClaudeAgent_1 = class ClaudeAgent {
+    constructor(model = 'claude-opus-4-6') {
+        this.logger = new common_1.Logger(ClaudeAgent_1.name);
+        this.name = 'ClaudeAgent';
+        this.provider = 'anthropic';
+        this.model = model;
+        const apiKey = process.env.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');
+        }
+    }
+    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.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)(),
+    (0, agent_registry_1.RegisterAgent)('anthropic'),
+    __param(0, (0, common_1.Optional)()),
+    __metadata("design:paramtypes", [String])
+], ClaudeAgent);

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

@@ -21,7 +21,7 @@ let GeminiAgent = GeminiAgent_1 = class GeminiAgent {
     constructor(model = 'gemini-2.5-flash-lite') {
         this.logger = new common_1.Logger(GeminiAgent_1.name);
         this.name = 'GeminiAgent';
-        this.provider = 'Google';
+        this.provider = 'google';
         this.enableGrounding = true;
         this.model = model;
         const apiKey = process.env.GEMINI_API_KEY;
@@ -173,7 +173,7 @@ Please perform the task and provide the output.
 exports.GeminiAgent = GeminiAgent;
 exports.GeminiAgent = GeminiAgent = GeminiAgent_1 = __decorate([
     (0, common_1.Injectable)(),
-    (0, agent_registry_1.RegisterAgent)('Google'),
+    (0, agent_registry_1.RegisterAgent)('google'),
     __param(0, (0, common_1.Optional)()),
     __metadata("design:paramtypes", [String])
 ], GeminiAgent);

package/dist/common/storage.service.js

@@ -48,6 +48,7 @@ const common_1 = require("@nestjs/common");
 const os = __importStar(require("os"));
 const path = __importStar(require("path"));
 const fs = __importStar(require("fs"));
+const crypto = __importStar(require("crypto"));
 let StorageService = StorageService_1 = class StorageService {
     constructor() {
         this.logger = new common_1.Logger(StorageService_1.name);
@@ -65,11 +66,34 @@ let StorageService = StorageService_1 = class StorageService {
         return path.join(this.baseDir, 'artifacts');
     }
     getFullPath(relativePath) {
-        // relativePath is like /uploads/artifacts/<uuid>.extension
-        // We only care about the filename part if the prefix is fixed
         const filename = path.basename(relativePath);
         return path.join(this.getArtifactsPath(), filename);
     }
+    async saveBase64(base64, mimeType, originalName) {
+        const buffer = Buffer.from(base64, 'base64');
+        return this.saveBuffer(buffer, mimeType, originalName);
+    }
+    async saveBuffer(buffer, mimeType, originalName) {
+        const id = crypto.randomUUID();
+        const extension = mimeType.split('/')[1] || 'bin';
+        const filename = `${id}.${extension}`;
+        const filePath = path.join(this.getArtifactsPath(), filename);
+        await fs.promises.writeFile(filePath, buffer);
+        this.logger.debug(`Saved artifact to: ${filePath}`);
+        return {
+            id,
+            originalName,
+            mimeType,
+            filePath: `uploads/artifacts/${filename}`, // Publicly accessible path via UploadsController
+        };
+    }
+    async delete(relativePath) {
+        const fullPath = this.getFullPath(relativePath);
+        if (fs.existsSync(fullPath)) {
+            await fs.promises.unlink(fullPath);
+            this.logger.debug(`Deleted artifact file: ${fullPath}`);
+        }
+    }
 };
 exports.StorageService = StorageService;
 exports.StorageService = StorageService = StorageService_1 = __decorate([

package/dist/providers/entities/provider.entity.js

@@ -28,7 +28,7 @@ __decorate([
     __metadata("design:type", String)
 ], Provider.prototype, "description", void 0);
 __decorate([
-    (0, typeorm_1.OneToMany)(() => model_entity_1.Model, (model) => model.provider),
+    (0, typeorm_1.OneToMany)(() => model_entity_1.Model, (model) => model.provider, { cascade: true }),
     __metadata("design:type", Array)
 ], Provider.prototype, "models", void 0);
 __decorate([

package/dist/providers/providers.service.js

@@ -21,6 +21,50 @@ let ProvidersService = class ProvidersService {
     constructor(providerRepository) {
         this.providerRepository = providerRepository;
     }
+    async onModuleInit() {
+        await this.seedProviders();
+    }
+    async seedProviders() {
+        const defaultProviders = [
+            {
+                name: 'google',
+                description: 'Google Generative AI (Gemini)',
+                models: [
+                    { name: 'gemini-2.5-flash-lite' },
+                    { name: 'gemini-2.5-flash-image' },
+                    { name: 'gemini-2.5-pro' },
+                ],
+            },
+            {
+                name: 'anthropic',
+                description: 'Anthropic Claude AI',
+                models: [{ name: 'claude-opus-4-6' }, { name: 'claude-haiku-4-5' }],
+            },
+        ];
+        for (const p of defaultProviders) {
+            let provider = await this.providerRepository.findOne({
+                where: { name: p.name },
+                relations: ['models'],
+            });
+            if (!provider) {
+                provider = this.providerRepository.create(p);
+                await this.providerRepository.save(provider);
+            }
+            else {
+                // Check for missing models
+                let changed = false;
+                for (const m of p.models) {
+                    if (!provider.models.some((existing) => existing.name === m.name)) {
+                        provider.models.push(m);
+                        changed = true;
+                    }
+                }
+                if (changed) {
+                    await this.providerRepository.save(provider);
+                }
+            }
+        }
+    }
     async create(createProviderDto) {
         const provider = this.providerRepository.create(createProviderDto);
         return this.providerRepository.save(provider);

package/dist/src/agents/agents.module.d.ts

@@ -1,2 +1,4 @@
+import './implementations/gemini.agent';
+import './implementations/claude.agent';
 export declare class AgentsModule {
 }

package/dist/src/agents/agents.module.js

@@ -12,7 +12,8 @@ const typeorm_1 = require("@nestjs/typeorm");
 const agents_controller_1 = require("./agents.controller");
 const agents_service_1 = require("./agents.service");
 const agent_entity_1 = require("./entities/agent.entity");
-const gemini_agent_1 = require("./implementations/gemini.agent");
+require("./implementations/gemini.agent");
+require("./implementations/claude.agent");
 let AgentsModule = class AgentsModule {
 };
 exports.AgentsModule = AgentsModule;
@@ -20,7 +21,7 @@ exports.AgentsModule = AgentsModule = __decorate([
     (0, common_1.Module)({
         imports: [typeorm_1.TypeOrmModule.forFeature([agent_entity_1.AgentEntity])],
         controllers: [agents_controller_1.AgentsController],
-        providers: [agents_service_1.AgentsService, gemini_agent_1.GeminiAgent],
+        providers: [agents_service_1.AgentsService],
         exports: [agents_service_1.AgentsService],
     })
 ], AgentsModule);

package/dist/src/agents/agents.module.js.map

@@ -1 +1 @@
-{"version":3,"file":"agents.module.js","sourceRoot":"","sources":["../../../src/agents/agents.module.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAAwC;AACxC,6CAAgD;AAChD,2DAAuD;AACvD,qDAAiD;AACjD,0DAAsD;AACtD,iEAA6D;AAQtD,IAAM,YAAY,GAAlB,MAAM,YAAY;CAAG,CAAA;AAAf,oCAAY;uBAAZ,YAAY;IANxB,IAAA,eAAM,EAAC;QACN,OAAO,EAAE,CAAC,uBAAa,CAAC,UAAU,CAAC,CAAC,0BAAW,CAAC,CAAC,CAAC;QAClD,WAAW,EAAE,CAAC,oCAAgB,CAAC;QAC/B,SAAS,EAAE,CAAC,8BAAa,EAAE,0BAAW,CAAC;QACvC,OAAO,EAAE,CAAC,8BAAa,CAAC;KACzB,CAAC;GACW,YAAY,CAAG"}
+{"version":3,"file":"agents.module.js","sourceRoot":"","sources":["../../../src/agents/agents.module.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAAwC;AACxC,6CAAgD;AAChD,2DAAuD;AACvD,qDAAiD;AACjD,0DAAsD;AAGtD,0CAAwC;AACxC,0CAAwC;AAQjC,IAAM,YAAY,GAAlB,MAAM,YAAY;CAAG,CAAA;AAAf,oCAAY;uBAAZ,YAAY;IANxB,IAAA,eAAM,EAAC;QACN,OAAO,EAAE,CAAC,uBAAa,CAAC,UAAU,CAAC,CAAC,0BAAW,CAAC,CAAC,CAAC;QAClD,WAAW,EAAE,CAAC,oCAAgB,CAAC;QAC/B,SAAS,EAAE,CAAC,8BAAa,CAAC;QAC1B,OAAO,EAAE,CAAC,8BAAa,CAAC;KACzB,CAAC;GACW,YAAY,CAAG"}

package/dist/src/agents/agents.service.js

@@ -19,6 +19,7 @@ const typeorm_1 = require("@nestjs/typeorm");
 const typeorm_2 = require("typeorm");
 const agent_entity_1 = require("./entities/agent.entity");
 const agent_registry_1 = require("./registry/agent.registry");
+const AGENT_RELATIONS = ['model', 'provider'];
 let AgentsService = AgentsService_1 = class AgentsService {
     agentRepository;
     logger = new common_1.Logger(AgentsService_1.name);
@@ -28,7 +29,9 @@ let AgentsService = AgentsService_1 = class AgentsService {
     }
     async onModuleInit() {
         this.logger.log('Initializing Agent instances from database...');
-        const agents = await this.agentRepository.find({ relations: ['model'] });
+        const agents = await this.agentRepository.find({
+            relations: AGENT_RELATIONS,
+        });
         for (const agentEntity of agents) {
             if (agentEntity.status !== 'inactive') {
                 try {
@@ -47,14 +50,14 @@ let AgentsService = AgentsService_1 = class AgentsService {
             this.agentInstances.delete(agentEntity.id);
             return;
         }
-        const provider = agentEntity.provider;
-        if (!provider) {
+        const providerName = agentEntity.provider?.name?.toLowerCase();
+        if (!providerName) {
             this.logger.warn(`No provider found for agent #${agentEntity.id}. Skipping instance creation.`);
             return;
         }
-        const AgentClass = (0, agent_registry_1.getAgentImplementation)(provider);
+        const AgentClass = (0, agent_registry_1.getAgentImplementation)(providerName);
         if (!AgentClass) {
-            this.logger.warn(`No implementation found for provider: ${provider}. Skipping instance creation.`);
+            this.logger.warn(`No implementation found for provider: ${providerName}. Skipping instance creation.`);
             return;
         }
         try {
@@ -64,7 +67,7 @@ let AgentsService = AgentsService_1 = class AgentsService {
                 description: agentEntity.description,
                 systemInstructions: agentEntity.systemInstructions,
                 role: agentEntity.role,
-                provider: agentEntity.provider,
+                provider: providerName,
             });
             this.agentInstances.set(agentEntity.id, instance);
             this.logger.debug(`Synchronized agent instance #${agentEntity.id}`);
@@ -72,23 +75,39 @@ let AgentsService = AgentsService_1 = class AgentsService {
         catch (error) {
             const errorMessage = error instanceof Error ? error.message : String(error);
             this.logger.error(`Error synchronizing agent #${agentEntity.id}: ${errorMessage}`);
-            throw error;
+            throw new common_1.BadRequestException(`Failed to initialize agent instance: ${errorMessage}`);
         }
     }
     async create(createAgentDto) {
-        const agent = this.agentRepository.create(createAgentDto);
-        const savedAgent = await this.agentRepository.save(agent);
-        const fullyLoadedAgent = await this.findOne(savedAgent.id);
-        this.syncAgentInstance(fullyLoadedAgent);
-        return fullyLoadedAgent;
+        const { modelId, providerId, ...rest } = createAgentDto;
+        const agentData = { ...rest };
+        if (modelId) {
+            agentData.model = { id: modelId };
+        }
+        if (providerId) {
+            agentData.provider = { id: providerId };
+        }
+        return await this.agentRepository.manager.transaction(async (manager) => {
+            const agent = manager.create(agent_entity_1.AgentEntity, agentData);
+            const savedAgent = await manager.save(agent);
+            const fullyLoadedAgent = await manager.findOne(agent_entity_1.AgentEntity, {
+                where: { id: savedAgent.id },
+                relations: AGENT_RELATIONS,
+            });
+            if (!fullyLoadedAgent) {
+                throw new Error(`Agent #${savedAgent.id} not found immediately after save`);
+            }
+            this.syncAgentInstance(fullyLoadedAgent);
+            return fullyLoadedAgent;
+        });
     }
     async findAll() {
-        return this.agentRepository.find({ relations: ['model'] });
+        return this.agentRepository.find({ relations: AGENT_RELATIONS });
     }
     async findOne(id) {
         const agent = await this.agentRepository.findOne({
             where: { id },
-            relations: ['model'],
+            relations: AGENT_RELATIONS,
         });
         if (!agent) {
             throw new common_1.NotFoundException(`Agent #${id} not found`);
@@ -96,21 +115,26 @@ let AgentsService = AgentsService_1 = class AgentsService {
         return agent;
     }
     async update(id, updateAgentDto) {
-        const { modelId, ...rest } = updateAgentDto;
+        const { modelId, providerId, ...rest } = updateAgentDto;
         const updateData = { ...rest };
         if (modelId) {
             updateData.model = { id: modelId };
         }
-        await this.agentRepository.update(id, updateData);
-        const updatedAgent = await this.agentRepository.findOne({
-            where: { id },
-            relations: ['model'],
-        });
-        if (!updatedAgent) {
-            throw new common_1.NotFoundException(`Agent #${id} not found`);
+        if (providerId) {
+            updateData.provider = { id: providerId };
         }
-        this.syncAgentInstance(updatedAgent);
-        return updatedAgent;
+        return await this.agentRepository.manager.transaction(async (manager) => {
+            await manager.update(agent_entity_1.AgentEntity, id, updateData);
+            const updatedAgent = await manager.findOne(agent_entity_1.AgentEntity, {
+                where: { id },
+                relations: AGENT_RELATIONS,
+            });
+            if (!updatedAgent) {
+                throw new common_1.NotFoundException(`Agent #${id} not found`);
+            }
+            this.syncAgentInstance(updatedAgent);
+            return updatedAgent;
+        });
     }
     async remove(id) {
         const agent = await this.findOne(id);
@@ -138,7 +162,7 @@ let AgentsService = AgentsService_1 = class AgentsService {
             this.logger.debug(`Agent instance #${agentId} not found in memory. Checking database...`);
             const agentEntity = await this.agentRepository.findOne({
                 where: { id: agentId },
-                relations: ['model'],
+                relations: AGENT_RELATIONS,
             });
             if (!agentEntity) {
                 throw new common_1.NotFoundException(`Agent #${agentId} not found`);