Proeycto de images-worker creado
This commit is contained in:
74
mvp/image-worker/dist/pipeline/prompt-builder.service.js
vendored
Normal file
74
mvp/image-worker/dist/pipeline/prompt-builder.service.js
vendored
Normal file
@@ -0,0 +1,74 @@
|
||||
"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 PromptBuilderService_1;
|
||||
Object.defineProperty(exports, "__esModule", { value: true });
|
||||
exports.PromptBuilderService = void 0;
|
||||
const common_1 = require("@nestjs/common");
|
||||
const config_1 = require("@nestjs/config");
|
||||
const fs = require("fs");
|
||||
const path = require("path");
|
||||
const axios_1 = require("axios");
|
||||
const OPENROUTER_URL = 'https://openrouter.ai/api/v1/chat/completions';
|
||||
let PromptBuilderService = PromptBuilderService_1 = class PromptBuilderService {
|
||||
constructor(config) {
|
||||
this.config = config;
|
||||
this.logger = new common_1.Logger(PromptBuilderService_1.name);
|
||||
this.systemPrompt = '';
|
||||
const ruta = path.join(process.cwd(), 'prompts', 'prompt-builder.txt');
|
||||
if (fs.existsSync(ruta)) {
|
||||
this.systemPrompt = fs.readFileSync(ruta, 'utf-8');
|
||||
}
|
||||
else {
|
||||
this.logger.warn('prompts/prompt-builder.txt no encontrado, usando default');
|
||||
}
|
||||
}
|
||||
async generarPrompt(tipoReforma, m2Suelo, calidad, notas) {
|
||||
const apiKey = this.config.get('OPENROUTER_API_KEY');
|
||||
const model = this.config.get('OPENROUTER_MODEL_TEXTO', 'anthropic/claude-3.5-haiku-20241022');
|
||||
const userContent = `Generate a render prompt for a ${tipoReforma} renovation.
|
||||
- Area: ${m2Suelo ?? 'unknown'} m²
|
||||
- Quality level: ${calidad}
|
||||
- Client notes: ${notas.join('; ') || 'none'}
|
||||
- Style: modern ${tipoReforma} renovation`;
|
||||
try {
|
||||
const response = await axios_1.default.post(OPENROUTER_URL, {
|
||||
model,
|
||||
messages: [
|
||||
{ role: 'system', content: this.systemPrompt },
|
||||
{ role: 'user', content: userContent },
|
||||
],
|
||||
max_tokens: 512,
|
||||
temperature: 0.5,
|
||||
}, {
|
||||
headers: {
|
||||
Authorization: `Bearer ${apiKey}`,
|
||||
'Content-Type': 'application/json',
|
||||
'HTTP-Referer': 'https://reformix.es',
|
||||
'X-Title': 'Reformix Image Worker',
|
||||
},
|
||||
});
|
||||
const prompt = response.data.choices?.[0]?.message?.content?.trim();
|
||||
if (!prompt)
|
||||
throw new Error('OpenRouter devolvio respuesta vacia');
|
||||
return prompt;
|
||||
}
|
||||
catch (err) {
|
||||
this.logger.error(`Error generando prompt: ${err.message}`);
|
||||
throw err;
|
||||
}
|
||||
}
|
||||
};
|
||||
exports.PromptBuilderService = PromptBuilderService;
|
||||
exports.PromptBuilderService = PromptBuilderService = PromptBuilderService_1 = __decorate([
|
||||
(0, common_1.Injectable)(),
|
||||
__metadata("design:paramtypes", [config_1.ConfigService])
|
||||
], PromptBuilderService);
|
||||
//# sourceMappingURL=prompt-builder.service.js.map
|
||||
Reference in New Issue
Block a user