@saltsquareio/nestjs-backend-platform 0.0.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/README.md +72 -0
- package/dist/cli/migrate.d.ts +2 -0
- package/dist/cli/migrate.js +34 -0
- package/dist/cli/migrate.js.map +1 -0
- package/dist/core/ai/ai.module.d.ts +5 -0
- package/dist/core/ai/ai.module.js +53 -0
- package/dist/core/ai/ai.module.js.map +1 -0
- package/dist/core/ai/ai.options.d.ts +33 -0
- package/dist/core/ai/ai.options.js +14 -0
- package/dist/core/ai/ai.options.js.map +1 -0
- package/dist/core/ai/ai.service.d.ts +49 -0
- package/dist/core/ai/ai.service.js +381 -0
- package/dist/core/ai/ai.service.js.map +1 -0
- package/dist/core/ai/ai.types.d.ts +133 -0
- package/dist/core/ai/ai.types.js +3 -0
- package/dist/core/ai/ai.types.js.map +1 -0
- package/dist/core/ai/context/context-formatter.service.d.ts +9 -0
- package/dist/core/ai/context/context-formatter.service.js +66 -0
- package/dist/core/ai/context/context-formatter.service.js.map +1 -0
- package/dist/core/ai/cost/cost-tracking.service.d.ts +12 -0
- package/dist/core/ai/cost/cost-tracking.service.js +65 -0
- package/dist/core/ai/cost/cost-tracking.service.js.map +1 -0
- package/dist/core/ai/entities/ai-invocation-log.entity.d.ts +19 -0
- package/dist/core/ai/entities/ai-invocation-log.entity.js +116 -0
- package/dist/core/ai/entities/ai-invocation-log.entity.js.map +1 -0
- package/dist/core/ai/entities/prompt-template.entity.d.ts +13 -0
- package/dist/core/ai/entities/prompt-template.entity.js +75 -0
- package/dist/core/ai/entities/prompt-template.entity.js.map +1 -0
- package/dist/core/ai/index.d.ts +18 -0
- package/dist/core/ai/index.js +36 -0
- package/dist/core/ai/index.js.map +1 -0
- package/dist/core/ai/llm/dummy-llm-model.d.ts +9 -0
- package/dist/core/ai/llm/dummy-llm-model.js +18 -0
- package/dist/core/ai/llm/dummy-llm-model.js.map +1 -0
- package/dist/core/ai/llm/index.d.ts +2 -0
- package/dist/core/ai/llm/index.js +19 -0
- package/dist/core/ai/llm/index.js.map +1 -0
- package/dist/core/ai/llm/openai-llm-model.d.ts +10 -0
- package/dist/core/ai/llm/openai-llm-model.js +46 -0
- package/dist/core/ai/llm/openai-llm-model.js.map +1 -0
- package/dist/core/ai/llm-model.d.ts +17 -0
- package/dist/core/ai/llm-model.js +3 -0
- package/dist/core/ai/llm-model.js.map +1 -0
- package/dist/core/ai/memory/memory.service.d.ts +18 -0
- package/dist/core/ai/memory/memory.service.js +157 -0
- package/dist/core/ai/memory/memory.service.js.map +1 -0
- package/dist/core/ai/memory/memory.types.d.ts +6 -0
- package/dist/core/ai/memory/memory.types.js +3 -0
- package/dist/core/ai/memory/memory.types.js.map +1 -0
- package/dist/core/ai/memory/memory.utils.d.ts +1 -0
- package/dist/core/ai/memory/memory.utils.js +17 -0
- package/dist/core/ai/memory/memory.utils.js.map +1 -0
- package/dist/core/ai/model-registry.service.d.ts +19 -0
- package/dist/core/ai/model-registry.service.js +60 -0
- package/dist/core/ai/model-registry.service.js.map +1 -0
- package/dist/core/ai/model-resolver.service.d.ts +6 -0
- package/dist/core/ai/model-resolver.service.js +28 -0
- package/dist/core/ai/model-resolver.service.js.map +1 -0
- package/dist/core/ai/prompt-builder.service.d.ts +16 -0
- package/dist/core/ai/prompt-builder.service.js +34 -0
- package/dist/core/ai/prompt-builder.service.js.map +1 -0
- package/dist/core/ai/provider/provider-factory.service.d.ts +18 -0
- package/dist/core/ai/provider/provider-factory.service.js +60 -0
- package/dist/core/ai/provider/provider-factory.service.js.map +1 -0
- package/dist/core/ai/safety/safety-guardrail.service.d.ts +9 -0
- package/dist/core/ai/safety/safety-guardrail.service.js +65 -0
- package/dist/core/ai/safety/safety-guardrail.service.js.map +1 -0
- package/dist/core/ai/summarization/summarization.service.d.ts +24 -0
- package/dist/core/ai/summarization/summarization.service.js +66 -0
- package/dist/core/ai/summarization/summarization.service.js.map +1 -0
- package/dist/core/ai/template/template.service.d.ts +14 -0
- package/dist/core/ai/template/template.service.js +88 -0
- package/dist/core/ai/template/template.service.js.map +1 -0
- package/dist/core/audit-log/audit-log.entity.d.ts +13 -0
- package/dist/core/audit-log/audit-log.entity.js +72 -0
- package/dist/core/audit-log/audit-log.entity.js.map +1 -0
- package/dist/core/audit-log/audit-log.module.d.ts +2 -0
- package/dist/core/audit-log/audit-log.module.js +25 -0
- package/dist/core/audit-log/audit-log.module.js.map +1 -0
- package/dist/core/audit-log/audit-log.service.d.ts +20 -0
- package/dist/core/audit-log/audit-log.service.js +59 -0
- package/dist/core/audit-log/audit-log.service.js.map +1 -0
- package/dist/core/audit-log/index.d.ts +6 -0
- package/dist/core/audit-log/index.js +11 -0
- package/dist/core/audit-log/index.js.map +1 -0
- package/dist/core/auth/auth.provider.d.ts +4 -0
- package/dist/core/auth/auth.provider.js +3 -0
- package/dist/core/auth/auth.provider.js.map +1 -0
- package/dist/core/auth/decorators.d.ts +4 -0
- package/dist/core/auth/decorators.js +11 -0
- package/dist/core/auth/decorators.js.map +1 -0
- package/dist/core/auth/index.d.ts +6 -0
- package/dist/core/auth/index.js +23 -0
- package/dist/core/auth/index.js.map +1 -0
- package/dist/core/auth/params.d.ts +2 -0
- package/dist/core/auth/params.js +23 -0
- package/dist/core/auth/params.js.map +1 -0
- package/dist/core/auth/role.can-activate.d.ts +10 -0
- package/dist/core/auth/role.can-activate.js +42 -0
- package/dist/core/auth/role.can-activate.js.map +1 -0
- package/dist/core/auth/types.d.ts +6 -0
- package/dist/core/auth/types.js +3 -0
- package/dist/core/auth/types.js.map +1 -0
- package/dist/core/auth/utils.d.ts +4 -0
- package/dist/core/auth/utils.js +22 -0
- package/dist/core/auth/utils.js.map +1 -0
- package/dist/core/configuration/coerce.d.ts +2 -0
- package/dist/core/configuration/coerce.js +8 -0
- package/dist/core/configuration/coerce.js.map +1 -0
- package/dist/core/configuration/configuration.module.d.ts +9 -0
- package/dist/core/configuration/configuration.module.js +38 -0
- package/dist/core/configuration/configuration.module.js.map +1 -0
- package/dist/core/configuration/define-config.d.ts +2 -0
- package/dist/core/configuration/define-config.js +20 -0
- package/dist/core/configuration/define-config.js.map +1 -0
- package/dist/core/configuration/env-files.d.ts +1 -0
- package/dist/core/configuration/env-files.js +7 -0
- package/dist/core/configuration/env-files.js.map +1 -0
- package/dist/core/configuration/index.d.ts +5 -0
- package/dist/core/configuration/index.js +22 -0
- package/dist/core/configuration/index.js.map +1 -0
- package/dist/core/configuration/schemas/app.schema.d.ts +16 -0
- package/dist/core/configuration/schemas/app.schema.js +23 -0
- package/dist/core/configuration/schemas/app.schema.js.map +1 -0
- package/dist/core/database/common.d.ts +12 -0
- package/dist/core/database/common.js +15 -0
- package/dist/core/database/common.js.map +1 -0
- package/dist/core/database/custom-snake-case-naming-strategy.d.ts +9 -0
- package/dist/core/database/custom-snake-case-naming-strategy.js +28 -0
- package/dist/core/database/custom-snake-case-naming-strategy.js.map +1 -0
- package/dist/core/database/database.config.d.ts +4 -0
- package/dist/core/database/database.config.js +39 -0
- package/dist/core/database/database.config.js.map +1 -0
- package/dist/core/database/database.module.d.ts +7 -0
- package/dist/core/database/database.module.js +58 -0
- package/dist/core/database/database.module.js.map +1 -0
- package/dist/core/database/database.schema.d.ts +41 -0
- package/dist/core/database/database.schema.js +33 -0
- package/dist/core/database/database.schema.js.map +1 -0
- package/dist/core/database/index.d.ts +5 -0
- package/dist/core/database/index.js +22 -0
- package/dist/core/database/index.js.map +1 -0
- package/dist/core/database/ssl-config.d.ts +3 -0
- package/dist/core/database/ssl-config.js +8 -0
- package/dist/core/database/ssl-config.js.map +1 -0
- package/dist/core/exception/all-exceptions.filter.d.ts +8 -0
- package/dist/core/exception/all-exceptions.filter.js +56 -0
- package/dist/core/exception/all-exceptions.filter.js.map +1 -0
- package/dist/core/exception/index.d.ts +1 -0
- package/dist/core/exception/index.js +18 -0
- package/dist/core/exception/index.js.map +1 -0
- package/dist/core/exception/utils.d.ts +6 -0
- package/dist/core/exception/utils.js +15 -0
- package/dist/core/exception/utils.js.map +1 -0
- package/dist/core/health/health.controller.d.ts +9 -0
- package/dist/core/health/health.controller.js +52 -0
- package/dist/core/health/health.controller.js.map +1 -0
- package/dist/core/health/health.module.d.ts +2 -0
- package/dist/core/health/health.module.js +24 -0
- package/dist/core/health/health.module.js.map +1 -0
- package/dist/core/health/index.d.ts +1 -0
- package/dist/core/health/index.js +18 -0
- package/dist/core/health/index.js.map +1 -0
- package/dist/core/health/shutdown-handler.d.ts +2 -0
- package/dist/core/health/shutdown-handler.js +20 -0
- package/dist/core/health/shutdown-handler.js.map +1 -0
- package/dist/core/health/shutdown.service.d.ts +11 -0
- package/dist/core/health/shutdown.service.js +54 -0
- package/dist/core/health/shutdown.service.js.map +1 -0
- package/dist/core/http/http.circuit-breaker.d.ts +4 -0
- package/dist/core/http/http.circuit-breaker.js +25 -0
- package/dist/core/http/http.circuit-breaker.js.map +1 -0
- package/dist/core/http/http.client.factory.d.ts +3 -0
- package/dist/core/http/http.client.factory.js +32 -0
- package/dist/core/http/http.client.factory.js.map +1 -0
- package/dist/core/http/http.module.d.ts +2 -0
- package/dist/core/http/http.module.js +19 -0
- package/dist/core/http/http.module.js.map +1 -0
- package/dist/core/http/http.open-api.d.ts +8 -0
- package/dist/core/http/http.open-api.js +22 -0
- package/dist/core/http/http.open-api.js.map +1 -0
- package/dist/core/http/http.types.d.ts +6 -0
- package/dist/core/http/http.types.js +3 -0
- package/dist/core/http/http.types.js.map +1 -0
- package/dist/core/http/index.d.ts +4 -0
- package/dist/core/http/index.js +21 -0
- package/dist/core/http/index.js.map +1 -0
- package/dist/core/info/index.d.ts +1 -0
- package/dist/core/info/index.js +18 -0
- package/dist/core/info/index.js.map +1 -0
- package/dist/core/info/info.controller.d.ts +3 -0
- package/dist/core/info/info.controller.js +33 -0
- package/dist/core/info/info.controller.js.map +1 -0
- package/dist/core/info/info.module.d.ts +6 -0
- package/dist/core/info/info.module.js +27 -0
- package/dist/core/info/info.module.js.map +1 -0
- package/dist/core/info/info.utils.d.ts +13 -0
- package/dist/core/info/info.utils.js +32 -0
- package/dist/core/info/info.utils.js.map +1 -0
- package/dist/core/logger/hash-pii.d.ts +1 -0
- package/dist/core/logger/hash-pii.js +13 -0
- package/dist/core/logger/hash-pii.js.map +1 -0
- package/dist/core/logger/index.d.ts +4 -0
- package/dist/core/logger/index.js +21 -0
- package/dist/core/logger/index.js.map +1 -0
- package/dist/core/logger/logger.module.d.ts +2 -0
- package/dist/core/logger/logger.module.js +30 -0
- package/dist/core/logger/logger.module.js.map +1 -0
- package/dist/core/logger/logging.interceptor.d.ts +8 -0
- package/dist/core/logger/logging.interceptor.js +51 -0
- package/dist/core/logger/logging.interceptor.js.map +1 -0
- package/dist/core/logger/pino.config.d.ts +9 -0
- package/dist/core/logger/pino.config.js +47 -0
- package/dist/core/logger/pino.config.js.map +1 -0
- package/dist/core/migrations/create-migration-data-source.d.ts +8 -0
- package/dist/core/migrations/create-migration-data-source.js +18 -0
- package/dist/core/migrations/create-migration-data-source.js.map +1 -0
- package/dist/core/migrations/index.d.ts +3 -0
- package/dist/core/migrations/index.js +20 -0
- package/dist/core/migrations/index.js.map +1 -0
- package/dist/core/migrations/platform-entities.d.ts +2 -0
- package/dist/core/migrations/platform-entities.js +10 -0
- package/dist/core/migrations/platform-entities.js.map +1 -0
- package/dist/core/migrations/run-migrations.d.ts +3 -0
- package/dist/core/migrations/run-migrations.js +80 -0
- package/dist/core/migrations/run-migrations.js.map +1 -0
- package/dist/core/newrelic/index.d.ts +1 -0
- package/dist/core/newrelic/index.js +18 -0
- package/dist/core/newrelic/index.js.map +1 -0
- package/dist/core/newrelic/newrelic.utils.d.ts +2 -0
- package/dist/core/newrelic/newrelic.utils.js +31 -0
- package/dist/core/newrelic/newrelic.utils.js.map +1 -0
- package/dist/core/scheduler/index.d.ts +3 -0
- package/dist/core/scheduler/index.js +20 -0
- package/dist/core/scheduler/index.js.map +1 -0
- package/dist/core/scheduler/scheduled-cron.decorator.d.ts +6 -0
- package/dist/core/scheduler/scheduled-cron.decorator.js +31 -0
- package/dist/core/scheduler/scheduled-cron.decorator.js.map +1 -0
- package/dist/core/scheduler/scheduler-control.service.d.ts +14 -0
- package/dist/core/scheduler/scheduler-control.service.js +42 -0
- package/dist/core/scheduler/scheduler-control.service.js.map +1 -0
- package/dist/core/scheduler/scheduler.module.d.ts +4 -0
- package/dist/core/scheduler/scheduler.module.js +27 -0
- package/dist/core/scheduler/scheduler.module.js.map +1 -0
- package/dist/core/throttler/index.d.ts +7 -0
- package/dist/core/throttler/index.js +25 -0
- package/dist/core/throttler/index.js.map +1 -0
- package/dist/core/throttler/pg-throttler.storage.d.ts +10 -0
- package/dist/core/throttler/pg-throttler.storage.js +58 -0
- package/dist/core/throttler/pg-throttler.storage.js.map +1 -0
- package/dist/core/throttler/rate-limit-hit.entity.d.ts +6 -0
- package/dist/core/throttler/rate-limit-hit.entity.js +36 -0
- package/dist/core/throttler/rate-limit-hit.entity.js.map +1 -0
- package/dist/core/throttler/throttler-cleanup.service.d.ts +7 -0
- package/dist/core/throttler/throttler-cleanup.service.js +41 -0
- package/dist/core/throttler/throttler-cleanup.service.js.map +1 -0
- package/dist/core/throttler/throttler-tracker.d.ts +2 -0
- package/dist/core/throttler/throttler-tracker.js +29 -0
- package/dist/core/throttler/throttler-tracker.js.map +1 -0
- package/dist/core/throttler/throttler.module.d.ts +2 -0
- package/dist/core/throttler/throttler.module.js +41 -0
- package/dist/core/throttler/throttler.module.js.map +1 -0
- package/dist/core/utils/app-initializer.d.ts +5 -0
- package/dist/core/utils/app-initializer.js +37 -0
- package/dist/core/utils/app-initializer.js.map +1 -0
- package/dist/core/utils/feature-flag.d.ts +1 -0
- package/dist/core/utils/feature-flag.js +10 -0
- package/dist/core/utils/feature-flag.js.map +1 -0
- package/dist/core/utils/index.d.ts +5 -0
- package/dist/core/utils/index.js +22 -0
- package/dist/core/utils/index.js.map +1 -0
- package/dist/core/utils/parse-allowed-origins.d.ts +1 -0
- package/dist/core/utils/parse-allowed-origins.js +21 -0
- package/dist/core/utils/parse-allowed-origins.js.map +1 -0
- package/dist/core/utils/security-middleware.d.ts +2 -0
- package/dist/core/utils/security-middleware.js +19 -0
- package/dist/core/utils/security-middleware.js.map +1 -0
- package/dist/core/utils/with-timeout.d.ts +4 -0
- package/dist/core/utils/with-timeout.js +23 -0
- package/dist/core/utils/with-timeout.js.map +1 -0
- package/dist/core/ws/index.d.ts +4 -0
- package/dist/core/ws/index.js +23 -0
- package/dist/core/ws/index.js.map +1 -0
- package/dist/core/ws/validate-ws-dto.d.ts +1 -0
- package/dist/core/ws/validate-ws-dto.js +14 -0
- package/dist/core/ws/validate-ws-dto.js.map +1 -0
- package/dist/core/ws/ws-token.module.d.ts +2 -0
- package/dist/core/ws/ws-token.module.js +21 -0
- package/dist/core/ws/ws-token.module.js.map +1 -0
- package/dist/core/ws/ws-token.service.d.ts +8 -0
- package/dist/core/ws/ws-token.service.js +57 -0
- package/dist/core/ws/ws-token.service.js.map +1 -0
- package/dist/core/ws/ws.constants.d.ts +7 -0
- package/dist/core/ws/ws.constants.js +11 -0
- package/dist/core/ws/ws.constants.js.map +1 -0
- package/dist/index.d.ts +15 -0
- package/dist/index.js +32 -0
- package/dist/index.js.map +1 -0
- package/package.json +167 -0
|
@@ -0,0 +1,157 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
3
|
+
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
4
|
+
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
5
|
+
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;
|
|
6
|
+
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
7
|
+
};
|
|
8
|
+
var __metadata = (this && this.__metadata) || function (k, v) {
|
|
9
|
+
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
|
|
10
|
+
};
|
|
11
|
+
var __param = (this && this.__param) || function (paramIndex, decorator) {
|
|
12
|
+
return function (target, key) { decorator(target, key, paramIndex); }
|
|
13
|
+
};
|
|
14
|
+
var MemoryService_1;
|
|
15
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
16
|
+
exports.MemoryService = void 0;
|
|
17
|
+
const common_1 = require("@nestjs/common");
|
|
18
|
+
const oss_1 = require("mem0ai/oss");
|
|
19
|
+
const ai_options_1 = require("../ai.options");
|
|
20
|
+
let MemoryService = MemoryService_1 = class MemoryService {
|
|
21
|
+
constructor(options) {
|
|
22
|
+
this.options = options;
|
|
23
|
+
this.logger = new common_1.Logger(MemoryService_1.name);
|
|
24
|
+
this.memory = null;
|
|
25
|
+
this.enabled = false;
|
|
26
|
+
}
|
|
27
|
+
onModuleInit() {
|
|
28
|
+
const mem = this.options.memory;
|
|
29
|
+
if (!mem?.enabled || !mem.config) {
|
|
30
|
+
this.logger.log('Memory disabled (no enabled mem0 config provided)');
|
|
31
|
+
return;
|
|
32
|
+
}
|
|
33
|
+
try {
|
|
34
|
+
const config = {
|
|
35
|
+
...mem.config,
|
|
36
|
+
...(mem.factExtractionPrompt
|
|
37
|
+
? { customInstructions: mem.factExtractionPrompt }
|
|
38
|
+
: {}),
|
|
39
|
+
};
|
|
40
|
+
this.memory = new oss_1.Memory(config);
|
|
41
|
+
this.enabled = true;
|
|
42
|
+
this.logger.log('Mem0 memory initialized');
|
|
43
|
+
}
|
|
44
|
+
catch (error) {
|
|
45
|
+
this.logger.warn(`Failed to initialize Mem0: ${error instanceof Error ? error.message : String(error)}. Memory disabled.`);
|
|
46
|
+
}
|
|
47
|
+
}
|
|
48
|
+
async search(userId, query, limit = 10) {
|
|
49
|
+
if (!this.enabled || !this.memory)
|
|
50
|
+
return [];
|
|
51
|
+
try {
|
|
52
|
+
const fetchLimit = Math.ceil(limit * 1.5);
|
|
53
|
+
const result = await this.memory.search(query, {
|
|
54
|
+
filters: { user_id: userId },
|
|
55
|
+
topK: fetchLimit,
|
|
56
|
+
});
|
|
57
|
+
return result.results
|
|
58
|
+
.map((item) => ({
|
|
59
|
+
id: item.id,
|
|
60
|
+
content: item.memory,
|
|
61
|
+
metadata: item.metadata ?? undefined,
|
|
62
|
+
createdAt: item.createdAt,
|
|
63
|
+
score: this.computeRankedScore(item.score ?? 0, item.createdAt),
|
|
64
|
+
}))
|
|
65
|
+
.sort((a, b) => b.score - a.score)
|
|
66
|
+
.slice(0, limit)
|
|
67
|
+
.map(({ score: _score, ...entry }) => entry);
|
|
68
|
+
}
|
|
69
|
+
catch (error) {
|
|
70
|
+
this.logger.error({
|
|
71
|
+
userId,
|
|
72
|
+
error: error instanceof Error ? error.message : String(error),
|
|
73
|
+
}, 'memory.search.failed');
|
|
74
|
+
return [];
|
|
75
|
+
}
|
|
76
|
+
}
|
|
77
|
+
computeRankedScore(similarity, createdAt) {
|
|
78
|
+
const SIMILARITY_WEIGHT = 0.7;
|
|
79
|
+
const RECENCY_WEIGHT = 0.3;
|
|
80
|
+
const HALF_LIFE_DAYS = 30;
|
|
81
|
+
if (!createdAt)
|
|
82
|
+
return similarity;
|
|
83
|
+
const days = (Date.now() - new Date(createdAt).getTime()) / (1000 * 60 * 60 * 24);
|
|
84
|
+
const recency = 1 / (1 + days / HALF_LIFE_DAYS);
|
|
85
|
+
return similarity * SIMILARITY_WEIGHT + recency * RECENCY_WEIGHT;
|
|
86
|
+
}
|
|
87
|
+
async add(userId, content, metadata) {
|
|
88
|
+
if (!this.enabled || !this.memory)
|
|
89
|
+
return;
|
|
90
|
+
try {
|
|
91
|
+
await this.memory.add([{ role: 'user', content }], { userId, metadata });
|
|
92
|
+
}
|
|
93
|
+
catch (error) {
|
|
94
|
+
this.logger.warn(`Mem0 add failed: ${error instanceof Error ? error.message : String(error)}`);
|
|
95
|
+
}
|
|
96
|
+
}
|
|
97
|
+
async addConversation(userId, userMessage, assistantResponse, metadata) {
|
|
98
|
+
if (!this.enabled || !this.memory)
|
|
99
|
+
return;
|
|
100
|
+
try {
|
|
101
|
+
await this.memory.add([
|
|
102
|
+
{ role: 'user', content: userMessage },
|
|
103
|
+
{ role: 'assistant', content: assistantResponse },
|
|
104
|
+
], { userId, metadata });
|
|
105
|
+
}
|
|
106
|
+
catch (error) {
|
|
107
|
+
this.logger.warn(`Mem0 addConversation failed: ${error instanceof Error ? error.message : String(error)}`);
|
|
108
|
+
}
|
|
109
|
+
}
|
|
110
|
+
async getAll(userId, limit = 50) {
|
|
111
|
+
if (!this.enabled || !this.memory)
|
|
112
|
+
return [];
|
|
113
|
+
try {
|
|
114
|
+
const result = await this.memory.getAll({
|
|
115
|
+
filters: { user_id: userId },
|
|
116
|
+
topK: limit,
|
|
117
|
+
});
|
|
118
|
+
return result.results.map((item) => ({
|
|
119
|
+
id: item.id,
|
|
120
|
+
content: item.memory,
|
|
121
|
+
metadata: item.metadata ?? undefined,
|
|
122
|
+
createdAt: item.createdAt,
|
|
123
|
+
}));
|
|
124
|
+
}
|
|
125
|
+
catch (error) {
|
|
126
|
+
this.logger.warn(`Mem0 getAll failed: ${error instanceof Error ? error.message : String(error)}`);
|
|
127
|
+
return [];
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
async delete(memoryId) {
|
|
131
|
+
if (!this.enabled || !this.memory)
|
|
132
|
+
return;
|
|
133
|
+
try {
|
|
134
|
+
await this.memory.delete(memoryId);
|
|
135
|
+
}
|
|
136
|
+
catch (error) {
|
|
137
|
+
this.logger.warn(`Mem0 delete failed: ${error instanceof Error ? error.message : String(error)}`);
|
|
138
|
+
}
|
|
139
|
+
}
|
|
140
|
+
async deleteAll(userId) {
|
|
141
|
+
if (!this.enabled || !this.memory)
|
|
142
|
+
return;
|
|
143
|
+
try {
|
|
144
|
+
await this.memory.deleteAll({ userId });
|
|
145
|
+
}
|
|
146
|
+
catch (error) {
|
|
147
|
+
this.logger.warn(`Mem0 deleteAll failed: ${error instanceof Error ? error.message : String(error)}`);
|
|
148
|
+
}
|
|
149
|
+
}
|
|
150
|
+
};
|
|
151
|
+
exports.MemoryService = MemoryService;
|
|
152
|
+
exports.MemoryService = MemoryService = MemoryService_1 = __decorate([
|
|
153
|
+
(0, common_1.Injectable)(),
|
|
154
|
+
__param(0, (0, common_1.Inject)(ai_options_1.AI_MODULE_OPTIONS)),
|
|
155
|
+
__metadata("design:paramtypes", [Object])
|
|
156
|
+
], MemoryService);
|
|
157
|
+
//# sourceMappingURL=memory.service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory.service.js","sourceRoot":"","sources":["../../../../src/core/ai/memory/memory.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CAA0E;AAC1E,oCAA8D;AAC9D,8CAAwE;AASjE,IAAM,aAAa,qBAAnB,MAAM,aAAa;IAKxB,YAC6B,OAAyC;QAAxB,YAAO,GAAP,OAAO,CAAiB;QALrD,WAAM,GAAG,IAAI,eAAM,CAAC,eAAa,CAAC,IAAI,CAAC,CAAC;QACjD,WAAM,GAAkB,IAAI,CAAC;QAC7B,YAAO,GAAG,KAAK,CAAC;IAIrB,CAAC;IAEJ,YAAY;QACV,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;QAChC,IAAI,CAAC,GAAG,EAAE,OAAO,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC;YACjC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,mDAAmD,CAAC,CAAC;YACrE,OAAO;QACT,CAAC;QACD,IAAI,CAAC;YACH,MAAM,MAAM,GAA0B;gBACpC,GAAG,GAAG,CAAC,MAAM;gBACb,GAAG,CAAC,GAAG,CAAC,oBAAoB;oBAC1B,CAAC,CAAC,EAAE,kBAAkB,EAAE,GAAG,CAAC,oBAAoB,EAAE;oBAClD,CAAC,CAAC,EAAE,CAAC;aACR,CAAC;YACF,IAAI,CAAC,MAAM,GAAG,IAAI,YAAM,CAAC,MAAM,CAAC,CAAC;YACjC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;YACpB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;QAC7C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,8BAA8B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,oBAAoB,CACzG,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CACV,MAAc,EACd,KAAa,EACb,KAAK,GAAG,EAAE;QAEV,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO,EAAE,CAAC;QAC7C,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,GAAG,CAAC,CAAC;YAC1C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE;gBAC7C,OAAO,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE;gBAC5B,IAAI,EAAE,UAAU;aACjB,CAAC,CAAC;YACH,OAAO,MAAM,CAAC,OAAO;iBAClB,GAAG,CAAC,CAAC,IAAgB,EAAE,EAAE,CAAC,CAAC;gBAC1B,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,OAAO,EAAE,IAAI,CAAC,MAAM;gBACpB,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,SAAS;gBACpC,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,KAAK,EAAE,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC;aAChE,CAAC,CAAC;iBACF,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;iBACjC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC;iBACf,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC;QACjD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CACf;gBACE,MAAM;gBACN,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC9D,EACD,sBAAsB,CACvB,CAAC;YACF,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAMO,kBAAkB,CAAC,UAAkB,EAAE,SAAkB;QAC/D,MAAM,iBAAiB,GAAG,GAAG,CAAC;QAC9B,MAAM,cAAc,GAAG,GAAG,CAAC;QAC3B,MAAM,cAAc,GAAG,EAAE,CAAC;QAC1B,IAAI,CAAC,SAAS;YAAE,OAAO,UAAU,CAAC;QAClC,MAAM,IAAI,GACR,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC,GAAG,CAAC,IAAI,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;QACvE,MAAM,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,GAAG,cAAc,CAAC,CAAC;QAChD,OAAO,UAAU,GAAG,iBAAiB,GAAG,OAAO,GAAG,cAAc,CAAC;IACnE,CAAC;IAED,KAAK,CAAC,GAAG,CACP,MAAc,EACd,OAAe,EACf,QAAkC;QAElC,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO;QAC1C,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC3E,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,oBAAoB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAC7E,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,MAAc,EACd,WAAmB,EACnB,iBAAyB,EACzB,QAAkC;QAElC,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO;QAC1C,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,GAAG,CACnB;gBACE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE;gBACtC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,iBAAiB,EAAE;aAClD,EACD,EAAE,MAAM,EAAE,QAAQ,EAAE,CACrB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,gCAAgC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CACzF,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAc,EAAE,KAAK,GAAG,EAAE;QACrC,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO,EAAE,CAAC;QAC7C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;gBACtC,OAAO,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE;gBAC5B,IAAI,EAAE,KAAK;aACZ,CAAC,CAAC;YACH,OAAO,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAgB,EAAE,EAAE,CAAC,CAAC;gBAC/C,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,OAAO,EAAE,IAAI,CAAC,MAAM;gBACpB,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,SAAS;gBACpC,SAAS,EAAE,IAAI,CAAC,SAAS;aAC1B,CAAC,CAAC,CAAC;QACN,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,uBAAuB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAChF,CAAC;YACF,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,QAAgB;QAC3B,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO;QAC1C,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACrC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,uBAAuB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAChF,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAc;QAC5B,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO;QAC1C,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;QAC1C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,0BAA0B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CACnF,CAAC;QACJ,CAAC;IACH,CAAC;CACF,CAAA;AAjKY,sCAAa;wBAAb,aAAa;IADzB,IAAA,mBAAU,GAAE;IAOR,WAAA,IAAA,eAAM,EAAC,8BAAiB,CAAC,CAAA;;GANjB,aAAa,CAiKzB"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory.types.js","sourceRoot":"","sources":["../../../../src/core/ai/memory/memory.types.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export declare function formatMemoryWithDate(content: string, createdAt?: string): string;
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.formatMemoryWithDate = formatMemoryWithDate;
|
|
4
|
+
function formatMemoryWithDate(content, createdAt) {
|
|
5
|
+
if (!createdAt)
|
|
6
|
+
return content;
|
|
7
|
+
const date = new Date(createdAt);
|
|
8
|
+
if (Number.isNaN(date.getTime()))
|
|
9
|
+
return content;
|
|
10
|
+
const dateStr = date.toLocaleDateString('en-US', {
|
|
11
|
+
month: 'short',
|
|
12
|
+
day: 'numeric',
|
|
13
|
+
year: 'numeric',
|
|
14
|
+
});
|
|
15
|
+
return `${content} (remembered ${dateStr})`;
|
|
16
|
+
}
|
|
17
|
+
//# sourceMappingURL=memory.utils.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory.utils.js","sourceRoot":"","sources":["../../../../src/core/ai/memory/memory.utils.ts"],"names":[],"mappings":";;AAIA,oDAaC;AAbD,SAAgB,oBAAoB,CAClC,OAAe,EACf,SAAkB;IAElB,IAAI,CAAC,SAAS;QAAE,OAAO,OAAO,CAAC;IAC/B,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC;IACjC,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QAAE,OAAO,OAAO,CAAC;IACjD,MAAM,OAAO,GAAG,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE;QAC/C,KAAK,EAAE,OAAO;QACd,GAAG,EAAE,SAAS;QACd,IAAI,EAAE,SAAS;KAChB,CAAC,CAAC;IACH,OAAO,GAAG,OAAO,gBAAgB,OAAO,GAAG,CAAC;AAC9C,CAAC"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import { OnModuleInit } from '@nestjs/common';
|
|
2
|
+
import { LlmModel } from './llm-model';
|
|
3
|
+
import { DummyLlmModel, OpenAiLlmModel } from './llm';
|
|
4
|
+
import { ConfigService } from '@nestjs/config';
|
|
5
|
+
interface EnvironmentVariables {
|
|
6
|
+
DEFAULT_MODEL: string;
|
|
7
|
+
OPENAI_API_KEY: string;
|
|
8
|
+
}
|
|
9
|
+
export declare class ModelRegistryService implements OnModuleInit {
|
|
10
|
+
private readonly configService;
|
|
11
|
+
private readonly logger;
|
|
12
|
+
private readonly models;
|
|
13
|
+
private defaultModel;
|
|
14
|
+
constructor(configService: ConfigService<EnvironmentVariables>);
|
|
15
|
+
onModuleInit(): Promise<void>;
|
|
16
|
+
getModelDefinitions(): (typeof OpenAiLlmModel | typeof DummyLlmModel)[];
|
|
17
|
+
get(name?: string): LlmModel;
|
|
18
|
+
}
|
|
19
|
+
export {};
|
|
@@ -0,0 +1,60 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
3
|
+
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
4
|
+
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
5
|
+
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;
|
|
6
|
+
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
7
|
+
};
|
|
8
|
+
var __metadata = (this && this.__metadata) || function (k, v) {
|
|
9
|
+
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
|
|
10
|
+
};
|
|
11
|
+
var ModelRegistryService_1;
|
|
12
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
13
|
+
exports.ModelRegistryService = void 0;
|
|
14
|
+
const common_1 = require("@nestjs/common");
|
|
15
|
+
const llm_1 = require("./llm");
|
|
16
|
+
const common_2 = require("@nestjs/common");
|
|
17
|
+
const config_1 = require("@nestjs/config");
|
|
18
|
+
let ModelRegistryService = ModelRegistryService_1 = class ModelRegistryService {
|
|
19
|
+
constructor(configService) {
|
|
20
|
+
this.configService = configService;
|
|
21
|
+
this.logger = new common_2.Logger(ModelRegistryService_1.name);
|
|
22
|
+
this.models = new Map();
|
|
23
|
+
}
|
|
24
|
+
async onModuleInit() {
|
|
25
|
+
const modelDefinitions = this.getModelDefinitions();
|
|
26
|
+
for (const Klass of Object.values(modelDefinitions)) {
|
|
27
|
+
const apiKey = this.configService.getOrThrow('OPENAI_API_KEY', {
|
|
28
|
+
infer: true,
|
|
29
|
+
});
|
|
30
|
+
const model = new Klass();
|
|
31
|
+
await model.init({
|
|
32
|
+
apiKey,
|
|
33
|
+
});
|
|
34
|
+
this.models.set(model.name, model);
|
|
35
|
+
}
|
|
36
|
+
const defaultModelName = this.configService.getOrThrow('DEFAULT_MODEL', {
|
|
37
|
+
infer: true,
|
|
38
|
+
});
|
|
39
|
+
const defaultModel = this.models.get(defaultModelName);
|
|
40
|
+
const firstModel = this.models.values().next().value;
|
|
41
|
+
this.defaultModel = defaultModel || firstModel;
|
|
42
|
+
this.logger.debug({ all: [...this.models.keys()], default: this.defaultModel.name }, 'Models registered');
|
|
43
|
+
}
|
|
44
|
+
getModelDefinitions() {
|
|
45
|
+
const models = { DummyLlmModel: llm_1.DummyLlmModel, OpenAiLlmModel: llm_1.OpenAiLlmModel };
|
|
46
|
+
return Object.values(models).filter((Klass) => typeof Klass === 'function');
|
|
47
|
+
}
|
|
48
|
+
get(name) {
|
|
49
|
+
if (!name) {
|
|
50
|
+
return this.defaultModel;
|
|
51
|
+
}
|
|
52
|
+
return this.models.get(name) || this.defaultModel;
|
|
53
|
+
}
|
|
54
|
+
};
|
|
55
|
+
exports.ModelRegistryService = ModelRegistryService;
|
|
56
|
+
exports.ModelRegistryService = ModelRegistryService = ModelRegistryService_1 = __decorate([
|
|
57
|
+
(0, common_1.Injectable)(),
|
|
58
|
+
__metadata("design:paramtypes", [config_1.ConfigService])
|
|
59
|
+
], ModelRegistryService);
|
|
60
|
+
//# sourceMappingURL=model-registry.service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model-registry.service.js","sourceRoot":"","sources":["../../../src/core/ai/model-registry.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;AAAA,2CAA0D;AAE1D,+BAAsD;AACtD,2CAAwC;AACxC,2CAA+C;AAQxC,IAAM,oBAAoB,4BAA1B,MAAM,oBAAoB;IAK/B,YACmB,aAAkD;QAAlD,kBAAa,GAAb,aAAa,CAAqC;QALpD,WAAM,GAAG,IAAI,eAAM,CAAC,sBAAoB,CAAC,IAAI,CAAC,CAAC;QAC/C,WAAM,GAAG,IAAI,GAAG,EAAoB,CAAC;IAKnD,CAAC;IAEJ,KAAK,CAAC,YAAY;QAChB,MAAM,gBAAgB,GAAG,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACpD,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,CAAC;YACpD,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,gBAAgB,EAAE;gBAC7D,KAAK,EAAE,IAAI;aACZ,CAAC,CAAC;YACH,MAAM,KAAK,GAAG,IAAI,KAAK,EAAE,CAAC;YAC1B,MAAM,KAAK,CAAC,IAAI,CAAC;gBACf,MAAM;aACP,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACrC,CAAC;QAED,MAAM,gBAAgB,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,eAAe,EAAE;YACtE,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QACvD,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,KAAiB,CAAC;QACjE,IAAI,CAAC,YAAY,GAAG,YAAY,IAAI,UAAU,CAAC;QAC/C,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,EAAE,GAAG,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,EACjE,mBAAmB,CACpB,CAAC;IACJ,CAAC;IAED,mBAAmB;QACjB,MAAM,MAAM,GAAG,EAAE,aAAa,EAAb,mBAAa,EAAE,cAAc,EAAd,oBAAc,EAAE,CAAC;QACjD,OAAO,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,KAAK,KAAK,UAAU,CAAC,CAAC;IAC9E,CAAC;IAED,GAAG,CAAC,IAAa;QACf,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,IAAI,CAAC,YAAY,CAAC;QAC3B,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,YAAY,CAAC;IACpD,CAAC;CACF,CAAA;AA9CY,oDAAoB;+BAApB,oBAAoB;IADhC,IAAA,mBAAU,GAAE;qCAOuB,sBAAa;GANpC,oBAAoB,CA8ChC"}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
import { ModelRegistryService } from './model-registry.service';
|
|
2
|
+
export declare class ModelResolverService {
|
|
3
|
+
private readonly modelRegistryService;
|
|
4
|
+
constructor(modelRegistryService: ModelRegistryService);
|
|
5
|
+
resolve(modelName: string): import("./llm-model").LlmModel;
|
|
6
|
+
}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
3
|
+
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
4
|
+
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
5
|
+
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;
|
|
6
|
+
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
7
|
+
};
|
|
8
|
+
var __metadata = (this && this.__metadata) || function (k, v) {
|
|
9
|
+
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
|
|
10
|
+
};
|
|
11
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
12
|
+
exports.ModelResolverService = void 0;
|
|
13
|
+
const common_1 = require("@nestjs/common");
|
|
14
|
+
const model_registry_service_1 = require("./model-registry.service");
|
|
15
|
+
let ModelResolverService = class ModelResolverService {
|
|
16
|
+
constructor(modelRegistryService) {
|
|
17
|
+
this.modelRegistryService = modelRegistryService;
|
|
18
|
+
}
|
|
19
|
+
resolve(modelName) {
|
|
20
|
+
return this.modelRegistryService.get(modelName);
|
|
21
|
+
}
|
|
22
|
+
};
|
|
23
|
+
exports.ModelResolverService = ModelResolverService;
|
|
24
|
+
exports.ModelResolverService = ModelResolverService = __decorate([
|
|
25
|
+
(0, common_1.Injectable)(),
|
|
26
|
+
__metadata("design:paramtypes", [model_registry_service_1.ModelRegistryService])
|
|
27
|
+
], ModelResolverService);
|
|
28
|
+
//# sourceMappingURL=model-resolver.service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model-resolver.service.js","sourceRoot":"","sources":["../../../src/core/ai/model-resolver.service.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,2CAA4C;AAC5C,qEAAgE;AAGzD,IAAM,oBAAoB,GAA1B,MAAM,oBAAoB;IAC/B,YAA6B,oBAA0C;QAA1C,yBAAoB,GAApB,oBAAoB,CAAsB;IAAG,CAAC;IAE3E,OAAO,CAAC,SAAiB;QACvB,OAAO,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAClD,CAAC;CACF,CAAA;AANY,oDAAoB;+BAApB,oBAAoB;IADhC,IAAA,mBAAU,GAAE;qCAEwC,6CAAoB;GAD5D,oBAAoB,CAMhC"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
export interface Prompt {
|
|
2
|
+
system: {
|
|
3
|
+
content: string;
|
|
4
|
+
};
|
|
5
|
+
user: {
|
|
6
|
+
content: string;
|
|
7
|
+
};
|
|
8
|
+
}
|
|
9
|
+
export declare class PromptBuilderService {
|
|
10
|
+
build(params: {
|
|
11
|
+
message: string;
|
|
12
|
+
language: string;
|
|
13
|
+
tone: string;
|
|
14
|
+
systemPrompt: string;
|
|
15
|
+
}): Promise<Prompt>;
|
|
16
|
+
}
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
3
|
+
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
4
|
+
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
5
|
+
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;
|
|
6
|
+
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
7
|
+
};
|
|
8
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
9
|
+
exports.PromptBuilderService = void 0;
|
|
10
|
+
const common_1 = require("@nestjs/common");
|
|
11
|
+
let PromptBuilderService = class PromptBuilderService {
|
|
12
|
+
async build(params) {
|
|
13
|
+
await Promise.resolve();
|
|
14
|
+
const prompt = {
|
|
15
|
+
system: {
|
|
16
|
+
content: `
|
|
17
|
+
${params.systemPrompt}
|
|
18
|
+
---
|
|
19
|
+
Language: ${params.language},
|
|
20
|
+
Tone: ${params.tone},
|
|
21
|
+
`.trim(),
|
|
22
|
+
},
|
|
23
|
+
user: {
|
|
24
|
+
content: params.message,
|
|
25
|
+
},
|
|
26
|
+
};
|
|
27
|
+
return prompt;
|
|
28
|
+
}
|
|
29
|
+
};
|
|
30
|
+
exports.PromptBuilderService = PromptBuilderService;
|
|
31
|
+
exports.PromptBuilderService = PromptBuilderService = __decorate([
|
|
32
|
+
(0, common_1.Injectable)()
|
|
33
|
+
], PromptBuilderService);
|
|
34
|
+
//# sourceMappingURL=prompt-builder.service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"prompt-builder.service.js","sourceRoot":"","sources":["../../../src/core/ai/prompt-builder.service.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAA4C;AAYrC,IAAM,oBAAoB,GAA1B,MAAM,oBAAoB;IAC/B,KAAK,CAAC,KAAK,CAAC,MAKX;QACC,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC;QACxB,MAAM,MAAM,GAAW;YACrB,MAAM,EAAE;gBACN,OAAO,EAAE;UACP,MAAM,CAAC,YAAY;;oBAET,MAAM,CAAC,QAAQ;gBACnB,MAAM,CAAC,IAAI;SAClB,CAAC,IAAI,EAAE;aACT;YACD,IAAI,EAAE;gBACJ,OAAO,EAAE,MAAM,CAAC,OAAO;aACxB;SACF,CAAC;QACF,OAAO,MAAM,CAAC;IAChB,CAAC;CACF,CAAA;AAvBY,oDAAoB;+BAApB,oBAAoB;IADhC,IAAA,mBAAU,GAAE;GACA,oBAAoB,CAuBhC"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import { OnModuleInit } from '@nestjs/common';
|
|
2
|
+
import { ConfigService } from '@nestjs/config';
|
|
3
|
+
import { LanguageModel } from 'ai';
|
|
4
|
+
import { AiServiceModelConfig } from '../ai.types';
|
|
5
|
+
interface EnvironmentVariables {
|
|
6
|
+
ANTHROPIC_API_KEY: string;
|
|
7
|
+
OPENAI_API_KEY: string;
|
|
8
|
+
}
|
|
9
|
+
export declare class ProviderFactory implements OnModuleInit {
|
|
10
|
+
private readonly configService;
|
|
11
|
+
private readonly logger;
|
|
12
|
+
private anthropicProvider?;
|
|
13
|
+
private openaiProvider?;
|
|
14
|
+
constructor(configService: ConfigService<EnvironmentVariables>);
|
|
15
|
+
onModuleInit(): void;
|
|
16
|
+
getModel(config: AiServiceModelConfig): LanguageModel;
|
|
17
|
+
}
|
|
18
|
+
export {};
|
|
@@ -0,0 +1,60 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
3
|
+
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
4
|
+
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
5
|
+
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;
|
|
6
|
+
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
7
|
+
};
|
|
8
|
+
var __metadata = (this && this.__metadata) || function (k, v) {
|
|
9
|
+
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
|
|
10
|
+
};
|
|
11
|
+
var ProviderFactory_1;
|
|
12
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
13
|
+
exports.ProviderFactory = void 0;
|
|
14
|
+
const common_1 = require("@nestjs/common");
|
|
15
|
+
const config_1 = require("@nestjs/config");
|
|
16
|
+
const anthropic_1 = require("@ai-sdk/anthropic");
|
|
17
|
+
const openai_1 = require("@ai-sdk/openai");
|
|
18
|
+
let ProviderFactory = ProviderFactory_1 = class ProviderFactory {
|
|
19
|
+
constructor(configService) {
|
|
20
|
+
this.configService = configService;
|
|
21
|
+
this.logger = new common_1.Logger(ProviderFactory_1.name);
|
|
22
|
+
}
|
|
23
|
+
onModuleInit() {
|
|
24
|
+
const anthropicApiKey = this.configService.get('ANTHROPIC_API_KEY', {
|
|
25
|
+
infer: true,
|
|
26
|
+
});
|
|
27
|
+
const openaiApiKey = this.configService.get('OPENAI_API_KEY', {
|
|
28
|
+
infer: true,
|
|
29
|
+
});
|
|
30
|
+
if (anthropicApiKey) {
|
|
31
|
+
this.anthropicProvider = (0, anthropic_1.createAnthropic)({ apiKey: anthropicApiKey });
|
|
32
|
+
this.logger.log('Anthropic provider initialized');
|
|
33
|
+
}
|
|
34
|
+
if (openaiApiKey) {
|
|
35
|
+
this.openaiProvider = (0, openai_1.createOpenAI)({ apiKey: openaiApiKey });
|
|
36
|
+
this.logger.log('OpenAI provider initialized');
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
getModel(config) {
|
|
40
|
+
if (config.provider === 'anthropic') {
|
|
41
|
+
if (!this.anthropicProvider) {
|
|
42
|
+
throw new Error('Anthropic provider not initialized. Check ANTHROPIC_API_KEY.');
|
|
43
|
+
}
|
|
44
|
+
return this.anthropicProvider(config.name);
|
|
45
|
+
}
|
|
46
|
+
if (config.provider === 'openai') {
|
|
47
|
+
if (!this.openaiProvider) {
|
|
48
|
+
throw new Error('OpenAI provider not initialized. Check OPENAI_API_KEY.');
|
|
49
|
+
}
|
|
50
|
+
return this.openaiProvider.chat(config.name);
|
|
51
|
+
}
|
|
52
|
+
throw new Error(`Unsupported provider: ${String(config.provider)}`);
|
|
53
|
+
}
|
|
54
|
+
};
|
|
55
|
+
exports.ProviderFactory = ProviderFactory;
|
|
56
|
+
exports.ProviderFactory = ProviderFactory = ProviderFactory_1 = __decorate([
|
|
57
|
+
(0, common_1.Injectable)(),
|
|
58
|
+
__metadata("design:paramtypes", [config_1.ConfigService])
|
|
59
|
+
], ProviderFactory);
|
|
60
|
+
//# sourceMappingURL=provider-factory.service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"provider-factory.service.js","sourceRoot":"","sources":["../../../../src/core/ai/provider/provider-factory.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;AAAA,2CAAkE;AAClE,2CAA+C;AAC/C,iDAAoD;AACpD,2CAA8C;AAcvC,IAAM,eAAe,uBAArB,MAAM,eAAe;IAK1B,YACmB,aAAkD;QAAlD,kBAAa,GAAb,aAAa,CAAqC;QALpD,WAAM,GAAG,IAAI,eAAM,CAAC,iBAAe,CAAC,IAAI,CAAC,CAAC;IAMxD,CAAC;IAEJ,YAAY;QACV,MAAM,eAAe,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,mBAAmB,EAAE;YAClE,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QACH,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,gBAAgB,EAAE;YAC5D,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QAEH,IAAI,eAAe,EAAE,CAAC;YACpB,IAAI,CAAC,iBAAiB,GAAG,IAAA,2BAAe,EAAC,EAAE,MAAM,EAAE,eAAe,EAAE,CAAC,CAAC;YACtE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;QACpD,CAAC;QACD,IAAI,YAAY,EAAE,CAAC;YACjB,IAAI,CAAC,cAAc,GAAG,IAAA,qBAAY,EAAC,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAC;YAC7D,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;QACjD,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,MAA4B;QACnC,IAAI,MAAM,CAAC,QAAQ,KAAK,WAAW,EAAE,CAAC;YACpC,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC5B,MAAM,IAAI,KAAK,CACb,8DAA8D,CAC/D,CAAC;YACJ,CAAC;YACD,OAAO,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC7C,CAAC;QACD,IAAI,MAAM,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;YACjC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;gBACzB,MAAM,IAAI,KAAK,CACb,wDAAwD,CACzD,CAAC;YACJ,CAAC;YACD,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC/C,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,yBAAyB,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACtE,CAAC;CACF,CAAA;AA9CY,0CAAe;0BAAf,eAAe;IAD3B,IAAA,mBAAU,GAAE;qCAOuB,sBAAa;GANpC,eAAe,CA8C3B"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import { type AiModuleOptions } from '../ai.options';
|
|
2
|
+
import { SafetyContext, SafetyResult } from '../ai.types';
|
|
3
|
+
export declare class SafetyGuardrailService {
|
|
4
|
+
private readonly logger;
|
|
5
|
+
private readonly patterns;
|
|
6
|
+
constructor(options: AiModuleOptions);
|
|
7
|
+
validateOutput(text: string, _context?: SafetyContext): SafetyResult;
|
|
8
|
+
sanitizeForLogging(data: Record<string, unknown>): Record<string, unknown>;
|
|
9
|
+
}
|
|
@@ -0,0 +1,65 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
3
|
+
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
4
|
+
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
5
|
+
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;
|
|
6
|
+
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
7
|
+
};
|
|
8
|
+
var __metadata = (this && this.__metadata) || function (k, v) {
|
|
9
|
+
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
|
|
10
|
+
};
|
|
11
|
+
var __param = (this && this.__param) || function (paramIndex, decorator) {
|
|
12
|
+
return function (target, key) { decorator(target, key, paramIndex); }
|
|
13
|
+
};
|
|
14
|
+
var SafetyGuardrailService_1;
|
|
15
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
16
|
+
exports.SafetyGuardrailService = void 0;
|
|
17
|
+
const common_1 = require("@nestjs/common");
|
|
18
|
+
const ai_options_1 = require("../ai.options");
|
|
19
|
+
let SafetyGuardrailService = SafetyGuardrailService_1 = class SafetyGuardrailService {
|
|
20
|
+
constructor(options) {
|
|
21
|
+
this.logger = new common_1.Logger(SafetyGuardrailService_1.name);
|
|
22
|
+
this.patterns = options.safety?.patterns ?? [];
|
|
23
|
+
}
|
|
24
|
+
validateOutput(text, _context) {
|
|
25
|
+
const flags = [];
|
|
26
|
+
for (const { flag, pattern } of this.patterns) {
|
|
27
|
+
if (pattern.test(text) && !flags.includes(flag)) {
|
|
28
|
+
flags.push(flag);
|
|
29
|
+
}
|
|
30
|
+
}
|
|
31
|
+
if (flags.length > 0) {
|
|
32
|
+
this.logger.warn({ flags }, 'Safety flags detected in AI output');
|
|
33
|
+
}
|
|
34
|
+
return { safe: flags.length === 0, flags };
|
|
35
|
+
}
|
|
36
|
+
sanitizeForLogging(data) {
|
|
37
|
+
const sanitized = {};
|
|
38
|
+
for (const key of Object.keys(data)) {
|
|
39
|
+
const value = data[key];
|
|
40
|
+
if (typeof value === 'string') {
|
|
41
|
+
sanitized[key] = `[string:${value.length}]`;
|
|
42
|
+
}
|
|
43
|
+
else if (typeof value === 'number' || typeof value === 'boolean') {
|
|
44
|
+
sanitized[key] = value;
|
|
45
|
+
}
|
|
46
|
+
else if (Array.isArray(value)) {
|
|
47
|
+
sanitized[key] = `[array:${value.length}]`;
|
|
48
|
+
}
|
|
49
|
+
else if (value && typeof value === 'object') {
|
|
50
|
+
sanitized[key] = `[object:${Object.keys(value).length} keys]`;
|
|
51
|
+
}
|
|
52
|
+
else {
|
|
53
|
+
sanitized[key] = '[redacted]';
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
return sanitized;
|
|
57
|
+
}
|
|
58
|
+
};
|
|
59
|
+
exports.SafetyGuardrailService = SafetyGuardrailService;
|
|
60
|
+
exports.SafetyGuardrailService = SafetyGuardrailService = SafetyGuardrailService_1 = __decorate([
|
|
61
|
+
(0, common_1.Injectable)(),
|
|
62
|
+
__param(0, (0, common_1.Inject)(ai_options_1.AI_MODULE_OPTIONS)),
|
|
63
|
+
__metadata("design:paramtypes", [Object])
|
|
64
|
+
], SafetyGuardrailService);
|
|
65
|
+
//# sourceMappingURL=safety-guardrail.service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"safety-guardrail.service.js","sourceRoot":"","sources":["../../../../src/core/ai/safety/safety-guardrail.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CAA4D;AAC5D,8CAIuB;AAShB,IAAM,sBAAsB,8BAA5B,MAAM,sBAAsB;IAIjC,YAAuC,OAAwB;QAH9C,WAAM,GAAG,IAAI,eAAM,CAAC,wBAAsB,CAAC,IAAI,CAAC,CAAC;QAIhE,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,MAAM,EAAE,QAAQ,IAAI,EAAE,CAAC;IACjD,CAAC;IAED,cAAc,CAAC,IAAY,EAAE,QAAwB;QACnD,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,KAAK,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC9C,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBAChD,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnB,CAAC;QACH,CAAC;QACD,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,oCAAoC,CAAC,CAAC;QACpE,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,KAAK,EAAE,CAAC;IAC7C,CAAC;IAGD,kBAAkB,CAAC,IAA6B;QAC9C,MAAM,SAAS,GAA4B,EAAE,CAAC;QAC9C,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YACpC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;YACxB,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gBAC9B,SAAS,CAAC,GAAG,CAAC,GAAG,WAAW,KAAK,CAAC,MAAM,GAAG,CAAC;YAC9C,CAAC;iBAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,SAAS,EAAE,CAAC;gBACnE,SAAS,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YACzB,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBAChC,SAAS,CAAC,GAAG,CAAC,GAAG,UAAU,KAAK,CAAC,MAAM,GAAG,CAAC;YAC7C,CAAC;iBAAM,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gBAC9C,SAAS,CAAC,GAAG,CAAC,GAAG,WAAW,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,QAAQ,CAAC;YAChE,CAAC;iBAAM,CAAC;gBACN,SAAS,CAAC,GAAG,CAAC,GAAG,YAAY,CAAC;YAChC,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;CACF,CAAA;AAxCY,wDAAsB;iCAAtB,sBAAsB;IADlC,IAAA,mBAAU,GAAE;IAKE,WAAA,IAAA,eAAM,EAAC,8BAAiB,CAAC,CAAA;;GAJ3B,sBAAsB,CAwClC"}
|
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
import { AiService } from '../ai.service';
|
|
2
|
+
import { type AiModuleOptions } from '../ai.options';
|
|
3
|
+
import { SummarizationConfig } from '../ai.types';
|
|
4
|
+
export declare class SummarizationService {
|
|
5
|
+
private readonly aiService;
|
|
6
|
+
private readonly options;
|
|
7
|
+
private readonly logger;
|
|
8
|
+
constructor(aiService: AiService, options: AiModuleOptions);
|
|
9
|
+
shouldSummarize(messages: Array<{
|
|
10
|
+
role: string;
|
|
11
|
+
content: string;
|
|
12
|
+
}>, threshold?: number): boolean;
|
|
13
|
+
summarize(messages: Array<{
|
|
14
|
+
role: string;
|
|
15
|
+
content: string;
|
|
16
|
+
}>, config?: SummarizationConfig): Promise<string | null>;
|
|
17
|
+
getMessageWindow(messages: Array<{
|
|
18
|
+
role: string;
|
|
19
|
+
content: string;
|
|
20
|
+
}>, windowSize?: number): Array<{
|
|
21
|
+
role: string;
|
|
22
|
+
content: string;
|
|
23
|
+
}>;
|
|
24
|
+
}
|