@lenne.tech/nest-server 11.25.6 → 11.26.0
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/.claude/rules/configurable-features.md +2 -0
- package/CLAUDE.md +18 -0
- package/FRAMEWORK-API.md +48 -1
- package/dist/config.env.js +35 -1
- package/dist/config.env.js.map +1 -1
- package/dist/core/common/helpers/config.helper.js +2 -2
- package/dist/core/common/helpers/config.helper.js.map +1 -1
- package/dist/core/common/interceptors/check-security.interceptor.js +9 -2
- package/dist/core/common/interceptors/check-security.interceptor.js.map +1 -1
- package/dist/core/common/interfaces/server-options.interface.d.ts +89 -0
- package/dist/core/modules/ai/core-ai-mcp.controller.d.ts +21 -0
- package/dist/core/modules/ai/core-ai-mcp.controller.js +191 -0
- package/dist/core/modules/ai/core-ai-mcp.controller.js.map +1 -0
- package/dist/core/modules/ai/core-ai.controller.d.ts +90 -0
- package/dist/core/modules/ai/core-ai.controller.js +576 -0
- package/dist/core/modules/ai/core-ai.controller.js.map +1 -0
- package/dist/core/modules/ai/core-ai.module.d.ts +43 -0
- package/dist/core/modules/ai/core-ai.module.js +164 -0
- package/dist/core/modules/ai/core-ai.module.js.map +1 -0
- package/dist/core/modules/ai/core-ai.resolver.d.ts +84 -0
- package/dist/core/modules/ai/core-ai.resolver.js +530 -0
- package/dist/core/modules/ai/core-ai.resolver.js.map +1 -0
- package/dist/core/modules/ai/helpers/ai-mcp-oauth.helper.d.ts +7 -0
- package/dist/core/modules/ai/helpers/ai-mcp-oauth.helper.js +16 -0
- package/dist/core/modules/ai/helpers/ai-mcp-oauth.helper.js.map +1 -0
- package/dist/core/modules/ai/hooks/ai-hook.base.d.ts +10 -0
- package/dist/core/modules/ai/hooks/ai-hook.base.js +30 -0
- package/dist/core/modules/ai/hooks/ai-hook.base.js.map +1 -0
- package/dist/core/modules/ai/hooks/ai-hook.registry.d.ts +22 -0
- package/dist/core/modules/ai/hooks/ai-hook.registry.js +98 -0
- package/dist/core/modules/ai/hooks/ai-hook.registry.js.map +1 -0
- package/dist/core/modules/ai/index.d.ts +69 -0
- package/dist/core/modules/ai/index.js +86 -0
- package/dist/core/modules/ai/index.js.map +1 -0
- package/dist/core/modules/ai/inputs/core-ai-budget-limit-create.input.d.ts +5 -0
- package/dist/core/modules/ai/inputs/core-ai-budget-limit-create.input.js +43 -0
- package/dist/core/modules/ai/inputs/core-ai-budget-limit-create.input.js.map +1 -0
- package/dist/core/modules/ai/inputs/core-ai-budget-limit.input.d.ts +7 -0
- package/dist/core/modules/ai/inputs/core-ai-budget-limit.input.js +74 -0
- package/dist/core/modules/ai/inputs/core-ai-budget-limit.input.js.map +1 -0
- package/dist/core/modules/ai/inputs/core-ai-connection-create.input.d.ts +6 -0
- package/dist/core/modules/ai/inputs/core-ai-connection-create.input.js +49 -0
- package/dist/core/modules/ai/inputs/core-ai-connection-create.input.js.map +1 -0
- package/dist/core/modules/ai/inputs/core-ai-connection-preference.input.d.ts +6 -0
- package/dist/core/modules/ai/inputs/core-ai-connection-preference.input.js +60 -0
- package/dist/core/modules/ai/inputs/core-ai-connection-preference.input.js.map +1 -0
- package/dist/core/modules/ai/inputs/core-ai-connection.input.d.ts +23 -0
- package/dist/core/modules/ai/inputs/core-ai-connection.input.js +229 -0
- package/dist/core/modules/ai/inputs/core-ai-connection.input.js.map +1 -0
- package/dist/core/modules/ai/inputs/core-ai-conversation-create.input.d.ts +3 -0
- package/dist/core/modules/ai/inputs/core-ai-conversation-create.input.js +21 -0
- package/dist/core/modules/ai/inputs/core-ai-conversation-create.input.js.map +1 -0
- package/dist/core/modules/ai/inputs/core-ai-conversation.input.d.ts +4 -0
- package/dist/core/modules/ai/inputs/core-ai-conversation.input.js +42 -0
- package/dist/core/modules/ai/inputs/core-ai-conversation.input.js.map +1 -0
- package/dist/core/modules/ai/inputs/core-ai-prompt-create.input.d.ts +10 -0
- package/dist/core/modules/ai/inputs/core-ai-prompt-create.input.js +74 -0
- package/dist/core/modules/ai/inputs/core-ai-prompt-create.input.js.map +1 -0
- package/dist/core/modules/ai/inputs/core-ai-prompt-hint-create.input.d.ts +5 -0
- package/dist/core/modules/ai/inputs/core-ai-prompt-hint-create.input.js +41 -0
- package/dist/core/modules/ai/inputs/core-ai-prompt-hint-create.input.js.map +1 -0
- package/dist/core/modules/ai/inputs/core-ai-prompt-hint.input.d.ts +6 -0
- package/dist/core/modules/ai/inputs/core-ai-prompt-hint.input.js +63 -0
- package/dist/core/modules/ai/inputs/core-ai-prompt-hint.input.js.map +1 -0
- package/dist/core/modules/ai/inputs/core-ai-prompt-update.input.d.ts +10 -0
- package/dist/core/modules/ai/inputs/core-ai-prompt-update.input.js +70 -0
- package/dist/core/modules/ai/inputs/core-ai-prompt-update.input.js.map +1 -0
- package/dist/core/modules/ai/inputs/core-ai-prompt.input.d.ts +18 -0
- package/dist/core/modules/ai/inputs/core-ai-prompt.input.js +131 -0
- package/dist/core/modules/ai/inputs/core-ai-prompt.input.js.map +1 -0
- package/dist/core/modules/ai/inputs/core-ai-slot-create.input.d.ts +5 -0
- package/dist/core/modules/ai/inputs/core-ai-slot-create.input.js +41 -0
- package/dist/core/modules/ai/inputs/core-ai-slot-create.input.js.map +1 -0
- package/dist/core/modules/ai/inputs/core-ai-slot-update.input.d.ts +9 -0
- package/dist/core/modules/ai/inputs/core-ai-slot-update.input.js +91 -0
- package/dist/core/modules/ai/inputs/core-ai-slot-update.input.js.map +1 -0
- package/dist/core/modules/ai/interfaces/ai-hook.interface.d.ts +28 -0
- package/dist/core/modules/ai/interfaces/ai-hook.interface.js +3 -0
- package/dist/core/modules/ai/interfaces/ai-hook.interface.js.map +1 -0
- package/dist/core/modules/ai/interfaces/ai-placeholder.interface.d.ts +21 -0
- package/dist/core/modules/ai/interfaces/ai-placeholder.interface.js +3 -0
- package/dist/core/modules/ai/interfaces/ai-placeholder.interface.js.map +1 -0
- package/dist/core/modules/ai/interfaces/ai-tool.interface.d.ts +30 -0
- package/dist/core/modules/ai/interfaces/ai-tool.interface.js +3 -0
- package/dist/core/modules/ai/interfaces/ai-tool.interface.js.map +1 -0
- package/dist/core/modules/ai/interfaces/llm-provider.interface.d.ts +55 -0
- package/dist/core/modules/ai/interfaces/llm-provider.interface.js +3 -0
- package/dist/core/modules/ai/interfaces/llm-provider.interface.js.map +1 -0
- package/dist/core/modules/ai/interfaces/resolved-ai-connection.interface.d.ts +16 -0
- package/dist/core/modules/ai/interfaces/resolved-ai-connection.interface.js +3 -0
- package/dist/core/modules/ai/interfaces/resolved-ai-connection.interface.js.map +1 -0
- package/dist/core/modules/ai/models/core-ai-action.model.d.ts +6 -0
- package/dist/core/modules/ai/models/core-ai-action.model.js +44 -0
- package/dist/core/modules/ai/models/core-ai-action.model.js.map +1 -0
- package/dist/core/modules/ai/models/core-ai-available-connection.model.d.ts +8 -0
- package/dist/core/modules/ai/models/core-ai-available-connection.model.js +53 -0
- package/dist/core/modules/ai/models/core-ai-available-connection.model.js.map +1 -0
- package/dist/core/modules/ai/models/core-ai-budget-limit.model.d.ts +97 -0
- package/dist/core/modules/ai/models/core-ai-budget-limit.model.js +79 -0
- package/dist/core/modules/ai/models/core-ai-budget-limit.model.js.map +1 -0
- package/dist/core/modules/ai/models/core-ai-connection-preference.model.d.ts +91 -0
- package/dist/core/modules/ai/models/core-ai-connection-preference.model.js +67 -0
- package/dist/core/modules/ai/models/core-ai-connection-preference.model.js.map +1 -0
- package/dist/core/modules/ai/models/core-ai-connection.model.d.ts +200 -0
- package/dist/core/modules/ai/models/core-ai-connection.model.js +263 -0
- package/dist/core/modules/ai/models/core-ai-connection.model.js.map +1 -0
- package/dist/core/modules/ai/models/core-ai-conversation.model.d.ts +93 -0
- package/dist/core/modules/ai/models/core-ai-conversation.model.js +82 -0
- package/dist/core/modules/ai/models/core-ai-conversation.model.js.map +1 -0
- package/dist/core/modules/ai/models/core-ai-interaction.model.d.ts +133 -0
- package/dist/core/modules/ai/models/core-ai-interaction.model.js +137 -0
- package/dist/core/modules/ai/models/core-ai-interaction.model.js.map +1 -0
- package/dist/core/modules/ai/models/core-ai-message.model.d.ts +5 -0
- package/dist/core/modules/ai/models/core-ai-message.model.js +38 -0
- package/dist/core/modules/ai/models/core-ai-message.model.js.map +1 -0
- package/dist/core/modules/ai/models/core-ai-mode.model.d.ts +109 -0
- package/dist/core/modules/ai/models/core-ai-mode.model.js +100 -0
- package/dist/core/modules/ai/models/core-ai-mode.model.js.map +1 -0
- package/dist/core/modules/ai/models/core-ai-prompt-hint.model.d.ts +103 -0
- package/dist/core/modules/ai/models/core-ai-prompt-hint.model.js +89 -0
- package/dist/core/modules/ai/models/core-ai-prompt-hint.model.js.map +1 -0
- package/dist/core/modules/ai/models/core-ai-prompt.model.d.ts +122 -0
- package/dist/core/modules/ai/models/core-ai-prompt.model.js +117 -0
- package/dist/core/modules/ai/models/core-ai-prompt.model.js.map +1 -0
- package/dist/core/modules/ai/models/core-ai-response.model.d.ts +29 -0
- package/dist/core/modules/ai/models/core-ai-response.model.js +105 -0
- package/dist/core/modules/ai/models/core-ai-response.model.js.map +1 -0
- package/dist/core/modules/ai/models/core-ai-slot.model.d.ts +121 -0
- package/dist/core/modules/ai/models/core-ai-slot.model.js +119 -0
- package/dist/core/modules/ai/models/core-ai-slot.model.js.map +1 -0
- package/dist/core/modules/ai/models/core-ai-tool-grant.model.d.ts +103 -0
- package/dist/core/modules/ai/models/core-ai-tool-grant.model.js +88 -0
- package/dist/core/modules/ai/models/core-ai-tool-grant.model.js.map +1 -0
- package/dist/core/modules/ai/models/core-ai-tool-policy.model.d.ts +122 -0
- package/dist/core/modules/ai/models/core-ai-tool-policy.model.js +121 -0
- package/dist/core/modules/ai/models/core-ai-tool-policy.model.js.map +1 -0
- package/dist/core/modules/ai/models/core-ai-usage-info.model.d.ts +24 -0
- package/dist/core/modules/ai/models/core-ai-usage-info.model.js +127 -0
- package/dist/core/modules/ai/models/core-ai-usage-info.model.js.map +1 -0
- package/dist/core/modules/ai/models/core-ai-usage.model.d.ts +5 -0
- package/dist/core/modules/ai/models/core-ai-usage.model.js +38 -0
- package/dist/core/modules/ai/models/core-ai-usage.model.js.map +1 -0
- package/dist/core/modules/ai/providers/claude-cli.provider.d.ts +16 -0
- package/dist/core/modules/ai/providers/claude-cli.provider.js +151 -0
- package/dist/core/modules/ai/providers/claude-cli.provider.js.map +1 -0
- package/dist/core/modules/ai/providers/llm-provider.factory.d.ts +10 -0
- package/dist/core/modules/ai/providers/llm-provider.factory.js +45 -0
- package/dist/core/modules/ai/providers/llm-provider.factory.js.map +1 -0
- package/dist/core/modules/ai/providers/openai-compatible.provider.d.ts +32 -0
- package/dist/core/modules/ai/providers/openai-compatible.provider.js +236 -0
- package/dist/core/modules/ai/providers/openai-compatible.provider.js.map +1 -0
- package/dist/core/modules/ai/services/ai-crypto.service.d.ts +11 -0
- package/dist/core/modules/ai/services/ai-crypto.service.js +84 -0
- package/dist/core/modules/ai/services/ai-crypto.service.js.map +1 -0
- package/dist/core/modules/ai/services/core-ai-budget.service.d.ts +42 -0
- package/dist/core/modules/ai/services/core-ai-budget.service.js +211 -0
- package/dist/core/modules/ai/services/core-ai-budget.service.js.map +1 -0
- package/dist/core/modules/ai/services/core-ai-connection-preference.service.d.ts +18 -0
- package/dist/core/modules/ai/services/core-ai-connection-preference.service.js +53 -0
- package/dist/core/modules/ai/services/core-ai-connection-preference.service.js.map +1 -0
- package/dist/core/modules/ai/services/core-ai-connection-resolver.service.d.ts +55 -0
- package/dist/core/modules/ai/services/core-ai-connection-resolver.service.js +177 -0
- package/dist/core/modules/ai/services/core-ai-connection-resolver.service.js.map +1 -0
- package/dist/core/modules/ai/services/core-ai-connection.service.d.ts +43 -0
- package/dist/core/modules/ai/services/core-ai-connection.service.js +272 -0
- package/dist/core/modules/ai/services/core-ai-connection.service.js.map +1 -0
- package/dist/core/modules/ai/services/core-ai-conversation.service.d.ts +26 -0
- package/dist/core/modules/ai/services/core-ai-conversation.service.js +72 -0
- package/dist/core/modules/ai/services/core-ai-conversation.service.js.map +1 -0
- package/dist/core/modules/ai/services/core-ai-interaction.service.d.ts +13 -0
- package/dist/core/modules/ai/services/core-ai-interaction.service.js +52 -0
- package/dist/core/modules/ai/services/core-ai-interaction.service.js.map +1 -0
- package/dist/core/modules/ai/services/core-ai-mcp-client.service.d.ts +51 -0
- package/dist/core/modules/ai/services/core-ai-mcp-client.service.js +123 -0
- package/dist/core/modules/ai/services/core-ai-mcp-client.service.js.map +1 -0
- package/dist/core/modules/ai/services/core-ai-mcp-oauth.service.d.ts +68 -0
- package/dist/core/modules/ai/services/core-ai-mcp-oauth.service.js +279 -0
- package/dist/core/modules/ai/services/core-ai-mcp-oauth.service.js.map +1 -0
- package/dist/core/modules/ai/services/core-ai-mcp.service.d.ts +22 -0
- package/dist/core/modules/ai/services/core-ai-mcp.service.js +59 -0
- package/dist/core/modules/ai/services/core-ai-mcp.service.js.map +1 -0
- package/dist/core/modules/ai/services/core-ai-mode.service.d.ts +14 -0
- package/dist/core/modules/ai/services/core-ai-mode.service.js +56 -0
- package/dist/core/modules/ai/services/core-ai-mode.service.js.map +1 -0
- package/dist/core/modules/ai/services/core-ai-placeholder.registry.d.ts +15 -0
- package/dist/core/modules/ai/services/core-ai-placeholder.registry.js +102 -0
- package/dist/core/modules/ai/services/core-ai-placeholder.registry.js.map +1 -0
- package/dist/core/modules/ai/services/core-ai-prompt-builder.service.d.ts +42 -0
- package/dist/core/modules/ai/services/core-ai-prompt-builder.service.js +136 -0
- package/dist/core/modules/ai/services/core-ai-prompt-builder.service.js.map +1 -0
- package/dist/core/modules/ai/services/core-ai-prompt-hint.service.d.ts +29 -0
- package/dist/core/modules/ai/services/core-ai-prompt-hint.service.js +98 -0
- package/dist/core/modules/ai/services/core-ai-prompt-hint.service.js.map +1 -0
- package/dist/core/modules/ai/services/core-ai-prompt.service.d.ts +21 -0
- package/dist/core/modules/ai/services/core-ai-prompt.service.js +143 -0
- package/dist/core/modules/ai/services/core-ai-prompt.service.js.map +1 -0
- package/dist/core/modules/ai/services/core-ai-slot.service.d.ts +51 -0
- package/dist/core/modules/ai/services/core-ai-slot.service.js +372 -0
- package/dist/core/modules/ai/services/core-ai-slot.service.js.map +1 -0
- package/dist/core/modules/ai/services/core-ai-tool-grant.service.d.ts +22 -0
- package/dist/core/modules/ai/services/core-ai-tool-grant.service.js +97 -0
- package/dist/core/modules/ai/services/core-ai-tool-grant.service.js.map +1 -0
- package/dist/core/modules/ai/services/core-ai-tool-policy.service.d.ts +23 -0
- package/dist/core/modules/ai/services/core-ai-tool-policy.service.js +102 -0
- package/dist/core/modules/ai/services/core-ai-tool-policy.service.js.map +1 -0
- package/dist/core/modules/ai/services/core-ai.service.d.ts +144 -0
- package/dist/core/modules/ai/services/core-ai.service.js +991 -0
- package/dist/core/modules/ai/services/core-ai.service.js.map +1 -0
- package/dist/core/modules/ai/tools/ai-tool.base.d.ts +15 -0
- package/dist/core/modules/ai/tools/ai-tool.base.js +30 -0
- package/dist/core/modules/ai/tools/ai-tool.base.js.map +1 -0
- package/dist/core/modules/ai/tools/ai-tool.registry.d.ts +19 -0
- package/dist/core/modules/ai/tools/ai-tool.registry.js +71 -0
- package/dist/core/modules/ai/tools/ai-tool.registry.js.map +1 -0
- package/dist/core/modules/ai/tools/ask-user-question.tool.d.ts +38 -0
- package/dist/core/modules/ai/tools/ask-user-question.tool.js +65 -0
- package/dist/core/modules/ai/tools/ask-user-question.tool.js.map +1 -0
- package/dist/core/modules/ai/tools/search-tools.tool.d.ts +24 -0
- package/dist/core/modules/ai/tools/search-tools.tool.js +68 -0
- package/dist/core/modules/ai/tools/search-tools.tool.js.map +1 -0
- package/dist/core/modules/error-code/error-codes.d.ts +81 -0
- package/dist/core/modules/error-code/error-codes.js +72 -0
- package/dist/core/modules/error-code/error-codes.js.map +1 -1
- package/dist/core.module.js +8 -0
- package/dist/core.module.js.map +1 -1
- package/dist/index.d.ts +1 -0
- package/dist/index.js +1 -0
- package/dist/index.js.map +1 -1
- package/dist/server/modules/ai/ai-tools.module.d.ts +5 -0
- package/dist/server/modules/ai/ai-tools.module.js +36 -0
- package/dist/server/modules/ai/ai-tools.module.js.map +1 -0
- package/dist/server/modules/ai/tools/delete-user.tool.d.ts +30 -0
- package/dist/server/modules/ai/tools/delete-user.tool.js +49 -0
- package/dist/server/modules/ai/tools/delete-user.tool.js.map +1 -0
- package/dist/server/modules/ai/tools/find-users.tool.d.ts +30 -0
- package/dist/server/modules/ai/tools/find-users.tool.js +52 -0
- package/dist/server/modules/ai/tools/find-users.tool.js.map +1 -0
- package/dist/server/modules/ai/tools/get-user.tool.d.ts +26 -0
- package/dist/server/modules/ai/tools/get-user.tool.js +48 -0
- package/dist/server/modules/ai/tools/get-user.tool.js.map +1 -0
- package/dist/server/modules/ai/tools/update-user-job-title.tool.d.ts +32 -0
- package/dist/server/modules/ai/tools/update-user-job-title.tool.js +54 -0
- package/dist/server/modules/ai/tools/update-user-job-title.tool.js.map +1 -0
- package/dist/server/modules/error-code/error-codes.d.ts +9 -0
- package/dist/server/server.module.js +2 -0
- package/dist/server/server.module.js.map +1 -1
- package/dist/tsconfig.build.tsbuildinfo +1 -1
- package/docs/REQUEST-LIFECYCLE.md +10 -0
- package/migration-guides/11.25.x-to-11.26.0.md +278 -0
- package/package.json +2 -1
- package/src/config.env.ts +48 -1
- package/src/core/common/helpers/config.helper.ts +5 -2
- package/src/core/common/interceptors/check-security.interceptor.ts +13 -3
- package/src/core/common/interfaces/server-options.interface.ts +312 -0
- package/src/core/modules/ai/INTEGRATION-CHECKLIST.md +220 -0
- package/src/core/modules/ai/README.md +526 -0
- package/src/core/modules/ai/core-ai-mcp.controller.ts +223 -0
- package/src/core/modules/ai/core-ai.controller.ts +547 -0
- package/src/core/modules/ai/core-ai.module.ts +269 -0
- package/src/core/modules/ai/core-ai.resolver.ts +526 -0
- package/src/core/modules/ai/helpers/ai-mcp-oauth.helper.ts +35 -0
- package/src/core/modules/ai/hooks/ai-hook.base.ts +23 -0
- package/src/core/modules/ai/hooks/ai-hook.registry.ts +119 -0
- package/src/core/modules/ai/index.ts +70 -0
- package/src/core/modules/ai/inputs/core-ai-budget-limit-create.input.ts +27 -0
- package/src/core/modules/ai/inputs/core-ai-budget-limit.input.ts +52 -0
- package/src/core/modules/ai/inputs/core-ai-connection-create.input.ts +32 -0
- package/src/core/modules/ai/inputs/core-ai-connection-preference.input.ts +42 -0
- package/src/core/modules/ai/inputs/core-ai-connection.input.ts +235 -0
- package/src/core/modules/ai/inputs/core-ai-conversation-create.input.ts +13 -0
- package/src/core/modules/ai/inputs/core-ai-conversation.input.ts +33 -0
- package/src/core/modules/ai/inputs/core-ai-prompt-create.input.ts +50 -0
- package/src/core/modules/ai/inputs/core-ai-prompt-hint-create.input.ts +26 -0
- package/src/core/modules/ai/inputs/core-ai-prompt-hint.input.ts +43 -0
- package/src/core/modules/ai/inputs/core-ai-prompt-update.input.ts +42 -0
- package/src/core/modules/ai/inputs/core-ai-prompt.input.ts +144 -0
- package/src/core/modules/ai/inputs/core-ai-slot-create.input.ts +26 -0
- package/src/core/modules/ai/inputs/core-ai-slot-update.input.ts +65 -0
- package/src/core/modules/ai/interfaces/ai-hook.interface.ts +74 -0
- package/src/core/modules/ai/interfaces/ai-placeholder.interface.ts +49 -0
- package/src/core/modules/ai/interfaces/ai-tool.interface.ts +112 -0
- package/src/core/modules/ai/interfaces/llm-provider.interface.ts +187 -0
- package/src/core/modules/ai/interfaces/resolved-ai-connection.interface.ts +55 -0
- package/src/core/modules/ai/models/core-ai-action.model.ts +37 -0
- package/src/core/modules/ai/models/core-ai-available-connection.model.ts +42 -0
- package/src/core/modules/ai/models/core-ai-budget-limit.model.ts +82 -0
- package/src/core/modules/ai/models/core-ai-connection-preference.model.ts +72 -0
- package/src/core/modules/ai/models/core-ai-connection.model.ts +330 -0
- package/src/core/modules/ai/models/core-ai-conversation.model.ts +87 -0
- package/src/core/modules/ai/models/core-ai-interaction.model.ts +145 -0
- package/src/core/modules/ai/models/core-ai-message.model.ts +29 -0
- package/src/core/modules/ai/models/core-ai-mode.model.ts +93 -0
- package/src/core/modules/ai/models/core-ai-prompt-hint.model.ts +98 -0
- package/src/core/modules/ai/models/core-ai-prompt.model.ts +120 -0
- package/src/core/modules/ai/models/core-ai-response.model.ts +119 -0
- package/src/core/modules/ai/models/core-ai-slot.model.ts +146 -0
- package/src/core/modules/ai/models/core-ai-tool-grant.model.ts +87 -0
- package/src/core/modules/ai/models/core-ai-tool-policy.model.ts +115 -0
- package/src/core/modules/ai/models/core-ai-usage-info.model.ts +107 -0
- package/src/core/modules/ai/models/core-ai-usage.model.ts +29 -0
- package/src/core/modules/ai/providers/claude-cli.provider.ts +240 -0
- package/src/core/modules/ai/providers/llm-provider.factory.ts +63 -0
- package/src/core/modules/ai/providers/openai-compatible.provider.ts +329 -0
- package/src/core/modules/ai/services/ai-crypto.service.ts +120 -0
- package/src/core/modules/ai/services/core-ai-budget.service.ts +297 -0
- package/src/core/modules/ai/services/core-ai-connection-preference.service.ts +74 -0
- package/src/core/modules/ai/services/core-ai-connection-resolver.service.ts +308 -0
- package/src/core/modules/ai/services/core-ai-connection.service.ts +387 -0
- package/src/core/modules/ai/services/core-ai-conversation.service.ts +105 -0
- package/src/core/modules/ai/services/core-ai-interaction.service.ts +56 -0
- package/src/core/modules/ai/services/core-ai-mcp-client.service.ts +169 -0
- package/src/core/modules/ai/services/core-ai-mcp-oauth.service.ts +445 -0
- package/src/core/modules/ai/services/core-ai-mcp.service.ts +94 -0
- package/src/core/modules/ai/services/core-ai-mode.service.ts +43 -0
- package/src/core/modules/ai/services/core-ai-placeholder.registry.ts +123 -0
- package/src/core/modules/ai/services/core-ai-prompt-builder.service.ts +228 -0
- package/src/core/modules/ai/services/core-ai-prompt-hint.service.ts +133 -0
- package/src/core/modules/ai/services/core-ai-prompt.service.ts +164 -0
- package/src/core/modules/ai/services/core-ai-slot.service.ts +526 -0
- package/src/core/modules/ai/services/core-ai-tool-grant.service.ts +107 -0
- package/src/core/modules/ai/services/core-ai-tool-policy.service.ts +124 -0
- package/src/core/modules/ai/services/core-ai.service.ts +1396 -0
- package/src/core/modules/ai/tools/ai-tool.base.ts +46 -0
- package/src/core/modules/ai/tools/ai-tool.registry.ts +124 -0
- package/src/core/modules/ai/tools/ask-user-question.tool.ts +76 -0
- package/src/core/modules/ai/tools/search-tools.tool.ts +70 -0
- package/src/core/modules/error-code/error-codes.ts +86 -0
- package/src/core.module.ts +12 -0
- package/src/index.ts +6 -0
- package/src/server/modules/ai/ai-tools.module.ts +33 -0
- package/src/server/modules/ai/tools/delete-user.tool.ts +44 -0
- package/src/server/modules/ai/tools/find-users.tool.ts +49 -0
- package/src/server/modules/ai/tools/get-user.tool.ts +43 -0
- package/src/server/modules/ai/tools/update-user-job-title.tool.ts +62 -0
- package/src/server/server.module.ts +5 -0
|
@@ -0,0 +1,84 @@
|
|
|
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 AiCryptoService_1;
|
|
9
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
10
|
+
exports.AiCryptoService = void 0;
|
|
11
|
+
const common_1 = require("@nestjs/common");
|
|
12
|
+
const node_crypto_1 = require("node:crypto");
|
|
13
|
+
const cookies_helper_1 = require("../../../common/helpers/cookies.helper");
|
|
14
|
+
const config_service_1 = require("../../../common/services/config.service");
|
|
15
|
+
let AiCryptoService = AiCryptoService_1 = class AiCryptoService {
|
|
16
|
+
logger = new common_1.Logger(AiCryptoService_1.name);
|
|
17
|
+
warned = false;
|
|
18
|
+
onModuleInit() {
|
|
19
|
+
this.assertProductionSafe();
|
|
20
|
+
}
|
|
21
|
+
assertProductionSafe() {
|
|
22
|
+
if ((0, cookies_helper_1.isProductionLikeEnv)(config_service_1.ConfigService.get('env')) && !this.resolveSecret()) {
|
|
23
|
+
throw new Error('AI encryption secret is required in production/staging but none is set ' +
|
|
24
|
+
'(ai.encryptionSecret / NSC__AI__ENCRYPTION_SECRET / SECRETS_ENCRYPTION_KEY). ' +
|
|
25
|
+
'Without it, stored AI API keys are encrypted with a public, insecure development ' +
|
|
26
|
+
'default. Set a random 32+ char value.');
|
|
27
|
+
}
|
|
28
|
+
}
|
|
29
|
+
decrypt(cipherText) {
|
|
30
|
+
if (!cipherText) {
|
|
31
|
+
return '';
|
|
32
|
+
}
|
|
33
|
+
const parts = cipherText.split('.');
|
|
34
|
+
if (parts.length !== 3) {
|
|
35
|
+
return cipherText;
|
|
36
|
+
}
|
|
37
|
+
const [ivB64, tagB64, dataB64] = parts;
|
|
38
|
+
try {
|
|
39
|
+
const iv = Buffer.from(ivB64, 'base64');
|
|
40
|
+
const tag = Buffer.from(tagB64, 'base64');
|
|
41
|
+
const data = Buffer.from(dataB64, 'base64');
|
|
42
|
+
const decipher = (0, node_crypto_1.createDecipheriv)('aes-256-gcm', this.getKey(), iv);
|
|
43
|
+
decipher.setAuthTag(tag);
|
|
44
|
+
const plain = Buffer.concat([decipher.update(data), decipher.final()]);
|
|
45
|
+
return plain.toString('utf8');
|
|
46
|
+
}
|
|
47
|
+
catch (err) {
|
|
48
|
+
this.logger.error(`Decrypt failed: ${err.message}`);
|
|
49
|
+
throw new Error('Secret decryption failed — key mismatch or corrupted value', { cause: err });
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
encrypt(plainText) {
|
|
53
|
+
if (plainText === '' || plainText === undefined || plainText === null) {
|
|
54
|
+
return '';
|
|
55
|
+
}
|
|
56
|
+
const iv = (0, node_crypto_1.randomBytes)(12);
|
|
57
|
+
const cipher = (0, node_crypto_1.createCipheriv)('aes-256-gcm', this.getKey(), iv);
|
|
58
|
+
const encrypted = Buffer.concat([cipher.update(plainText, 'utf8'), cipher.final()]);
|
|
59
|
+
const tag = cipher.getAuthTag();
|
|
60
|
+
return [iv.toString('base64'), tag.toString('base64'), encrypted.toString('base64')].join('.');
|
|
61
|
+
}
|
|
62
|
+
resolveSecret() {
|
|
63
|
+
return (config_service_1.ConfigService.get('ai.encryptionSecret') ||
|
|
64
|
+
process.env.NSC__AI__ENCRYPTION_SECRET ||
|
|
65
|
+
process.env.SECRETS_ENCRYPTION_KEY);
|
|
66
|
+
}
|
|
67
|
+
getKey() {
|
|
68
|
+
const raw = this.resolveSecret();
|
|
69
|
+
if (!raw) {
|
|
70
|
+
if (!this.warned) {
|
|
71
|
+
this.logger.warn('No AI encryption secret set (ai.encryptionSecret / NSC__AI__ENCRYPTION_SECRET / SECRETS_ENCRYPTION_KEY) — ' +
|
|
72
|
+
'using an insecure development default. DO NOT run this in production; set a random 32+ char value.');
|
|
73
|
+
this.warned = true;
|
|
74
|
+
}
|
|
75
|
+
return (0, node_crypto_1.createHash)('sha256').update('lt-nest-server-ai-dev-only-insecure-default').digest();
|
|
76
|
+
}
|
|
77
|
+
return (0, node_crypto_1.createHash)('sha256').update(raw, 'utf8').digest();
|
|
78
|
+
}
|
|
79
|
+
};
|
|
80
|
+
exports.AiCryptoService = AiCryptoService;
|
|
81
|
+
exports.AiCryptoService = AiCryptoService = AiCryptoService_1 = __decorate([
|
|
82
|
+
(0, common_1.Injectable)()
|
|
83
|
+
], AiCryptoService);
|
|
84
|
+
//# sourceMappingURL=ai-crypto.service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ai-crypto.service.js","sourceRoot":"","sources":["../../../../../src/core/modules/ai/services/ai-crypto.service.ts"],"names":[],"mappings":";;;;;;;;;;AAAA,2CAAkE;AAClE,6CAAwF;AAExF,2EAA6E;AAC7E,4EAAwE;AAkBjE,IAAM,eAAe,uBAArB,MAAM,eAAe;IACT,MAAM,GAAG,IAAI,eAAM,CAAC,iBAAe,CAAC,IAAI,CAAC,CAAC;IACnD,MAAM,GAAG,KAAK,CAAC;IAOvB,YAAY;QACV,IAAI,CAAC,oBAAoB,EAAE,CAAC;IAC9B,CAAC;IAMD,oBAAoB;QAClB,IAAI,IAAA,oCAAmB,EAAC,8BAAa,CAAC,GAAG,CAAS,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC;YACnF,MAAM,IAAI,KAAK,CACb,yEAAyE;gBACvE,+EAA+E;gBAC/E,mFAAmF;gBACnF,uCAAuC,CAC1C,CAAC;QACJ,CAAC;IACH,CAAC;IAMD,OAAO,CAAC,UAAkB;QACxB,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACpC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,OAAO,UAAU,CAAC;QACpB,CAAC;QACD,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,GAAG,KAAK,CAAC;QACvC,IAAI,CAAC;YACH,MAAM,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;YACxC,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YAC1C,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC5C,MAAM,QAAQ,GAAG,IAAA,8BAAgB,EAAC,aAAa,EAAE,IAAI,CAAC,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;YACpE,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YACzB,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YACvE,OAAO,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,mBAAoB,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;YAC/D,MAAM,IAAI,KAAK,CAAC,4DAA4D,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QAChG,CAAC;IACH,CAAC;IAMD,OAAO,CAAC,SAAiB;QACvB,IAAI,SAAS,KAAK,EAAE,IAAI,SAAS,KAAK,SAAS,IAAI,SAAS,KAAK,IAAI,EAAE,CAAC;YACtE,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,EAAE,GAAG,IAAA,yBAAW,EAAC,EAAE,CAAC,CAAC;QAC3B,MAAM,MAAM,GAAG,IAAA,4BAAc,EAAC,aAAa,EAAE,IAAI,CAAC,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAChE,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACpF,MAAM,GAAG,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;QAChC,OAAO,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACjG,CAAC;IAOS,aAAa;QACrB,OAAO,CACL,8BAAa,CAAC,GAAG,CAAS,qBAAqB,CAAC;YAChD,OAAO,CAAC,GAAG,CAAC,0BAA0B;YACtC,OAAO,CAAC,GAAG,CAAC,sBAAsB,CACnC,CAAC;IACJ,CAAC;IAES,MAAM;QACd,MAAM,GAAG,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;QACjC,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACjB,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,4GAA4G;oBAC1G,oGAAoG,CACvG,CAAC;gBACF,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;YACrB,CAAC;YACD,OAAO,IAAA,wBAAU,EAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,6CAA6C,CAAC,CAAC,MAAM,EAAE,CAAC;QAC7F,CAAC;QACD,OAAO,IAAA,wBAAU,EAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC,MAAM,EAAE,CAAC;IAC3D,CAAC;CACF,CAAA;AAjGY,0CAAe;0BAAf,eAAe;IAD3B,IAAA,mBAAU,GAAE;GACA,eAAe,CAiG3B"}
|
|
@@ -0,0 +1,42 @@
|
|
|
1
|
+
import { Connection, Model } from 'mongoose';
|
|
2
|
+
import { CrudService } from '../../../common/services/crud.service';
|
|
3
|
+
import { CoreModelConstructor } from '../../../common/types/core-model-constructor.type';
|
|
4
|
+
import { CoreAiBudgetLimitCreateInput } from '../inputs/core-ai-budget-limit-create.input';
|
|
5
|
+
import { CoreAiBudgetLimitInput } from '../inputs/core-ai-budget-limit.input';
|
|
6
|
+
import { AiBudgetLimitDocument, CoreAiBudgetLimit } from '../models/core-ai-budget-limit.model';
|
|
7
|
+
import { CoreAiBudgetSummary, CoreAiUsageInfo, CoreAiUsageScope } from '../models/core-ai-usage-info.model';
|
|
8
|
+
export declare const AI_BUDGET_LIMIT_MODEL = "AiBudgetLimit";
|
|
9
|
+
export declare const AI_BUDGET_LIMIT_CLASS = "AI_BUDGET_LIMIT_CLASS";
|
|
10
|
+
export interface ResolvedAiBudgetLimit {
|
|
11
|
+
maxPrompts?: number;
|
|
12
|
+
maxTokens?: number;
|
|
13
|
+
period: string;
|
|
14
|
+
}
|
|
15
|
+
export interface AiBudgetUsage {
|
|
16
|
+
resetAt: Date | null;
|
|
17
|
+
usedPrompts: number;
|
|
18
|
+
usedTokens: number;
|
|
19
|
+
}
|
|
20
|
+
export declare class CoreAiBudgetService extends CrudService<CoreAiBudgetLimit, CoreAiBudgetLimitCreateInput, CoreAiBudgetLimitInput> {
|
|
21
|
+
protected readonly connection: Connection;
|
|
22
|
+
protected readonly mainDbModel: Model<AiBudgetLimitDocument>;
|
|
23
|
+
protected readonly mainModelConstructor: CoreModelConstructor<CoreAiBudgetLimit>;
|
|
24
|
+
protected readonly interactionsCollection = "aiInteractions";
|
|
25
|
+
constructor(connection: Connection, mainDbModel: Model<AiBudgetLimitDocument>, mainModelConstructor: CoreModelConstructor<CoreAiBudgetLimit>);
|
|
26
|
+
resolveLimit(scope: 'tenant' | 'user', refId: string): Promise<ResolvedAiBudgetLimit>;
|
|
27
|
+
getUsage(filter: {
|
|
28
|
+
tenantId?: string;
|
|
29
|
+
userId?: string;
|
|
30
|
+
}, period: string): Promise<AiBudgetUsage>;
|
|
31
|
+
assertWithinBudget(userId?: string, tenantId?: string, language?: string): Promise<void>;
|
|
32
|
+
getUsageInfo(userId?: string, tenantId?: string): Promise<CoreAiUsageInfo>;
|
|
33
|
+
buildSummary(userId: string | undefined, tenantId: string | undefined, promptTokens: number, llmContextWindow?: number, providerQuota?: {
|
|
34
|
+
maxTokens?: number;
|
|
35
|
+
period?: string;
|
|
36
|
+
}): Promise<CoreAiBudgetSummary>;
|
|
37
|
+
protected buildScope(scope: 'tenant' | 'user', refId: string): Promise<CoreAiUsageScope>;
|
|
38
|
+
protected finite(value?: number): boolean;
|
|
39
|
+
protected exceeded(usage: AiBudgetUsage, limit: ResolvedAiBudgetLimit): boolean;
|
|
40
|
+
protected periodStart(period: string): Date;
|
|
41
|
+
protected nextReset(period: string): Date | null;
|
|
42
|
+
}
|
|
@@ -0,0 +1,211 @@
|
|
|
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
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
15
|
+
exports.CoreAiBudgetService = exports.AI_BUDGET_LIMIT_CLASS = exports.AI_BUDGET_LIMIT_MODEL = void 0;
|
|
16
|
+
const common_1 = require("@nestjs/common");
|
|
17
|
+
const mongoose_1 = require("@nestjs/mongoose");
|
|
18
|
+
const mongoose_2 = require("mongoose");
|
|
19
|
+
const config_service_1 = require("../../../common/services/config.service");
|
|
20
|
+
const crud_service_1 = require("../../../common/services/crud.service");
|
|
21
|
+
const error_code_1 = require("../../error-code");
|
|
22
|
+
const core_ai_usage_info_model_1 = require("../models/core-ai-usage-info.model");
|
|
23
|
+
exports.AI_BUDGET_LIMIT_MODEL = 'AiBudgetLimit';
|
|
24
|
+
exports.AI_BUDGET_LIMIT_CLASS = 'AI_BUDGET_LIMIT_CLASS';
|
|
25
|
+
let CoreAiBudgetService = class CoreAiBudgetService extends crud_service_1.CrudService {
|
|
26
|
+
connection;
|
|
27
|
+
mainDbModel;
|
|
28
|
+
mainModelConstructor;
|
|
29
|
+
interactionsCollection = 'aiInteractions';
|
|
30
|
+
constructor(connection, mainDbModel, mainModelConstructor) {
|
|
31
|
+
super();
|
|
32
|
+
this.connection = connection;
|
|
33
|
+
this.mainDbModel = mainDbModel;
|
|
34
|
+
this.mainModelConstructor = mainModelConstructor;
|
|
35
|
+
}
|
|
36
|
+
async resolveLimit(scope, refId) {
|
|
37
|
+
const override = await this.mainDbModel.findOne({ refId, scope }).lean().exec();
|
|
38
|
+
const cfg = config_service_1.ConfigService.get('ai.budget');
|
|
39
|
+
const defaults = scope === 'tenant' ? cfg?.tenant : cfg?.user;
|
|
40
|
+
return {
|
|
41
|
+
maxPrompts: override?.maxPrompts ?? defaults?.maxPrompts,
|
|
42
|
+
maxTokens: override?.maxTokens ?? defaults?.maxTokens,
|
|
43
|
+
period: override?.period || cfg?.period || 'day',
|
|
44
|
+
};
|
|
45
|
+
}
|
|
46
|
+
async getUsage(filter, period) {
|
|
47
|
+
const periodStart = this.periodStart(period);
|
|
48
|
+
const match = { createdAt: { $gte: periodStart } };
|
|
49
|
+
if (filter.userId) {
|
|
50
|
+
match.userId = filter.userId;
|
|
51
|
+
}
|
|
52
|
+
if (filter.tenantId) {
|
|
53
|
+
match.tenantId = filter.tenantId;
|
|
54
|
+
}
|
|
55
|
+
const db = this.connection.db;
|
|
56
|
+
if (!db) {
|
|
57
|
+
return { resetAt: this.nextReset(period), usedPrompts: 0, usedTokens: 0 };
|
|
58
|
+
}
|
|
59
|
+
const [agg] = await db
|
|
60
|
+
.collection(this.interactionsCollection)
|
|
61
|
+
.aggregate([
|
|
62
|
+
{ $match: match },
|
|
63
|
+
{ $group: { _id: null, usedPrompts: { $sum: 1 }, usedTokens: { $sum: { $ifNull: ['$totalTokens', 0] } } } },
|
|
64
|
+
])
|
|
65
|
+
.toArray();
|
|
66
|
+
return { resetAt: this.nextReset(period), usedPrompts: agg?.usedPrompts ?? 0, usedTokens: agg?.usedTokens ?? 0 };
|
|
67
|
+
}
|
|
68
|
+
async assertWithinBudget(userId, tenantId, language) {
|
|
69
|
+
for (const [scope, refId] of [
|
|
70
|
+
['user', userId],
|
|
71
|
+
['tenant', tenantId],
|
|
72
|
+
]) {
|
|
73
|
+
if (!refId) {
|
|
74
|
+
continue;
|
|
75
|
+
}
|
|
76
|
+
const limit = await this.resolveLimit(scope, refId);
|
|
77
|
+
if (!this.finite(limit.maxTokens) && !this.finite(limit.maxPrompts)) {
|
|
78
|
+
continue;
|
|
79
|
+
}
|
|
80
|
+
const usage = await this.getUsage(scope === 'user' ? { userId: refId } : { tenantId: refId }, limit.period);
|
|
81
|
+
if (this.exceeded(usage, limit)) {
|
|
82
|
+
void language;
|
|
83
|
+
throw new common_1.HttpException(error_code_1.ErrorCode.AI_BUDGET_EXCEEDED, common_1.HttpStatus.TOO_MANY_REQUESTS);
|
|
84
|
+
}
|
|
85
|
+
}
|
|
86
|
+
}
|
|
87
|
+
async getUsageInfo(userId, tenantId) {
|
|
88
|
+
const info = new core_ai_usage_info_model_1.CoreAiUsageInfo();
|
|
89
|
+
if (userId) {
|
|
90
|
+
info.user = await this.buildScope('user', userId);
|
|
91
|
+
}
|
|
92
|
+
if (tenantId) {
|
|
93
|
+
info.tenant = await this.buildScope('tenant', tenantId);
|
|
94
|
+
}
|
|
95
|
+
return info;
|
|
96
|
+
}
|
|
97
|
+
async buildSummary(userId, tenantId, promptTokens, llmContextWindow, providerQuota) {
|
|
98
|
+
const summary = new core_ai_usage_info_model_1.CoreAiBudgetSummary();
|
|
99
|
+
summary.promptTokens = promptTokens;
|
|
100
|
+
if (!userId) {
|
|
101
|
+
if (this.finite(llmContextWindow)) {
|
|
102
|
+
summary.maxTokens = llmContextWindow;
|
|
103
|
+
summary.scope = 'llm';
|
|
104
|
+
}
|
|
105
|
+
return summary;
|
|
106
|
+
}
|
|
107
|
+
const userLimit = await this.resolveLimit('user', userId);
|
|
108
|
+
if (this.finite(userLimit.maxTokens) || this.finite(userLimit.maxPrompts)) {
|
|
109
|
+
const usage = await this.getUsage({ userId }, userLimit.period);
|
|
110
|
+
summary.usedTokens = usage.usedTokens;
|
|
111
|
+
summary.remainingTokens = this.finite(userLimit.maxTokens)
|
|
112
|
+
? Math.max(0, userLimit.maxTokens - usage.usedTokens)
|
|
113
|
+
: undefined;
|
|
114
|
+
summary.resetAt = usage.resetAt ?? undefined;
|
|
115
|
+
if (this.finite(userLimit.maxTokens)) {
|
|
116
|
+
summary.maxTokens = userLimit.maxTokens;
|
|
117
|
+
summary.scope = 'user';
|
|
118
|
+
}
|
|
119
|
+
return summary;
|
|
120
|
+
}
|
|
121
|
+
if (tenantId) {
|
|
122
|
+
const tenantLimit = await this.resolveLimit('tenant', tenantId);
|
|
123
|
+
if (this.finite(tenantLimit.maxTokens)) {
|
|
124
|
+
const usage = await this.getUsage({ tenantId }, tenantLimit.period);
|
|
125
|
+
summary.usedTokens = usage.usedTokens;
|
|
126
|
+
summary.remainingTokens = Math.max(0, tenantLimit.maxTokens - usage.usedTokens);
|
|
127
|
+
summary.resetAt = usage.resetAt ?? undefined;
|
|
128
|
+
summary.maxTokens = tenantLimit.maxTokens;
|
|
129
|
+
summary.scope = 'tenant';
|
|
130
|
+
return summary;
|
|
131
|
+
}
|
|
132
|
+
}
|
|
133
|
+
const period = providerQuota?.period || userLimit.period;
|
|
134
|
+
const usage = await this.getUsage({ userId }, period);
|
|
135
|
+
summary.usedTokens = usage.usedTokens;
|
|
136
|
+
summary.resetAt = usage.resetAt ?? undefined;
|
|
137
|
+
if (this.finite(providerQuota?.maxTokens)) {
|
|
138
|
+
summary.maxTokens = providerQuota.maxTokens;
|
|
139
|
+
summary.remainingTokens = Math.max(0, providerQuota.maxTokens - usage.usedTokens);
|
|
140
|
+
summary.scope = 'llm';
|
|
141
|
+
return summary;
|
|
142
|
+
}
|
|
143
|
+
if (this.finite(llmContextWindow)) {
|
|
144
|
+
summary.maxTokens = llmContextWindow;
|
|
145
|
+
summary.scope = 'llm';
|
|
146
|
+
}
|
|
147
|
+
return summary;
|
|
148
|
+
}
|
|
149
|
+
async buildScope(scope, refId) {
|
|
150
|
+
const limit = await this.resolveLimit(scope, refId);
|
|
151
|
+
const usage = await this.getUsage(scope === 'user' ? { userId: refId } : { tenantId: refId }, limit.period);
|
|
152
|
+
const result = new core_ai_usage_info_model_1.CoreAiUsageScope();
|
|
153
|
+
result.maxPrompts = this.finite(limit.maxPrompts) ? limit.maxPrompts : undefined;
|
|
154
|
+
result.maxTokens = this.finite(limit.maxTokens) ? limit.maxTokens : undefined;
|
|
155
|
+
result.period = limit.period;
|
|
156
|
+
result.refId = refId;
|
|
157
|
+
result.remainingPrompts = this.finite(limit.maxPrompts)
|
|
158
|
+
? Math.max(0, limit.maxPrompts - usage.usedPrompts)
|
|
159
|
+
: undefined;
|
|
160
|
+
result.remainingTokens = this.finite(limit.maxTokens)
|
|
161
|
+
? Math.max(0, limit.maxTokens - usage.usedTokens)
|
|
162
|
+
: undefined;
|
|
163
|
+
result.resetAt = usage.resetAt ?? undefined;
|
|
164
|
+
result.scope = scope;
|
|
165
|
+
result.usedPrompts = usage.usedPrompts;
|
|
166
|
+
result.usedTokens = usage.usedTokens;
|
|
167
|
+
return result;
|
|
168
|
+
}
|
|
169
|
+
finite(value) {
|
|
170
|
+
return typeof value === 'number' && value > 0;
|
|
171
|
+
}
|
|
172
|
+
exceeded(usage, limit) {
|
|
173
|
+
return ((this.finite(limit.maxTokens) && usage.usedTokens >= limit.maxTokens) ||
|
|
174
|
+
(this.finite(limit.maxPrompts) && usage.usedPrompts >= limit.maxPrompts));
|
|
175
|
+
}
|
|
176
|
+
periodStart(period) {
|
|
177
|
+
const now = new Date();
|
|
178
|
+
if (period === 'month') {
|
|
179
|
+
return new Date(now.getFullYear(), now.getMonth(), 1, 0, 0, 0, 0);
|
|
180
|
+
}
|
|
181
|
+
if (period === 'none') {
|
|
182
|
+
return new Date(0);
|
|
183
|
+
}
|
|
184
|
+
const start = new Date(now);
|
|
185
|
+
start.setHours(0, 0, 0, 0);
|
|
186
|
+
return start;
|
|
187
|
+
}
|
|
188
|
+
nextReset(period) {
|
|
189
|
+
const now = new Date();
|
|
190
|
+
if (period === 'month') {
|
|
191
|
+
return new Date(now.getFullYear(), now.getMonth() + 1, 1, 0, 0, 0, 0);
|
|
192
|
+
}
|
|
193
|
+
if (period === 'none') {
|
|
194
|
+
return null;
|
|
195
|
+
}
|
|
196
|
+
const next = new Date(now);
|
|
197
|
+
next.setHours(0, 0, 0, 0);
|
|
198
|
+
next.setDate(next.getDate() + 1);
|
|
199
|
+
return next;
|
|
200
|
+
}
|
|
201
|
+
};
|
|
202
|
+
exports.CoreAiBudgetService = CoreAiBudgetService;
|
|
203
|
+
exports.CoreAiBudgetService = CoreAiBudgetService = __decorate([
|
|
204
|
+
(0, common_1.Injectable)(),
|
|
205
|
+
__param(0, (0, mongoose_1.InjectConnection)()),
|
|
206
|
+
__param(1, (0, mongoose_1.InjectModel)(exports.AI_BUDGET_LIMIT_MODEL)),
|
|
207
|
+
__param(2, (0, common_1.Inject)(exports.AI_BUDGET_LIMIT_CLASS)),
|
|
208
|
+
__metadata("design:paramtypes", [mongoose_2.Connection,
|
|
209
|
+
mongoose_2.Model, Object])
|
|
210
|
+
], CoreAiBudgetService);
|
|
211
|
+
//# sourceMappingURL=core-ai-budget.service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"core-ai-budget.service.js","sourceRoot":"","sources":["../../../../../src/core/modules/ai/services/core-ai-budget.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,2CAA+E;AAC/E,+CAAiE;AACjE,uCAA6C;AAE7C,4EAAwE;AACxE,wEAAoE;AAEpE,iDAA6C;AAI7C,iFAA4G;AAG/F,QAAA,qBAAqB,GAAG,eAAe,CAAC;AAExC,QAAA,qBAAqB,GAAG,uBAAuB,CAAC;AA6BtD,IAAM,mBAAmB,GAAzB,MAAM,mBAAoB,SAAQ,0BAIxC;IAK0C;IACyB;IAEpC;IANX,sBAAsB,GAAG,gBAAgB,CAAC;IAE7D,YACyC,UAAsB,EACG,WAAyC,EAE7E,oBAA6D;QAEzF,KAAK,EAAE,CAAC;QAL+B,eAAU,GAAV,UAAU,CAAY;QACG,gBAAW,GAAX,WAAW,CAA8B;QAE7E,yBAAoB,GAApB,oBAAoB,CAAyC;IAG3F,CAAC;IAKD,KAAK,CAAC,YAAY,CAAC,KAAwB,EAAE,KAAa;QACxD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;QAChF,MAAM,GAAG,GAAG,8BAAa,CAAC,GAAG,CAI1B,WAAW,CAAC,CAAC;QAChB,MAAM,QAAQ,GAAG,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,IAAI,CAAC;QAC9D,OAAO;YACL,UAAU,EAAE,QAAQ,EAAE,UAAU,IAAI,QAAQ,EAAE,UAAU;YACxD,SAAS,EAAE,QAAQ,EAAE,SAAS,IAAI,QAAQ,EAAE,SAAS;YACrD,MAAM,EAAE,QAAQ,EAAE,MAAM,IAAI,GAAG,EAAE,MAAM,IAAI,KAAK;SACjD,CAAC;IACJ,CAAC;IAOD,KAAK,CAAC,QAAQ,CAAC,MAA8C,EAAE,MAAc;QAC3E,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAwB,EAAE,SAAS,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC;QACxE,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC/B,CAAC;QACD,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YACpB,KAAK,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QACnC,CAAC;QACD,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;QAC9B,IAAI,CAAC,EAAE,EAAE,CAAC;YACR,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC;QAC5E,CAAC;QAID,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,EAAE;aACnB,UAAU,CAAC,IAAI,CAAC,sBAAsB,CAAC;aACvC,SAAS,CAA8C;YACtD,EAAE,MAAM,EAAE,KAAK,EAAE;YACjB,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,CAAC,cAAc,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE;SAC5G,CAAC;aACD,OAAO,EAAE,CAAC;QACb,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,WAAW,IAAI,CAAC,EAAE,UAAU,EAAE,GAAG,EAAE,UAAU,IAAI,CAAC,EAAE,CAAC;IACnH,CAAC;IAQD,KAAK,CAAC,kBAAkB,CAAC,MAAe,EAAE,QAAiB,EAAE,QAAiB;QAC5E,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI;YAC3B,CAAC,MAAM,EAAE,MAAM,CAAC;YAChB,CAAC,QAAQ,EAAE,QAAQ,CAAC;SACZ,EAAE,CAAC;YACX,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,SAAS;YACX,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YACpD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;gBACpE,SAAS;YACX,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;YAC5G,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC;gBAEhC,KAAK,QAAQ,CAAC;gBACd,MAAM,IAAI,sBAAa,CAAC,sBAAS,CAAC,kBAAkB,EAAE,mBAAU,CAAC,iBAAiB,CAAC,CAAC;YACtF,CAAC;QACH,CAAC;IACH,CAAC;IAKD,KAAK,CAAC,YAAY,CAAC,MAAe,EAAE,QAAiB;QACnD,MAAM,IAAI,GAAG,IAAI,0CAAe,EAAE,CAAC;QACnC,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,IAAI,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACpD,CAAC;QACD,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC1D,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAKD,KAAK,CAAC,YAAY,CAChB,MAA0B,EAC1B,QAA4B,EAC5B,YAAoB,EACpB,gBAAyB,EACzB,aAAuD;QAEvD,MAAM,OAAO,GAAG,IAAI,8CAAmB,EAAE,CAAC;QAC1C,OAAO,CAAC,YAAY,GAAG,YAAY,CAAC;QACpC,IAAI,CAAC,MAAM,EAAE,CAAC;YAEZ,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,CAAC;gBAClC,OAAO,CAAC,SAAS,GAAG,gBAAgB,CAAC;gBACrC,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC;YACxB,CAAC;YACD,OAAO,OAAO,CAAC;QACjB,CAAC;QAGD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC1D,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE,CAAC;YAC1E,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;YAChE,OAAO,CAAC,UAAU,GAAG,KAAK,CAAC,UAAU,CAAC;YACtC,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC;gBACxD,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAG,SAAS,CAAC,SAAoB,GAAG,KAAK,CAAC,UAAU,CAAC;gBACjE,CAAC,CAAC,SAAS,CAAC;YACd,OAAO,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,IAAI,SAAS,CAAC;YAC7C,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC;gBACrC,OAAO,CAAC,SAAS,GAAG,SAAS,CAAC,SAAS,CAAC;gBACxC,OAAO,CAAC,KAAK,GAAG,MAAM,CAAC;YACzB,CAAC;YACD,OAAO,OAAO,CAAC;QACjB,CAAC;QAGD,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAChE,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,EAAE,CAAC;gBACvC,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;gBACpE,OAAO,CAAC,UAAU,GAAG,KAAK,CAAC,UAAU,CAAC;gBACtC,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAG,WAAW,CAAC,SAAoB,GAAG,KAAK,CAAC,UAAU,CAAC,CAAC;gBAC5F,OAAO,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,IAAI,SAAS,CAAC;gBAC7C,OAAO,CAAC,SAAS,GAAG,WAAW,CAAC,SAAS,CAAC;gBAC1C,OAAO,CAAC,KAAK,GAAG,QAAQ,CAAC;gBACzB,OAAO,OAAO,CAAC;YACjB,CAAC;QACH,CAAC;QAKD,MAAM,MAAM,GAAG,aAAa,EAAE,MAAM,IAAI,SAAS,CAAC,MAAM,CAAC;QACzD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,EAAE,MAAM,CAAC,CAAC;QACtD,OAAO,CAAC,UAAU,GAAG,KAAK,CAAC,UAAU,CAAC;QACtC,OAAO,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,IAAI,SAAS,CAAC;QAK7C,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,SAAS,CAAC,EAAE,CAAC;YAC1C,OAAO,CAAC,SAAS,GAAG,aAAc,CAAC,SAAS,CAAC;YAC7C,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAG,aAAc,CAAC,SAAoB,GAAG,KAAK,CAAC,UAAU,CAAC,CAAC;YAC/F,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC;YACtB,OAAO,OAAO,CAAC;QACjB,CAAC;QAID,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,CAAC;YAClC,OAAO,CAAC,SAAS,GAAG,gBAAgB,CAAC;YACrC,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC;QACxB,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAMS,KAAK,CAAC,UAAU,CAAC,KAAwB,EAAE,KAAa;QAChE,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACpD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;QAC5G,MAAM,MAAM,GAAG,IAAI,2CAAgB,EAAE,CAAC;QACtC,MAAM,CAAC,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;QACjF,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QAC9E,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;QAC7B,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;QACrB,MAAM,CAAC,gBAAgB,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC;YACrD,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAG,KAAK,CAAC,UAAqB,GAAG,KAAK,CAAC,WAAW,CAAC;YAC/D,CAAC,CAAC,SAAS,CAAC;QACd,MAAM,CAAC,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC;YACnD,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAG,KAAK,CAAC,SAAoB,GAAG,KAAK,CAAC,UAAU,CAAC;YAC7D,CAAC,CAAC,SAAS,CAAC;QACd,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,IAAI,SAAS,CAAC;QAC5C,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;QACrB,MAAM,CAAC,WAAW,GAAG,KAAK,CAAC,WAAW,CAAC;QACvC,MAAM,CAAC,UAAU,GAAG,KAAK,CAAC,UAAU,CAAC;QACrC,OAAO,MAAM,CAAC;IAChB,CAAC;IAGS,MAAM,CAAC,KAAc;QAC7B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,GAAG,CAAC,CAAC;IAChD,CAAC;IAES,QAAQ,CAAC,KAAoB,EAAE,KAA4B;QACnE,OAAO,CACL,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,UAAU,IAAK,KAAK,CAAC,SAAoB,CAAC;YACjF,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,KAAK,CAAC,WAAW,IAAK,KAAK,CAAC,UAAqB,CAAC,CACrF,CAAC;IACJ,CAAC;IAGS,WAAW,CAAC,MAAc;QAClC,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;YACvB,OAAO,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QACpE,CAAC;QACD,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;YACtB,OAAO,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC;QACrB,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC;QAC5B,KAAK,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC3B,OAAO,KAAK,CAAC;IACf,CAAC;IAGS,SAAS,CAAC,MAAc;QAChC,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;YACvB,OAAO,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QACxE,CAAC;QACD,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;YACtB,OAAO,IAAI,CAAC;QACd,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC;QAC3B,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC1B,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QACjC,OAAO,IAAI,CAAC;IACd,CAAC;CACF,CAAA;AA3PY,kDAAmB;8BAAnB,mBAAmB;IAD/B,IAAA,mBAAU,GAAE;IAUR,WAAA,IAAA,2BAAgB,GAAE,CAAA;IAClB,WAAA,IAAA,sBAAW,EAAC,6BAAqB,CAAC,CAAA;IAClC,WAAA,IAAA,eAAM,EAAC,6BAAqB,CAAC,CAAA;qCAFqB,qBAAU;QACgB,gBAAK;GAVzE,mBAAmB,CA2P/B"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import { Model } from 'mongoose';
|
|
2
|
+
import { CrudService } from '../../../common/services/crud.service';
|
|
3
|
+
import { CoreModelConstructor } from '../../../common/types/core-model-constructor.type';
|
|
4
|
+
import { CoreAiConnectionPreferenceInput } from '../inputs/core-ai-connection-preference.input';
|
|
5
|
+
import { AiConnectionPreferenceDocument, CoreAiConnectionPreference } from '../models/core-ai-connection-preference.model';
|
|
6
|
+
export declare const AI_CONNECTION_PREFERENCE_MODEL = "AiConnectionPreference";
|
|
7
|
+
export declare const AI_CONNECTION_PREFERENCE_CLASS = "AI_CONNECTION_PREFERENCE_CLASS";
|
|
8
|
+
export declare class CoreAiConnectionPreferenceService extends CrudService<CoreAiConnectionPreference, CoreAiConnectionPreferenceInput, CoreAiConnectionPreferenceInput> {
|
|
9
|
+
protected readonly mainDbModel: Model<AiConnectionPreferenceDocument>;
|
|
10
|
+
protected readonly mainModelConstructor: CoreModelConstructor<CoreAiConnectionPreference>;
|
|
11
|
+
constructor(mainDbModel: Model<AiConnectionPreferenceDocument>, mainModelConstructor: CoreModelConstructor<CoreAiConnectionPreference>);
|
|
12
|
+
getPreference(scope: 'tenant' | 'user', refId: string): Promise<{
|
|
13
|
+
connectionId: string;
|
|
14
|
+
enforced?: boolean;
|
|
15
|
+
} | null>;
|
|
16
|
+
deleteByConnectionId(connectionId: string): Promise<number>;
|
|
17
|
+
upsertPreference(scope: 'tenant' | 'user', refId: string, connectionId: string, enforced?: boolean): Promise<CoreAiConnectionPreference>;
|
|
18
|
+
}
|
|
@@ -0,0 +1,53 @@
|
|
|
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
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
15
|
+
exports.CoreAiConnectionPreferenceService = exports.AI_CONNECTION_PREFERENCE_CLASS = exports.AI_CONNECTION_PREFERENCE_MODEL = void 0;
|
|
16
|
+
const common_1 = require("@nestjs/common");
|
|
17
|
+
const mongoose_1 = require("@nestjs/mongoose");
|
|
18
|
+
const mongoose_2 = require("mongoose");
|
|
19
|
+
const crud_service_1 = require("../../../common/services/crud.service");
|
|
20
|
+
exports.AI_CONNECTION_PREFERENCE_MODEL = 'AiConnectionPreference';
|
|
21
|
+
exports.AI_CONNECTION_PREFERENCE_CLASS = 'AI_CONNECTION_PREFERENCE_CLASS';
|
|
22
|
+
let CoreAiConnectionPreferenceService = class CoreAiConnectionPreferenceService extends crud_service_1.CrudService {
|
|
23
|
+
mainDbModel;
|
|
24
|
+
mainModelConstructor;
|
|
25
|
+
constructor(mainDbModel, mainModelConstructor) {
|
|
26
|
+
super();
|
|
27
|
+
this.mainDbModel = mainDbModel;
|
|
28
|
+
this.mainModelConstructor = mainModelConstructor;
|
|
29
|
+
}
|
|
30
|
+
async getPreference(scope, refId) {
|
|
31
|
+
const doc = await this.mainDbModel.findOne({ refId, scope }).lean().exec();
|
|
32
|
+
return doc ? { connectionId: doc.connectionId, enforced: doc.enforced } : null;
|
|
33
|
+
}
|
|
34
|
+
async deleteByConnectionId(connectionId) {
|
|
35
|
+
const result = await this.mainDbModel.deleteMany({ connectionId }).exec();
|
|
36
|
+
return result.deletedCount ?? 0;
|
|
37
|
+
}
|
|
38
|
+
async upsertPreference(scope, refId, connectionId, enforced = false) {
|
|
39
|
+
const doc = await this.mainDbModel
|
|
40
|
+
.findOneAndUpdate({ refId, scope }, { $set: { connectionId, enforced } }, { new: true, upsert: true })
|
|
41
|
+
.lean()
|
|
42
|
+
.exec();
|
|
43
|
+
return this.mainModelConstructor.map(doc);
|
|
44
|
+
}
|
|
45
|
+
};
|
|
46
|
+
exports.CoreAiConnectionPreferenceService = CoreAiConnectionPreferenceService;
|
|
47
|
+
exports.CoreAiConnectionPreferenceService = CoreAiConnectionPreferenceService = __decorate([
|
|
48
|
+
(0, common_1.Injectable)(),
|
|
49
|
+
__param(0, (0, mongoose_1.InjectModel)(exports.AI_CONNECTION_PREFERENCE_MODEL)),
|
|
50
|
+
__param(1, (0, common_1.Inject)(exports.AI_CONNECTION_PREFERENCE_CLASS)),
|
|
51
|
+
__metadata("design:paramtypes", [mongoose_2.Model, Object])
|
|
52
|
+
], CoreAiConnectionPreferenceService);
|
|
53
|
+
//# sourceMappingURL=core-ai-connection-preference.service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"core-ai-connection-preference.service.js","sourceRoot":"","sources":["../../../../../src/core/modules/ai/services/core-ai-connection-preference.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,2CAAoD;AACpD,+CAA+C;AAC/C,uCAAiC;AAEjC,wEAAoE;AASvD,QAAA,8BAA8B,GAAG,wBAAwB,CAAC;AAE1D,QAAA,8BAA8B,GAAG,gCAAgC,CAAC;AAOxE,IAAM,iCAAiC,GAAvC,MAAM,iCAAkC,SAAQ,0BAItD;IAG+B;IAEA;IAJ9B,YAE8B,WAAkD,EAElD,oBAAsE;QAElG,KAAK,EAAE,CAAC;QAJoB,gBAAW,GAAX,WAAW,CAAuC;QAElD,yBAAoB,GAApB,oBAAoB,CAAkD;IAGpG,CAAC;IAKD,KAAK,CAAC,aAAa,CACjB,KAAwB,EACxB,KAAa;QAEb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;QAC3E,OAAO,GAAG,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,GAAG,CAAC,YAAY,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;IACjF,CAAC;IAMD,KAAK,CAAC,oBAAoB,CAAC,YAAoB;QAC7C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAC1E,OAAO,MAAM,CAAC,YAAY,IAAI,CAAC,CAAC;IAClC,CAAC;IAOD,KAAK,CAAC,gBAAgB,CACpB,KAAwB,EACxB,KAAa,EACb,YAAoB,EACpB,QAAQ,GAAG,KAAK;QAEhB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,WAAW;aAC/B,gBAAgB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,YAAY,EAAE,QAAQ,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;aACrG,IAAI,EAAE;aACN,IAAI,EAAE,CAAC;QACV,OAAO,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC5C,CAAC;CACF,CAAA;AAnDY,8EAAiC;4CAAjC,iCAAiC;IAD7C,IAAA,mBAAU,GAAE;IAOR,WAAA,IAAA,sBAAW,EAAC,sCAA8B,CAAC,CAAA;IAE3C,WAAA,IAAA,eAAM,EAAC,sCAA8B,CAAC,CAAA;qCADE,gBAAK;GAPrC,iCAAiC,CAmD7C"}
|
|
@@ -0,0 +1,55 @@
|
|
|
1
|
+
import { Logger } from '@nestjs/common';
|
|
2
|
+
import { ResolvedAiConnection } from '../interfaces/resolved-ai-connection.interface';
|
|
3
|
+
import { CoreAiAvailableConnection } from '../models/core-ai-available-connection.model';
|
|
4
|
+
import { CoreAiConnectionPreference } from '../models/core-ai-connection-preference.model';
|
|
5
|
+
import { CoreAiConnectionPreferenceService } from './core-ai-connection-preference.service';
|
|
6
|
+
import { CoreAiConnectionService } from './core-ai-connection.service';
|
|
7
|
+
export interface AiResolvableConnection {
|
|
8
|
+
enforced?: boolean;
|
|
9
|
+
enforcedTenantIds?: string[];
|
|
10
|
+
id: string;
|
|
11
|
+
isDefault?: boolean;
|
|
12
|
+
model?: string;
|
|
13
|
+
name?: string;
|
|
14
|
+
tenantIds?: string[];
|
|
15
|
+
}
|
|
16
|
+
export interface AiConnectionResolveContext {
|
|
17
|
+
codeOverride?: string;
|
|
18
|
+
requested?: string;
|
|
19
|
+
tenantId?: string;
|
|
20
|
+
userId?: string;
|
|
21
|
+
}
|
|
22
|
+
export declare class CoreAiConnectionResolverService {
|
|
23
|
+
protected readonly connectionService: CoreAiConnectionService;
|
|
24
|
+
protected readonly preferenceService?: CoreAiConnectionPreferenceService;
|
|
25
|
+
protected readonly logger: Logger;
|
|
26
|
+
private readonly tenantPrefCache;
|
|
27
|
+
constructor(connectionService: CoreAiConnectionService, preferenceService?: CoreAiConnectionPreferenceService);
|
|
28
|
+
resolveConnection(ctx: AiConnectionResolveContext): Promise<ResolvedAiConnection | undefined>;
|
|
29
|
+
resolveConnectionId(ctx: AiConnectionResolveContext): Promise<string | undefined>;
|
|
30
|
+
listAvailable(ctx: AiConnectionResolveContext): Promise<CoreAiAvailableConnection[]>;
|
|
31
|
+
setUserConnection(userId: string, connectionId: string, tenantId?: string): Promise<void>;
|
|
32
|
+
setPreference(scope: 'tenant' | 'user', refId: string, connectionId: string, enforced?: boolean): Promise<CoreAiConnectionPreference>;
|
|
33
|
+
protected assertConnectionUsable(connectionId: string): Promise<void>;
|
|
34
|
+
protected resolveFrom(connections: AiResolvableConnection[], ctx: AiConnectionResolveContext): Promise<{
|
|
35
|
+
locked: boolean;
|
|
36
|
+
selected?: string;
|
|
37
|
+
}>;
|
|
38
|
+
protected resolutionLayers(): {
|
|
39
|
+
resolve: (all: AiResolvableConnection[], available: AiResolvableConnection[], ctx: AiConnectionResolveContext) => Promise<string | undefined> | string | undefined;
|
|
40
|
+
soft: boolean;
|
|
41
|
+
}[];
|
|
42
|
+
protected availableConnections(connections: AiResolvableConnection[], tenantId?: string): AiResolvableConnection[];
|
|
43
|
+
protected globalDefault(available: AiResolvableConnection[]): string | undefined;
|
|
44
|
+
protected tenantDefault(ctx: AiConnectionResolveContext): Promise<string | undefined>;
|
|
45
|
+
protected userDefault(ctx: AiConnectionResolveContext): Promise<string | undefined>;
|
|
46
|
+
protected clientSelection(ctx: AiConnectionResolveContext): string | undefined;
|
|
47
|
+
protected tenantEnforced(ctx: AiConnectionResolveContext): Promise<string | undefined>;
|
|
48
|
+
protected loadTenantPreference(ctx: AiConnectionResolveContext): Promise<{
|
|
49
|
+
connectionId: string;
|
|
50
|
+
enforced?: boolean;
|
|
51
|
+
} | null>;
|
|
52
|
+
protected adminEnforcedGlobal(all: AiResolvableConnection[]): string | undefined;
|
|
53
|
+
protected adminEnforcedForTenant(all: AiResolvableConnection[], ctx: AiConnectionResolveContext): string | undefined;
|
|
54
|
+
protected codeOverride(ctx: AiConnectionResolveContext): string | undefined;
|
|
55
|
+
}
|