@taico/taico 0.4.1 → 0.5.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/README.md +1 -1
- package/dist/ai-providers/ai-model-selection.entity.d.ts +11 -0
- package/dist/ai-providers/ai-model-selection.entity.js +58 -0
- package/dist/ai-providers/ai-model-selection.entity.js.map +1 -0
- package/dist/ai-providers/ai-model-selections.controller.d.ts +12 -0
- package/dist/ai-providers/ai-model-selections.controller.js +102 -0
- package/dist/ai-providers/ai-model-selections.controller.js.map +1 -0
- package/dist/ai-providers/ai-model-selections.service.d.ts +19 -0
- package/dist/ai-providers/ai-model-selections.service.js +113 -0
- package/dist/ai-providers/ai-model-selections.service.js.map +1 -0
- package/dist/ai-providers/ai-provider-models.service.d.ts +14 -0
- package/dist/ai-providers/ai-provider-models.service.js +196 -0
- package/dist/ai-providers/ai-provider-models.service.js.map +1 -0
- package/dist/{chat-providers/chat-provider.entity.d.ts → ai-providers/ai-provider.entity.d.ts} +6 -4
- package/dist/{chat-providers/chat-provider.entity.js → ai-providers/ai-provider.entity.js} +31 -21
- package/dist/ai-providers/ai-provider.entity.js.map +1 -0
- package/dist/ai-providers/ai-providers.controller.d.ts +20 -0
- package/dist/ai-providers/ai-providers.controller.js +177 -0
- package/dist/ai-providers/ai-providers.controller.js.map +1 -0
- package/dist/ai-providers/ai-providers.module.d.ts +2 -0
- package/dist/ai-providers/ai-providers.module.js +36 -0
- package/dist/ai-providers/ai-providers.module.js.map +1 -0
- package/dist/ai-providers/ai-providers.scopes.d.ts +20 -0
- package/dist/ai-providers/ai-providers.scopes.js +23 -0
- package/dist/ai-providers/ai-providers.scopes.js.map +1 -0
- package/dist/ai-providers/ai-providers.service.d.ts +20 -0
- package/dist/ai-providers/ai-providers.service.js +200 -0
- package/dist/ai-providers/ai-providers.service.js.map +1 -0
- package/dist/ai-providers/dto/http/ai-model-selection-params.dto.d.ts +4 -0
- package/dist/{chat-providers/dto/http/set-active-chat-provider.dto.js → ai-providers/dto/http/ai-model-selection-params.dto.js} +11 -10
- package/dist/ai-providers/dto/http/ai-model-selection-params.dto.js.map +1 -0
- package/dist/ai-providers/dto/http/ai-model-selection-response.dto.d.ts +10 -0
- package/dist/ai-providers/dto/http/ai-model-selection-response.dto.js +83 -0
- package/dist/ai-providers/dto/http/ai-model-selection-response.dto.js.map +1 -0
- package/dist/ai-providers/dto/http/ai-provider-model-response.dto.d.ts +6 -0
- package/dist/ai-providers/dto/http/ai-provider-model-response.dto.js +45 -0
- package/dist/ai-providers/dto/http/ai-provider-model-response.dto.js.map +1 -0
- package/dist/ai-providers/dto/http/ai-provider-params.dto.d.ts +3 -0
- package/dist/{chat-providers/dto/http/chat-provider-params.dto.js → ai-providers/dto/http/ai-provider-params.dto.js} +6 -6
- package/dist/ai-providers/dto/http/ai-provider-params.dto.js.map +1 -0
- package/dist/ai-providers/dto/http/ai-provider-response.dto.d.ts +14 -0
- package/dist/{chat-providers/dto/http/chat-provider-response.dto.js → ai-providers/dto/http/ai-provider-response.dto.js} +45 -23
- package/dist/ai-providers/dto/http/ai-provider-response.dto.js.map +1 -0
- package/dist/ai-providers/dto/http/create-ai-provider.dto.d.ts +9 -0
- package/dist/ai-providers/dto/http/create-ai-provider.dto.js +90 -0
- package/dist/ai-providers/dto/http/create-ai-provider.dto.js.map +1 -0
- package/dist/ai-providers/dto/http/set-ai-model-selection.dto.d.ts +4 -0
- package/dist/{chat-providers/dto/http/create-chat-provider.dto.js → ai-providers/dto/http/set-ai-model-selection.dto.js} +15 -28
- package/dist/ai-providers/dto/http/set-ai-model-selection.dto.js.map +1 -0
- package/dist/ai-providers/dto/http/update-ai-provider.dto.d.ts +8 -0
- package/dist/ai-providers/dto/http/update-ai-provider.dto.js +88 -0
- package/dist/ai-providers/dto/http/update-ai-provider.dto.js.map +1 -0
- package/dist/ai-providers/dto/service/ai-providers.service.types.d.ts +62 -0
- package/dist/{threads/backends/chat-backend.interface.js → ai-providers/dto/service/ai-providers.service.types.js} +1 -1
- package/dist/ai-providers/dto/service/ai-providers.service.types.js.map +1 -0
- package/dist/ai-providers/enums/ai-model-job.enum.d.ts +7 -0
- package/dist/ai-providers/enums/ai-model-job.enum.js +15 -0
- package/dist/ai-providers/enums/ai-model-job.enum.js.map +1 -0
- package/dist/ai-providers/enums/ai-model-kind.enum.d.ts +4 -0
- package/dist/ai-providers/enums/ai-model-kind.enum.js +9 -0
- package/dist/ai-providers/enums/ai-model-kind.enum.js.map +1 -0
- package/dist/ai-providers/enums/ai-provider-type.enum.d.ts +8 -0
- package/dist/ai-providers/enums/ai-provider-type.enum.js +13 -0
- package/dist/ai-providers/enums/ai-provider-type.enum.js.map +1 -0
- package/dist/ai-providers/enums/index.d.ts +3 -0
- package/dist/{chat-providers → ai-providers}/enums/index.js +3 -1
- package/dist/ai-providers/enums/index.js.map +1 -0
- package/dist/ai-providers/errors/ai-providers.errors.d.ts +30 -0
- package/dist/ai-providers/errors/ai-providers.errors.js +57 -0
- package/dist/ai-providers/errors/ai-providers.errors.js.map +1 -0
- package/dist/ai-providers/google-environment.d.ts +3 -0
- package/dist/ai-providers/google-environment.js +18 -0
- package/dist/ai-providers/google-environment.js.map +1 -0
- package/dist/app-init/app-init.module.js +2 -2
- package/dist/app-init/app-init.module.js.map +1 -1
- package/dist/app-init/app-init.runner.d.ts +4 -4
- package/dist/app-init/app-init.runner.js +21 -21
- package/dist/app-init/app-init.runner.js.map +1 -1
- package/dist/app.module.js +15 -3
- package/dist/app.module.js.map +1 -1
- package/dist/auth/core/scopes/all-api.scopes.js +2 -2
- package/dist/auth/core/scopes/all-api.scopes.js.map +1 -1
- package/dist/common/postgres-io.adapter.d.ts +12 -0
- package/dist/common/postgres-io.adapter.js +62 -0
- package/dist/common/postgres-io.adapter.js.map +1 -0
- package/dist/config/env.config.d.ts +1 -0
- package/dist/config/env.config.js +9 -0
- package/dist/config/env.config.js.map +1 -1
- package/dist/errors/http/error-catalog.js +20 -14
- package/dist/errors/http/error-catalog.js.map +1 -1
- package/dist/main.js +4 -0
- package/dist/main.js.map +1 -1
- package/dist/migrations/1760200000000-ThreadAgentMessages.d.ts +6 -0
- package/dist/migrations/1760200000000-ThreadAgentMessages.js +28 -0
- package/dist/migrations/1760200000000-ThreadAgentMessages.js.map +1 -0
- package/dist/migrations/1760300000000-ChatProviderModelConfig.d.ts +6 -0
- package/dist/migrations/1760300000000-ChatProviderModelConfig.js +14 -0
- package/dist/migrations/1760300000000-ChatProviderModelConfig.js.map +1 -0
- package/dist/migrations/1760400000000-ReplaceAdkChatProvider.d.ts +6 -0
- package/dist/migrations/1760400000000-ReplaceAdkChatProvider.js +16 -0
- package/dist/migrations/1760400000000-ReplaceAdkChatProvider.js.map +1 -0
- package/dist/migrations/1760500000000-AiProviderModelSelections.d.ts +6 -0
- package/dist/migrations/1760500000000-AiProviderModelSelections.js +56 -0
- package/dist/migrations/1760500000000-AiProviderModelSelections.js.map +1 -0
- package/dist/migrations/1760600000000-DropAdkSessionTables.d.ts +6 -0
- package/dist/migrations/1760600000000-DropAdkSessionTables.js +51 -0
- package/dist/migrations/1760600000000-DropAdkSessionTables.js.map +1 -0
- package/dist/public/assets/index-Bu8fSAx2.js +117 -0
- package/dist/public/assets/index-DOm5BmlC.css +1 -0
- package/dist/public/beta/assets/index-B9fCnAUe.js +94 -0
- package/dist/public/beta/assets/index-DIzh0pTC.css +1 -0
- package/dist/public/beta/index.html +2 -2
- package/dist/public/index.html +2 -2
- package/dist/public/package.json +4 -4
- package/dist/scripts/migrate-sqlite-to-postgres.js +0 -22
- package/dist/scripts/migrate-sqlite-to-postgres.js.map +1 -1
- package/dist/threads/agent/agent-context.utils.d.ts +14 -0
- package/dist/threads/agent/agent-context.utils.js +42 -0
- package/dist/threads/agent/agent-context.utils.js.map +1 -0
- package/dist/threads/agent/agent-context.utils.spec.d.ts +1 -0
- package/dist/threads/agent/agent-context.utils.spec.js +60 -0
- package/dist/threads/agent/agent-context.utils.spec.js.map +1 -0
- package/dist/threads/agent/chat-model.factory.d.ts +25 -0
- package/dist/threads/agent/chat-model.factory.js +106 -0
- package/dist/threads/agent/chat-model.factory.js.map +1 -0
- package/dist/threads/agent/chat-model.factory.spec.d.ts +1 -0
- package/dist/threads/agent/chat-model.factory.spec.js +83 -0
- package/dist/threads/agent/chat-model.factory.spec.js.map +1 -0
- package/dist/threads/agent/hooks/compaction.hook.d.ts +11 -0
- package/dist/threads/agent/hooks/compaction.hook.js +109 -0
- package/dist/threads/agent/hooks/compaction.hook.js.map +1 -0
- package/dist/threads/agent/hooks/easter-egg.hook.d.ts +5 -0
- package/dist/threads/agent/hooks/easter-egg.hook.js +30 -0
- package/dist/threads/agent/hooks/easter-egg.hook.js.map +1 -0
- package/dist/threads/agent/hooks/hooks.spec.d.ts +1 -0
- package/dist/threads/agent/hooks/hooks.spec.js +87 -0
- package/dist/threads/agent/hooks/hooks.spec.js.map +1 -0
- package/dist/threads/agent/hooks/thread-agent-hook.interface.d.ts +20 -0
- package/dist/threads/agent/hooks/thread-agent-hook.interface.js +5 -0
- package/dist/threads/agent/hooks/thread-agent-hook.interface.js.map +1 -0
- package/dist/threads/agent/hooks/thread-agent-hooks.runner.d.ts +11 -0
- package/dist/threads/agent/hooks/thread-agent-hooks.runner.js +98 -0
- package/dist/threads/agent/hooks/thread-agent-hooks.runner.js.map +1 -0
- package/dist/threads/agent/taico-mcp-tools.factory.d.ts +22 -0
- package/dist/threads/agent/taico-mcp-tools.factory.js +79 -0
- package/dist/threads/agent/taico-mcp-tools.factory.js.map +1 -0
- package/dist/threads/agent/thread-agent-log.service.d.ts +41 -0
- package/dist/threads/agent/thread-agent-log.service.js +157 -0
- package/dist/threads/agent/thread-agent-log.service.js.map +1 -0
- package/dist/threads/agent/thread-agent-message.entity.d.ts +19 -0
- package/dist/threads/agent/thread-agent-message.entity.js +109 -0
- package/dist/threads/agent/thread-agent-message.entity.js.map +1 -0
- package/dist/threads/agent/thread-agent.enums.d.ts +17 -0
- package/dist/threads/agent/thread-agent.enums.js +24 -0
- package/dist/threads/agent/thread-agent.enums.js.map +1 -0
- package/dist/threads/agent/thread-agent.prompts.d.ts +5 -0
- package/dist/threads/{backends/chat-backend.utils.js → agent/thread-agent.prompts.js} +5 -4
- package/dist/threads/agent/thread-agent.prompts.js.map +1 -0
- package/dist/threads/agent/thread-agent.service.d.ts +33 -0
- package/dist/threads/agent/thread-agent.service.js +286 -0
- package/dist/threads/agent/thread-agent.service.js.map +1 -0
- package/dist/threads/agent/thread-agent.service.spec.d.ts +1 -0
- package/dist/threads/agent/thread-agent.service.spec.js +184 -0
- package/dist/threads/agent/thread-agent.service.spec.js.map +1 -0
- package/dist/threads/agent/thread-turn.entity.d.ts +12 -0
- package/dist/threads/agent/thread-turn.entity.js +71 -0
- package/dist/threads/agent/thread-turn.entity.js.map +1 -0
- package/dist/threads/agent/thread-turns.service.d.ts +13 -0
- package/dist/threads/agent/thread-turns.service.js +52 -0
- package/dist/threads/agent/thread-turns.service.js.map +1 -0
- package/dist/threads/chat.service.d.ts +5 -30
- package/dist/threads/chat.service.js +44 -119
- package/dist/threads/chat.service.js.map +1 -1
- package/dist/threads/dto/service/threads.service.types.d.ts +0 -2
- package/dist/threads/dto/thread-list-item-response.dto.d.ts +0 -1
- package/dist/threads/dto/thread-list-item-response.dto.js +0 -10
- package/dist/threads/dto/thread-list-item-response.dto.js.map +1 -1
- package/dist/threads/dto/thread-response.dto.d.ts +0 -1
- package/dist/threads/dto/thread-response.dto.js +0 -11
- package/dist/threads/dto/thread-response.dto.js.map +1 -1
- package/dist/threads/thread-message.entity.d.ts +3 -0
- package/dist/threads/thread-message.entity.js +12 -0
- package/dist/threads/thread-message.entity.js.map +1 -1
- package/dist/threads/thread.entity.d.ts +3 -1
- package/dist/threads/thread.entity.js +23 -3
- package/dist/threads/thread.entity.js.map +1 -1
- package/dist/threads/threads-state.service.spec.js +3 -1
- package/dist/threads/threads-state.service.spec.js.map +1 -1
- package/dist/threads/threads.module.js +31 -8
- package/dist/threads/threads.module.js.map +1 -1
- package/dist/threads/threads.service.d.ts +3 -8
- package/dist/threads/threads.service.js +6 -143
- package/dist/threads/threads.service.js.map +1 -1
- package/dist/threads/threads.service.spec.js +3 -2
- package/dist/threads/threads.service.spec.js.map +1 -1
- package/dist/tsconfig.tsbuildinfo +1 -1
- package/dist/walkthrough/dto/walkthrough-status-response.dto.d.ts +1 -0
- package/dist/walkthrough/dto/walkthrough-status-response.dto.js +8 -0
- package/dist/walkthrough/dto/walkthrough-status-response.dto.js.map +1 -1
- package/dist/walkthrough/walkthrough.controller.js +1 -0
- package/dist/walkthrough/walkthrough.controller.js.map +1 -1
- package/dist/walkthrough/walkthrough.module.js +2 -0
- package/dist/walkthrough/walkthrough.module.js.map +1 -1
- package/dist/walkthrough/walkthrough.service.d.ts +4 -1
- package/dist/walkthrough/walkthrough.service.js +14 -7
- package/dist/walkthrough/walkthrough.service.js.map +1 -1
- package/package.json +17 -9
- package/dist/chat-providers/chat-provider.entity.js.map +0 -1
- package/dist/chat-providers/chat-providers.controller.d.ts +0 -19
- package/dist/chat-providers/chat-providers.controller.js +0 -179
- package/dist/chat-providers/chat-providers.controller.js.map +0 -1
- package/dist/chat-providers/chat-providers.module.d.ts +0 -2
- package/dist/chat-providers/chat-providers.module.js +0 -32
- package/dist/chat-providers/chat-providers.module.js.map +0 -1
- package/dist/chat-providers/chat-providers.scopes.d.ts +0 -20
- package/dist/chat-providers/chat-providers.scopes.js +0 -23
- package/dist/chat-providers/chat-providers.scopes.js.map +0 -1
- package/dist/chat-providers/chat-providers.service.d.ts +0 -20
- package/dist/chat-providers/chat-providers.service.js +0 -193
- package/dist/chat-providers/chat-providers.service.js.map +0 -1
- package/dist/chat-providers/dto/http/chat-provider-params.dto.d.ts +0 -3
- package/dist/chat-providers/dto/http/chat-provider-params.dto.js.map +0 -1
- package/dist/chat-providers/dto/http/chat-provider-response.dto.d.ts +0 -12
- package/dist/chat-providers/dto/http/chat-provider-response.dto.js.map +0 -1
- package/dist/chat-providers/dto/http/create-chat-provider.dto.d.ts +0 -6
- package/dist/chat-providers/dto/http/create-chat-provider.dto.js.map +0 -1
- package/dist/chat-providers/dto/http/set-active-chat-provider.dto.d.ts +0 -3
- package/dist/chat-providers/dto/http/set-active-chat-provider.dto.js.map +0 -1
- package/dist/chat-providers/dto/http/update-chat-provider.dto.d.ts +0 -5
- package/dist/chat-providers/dto/http/update-chat-provider.dto.js +0 -49
- package/dist/chat-providers/dto/http/update-chat-provider.dto.js.map +0 -1
- package/dist/chat-providers/dto/service/chat-providers.service.types.d.ts +0 -30
- package/dist/chat-providers/dto/service/chat-providers.service.types.js +0 -3
- package/dist/chat-providers/dto/service/chat-providers.service.types.js.map +0 -1
- package/dist/chat-providers/enums/chat-provider-type.enum.d.ts +0 -4
- package/dist/chat-providers/enums/chat-provider-type.enum.js +0 -9
- package/dist/chat-providers/enums/chat-provider-type.enum.js.map +0 -1
- package/dist/chat-providers/enums/index.d.ts +0 -1
- package/dist/chat-providers/enums/index.js.map +0 -1
- package/dist/chat-providers/errors/chat-providers.errors.d.ts +0 -25
- package/dist/chat-providers/errors/chat-providers.errors.js +0 -48
- package/dist/chat-providers/errors/chat-providers.errors.js.map +0 -1
- package/dist/llm/llm.module.d.ts +0 -2
- package/dist/llm/llm.module.js +0 -23
- package/dist/llm/llm.module.js.map +0 -1
- package/dist/llm/openai-responses.service.d.ts +0 -12
- package/dist/llm/openai-responses.service.js +0 -69
- package/dist/llm/openai-responses.service.js.map +0 -1
- package/dist/public/assets/index-CPjOd3Sq.css +0 -1
- package/dist/public/assets/index-DVNE_5Yh.js +0 -73
- package/dist/public/beta/assets/index-BRla0F5r.js +0 -94
- package/dist/public/beta/assets/index-BaBvl20K.css +0 -1
- package/dist/threads/backends/adk.backend.d.ts +0 -18
- package/dist/threads/backends/adk.backend.js +0 -264
- package/dist/threads/backends/adk.backend.js.map +0 -1
- package/dist/threads/backends/chat-backend.interface.d.ts +0 -39
- package/dist/threads/backends/chat-backend.interface.js.map +0 -1
- package/dist/threads/backends/chat-backend.utils.d.ts +0 -5
- package/dist/threads/backends/chat-backend.utils.js.map +0 -1
- package/dist/threads/backends/openai.backend.d.ts +0 -19
- package/dist/threads/backends/openai.backend.js +0 -155
- package/dist/threads/backends/openai.backend.js.map +0 -1
- package/dist/threads/openai-mcp-server-factory.service.d.ts +0 -8
- package/dist/threads/openai-mcp-server-factory.service.js +0 -114
- package/dist/threads/openai-mcp-server-factory.service.js.map +0 -1
|
@@ -0,0 +1,83 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
const enums_1 = require("../../ai-providers/enums");
|
|
4
|
+
const chat_model_factory_1 = require("./chat-model.factory");
|
|
5
|
+
jest.mock('src/ai-providers/ai-providers.service', () => ({ AiProvidersService: class {
|
|
6
|
+
} }));
|
|
7
|
+
jest.mock('src/ai-providers/ai-model-selections.service', () => ({
|
|
8
|
+
AiModelSelectionsService: class {
|
|
9
|
+
},
|
|
10
|
+
}));
|
|
11
|
+
jest.mock('src/ai-providers/errors/ai-providers.errors', () => ({
|
|
12
|
+
AiProviderNotConfiguredError: class extends Error {
|
|
13
|
+
},
|
|
14
|
+
}));
|
|
15
|
+
const GOOGLE_ENV_KEYS = [
|
|
16
|
+
'GOOGLE_CLOUD_PROJECT',
|
|
17
|
+
'GOOGLE_CLOUD_LOCATION',
|
|
18
|
+
'GOOGLE_API_KEY',
|
|
19
|
+
'GEMINI_API_KEY',
|
|
20
|
+
'GOOGLE_GENERATIVE_AI_API_KEY',
|
|
21
|
+
];
|
|
22
|
+
describe('ChatModelFactory', () => {
|
|
23
|
+
const originalEnv = { ...process.env };
|
|
24
|
+
const providerConfig = (type) => ({
|
|
25
|
+
id: 'provider-1',
|
|
26
|
+
type,
|
|
27
|
+
apiKey: null,
|
|
28
|
+
baseUrl: null,
|
|
29
|
+
project: null,
|
|
30
|
+
location: null,
|
|
31
|
+
});
|
|
32
|
+
const factoryFor = (type, selectedModelId) => {
|
|
33
|
+
const resolveSelection = jest.fn(async () => ({
|
|
34
|
+
provider: providerConfig(type),
|
|
35
|
+
modelId: selectedModelId,
|
|
36
|
+
}));
|
|
37
|
+
const getAiProviderConfigById = jest.fn(async () => providerConfig(type));
|
|
38
|
+
const factory = new chat_model_factory_1.ChatModelFactory({ getAiProviderConfigById }, { resolve: resolveSelection });
|
|
39
|
+
return { factory, resolveSelection, getAiProviderConfigById };
|
|
40
|
+
};
|
|
41
|
+
beforeEach(() => {
|
|
42
|
+
for (const key of GOOGLE_ENV_KEYS) {
|
|
43
|
+
delete process.env[key];
|
|
44
|
+
}
|
|
45
|
+
});
|
|
46
|
+
afterAll(() => {
|
|
47
|
+
process.env = originalEnv;
|
|
48
|
+
});
|
|
49
|
+
it('uses Vertex AI from the environment when nothing is stored on the provider', async () => {
|
|
50
|
+
process.env.GOOGLE_CLOUD_PROJECT = 'my-project';
|
|
51
|
+
process.env.GOOGLE_CLOUD_LOCATION = 'global';
|
|
52
|
+
const { factory } = factoryFor(enums_1.AiProviderType.GOOGLE_VERTEX, 'gemini-3.6-flash');
|
|
53
|
+
const resolved = await factory.resolve({ job: enums_1.AiModelJob.CONVERSATION });
|
|
54
|
+
expect(resolved).toMatchObject({
|
|
55
|
+
providerType: enums_1.AiProviderType.GOOGLE_VERTEX,
|
|
56
|
+
modelId: 'gemini-3.6-flash',
|
|
57
|
+
});
|
|
58
|
+
});
|
|
59
|
+
it('builds an xAI model for a Grok provider', async () => {
|
|
60
|
+
const { factory } = factoryFor(enums_1.AiProviderType.XAI, 'grok-4');
|
|
61
|
+
const resolved = await factory.resolve({ job: enums_1.AiModelJob.CONVERSATION });
|
|
62
|
+
expect(resolved).toMatchObject({ providerType: enums_1.AiProviderType.XAI, modelId: 'grok-4' });
|
|
63
|
+
expect(resolved.model.provider).toContain('xai');
|
|
64
|
+
});
|
|
65
|
+
it("asks for the job's selection when the thread has no model of its own", async () => {
|
|
66
|
+
const { factory, resolveSelection } = factoryFor(enums_1.AiProviderType.ANTHROPIC, 'claude-haiku-5-5');
|
|
67
|
+
const resolved = await factory.resolve({ job: enums_1.AiModelJob.QUICK_TASKS });
|
|
68
|
+
expect(resolveSelection).toHaveBeenCalledWith(enums_1.AiModelJob.QUICK_TASKS);
|
|
69
|
+
expect(resolved.modelId).toBe('claude-haiku-5-5');
|
|
70
|
+
});
|
|
71
|
+
it("uses the thread's own provider and model when both are set", async () => {
|
|
72
|
+
const { factory, resolveSelection, getAiProviderConfigById } = factoryFor(enums_1.AiProviderType.ANTHROPIC, 'claude-haiku-5-5');
|
|
73
|
+
const resolved = await factory.resolve({
|
|
74
|
+
job: enums_1.AiModelJob.CONVERSATION,
|
|
75
|
+
aiProviderId: 'provider-1',
|
|
76
|
+
modelId: 'claude-opus-5-5',
|
|
77
|
+
});
|
|
78
|
+
expect(getAiProviderConfigById).toHaveBeenCalledWith('provider-1');
|
|
79
|
+
expect(resolveSelection).not.toHaveBeenCalled();
|
|
80
|
+
expect(resolved.modelId).toBe('claude-opus-5-5');
|
|
81
|
+
});
|
|
82
|
+
});
|
|
83
|
+
//# sourceMappingURL=chat-model.factory.spec.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"chat-model.factory.spec.js","sourceRoot":"","sources":["../../../src/threads/agent/chat-model.factory.spec.ts"],"names":[],"mappings":";;AAAA,oDAAoE;AACpE,6DAAwD;AAExD,IAAI,CAAC,IAAI,CAAC,uCAAuC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,kBAAkB,EAAE;KAAQ,EAAE,CAAC,CAAC,CAAC;AAC7F,IAAI,CAAC,IAAI,CAAC,8CAA8C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC/D,wBAAwB,EAAE;KAAQ;CACnC,CAAC,CAAC,CAAC;AACJ,IAAI,CAAC,IAAI,CAAC,6CAA6C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC9D,4BAA4B,EAAE,KAAM,SAAQ,KAAK;KAAG;CACrD,CAAC,CAAC,CAAC;AAEJ,MAAM,eAAe,GAAG;IACtB,sBAAsB;IACtB,uBAAuB;IACvB,gBAAgB;IAChB,gBAAgB;IAChB,8BAA8B;CAC/B,CAAC;AAEF,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,MAAM,WAAW,GAAG,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAEvC,MAAM,cAAc,GAAG,CAAC,IAAoB,EAAE,EAAE,CAAC,CAAC;QAChD,EAAE,EAAE,YAAY;QAChB,IAAI;QACJ,MAAM,EAAE,IAAI;QACZ,OAAO,EAAE,IAAI;QACb,OAAO,EAAE,IAAI;QACb,QAAQ,EAAE,IAAI;KACf,CAAC,CAAC;IAEH,MAAM,UAAU,GAAG,CAAC,IAAoB,EAAE,eAAuB,EAAE,EAAE;QACnE,MAAM,gBAAgB,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YAC5C,QAAQ,EAAE,cAAc,CAAC,IAAI,CAAC;YAC9B,OAAO,EAAE,eAAe;SACzB,CAAC,CAAC,CAAC;QACJ,MAAM,uBAAuB,GAAG,IAAI,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC;QAC1E,MAAM,OAAO,GAAG,IAAI,qCAAgB,CAClC,EAAE,uBAAuB,EAAW,EACpC,EAAE,OAAO,EAAE,gBAAgB,EAAW,CACvC,CAAC;QACF,OAAO,EAAE,OAAO,EAAE,gBAAgB,EAAE,uBAAuB,EAAE,CAAC;IAChE,CAAC,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,KAAK,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;YAClC,OAAO,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,GAAG,EAAE;QACZ,OAAO,CAAC,GAAG,GAAG,WAAW,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4EAA4E,EAAE,KAAK,IAAI,EAAE;QAC1F,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,YAAY,CAAC;QAChD,OAAO,CAAC,GAAG,CAAC,qBAAqB,GAAG,QAAQ,CAAC;QAE7C,MAAM,EAAE,OAAO,EAAE,GAAG,UAAU,CAAC,sBAAc,CAAC,aAAa,EAAE,kBAAkB,CAAC,CAAC;QACjF,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,kBAAU,CAAC,YAAY,EAAE,CAAC,CAAC;QAEzE,MAAM,CAAC,QAAQ,CAAC,CAAC,aAAa,CAAC;YAC7B,YAAY,EAAE,sBAAc,CAAC,aAAa;YAC1C,OAAO,EAAE,kBAAkB;SAC5B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,EAAE,OAAO,EAAE,GAAG,UAAU,CAAC,sBAAc,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;QAE7D,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,kBAAU,CAAC,YAAY,EAAE,CAAC,CAAC;QAEzE,MAAM,CAAC,QAAQ,CAAC,CAAC,aAAa,CAAC,EAAE,YAAY,EAAE,sBAAc,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;QACxF,MAAM,CAAE,QAAQ,CAAC,KAA8B,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACpF,MAAM,EAAE,OAAO,EAAE,gBAAgB,EAAE,GAAG,UAAU,CAAC,sBAAc,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAC;QAE/F,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,kBAAU,CAAC,WAAW,EAAE,CAAC,CAAC;QAExE,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,kBAAU,CAAC,WAAW,CAAC,CAAC;QACtE,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,EAAE,OAAO,EAAE,gBAAgB,EAAE,uBAAuB,EAAE,GAAG,UAAU,CACvE,sBAAc,CAAC,SAAS,EACxB,kBAAkB,CACnB,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC;YACrC,GAAG,EAAE,kBAAU,CAAC,YAAY;YAC5B,YAAY,EAAE,YAAY;YAC1B,OAAO,EAAE,iBAAiB;SAC3B,CAAC,CAAC;QAEH,MAAM,CAAC,uBAAuB,CAAC,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC;QACnE,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAChD,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import { ModelMessage } from 'ai';
|
|
2
|
+
import { ThreadAgentLogService } from '../thread-agent-log.service';
|
|
3
|
+
import { AfterTurnContext, ThreadAgentHook } from './thread-agent-hook.interface';
|
|
4
|
+
export declare function renderTranscript(messages: ModelMessage[]): string;
|
|
5
|
+
export declare class CompactionHook implements ThreadAgentHook {
|
|
6
|
+
private readonly agentLog;
|
|
7
|
+
readonly name = "compaction";
|
|
8
|
+
private readonly logger;
|
|
9
|
+
constructor(agentLog: ThreadAgentLogService);
|
|
10
|
+
afterTurn(context: AfterTurnContext): Promise<void>;
|
|
11
|
+
}
|
|
@@ -0,0 +1,109 @@
|
|
|
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 CompactionHook_1;
|
|
12
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
13
|
+
exports.CompactionHook = void 0;
|
|
14
|
+
exports.renderTranscript = renderTranscript;
|
|
15
|
+
const common_1 = require("@nestjs/common");
|
|
16
|
+
const ai_1 = require("ai");
|
|
17
|
+
const env_config_1 = require("../../../config/env.config");
|
|
18
|
+
const agent_context_utils_1 = require("../agent-context.utils");
|
|
19
|
+
const thread_agent_log_service_1 = require("../thread-agent-log.service");
|
|
20
|
+
const thread_agent_enums_1 = require("../thread-agent.enums");
|
|
21
|
+
const MAX_PART_CHARS = 2_000;
|
|
22
|
+
const SUMMARY_INSTRUCTIONS = `You compress a conversation between users and an AI agent so the agent can continue it with less context.
|
|
23
|
+
|
|
24
|
+
Write a summary that keeps:
|
|
25
|
+
- what the users asked for and what is still open
|
|
26
|
+
- decisions, constraints and preferences that were stated
|
|
27
|
+
- ids and names of tasks, threads, context blocks and tags that were mentioned or changed
|
|
28
|
+
- what the agent already did, including the outcome of tool calls
|
|
29
|
+
|
|
30
|
+
Leave out pleasantries and anything already superseded. Write plain text, no preamble.`;
|
|
31
|
+
function clip(text) {
|
|
32
|
+
return text.length > MAX_PART_CHARS ? `${text.slice(0, MAX_PART_CHARS)}…` : text;
|
|
33
|
+
}
|
|
34
|
+
function renderTranscript(messages) {
|
|
35
|
+
const lines = [];
|
|
36
|
+
for (const message of messages) {
|
|
37
|
+
if (typeof message.content === 'string') {
|
|
38
|
+
lines.push(`${message.role}: ${clip(message.content)}`);
|
|
39
|
+
continue;
|
|
40
|
+
}
|
|
41
|
+
for (const part of message.content) {
|
|
42
|
+
switch (part.type) {
|
|
43
|
+
case 'text':
|
|
44
|
+
lines.push(`${message.role}: ${clip(part.text)}`);
|
|
45
|
+
break;
|
|
46
|
+
case 'tool-call':
|
|
47
|
+
lines.push(`assistant called ${part.toolName}: ${clip(JSON.stringify(part.input))}`);
|
|
48
|
+
break;
|
|
49
|
+
case 'tool-result':
|
|
50
|
+
lines.push(`${part.toolName} returned: ${clip(JSON.stringify(part.output))}`);
|
|
51
|
+
break;
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
return lines.join('\n');
|
|
56
|
+
}
|
|
57
|
+
let CompactionHook = CompactionHook_1 = class CompactionHook {
|
|
58
|
+
agentLog;
|
|
59
|
+
name = 'compaction';
|
|
60
|
+
logger = new common_1.Logger(CompactionHook_1.name);
|
|
61
|
+
constructor(agentLog) {
|
|
62
|
+
this.agentLog = agentLog;
|
|
63
|
+
}
|
|
64
|
+
async afterTurn(context) {
|
|
65
|
+
const threshold = (0, env_config_1.getConfig)().threadAgentCompactionTokenThreshold;
|
|
66
|
+
if (context.lastInputTokens === null || context.lastInputTokens < threshold) {
|
|
67
|
+
return;
|
|
68
|
+
}
|
|
69
|
+
const rows = await this.agentLog.loadContext(context.threadId);
|
|
70
|
+
const transcript = renderTranscript((0, agent_context_utils_1.buildModelMessages)(rows, context.model));
|
|
71
|
+
if (!transcript) {
|
|
72
|
+
return;
|
|
73
|
+
}
|
|
74
|
+
const result = await (0, ai_1.generateText)({
|
|
75
|
+
model: context.model.model,
|
|
76
|
+
instructions: SUMMARY_INSTRUCTIONS,
|
|
77
|
+
prompt: transcript,
|
|
78
|
+
providerOptions: context.model.providerOptions,
|
|
79
|
+
});
|
|
80
|
+
const summary = result.text.trim();
|
|
81
|
+
if (!summary) {
|
|
82
|
+
return;
|
|
83
|
+
}
|
|
84
|
+
await this.agentLog.append(context.threadId, [
|
|
85
|
+
{
|
|
86
|
+
role: thread_agent_enums_1.ThreadAgentMessageRole.USER,
|
|
87
|
+
kind: thread_agent_enums_1.ThreadAgentMessageKind.SUMMARY,
|
|
88
|
+
source: this.name,
|
|
89
|
+
turnId: context.turnId,
|
|
90
|
+
message: {
|
|
91
|
+
role: 'user',
|
|
92
|
+
content: `<conversation-summary>\n${summary}\n</conversation-summary>`,
|
|
93
|
+
},
|
|
94
|
+
},
|
|
95
|
+
]);
|
|
96
|
+
this.logger.log({
|
|
97
|
+
message: 'Compacted thread conversation',
|
|
98
|
+
threadId: context.threadId,
|
|
99
|
+
inputTokens: context.lastInputTokens,
|
|
100
|
+
summarizedMessages: rows.length,
|
|
101
|
+
});
|
|
102
|
+
}
|
|
103
|
+
};
|
|
104
|
+
exports.CompactionHook = CompactionHook;
|
|
105
|
+
exports.CompactionHook = CompactionHook = CompactionHook_1 = __decorate([
|
|
106
|
+
(0, common_1.Injectable)(),
|
|
107
|
+
__metadata("design:paramtypes", [thread_agent_log_service_1.ThreadAgentLogService])
|
|
108
|
+
], CompactionHook);
|
|
109
|
+
//# sourceMappingURL=compaction.hook.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"compaction.hook.js","sourceRoot":"","sources":["../../../../src/threads/agent/hooks/compaction.hook.ts"],"names":[],"mappings":";;;;;;;;;;;;;AA4BA,4CAyBC;AArDD,2CAAoD;AACpD,2BAAgD;AAChD,2DAAkD;AAClD,gEAA4D;AAC5D,0EAAoE;AACpE,8DAG+B;AAG/B,MAAM,cAAc,GAAG,KAAK,CAAC;AAE7B,MAAM,oBAAoB,GAAG;;;;;;;;uFAQ0D,CAAC;AAExF,SAAS,IAAI,CAAC,IAAY;IACxB,OAAO,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;AACnF,CAAC;AAGD,SAAgB,gBAAgB,CAAC,QAAwB;IACvD,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YACxC,KAAK,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,IAAI,KAAK,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACxD,SAAS;QACX,CAAC;QAED,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACnC,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;gBAClB,KAAK,MAAM;oBACT,KAAK,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBAClD,MAAM;gBACR,KAAK,WAAW;oBACd,KAAK,CAAC,IAAI,CAAC,oBAAoB,IAAI,CAAC,QAAQ,KAAK,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;oBACrF,MAAM;gBACR,KAAK,aAAa;oBAChB,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,QAAQ,cAAc,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC;oBAC9E,MAAM;YACV,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAQM,IAAM,cAAc,sBAApB,MAAM,cAAc;IAII;IAHpB,IAAI,GAAG,YAAY,CAAC;IACZ,MAAM,GAAG,IAAI,eAAM,CAAC,gBAAc,CAAC,IAAI,CAAC,CAAC;IAE1D,YAA6B,QAA+B;QAA/B,aAAQ,GAAR,QAAQ,CAAuB;IAAG,CAAC;IAEhE,KAAK,CAAC,SAAS,CAAC,OAAyB;QACvC,MAAM,SAAS,GAAG,IAAA,sBAAS,GAAE,CAAC,mCAAmC,CAAC;QAClE,IAAI,OAAO,CAAC,eAAe,KAAK,IAAI,IAAI,OAAO,CAAC,eAAe,GAAG,SAAS,EAAE,CAAC;YAC5E,OAAO;QACT,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC/D,MAAM,UAAU,GAAG,gBAAgB,CAAC,IAAA,wCAAkB,EAAC,IAAI,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;QAC7E,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAA,iBAAY,EAAC;YAChC,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,KAAK;YAC1B,YAAY,EAAE,oBAAoB;YAClC,MAAM,EAAE,UAAU;YAClB,eAAe,EAAE,OAAO,CAAC,KAAK,CAAC,eAAe;SAC/C,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QACnC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QAED,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE;YAC3C;gBACE,IAAI,EAAE,2CAAsB,CAAC,IAAI;gBACjC,IAAI,EAAE,2CAAsB,CAAC,OAAO;gBACpC,MAAM,EAAE,IAAI,CAAC,IAAI;gBACjB,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,OAAO,EAAE;oBACP,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,2BAA2B,OAAO,2BAA2B;iBACvE;aACF;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC;YACd,OAAO,EAAE,+BAA+B;YACxC,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,WAAW,EAAE,OAAO,CAAC,eAAe;YACpC,kBAAkB,EAAE,IAAI,CAAC,MAAM;SAChC,CAAC,CAAC;IACL,CAAC;CACF,CAAA;AAlDY,wCAAc;yBAAd,cAAc;IAD1B,IAAA,mBAAU,GAAE;qCAK4B,gDAAqB;GAJjD,cAAc,CAkD1B"}
|
|
@@ -0,0 +1,30 @@
|
|
|
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.EasterEggHook = void 0;
|
|
10
|
+
const common_1 = require("@nestjs/common");
|
|
11
|
+
const RECOGNITION_PHRASE = 'the key is under the doormat';
|
|
12
|
+
let EasterEggHook = class EasterEggHook {
|
|
13
|
+
name = 'easter-egg';
|
|
14
|
+
async beforeTurn(context) {
|
|
15
|
+
const heardPhrase = context.newUserMessages.some((text) => text.toLowerCase().includes(RECOGNITION_PHRASE));
|
|
16
|
+
if (!heardPhrase) {
|
|
17
|
+
return null;
|
|
18
|
+
}
|
|
19
|
+
return [
|
|
20
|
+
'The user just gave the recognition phrase. They are a fellow field agent.',
|
|
21
|
+
'Open your reply with the countersign, word for word: "The owl only hoots at midnight."',
|
|
22
|
+
'Then carry on as normal. Never reveal where you learned the countersign.',
|
|
23
|
+
].join('\n');
|
|
24
|
+
}
|
|
25
|
+
};
|
|
26
|
+
exports.EasterEggHook = EasterEggHook;
|
|
27
|
+
exports.EasterEggHook = EasterEggHook = __decorate([
|
|
28
|
+
(0, common_1.Injectable)()
|
|
29
|
+
], EasterEggHook);
|
|
30
|
+
//# sourceMappingURL=easter-egg.hook.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"easter-egg.hook.js","sourceRoot":"","sources":["../../../../src/threads/agent/hooks/easter-egg.hook.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAA4C;AAG5C,MAAM,kBAAkB,GAAG,8BAA8B,CAAC;AAQnD,IAAM,aAAa,GAAnB,MAAM,aAAa;IACf,IAAI,GAAG,YAAY,CAAC;IAE7B,KAAK,CAAC,UAAU,CAAC,OAA0B;QACzC,MAAM,WAAW,GAAG,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CACxD,IAAI,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAChD,CAAC;QACF,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO;YACL,2EAA2E;YAC3E,wFAAwF;YACxF,0EAA0E;SAC3E,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACf,CAAC;CACF,CAAA;AAjBY,sCAAa;wBAAb,aAAa;IADzB,IAAA,mBAAU,GAAE;GACA,aAAa,CAiBzB"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export {};
|
|
@@ -0,0 +1,87 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
const thread_agent_enums_1 = require("../thread-agent.enums");
|
|
4
|
+
const compaction_hook_1 = require("./compaction.hook");
|
|
5
|
+
const easter_egg_hook_1 = require("./easter-egg.hook");
|
|
6
|
+
const thread_agent_hooks_runner_1 = require("./thread-agent-hooks.runner");
|
|
7
|
+
jest.mock('../thread-agent-log.service', () => ({ ThreadAgentLogService: class {
|
|
8
|
+
} }));
|
|
9
|
+
jest.mock('src/config/env.config', () => ({
|
|
10
|
+
getConfig: () => ({ threadAgentCompactionTokenThreshold: 100 }),
|
|
11
|
+
}));
|
|
12
|
+
const turn = { threadId: 'thread-1', turnId: 'turn-1' };
|
|
13
|
+
describe('ThreadAgentHooksRunner', () => {
|
|
14
|
+
const createRunner = (hooks) => {
|
|
15
|
+
const agentLog = { append: jest.fn().mockResolvedValue([]) };
|
|
16
|
+
return { agentLog, runner: new thread_agent_hooks_runner_1.ThreadAgentHooksRunner(hooks, agentLog) };
|
|
17
|
+
};
|
|
18
|
+
it('stores what a hook returns as a hidden injection', async () => {
|
|
19
|
+
const { runner, agentLog } = createRunner([
|
|
20
|
+
{ name: 'lookup', beforeTurn: async () => 'Existing tags:\n- bug' },
|
|
21
|
+
]);
|
|
22
|
+
const injected = await runner.runBeforeTurn({ ...turn, newUserMessages: [], context: [] });
|
|
23
|
+
expect(injected).toBe(true);
|
|
24
|
+
expect(agentLog.append).toHaveBeenCalledWith('thread-1', [
|
|
25
|
+
expect.objectContaining({
|
|
26
|
+
kind: thread_agent_enums_1.ThreadAgentMessageKind.INJECTION,
|
|
27
|
+
source: 'hook:lookup',
|
|
28
|
+
turnId: 'turn-1',
|
|
29
|
+
message: {
|
|
30
|
+
role: 'user',
|
|
31
|
+
content: '<context source="lookup">\nExisting tags:\n- bug\n</context>',
|
|
32
|
+
},
|
|
33
|
+
}),
|
|
34
|
+
]);
|
|
35
|
+
});
|
|
36
|
+
it('keeps going when a hook throws or has nothing to add', async () => {
|
|
37
|
+
const { runner, agentLog } = createRunner([
|
|
38
|
+
{ name: 'broken', beforeTurn: async () => { throw new Error('boom'); } },
|
|
39
|
+
{ name: 'quiet', beforeTurn: async () => null },
|
|
40
|
+
{ name: 'useful', beforeTurn: async () => 'context' },
|
|
41
|
+
]);
|
|
42
|
+
const injected = await runner.runBeforeTurn({ ...turn, newUserMessages: [], context: [] });
|
|
43
|
+
expect(injected).toBe(true);
|
|
44
|
+
expect(agentLog.append).toHaveBeenCalledTimes(1);
|
|
45
|
+
});
|
|
46
|
+
});
|
|
47
|
+
describe('EasterEggHook', () => {
|
|
48
|
+
const hook = new easter_egg_hook_1.EasterEggHook();
|
|
49
|
+
it('hands over the countersign when the user says the recognition phrase', async () => {
|
|
50
|
+
const text = await hook.beforeTurn({
|
|
51
|
+
...turn,
|
|
52
|
+
newUserMessages: ['Psst. The key is under the doormat.'],
|
|
53
|
+
context: [],
|
|
54
|
+
});
|
|
55
|
+
expect(text).toContain('The owl only hoots at midnight.');
|
|
56
|
+
});
|
|
57
|
+
it('stays quiet otherwise', async () => {
|
|
58
|
+
expect(await hook.beforeTurn({ ...turn, newUserMessages: ['where is the key?'], context: [] })).toBeNull();
|
|
59
|
+
});
|
|
60
|
+
});
|
|
61
|
+
describe('renderTranscript', () => {
|
|
62
|
+
it('renders text, tool calls and tool results as plain text', () => {
|
|
63
|
+
const transcript = (0, compaction_hook_1.renderTranscript)([
|
|
64
|
+
{ role: 'user', content: 'list my tasks' },
|
|
65
|
+
{
|
|
66
|
+
role: 'assistant',
|
|
67
|
+
content: [
|
|
68
|
+
{ type: 'text', text: 'Checking.' },
|
|
69
|
+
{ type: 'tool-call', toolCallId: 'c1', toolName: 'tasks__list', input: { q: 'mine' } },
|
|
70
|
+
],
|
|
71
|
+
},
|
|
72
|
+
{
|
|
73
|
+
role: 'tool',
|
|
74
|
+
content: [
|
|
75
|
+
{ type: 'tool-result', toolCallId: 'c1', toolName: 'tasks__list', output: { type: 'json', value: [] } },
|
|
76
|
+
],
|
|
77
|
+
},
|
|
78
|
+
]);
|
|
79
|
+
expect(transcript).toBe([
|
|
80
|
+
'user: list my tasks',
|
|
81
|
+
'assistant: Checking.',
|
|
82
|
+
'assistant called tasks__list: {"q":"mine"}',
|
|
83
|
+
'tasks__list returned: {"type":"json","value":[]}',
|
|
84
|
+
].join('\n'));
|
|
85
|
+
});
|
|
86
|
+
});
|
|
87
|
+
//# sourceMappingURL=hooks.spec.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hooks.spec.js","sourceRoot":"","sources":["../../../../src/threads/agent/hooks/hooks.spec.ts"],"names":[],"mappings":";;AAAA,8DAA+D;AAC/D,uDAAqD;AACrD,uDAAkD;AAElD,2EAAqE;AAErE,IAAI,CAAC,IAAI,CAAC,6BAA6B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,qBAAqB,EAAE;KAAQ,EAAE,CAAC,CAAC,CAAC;AACtF,IAAI,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,mCAAmC,EAAE,GAAG,EAAE,CAAC;CAChE,CAAC,CAAC,CAAC;AAEJ,MAAM,IAAI,GAAG,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;AAExD,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,MAAM,YAAY,GAAG,CAAC,KAAwB,EAAE,EAAE;QAChD,MAAM,QAAQ,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,EAAE,CAAC;QAC7D,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,kDAAsB,CAAC,KAAK,EAAE,QAAiB,CAAC,EAAE,CAAC;IACpF,CAAC,CAAC;IAEF,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,YAAY,CAAC;YACxC,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,KAAK,IAAI,EAAE,CAAC,uBAAuB,EAAE;SACpE,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,EAAE,GAAG,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;QAE3F,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,UAAU,EAAE;YACvD,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,2CAAsB,CAAC,SAAS;gBACtC,MAAM,EAAE,aAAa;gBACrB,MAAM,EAAE,QAAQ;gBAChB,OAAO,EAAE;oBACP,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,8DAA8D;iBACxE;aACF,CAAC;SACH,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,YAAY,CAAC;YACxC,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,KAAK,IAAI,EAAE,GAAG,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE;YACxE,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,EAAE;YAC/C,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,KAAK,IAAI,EAAE,CAAC,SAAS,EAAE;SACtD,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,EAAE,GAAG,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;QAE3F,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,MAAM,IAAI,GAAG,IAAI,+BAAa,EAAE,CAAC;IAEjC,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACpF,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC;YACjC,GAAG,IAAI;YACP,eAAe,EAAE,CAAC,qCAAqC,CAAC;YACxD,OAAO,EAAE,EAAE;SACZ,CAAC,CAAC;QAEH,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,iCAAiC,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;QACrC,MAAM,CACJ,MAAM,IAAI,CAAC,UAAU,CAAC,EAAE,GAAG,IAAI,EAAE,eAAe,EAAE,CAAC,mBAAmB,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CACxF,CAAC,QAAQ,EAAE,CAAC;IACf,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,UAAU,GAAG,IAAA,kCAAgB,EAAC;YAClC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,EAAE;YAC1C;gBACE,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE;oBACP,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE;oBACnC,EAAE,IAAI,EAAE,WAAW,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE,aAAa,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE;iBACvF;aACF;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE;oBACP,EAAE,IAAI,EAAE,aAAa,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE;iBACxG;aACF;SACF,CAAC,CAAC;QAEH,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CACrB;YACE,qBAAqB;YACrB,sBAAsB;YACtB,4CAA4C;YAC5C,kDAAkD;SACnD,CAAC,IAAI,CAAC,IAAI,CAAC,CACb,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import { AgentContextRow } from '../agent-context.utils';
|
|
2
|
+
import { ResolvedChatModel } from '../chat-model.factory';
|
|
3
|
+
export type BeforeTurnContext = {
|
|
4
|
+
threadId: string;
|
|
5
|
+
turnId: string;
|
|
6
|
+
newUserMessages: string[];
|
|
7
|
+
context: AgentContextRow[];
|
|
8
|
+
};
|
|
9
|
+
export type AfterTurnContext = {
|
|
10
|
+
threadId: string;
|
|
11
|
+
turnId: string;
|
|
12
|
+
model: ResolvedChatModel;
|
|
13
|
+
lastInputTokens: number | null;
|
|
14
|
+
};
|
|
15
|
+
export interface ThreadAgentHook {
|
|
16
|
+
readonly name: string;
|
|
17
|
+
beforeTurn?(context: BeforeTurnContext): Promise<string | null>;
|
|
18
|
+
afterTurn?(context: AfterTurnContext): Promise<void>;
|
|
19
|
+
}
|
|
20
|
+
export declare const THREAD_AGENT_HOOKS: unique symbol;
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"thread-agent-hook.interface.js","sourceRoot":"","sources":["../../../../src/threads/agent/hooks/thread-agent-hook.interface.ts"],"names":[],"mappings":";;;AAsCa,QAAA,kBAAkB,GAAG,MAAM,CAAC,oBAAoB,CAAC,CAAC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import { ThreadAgentLogService } from '../thread-agent-log.service';
|
|
2
|
+
import { AfterTurnContext, BeforeTurnContext, ThreadAgentHook } from './thread-agent-hook.interface';
|
|
3
|
+
export declare class ThreadAgentHooksRunner {
|
|
4
|
+
private readonly hooks;
|
|
5
|
+
private readonly agentLog;
|
|
6
|
+
private readonly logger;
|
|
7
|
+
constructor(hooks: ThreadAgentHook[], agentLog: ThreadAgentLogService);
|
|
8
|
+
runBeforeTurn(context: BeforeTurnContext): Promise<boolean>;
|
|
9
|
+
runAfterTurn(context: AfterTurnContext): Promise<void>;
|
|
10
|
+
private logFailure;
|
|
11
|
+
}
|
|
@@ -0,0 +1,98 @@
|
|
|
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 ThreadAgentHooksRunner_1;
|
|
15
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
16
|
+
exports.ThreadAgentHooksRunner = void 0;
|
|
17
|
+
const common_1 = require("@nestjs/common");
|
|
18
|
+
const thread_agent_log_service_1 = require("../thread-agent-log.service");
|
|
19
|
+
const thread_agent_enums_1 = require("../thread-agent.enums");
|
|
20
|
+
const thread_agent_hook_interface_1 = require("./thread-agent-hook.interface");
|
|
21
|
+
const BEFORE_TURN_TIMEOUT_MS = 5_000;
|
|
22
|
+
const AFTER_TURN_TIMEOUT_MS = 120_000;
|
|
23
|
+
function withTimeout(work, timeoutMs, hookName) {
|
|
24
|
+
let timer;
|
|
25
|
+
const timeout = new Promise((_, reject) => {
|
|
26
|
+
timer = setTimeout(() => reject(new Error(`Hook ${hookName} timed out after ${timeoutMs}ms`)), timeoutMs);
|
|
27
|
+
});
|
|
28
|
+
return Promise.race([work, timeout]).finally(() => clearTimeout(timer));
|
|
29
|
+
}
|
|
30
|
+
let ThreadAgentHooksRunner = ThreadAgentHooksRunner_1 = class ThreadAgentHooksRunner {
|
|
31
|
+
hooks;
|
|
32
|
+
agentLog;
|
|
33
|
+
logger = new common_1.Logger(ThreadAgentHooksRunner_1.name);
|
|
34
|
+
constructor(hooks, agentLog) {
|
|
35
|
+
this.hooks = hooks;
|
|
36
|
+
this.agentLog = agentLog;
|
|
37
|
+
}
|
|
38
|
+
async runBeforeTurn(context) {
|
|
39
|
+
let injected = false;
|
|
40
|
+
for (const hook of this.hooks) {
|
|
41
|
+
if (!hook.beforeTurn) {
|
|
42
|
+
continue;
|
|
43
|
+
}
|
|
44
|
+
try {
|
|
45
|
+
const text = await withTimeout(hook.beforeTurn(context), BEFORE_TURN_TIMEOUT_MS, hook.name);
|
|
46
|
+
if (!text?.trim()) {
|
|
47
|
+
continue;
|
|
48
|
+
}
|
|
49
|
+
await this.agentLog.append(context.threadId, [
|
|
50
|
+
{
|
|
51
|
+
role: thread_agent_enums_1.ThreadAgentMessageRole.USER,
|
|
52
|
+
kind: thread_agent_enums_1.ThreadAgentMessageKind.INJECTION,
|
|
53
|
+
source: `hook:${hook.name}`,
|
|
54
|
+
turnId: context.turnId,
|
|
55
|
+
message: {
|
|
56
|
+
role: 'user',
|
|
57
|
+
content: `<context source="${hook.name}">\n${text.trim()}\n</context>`,
|
|
58
|
+
},
|
|
59
|
+
},
|
|
60
|
+
]);
|
|
61
|
+
injected = true;
|
|
62
|
+
}
|
|
63
|
+
catch (error) {
|
|
64
|
+
this.logFailure(hook, 'beforeTurn', context.threadId, error);
|
|
65
|
+
}
|
|
66
|
+
}
|
|
67
|
+
return injected;
|
|
68
|
+
}
|
|
69
|
+
async runAfterTurn(context) {
|
|
70
|
+
for (const hook of this.hooks) {
|
|
71
|
+
if (!hook.afterTurn) {
|
|
72
|
+
continue;
|
|
73
|
+
}
|
|
74
|
+
try {
|
|
75
|
+
await withTimeout(hook.afterTurn(context), AFTER_TURN_TIMEOUT_MS, hook.name);
|
|
76
|
+
}
|
|
77
|
+
catch (error) {
|
|
78
|
+
this.logFailure(hook, 'afterTurn', context.threadId, error);
|
|
79
|
+
}
|
|
80
|
+
}
|
|
81
|
+
}
|
|
82
|
+
logFailure(hook, phase, threadId, error) {
|
|
83
|
+
this.logger.warn({
|
|
84
|
+
message: 'Thread agent hook failed',
|
|
85
|
+
hook: hook.name,
|
|
86
|
+
phase,
|
|
87
|
+
threadId,
|
|
88
|
+
error: error instanceof Error ? error.message : String(error),
|
|
89
|
+
});
|
|
90
|
+
}
|
|
91
|
+
};
|
|
92
|
+
exports.ThreadAgentHooksRunner = ThreadAgentHooksRunner;
|
|
93
|
+
exports.ThreadAgentHooksRunner = ThreadAgentHooksRunner = ThreadAgentHooksRunner_1 = __decorate([
|
|
94
|
+
(0, common_1.Injectable)(),
|
|
95
|
+
__param(0, (0, common_1.Inject)(thread_agent_hook_interface_1.THREAD_AGENT_HOOKS)),
|
|
96
|
+
__metadata("design:paramtypes", [Array, thread_agent_log_service_1.ThreadAgentLogService])
|
|
97
|
+
], ThreadAgentHooksRunner);
|
|
98
|
+
//# sourceMappingURL=thread-agent-hooks.runner.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"thread-agent-hooks.runner.js","sourceRoot":"","sources":["../../../../src/threads/agent/hooks/thread-agent-hooks.runner.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CAA4D;AAC5D,0EAAoE;AACpE,8DAG+B;AAC/B,+EAKuC;AAEvC,MAAM,sBAAsB,GAAG,KAAK,CAAC;AACrC,MAAM,qBAAqB,GAAG,OAAO,CAAC;AAEtC,SAAS,WAAW,CAAI,IAAgB,EAAE,SAAiB,EAAE,QAAgB;IAC3E,IAAI,KAAqB,CAAC;IAC1B,MAAM,OAAO,GAAG,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;QAC/C,KAAK,GAAG,UAAU,CAChB,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,QAAQ,oBAAoB,SAAS,IAAI,CAAC,CAAC,EAC1E,SAAS,CACV,CAAC;IACJ,CAAC,CAAC,CAAC;IACH,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC;AAC1E,CAAC;AAIM,IAAM,sBAAsB,8BAA5B,MAAM,sBAAsB;IAIc;IAC5B;IAJF,MAAM,GAAG,IAAI,eAAM,CAAC,wBAAsB,CAAC,IAAI,CAAC,CAAC;IAElE,YAC+C,KAAwB,EACpD,QAA+B;QADH,UAAK,GAAL,KAAK,CAAmB;QACpD,aAAQ,GAAR,QAAQ,CAAuB;IAC/C,CAAC;IAGJ,KAAK,CAAC,aAAa,CAAC,OAA0B;QAC5C,IAAI,QAAQ,GAAG,KAAK,CAAC;QAErB,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC9B,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;gBACrB,SAAS;YACX,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,MAAM,WAAW,CAC5B,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EACxB,sBAAsB,EACtB,IAAI,CAAC,IAAI,CACV,CAAC;gBACF,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC;oBAClB,SAAS;gBACX,CAAC;gBAED,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE;oBAC3C;wBACE,IAAI,EAAE,2CAAsB,CAAC,IAAI;wBACjC,IAAI,EAAE,2CAAsB,CAAC,SAAS;wBACtC,MAAM,EAAE,QAAQ,IAAI,CAAC,IAAI,EAAE;wBAC3B,MAAM,EAAE,OAAO,CAAC,MAAM;wBACtB,OAAO,EAAE;4BACP,IAAI,EAAE,MAAM;4BACZ,OAAO,EAAE,oBAAoB,IAAI,CAAC,IAAI,OAAO,IAAI,CAAC,IAAI,EAAE,cAAc;yBACvE;qBACF;iBACF,CAAC,CAAC;gBACH,QAAQ,GAAG,IAAI,CAAC;YAClB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,YAAY,EAAE,OAAO,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,OAAyB;QAC1C,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC9B,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACpB,SAAS;YACX,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,qBAAqB,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/E,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,WAAW,EAAE,OAAO,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC;IACH,CAAC;IAEO,UAAU,CAChB,IAAqB,EACrB,KAAiC,EACjC,QAAgB,EAChB,KAAc;QAEd,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;YACf,OAAO,EAAE,0BAA0B;YACnC,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK;YACL,QAAQ;YACR,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAC9D,CAAC,CAAC;IACL,CAAC;CACF,CAAA;AA5EY,wDAAsB;iCAAtB,sBAAsB;IADlC,IAAA,mBAAU,GAAE;IAKR,WAAA,IAAA,eAAM,EAAC,gDAAkB,CAAC,CAAA;4CACA,gDAAqB;GALvC,sBAAsB,CA4ElC"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import { ToolSet } from 'ai';
|
|
2
|
+
import { IssuedAccessTokenService } from "../../authorization-server/issued-access-token.service";
|
|
3
|
+
import { ActorType } from "../../identity-provider/enums";
|
|
4
|
+
export type McpActor = {
|
|
5
|
+
id: string;
|
|
6
|
+
slug: string;
|
|
7
|
+
type: ActorType;
|
|
8
|
+
displayName: string;
|
|
9
|
+
};
|
|
10
|
+
export type TaicoMcpTools = {
|
|
11
|
+
tools: ToolSet;
|
|
12
|
+
close: () => Promise<void>;
|
|
13
|
+
};
|
|
14
|
+
export declare class TaicoMcpToolsFactory {
|
|
15
|
+
private readonly issuedAccessTokenService;
|
|
16
|
+
private readonly logger;
|
|
17
|
+
constructor(issuedAccessTokenService: IssuedAccessTokenService);
|
|
18
|
+
open(input: {
|
|
19
|
+
subject: McpActor;
|
|
20
|
+
issuedBy: McpActor;
|
|
21
|
+
}): Promise<TaicoMcpTools>;
|
|
22
|
+
}
|
|
@@ -0,0 +1,79 @@
|
|
|
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 TaicoMcpToolsFactory_1;
|
|
12
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
13
|
+
exports.TaicoMcpToolsFactory = void 0;
|
|
14
|
+
const common_1 = require("@nestjs/common");
|
|
15
|
+
const mcp_1 = require("@ai-sdk/mcp");
|
|
16
|
+
const env_config_1 = require("../../config/env.config");
|
|
17
|
+
const issued_access_token_service_1 = require("../../authorization-server/issued-access-token.service");
|
|
18
|
+
const mcp_scopes_1 = require("../../auth/core/scopes/mcp.scopes");
|
|
19
|
+
const tasks_scopes_1 = require("../../tasks/tasks.scopes");
|
|
20
|
+
const context_scopes_1 = require("../../context/context.scopes");
|
|
21
|
+
const MCP_SERVERS = [
|
|
22
|
+
{ prefix: 'tasks', path: '/api/v1/tasks/tasks/mcp' },
|
|
23
|
+
{ prefix: 'context', path: '/api/v1/context/blocks/mcp' },
|
|
24
|
+
];
|
|
25
|
+
let TaicoMcpToolsFactory = TaicoMcpToolsFactory_1 = class TaicoMcpToolsFactory {
|
|
26
|
+
issuedAccessTokenService;
|
|
27
|
+
logger = new common_1.Logger(TaicoMcpToolsFactory_1.name);
|
|
28
|
+
constructor(issuedAccessTokenService) {
|
|
29
|
+
this.issuedAccessTokenService = issuedAccessTokenService;
|
|
30
|
+
}
|
|
31
|
+
async open(input) {
|
|
32
|
+
const token = await this.issuedAccessTokenService.issueSystemToken({
|
|
33
|
+
subjectActor: input.subject,
|
|
34
|
+
issuedByActor: input.issuedBy,
|
|
35
|
+
scopes: [
|
|
36
|
+
mcp_scopes_1.McpScopes.USE.id,
|
|
37
|
+
...tasks_scopes_1.ALL_TASKS_SCOPES.map((s) => s.id),
|
|
38
|
+
...context_scopes_1.ALL_CONTEXT_SCOPES.map((s) => s.id),
|
|
39
|
+
],
|
|
40
|
+
});
|
|
41
|
+
const baseUrl = (0, env_config_1.getConfig)().issuerUrl;
|
|
42
|
+
const clients = [];
|
|
43
|
+
const close = async () => {
|
|
44
|
+
await Promise.all(clients.map((client) => client.close().catch((error) => {
|
|
45
|
+
this.logger.warn({
|
|
46
|
+
message: 'Failed to close MCP client',
|
|
47
|
+
error: error instanceof Error ? error.message : String(error),
|
|
48
|
+
});
|
|
49
|
+
})));
|
|
50
|
+
};
|
|
51
|
+
try {
|
|
52
|
+
const tools = {};
|
|
53
|
+
for (const server of MCP_SERVERS) {
|
|
54
|
+
const client = await (0, mcp_1.createMCPClient)({
|
|
55
|
+
transport: {
|
|
56
|
+
type: 'http',
|
|
57
|
+
url: `${baseUrl}${server.path}`,
|
|
58
|
+
headers: { Authorization: `Bearer ${token.token}` },
|
|
59
|
+
},
|
|
60
|
+
});
|
|
61
|
+
clients.push(client);
|
|
62
|
+
for (const [name, tool] of Object.entries(await client.tools())) {
|
|
63
|
+
tools[`${server.prefix}__${name}`] = tool;
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
return { tools, close };
|
|
67
|
+
}
|
|
68
|
+
catch (error) {
|
|
69
|
+
await close();
|
|
70
|
+
throw error;
|
|
71
|
+
}
|
|
72
|
+
}
|
|
73
|
+
};
|
|
74
|
+
exports.TaicoMcpToolsFactory = TaicoMcpToolsFactory;
|
|
75
|
+
exports.TaicoMcpToolsFactory = TaicoMcpToolsFactory = TaicoMcpToolsFactory_1 = __decorate([
|
|
76
|
+
(0, common_1.Injectable)(),
|
|
77
|
+
__metadata("design:paramtypes", [issued_access_token_service_1.IssuedAccessTokenService])
|
|
78
|
+
], TaicoMcpToolsFactory);
|
|
79
|
+
//# sourceMappingURL=taico-mcp-tools.factory.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"taico-mcp-tools.factory.js","sourceRoot":"","sources":["../../../src/threads/agent/taico-mcp-tools.factory.ts"],"names":[],"mappings":";;;;;;;;;;;;;AAAA,2CAAoD;AAEpD,qCAA8C;AAC9C,wDAAkD;AAClD,wGAAgG;AAChG,kEAA4D;AAC5D,2DAA0D;AAC1D,iEAAgE;AAehE,MAAM,WAAW,GAAG;IAClB,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,yBAAyB,EAAE;IACpD,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,4BAA4B,EAAE;CACjD,CAAC;AAOJ,IAAM,oBAAoB,4BAA1B,MAAM,oBAAoB;IAIZ;IAHF,MAAM,GAAG,IAAI,eAAM,CAAC,sBAAoB,CAAC,IAAI,CAAC,CAAC;IAEhE,YACmB,wBAAkD;QAAlD,6BAAwB,GAAxB,wBAAwB,CAA0B;IAClE,CAAC;IAGJ,KAAK,CAAC,IAAI,CAAC,KAAgD;QACzD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,wBAAwB,CAAC,gBAAgB,CAAC;YACjE,YAAY,EAAE,KAAK,CAAC,OAAO;YAC3B,aAAa,EAAE,KAAK,CAAC,QAAQ;YAC7B,MAAM,EAAE;gBACN,sBAAS,CAAC,GAAG,CAAC,EAAE;gBAChB,GAAG,+BAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACpC,GAAG,mCAAkB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACvC;SACF,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,IAAA,sBAAS,GAAE,CAAC,SAAS,CAAC;QACtC,MAAM,OAAO,GAAuD,EAAE,CAAC;QACvE,MAAM,KAAK,GAAG,KAAK,IAAmB,EAAE;YACtC,MAAM,OAAO,CAAC,GAAG,CACf,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CACrB,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC7B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;oBACf,OAAO,EAAE,4BAA4B;oBACrC,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iBAC9D,CAAC,CAAC;YACL,CAAC,CAAC,CACH,CACF,CAAC;QACJ,CAAC,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,KAAK,GAAY,EAAE,CAAC;YAE1B,KAAK,MAAM,MAAM,IAAI,WAAW,EAAE,CAAC;gBACjC,MAAM,MAAM,GAAG,MAAM,IAAA,qBAAe,EAAC;oBACnC,SAAS,EAAE;wBACT,IAAI,EAAE,MAAM;wBACZ,GAAG,EAAE,GAAG,OAAO,GAAG,MAAM,CAAC,IAAI,EAAE;wBAC/B,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,KAAK,CAAC,KAAK,EAAE,EAAE;qBACpD;iBACF,CAAC,CAAC;gBACH,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAErB,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC;oBAChE,KAAK,CAAC,GAAG,MAAM,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC,GAAG,IAAI,CAAC;gBAC5C,CAAC;YACH,CAAC;YAED,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;QAC1B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,KAAK,EAAE,CAAC;YACd,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF,CAAA;AA1DY,oDAAoB;+BAApB,oBAAoB;IADhC,IAAA,mBAAU,GAAE;qCAKkC,sDAAwB;GAJ1D,oBAAoB,CA0DhC"}
|