@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.
Files changed (265) hide show
  1. package/README.md +1 -1
  2. package/dist/ai-providers/ai-model-selection.entity.d.ts +11 -0
  3. package/dist/ai-providers/ai-model-selection.entity.js +58 -0
  4. package/dist/ai-providers/ai-model-selection.entity.js.map +1 -0
  5. package/dist/ai-providers/ai-model-selections.controller.d.ts +12 -0
  6. package/dist/ai-providers/ai-model-selections.controller.js +102 -0
  7. package/dist/ai-providers/ai-model-selections.controller.js.map +1 -0
  8. package/dist/ai-providers/ai-model-selections.service.d.ts +19 -0
  9. package/dist/ai-providers/ai-model-selections.service.js +113 -0
  10. package/dist/ai-providers/ai-model-selections.service.js.map +1 -0
  11. package/dist/ai-providers/ai-provider-models.service.d.ts +14 -0
  12. package/dist/ai-providers/ai-provider-models.service.js +196 -0
  13. package/dist/ai-providers/ai-provider-models.service.js.map +1 -0
  14. package/dist/{chat-providers/chat-provider.entity.d.ts → ai-providers/ai-provider.entity.d.ts} +6 -4
  15. package/dist/{chat-providers/chat-provider.entity.js → ai-providers/ai-provider.entity.js} +31 -21
  16. package/dist/ai-providers/ai-provider.entity.js.map +1 -0
  17. package/dist/ai-providers/ai-providers.controller.d.ts +20 -0
  18. package/dist/ai-providers/ai-providers.controller.js +177 -0
  19. package/dist/ai-providers/ai-providers.controller.js.map +1 -0
  20. package/dist/ai-providers/ai-providers.module.d.ts +2 -0
  21. package/dist/ai-providers/ai-providers.module.js +36 -0
  22. package/dist/ai-providers/ai-providers.module.js.map +1 -0
  23. package/dist/ai-providers/ai-providers.scopes.d.ts +20 -0
  24. package/dist/ai-providers/ai-providers.scopes.js +23 -0
  25. package/dist/ai-providers/ai-providers.scopes.js.map +1 -0
  26. package/dist/ai-providers/ai-providers.service.d.ts +20 -0
  27. package/dist/ai-providers/ai-providers.service.js +200 -0
  28. package/dist/ai-providers/ai-providers.service.js.map +1 -0
  29. package/dist/ai-providers/dto/http/ai-model-selection-params.dto.d.ts +4 -0
  30. package/dist/{chat-providers/dto/http/set-active-chat-provider.dto.js → ai-providers/dto/http/ai-model-selection-params.dto.js} +11 -10
  31. package/dist/ai-providers/dto/http/ai-model-selection-params.dto.js.map +1 -0
  32. package/dist/ai-providers/dto/http/ai-model-selection-response.dto.d.ts +10 -0
  33. package/dist/ai-providers/dto/http/ai-model-selection-response.dto.js +83 -0
  34. package/dist/ai-providers/dto/http/ai-model-selection-response.dto.js.map +1 -0
  35. package/dist/ai-providers/dto/http/ai-provider-model-response.dto.d.ts +6 -0
  36. package/dist/ai-providers/dto/http/ai-provider-model-response.dto.js +45 -0
  37. package/dist/ai-providers/dto/http/ai-provider-model-response.dto.js.map +1 -0
  38. package/dist/ai-providers/dto/http/ai-provider-params.dto.d.ts +3 -0
  39. package/dist/{chat-providers/dto/http/chat-provider-params.dto.js → ai-providers/dto/http/ai-provider-params.dto.js} +6 -6
  40. package/dist/ai-providers/dto/http/ai-provider-params.dto.js.map +1 -0
  41. package/dist/ai-providers/dto/http/ai-provider-response.dto.d.ts +14 -0
  42. package/dist/{chat-providers/dto/http/chat-provider-response.dto.js → ai-providers/dto/http/ai-provider-response.dto.js} +45 -23
  43. package/dist/ai-providers/dto/http/ai-provider-response.dto.js.map +1 -0
  44. package/dist/ai-providers/dto/http/create-ai-provider.dto.d.ts +9 -0
  45. package/dist/ai-providers/dto/http/create-ai-provider.dto.js +90 -0
  46. package/dist/ai-providers/dto/http/create-ai-provider.dto.js.map +1 -0
  47. package/dist/ai-providers/dto/http/set-ai-model-selection.dto.d.ts +4 -0
  48. package/dist/{chat-providers/dto/http/create-chat-provider.dto.js → ai-providers/dto/http/set-ai-model-selection.dto.js} +15 -28
  49. package/dist/ai-providers/dto/http/set-ai-model-selection.dto.js.map +1 -0
  50. package/dist/ai-providers/dto/http/update-ai-provider.dto.d.ts +8 -0
  51. package/dist/ai-providers/dto/http/update-ai-provider.dto.js +88 -0
  52. package/dist/ai-providers/dto/http/update-ai-provider.dto.js.map +1 -0
  53. package/dist/ai-providers/dto/service/ai-providers.service.types.d.ts +62 -0
  54. package/dist/{threads/backends/chat-backend.interface.js → ai-providers/dto/service/ai-providers.service.types.js} +1 -1
  55. package/dist/ai-providers/dto/service/ai-providers.service.types.js.map +1 -0
  56. package/dist/ai-providers/enums/ai-model-job.enum.d.ts +7 -0
  57. package/dist/ai-providers/enums/ai-model-job.enum.js +15 -0
  58. package/dist/ai-providers/enums/ai-model-job.enum.js.map +1 -0
  59. package/dist/ai-providers/enums/ai-model-kind.enum.d.ts +4 -0
  60. package/dist/ai-providers/enums/ai-model-kind.enum.js +9 -0
  61. package/dist/ai-providers/enums/ai-model-kind.enum.js.map +1 -0
  62. package/dist/ai-providers/enums/ai-provider-type.enum.d.ts +8 -0
  63. package/dist/ai-providers/enums/ai-provider-type.enum.js +13 -0
  64. package/dist/ai-providers/enums/ai-provider-type.enum.js.map +1 -0
  65. package/dist/ai-providers/enums/index.d.ts +3 -0
  66. package/dist/{chat-providers → ai-providers}/enums/index.js +3 -1
  67. package/dist/ai-providers/enums/index.js.map +1 -0
  68. package/dist/ai-providers/errors/ai-providers.errors.d.ts +30 -0
  69. package/dist/ai-providers/errors/ai-providers.errors.js +57 -0
  70. package/dist/ai-providers/errors/ai-providers.errors.js.map +1 -0
  71. package/dist/ai-providers/google-environment.d.ts +3 -0
  72. package/dist/ai-providers/google-environment.js +18 -0
  73. package/dist/ai-providers/google-environment.js.map +1 -0
  74. package/dist/app-init/app-init.module.js +2 -2
  75. package/dist/app-init/app-init.module.js.map +1 -1
  76. package/dist/app-init/app-init.runner.d.ts +4 -4
  77. package/dist/app-init/app-init.runner.js +21 -21
  78. package/dist/app-init/app-init.runner.js.map +1 -1
  79. package/dist/app.module.js +15 -3
  80. package/dist/app.module.js.map +1 -1
  81. package/dist/auth/core/scopes/all-api.scopes.js +2 -2
  82. package/dist/auth/core/scopes/all-api.scopes.js.map +1 -1
  83. package/dist/common/postgres-io.adapter.d.ts +12 -0
  84. package/dist/common/postgres-io.adapter.js +62 -0
  85. package/dist/common/postgres-io.adapter.js.map +1 -0
  86. package/dist/config/env.config.d.ts +1 -0
  87. package/dist/config/env.config.js +9 -0
  88. package/dist/config/env.config.js.map +1 -1
  89. package/dist/errors/http/error-catalog.js +20 -14
  90. package/dist/errors/http/error-catalog.js.map +1 -1
  91. package/dist/main.js +4 -0
  92. package/dist/main.js.map +1 -1
  93. package/dist/migrations/1760200000000-ThreadAgentMessages.d.ts +6 -0
  94. package/dist/migrations/1760200000000-ThreadAgentMessages.js +28 -0
  95. package/dist/migrations/1760200000000-ThreadAgentMessages.js.map +1 -0
  96. package/dist/migrations/1760300000000-ChatProviderModelConfig.d.ts +6 -0
  97. package/dist/migrations/1760300000000-ChatProviderModelConfig.js +14 -0
  98. package/dist/migrations/1760300000000-ChatProviderModelConfig.js.map +1 -0
  99. package/dist/migrations/1760400000000-ReplaceAdkChatProvider.d.ts +6 -0
  100. package/dist/migrations/1760400000000-ReplaceAdkChatProvider.js +16 -0
  101. package/dist/migrations/1760400000000-ReplaceAdkChatProvider.js.map +1 -0
  102. package/dist/migrations/1760500000000-AiProviderModelSelections.d.ts +6 -0
  103. package/dist/migrations/1760500000000-AiProviderModelSelections.js +56 -0
  104. package/dist/migrations/1760500000000-AiProviderModelSelections.js.map +1 -0
  105. package/dist/migrations/1760600000000-DropAdkSessionTables.d.ts +6 -0
  106. package/dist/migrations/1760600000000-DropAdkSessionTables.js +51 -0
  107. package/dist/migrations/1760600000000-DropAdkSessionTables.js.map +1 -0
  108. package/dist/public/assets/index-Bu8fSAx2.js +117 -0
  109. package/dist/public/assets/index-DOm5BmlC.css +1 -0
  110. package/dist/public/beta/assets/index-B9fCnAUe.js +94 -0
  111. package/dist/public/beta/assets/index-DIzh0pTC.css +1 -0
  112. package/dist/public/beta/index.html +2 -2
  113. package/dist/public/index.html +2 -2
  114. package/dist/public/package.json +4 -4
  115. package/dist/scripts/migrate-sqlite-to-postgres.js +0 -22
  116. package/dist/scripts/migrate-sqlite-to-postgres.js.map +1 -1
  117. package/dist/threads/agent/agent-context.utils.d.ts +14 -0
  118. package/dist/threads/agent/agent-context.utils.js +42 -0
  119. package/dist/threads/agent/agent-context.utils.js.map +1 -0
  120. package/dist/threads/agent/agent-context.utils.spec.d.ts +1 -0
  121. package/dist/threads/agent/agent-context.utils.spec.js +60 -0
  122. package/dist/threads/agent/agent-context.utils.spec.js.map +1 -0
  123. package/dist/threads/agent/chat-model.factory.d.ts +25 -0
  124. package/dist/threads/agent/chat-model.factory.js +106 -0
  125. package/dist/threads/agent/chat-model.factory.js.map +1 -0
  126. package/dist/threads/agent/chat-model.factory.spec.d.ts +1 -0
  127. package/dist/threads/agent/chat-model.factory.spec.js +83 -0
  128. package/dist/threads/agent/chat-model.factory.spec.js.map +1 -0
  129. package/dist/threads/agent/hooks/compaction.hook.d.ts +11 -0
  130. package/dist/threads/agent/hooks/compaction.hook.js +109 -0
  131. package/dist/threads/agent/hooks/compaction.hook.js.map +1 -0
  132. package/dist/threads/agent/hooks/easter-egg.hook.d.ts +5 -0
  133. package/dist/threads/agent/hooks/easter-egg.hook.js +30 -0
  134. package/dist/threads/agent/hooks/easter-egg.hook.js.map +1 -0
  135. package/dist/threads/agent/hooks/hooks.spec.d.ts +1 -0
  136. package/dist/threads/agent/hooks/hooks.spec.js +87 -0
  137. package/dist/threads/agent/hooks/hooks.spec.js.map +1 -0
  138. package/dist/threads/agent/hooks/thread-agent-hook.interface.d.ts +20 -0
  139. package/dist/threads/agent/hooks/thread-agent-hook.interface.js +5 -0
  140. package/dist/threads/agent/hooks/thread-agent-hook.interface.js.map +1 -0
  141. package/dist/threads/agent/hooks/thread-agent-hooks.runner.d.ts +11 -0
  142. package/dist/threads/agent/hooks/thread-agent-hooks.runner.js +98 -0
  143. package/dist/threads/agent/hooks/thread-agent-hooks.runner.js.map +1 -0
  144. package/dist/threads/agent/taico-mcp-tools.factory.d.ts +22 -0
  145. package/dist/threads/agent/taico-mcp-tools.factory.js +79 -0
  146. package/dist/threads/agent/taico-mcp-tools.factory.js.map +1 -0
  147. package/dist/threads/agent/thread-agent-log.service.d.ts +41 -0
  148. package/dist/threads/agent/thread-agent-log.service.js +157 -0
  149. package/dist/threads/agent/thread-agent-log.service.js.map +1 -0
  150. package/dist/threads/agent/thread-agent-message.entity.d.ts +19 -0
  151. package/dist/threads/agent/thread-agent-message.entity.js +109 -0
  152. package/dist/threads/agent/thread-agent-message.entity.js.map +1 -0
  153. package/dist/threads/agent/thread-agent.enums.d.ts +17 -0
  154. package/dist/threads/agent/thread-agent.enums.js +24 -0
  155. package/dist/threads/agent/thread-agent.enums.js.map +1 -0
  156. package/dist/threads/agent/thread-agent.prompts.d.ts +5 -0
  157. package/dist/threads/{backends/chat-backend.utils.js → agent/thread-agent.prompts.js} +5 -4
  158. package/dist/threads/agent/thread-agent.prompts.js.map +1 -0
  159. package/dist/threads/agent/thread-agent.service.d.ts +33 -0
  160. package/dist/threads/agent/thread-agent.service.js +286 -0
  161. package/dist/threads/agent/thread-agent.service.js.map +1 -0
  162. package/dist/threads/agent/thread-agent.service.spec.d.ts +1 -0
  163. package/dist/threads/agent/thread-agent.service.spec.js +184 -0
  164. package/dist/threads/agent/thread-agent.service.spec.js.map +1 -0
  165. package/dist/threads/agent/thread-turn.entity.d.ts +12 -0
  166. package/dist/threads/agent/thread-turn.entity.js +71 -0
  167. package/dist/threads/agent/thread-turn.entity.js.map +1 -0
  168. package/dist/threads/agent/thread-turns.service.d.ts +13 -0
  169. package/dist/threads/agent/thread-turns.service.js +52 -0
  170. package/dist/threads/agent/thread-turns.service.js.map +1 -0
  171. package/dist/threads/chat.service.d.ts +5 -30
  172. package/dist/threads/chat.service.js +44 -119
  173. package/dist/threads/chat.service.js.map +1 -1
  174. package/dist/threads/dto/service/threads.service.types.d.ts +0 -2
  175. package/dist/threads/dto/thread-list-item-response.dto.d.ts +0 -1
  176. package/dist/threads/dto/thread-list-item-response.dto.js +0 -10
  177. package/dist/threads/dto/thread-list-item-response.dto.js.map +1 -1
  178. package/dist/threads/dto/thread-response.dto.d.ts +0 -1
  179. package/dist/threads/dto/thread-response.dto.js +0 -11
  180. package/dist/threads/dto/thread-response.dto.js.map +1 -1
  181. package/dist/threads/thread-message.entity.d.ts +3 -0
  182. package/dist/threads/thread-message.entity.js +12 -0
  183. package/dist/threads/thread-message.entity.js.map +1 -1
  184. package/dist/threads/thread.entity.d.ts +3 -1
  185. package/dist/threads/thread.entity.js +23 -3
  186. package/dist/threads/thread.entity.js.map +1 -1
  187. package/dist/threads/threads-state.service.spec.js +3 -1
  188. package/dist/threads/threads-state.service.spec.js.map +1 -1
  189. package/dist/threads/threads.module.js +31 -8
  190. package/dist/threads/threads.module.js.map +1 -1
  191. package/dist/threads/threads.service.d.ts +3 -8
  192. package/dist/threads/threads.service.js +6 -143
  193. package/dist/threads/threads.service.js.map +1 -1
  194. package/dist/threads/threads.service.spec.js +3 -2
  195. package/dist/threads/threads.service.spec.js.map +1 -1
  196. package/dist/tsconfig.tsbuildinfo +1 -1
  197. package/dist/walkthrough/dto/walkthrough-status-response.dto.d.ts +1 -0
  198. package/dist/walkthrough/dto/walkthrough-status-response.dto.js +8 -0
  199. package/dist/walkthrough/dto/walkthrough-status-response.dto.js.map +1 -1
  200. package/dist/walkthrough/walkthrough.controller.js +1 -0
  201. package/dist/walkthrough/walkthrough.controller.js.map +1 -1
  202. package/dist/walkthrough/walkthrough.module.js +2 -0
  203. package/dist/walkthrough/walkthrough.module.js.map +1 -1
  204. package/dist/walkthrough/walkthrough.service.d.ts +4 -1
  205. package/dist/walkthrough/walkthrough.service.js +14 -7
  206. package/dist/walkthrough/walkthrough.service.js.map +1 -1
  207. package/package.json +17 -9
  208. package/dist/chat-providers/chat-provider.entity.js.map +0 -1
  209. package/dist/chat-providers/chat-providers.controller.d.ts +0 -19
  210. package/dist/chat-providers/chat-providers.controller.js +0 -179
  211. package/dist/chat-providers/chat-providers.controller.js.map +0 -1
  212. package/dist/chat-providers/chat-providers.module.d.ts +0 -2
  213. package/dist/chat-providers/chat-providers.module.js +0 -32
  214. package/dist/chat-providers/chat-providers.module.js.map +0 -1
  215. package/dist/chat-providers/chat-providers.scopes.d.ts +0 -20
  216. package/dist/chat-providers/chat-providers.scopes.js +0 -23
  217. package/dist/chat-providers/chat-providers.scopes.js.map +0 -1
  218. package/dist/chat-providers/chat-providers.service.d.ts +0 -20
  219. package/dist/chat-providers/chat-providers.service.js +0 -193
  220. package/dist/chat-providers/chat-providers.service.js.map +0 -1
  221. package/dist/chat-providers/dto/http/chat-provider-params.dto.d.ts +0 -3
  222. package/dist/chat-providers/dto/http/chat-provider-params.dto.js.map +0 -1
  223. package/dist/chat-providers/dto/http/chat-provider-response.dto.d.ts +0 -12
  224. package/dist/chat-providers/dto/http/chat-provider-response.dto.js.map +0 -1
  225. package/dist/chat-providers/dto/http/create-chat-provider.dto.d.ts +0 -6
  226. package/dist/chat-providers/dto/http/create-chat-provider.dto.js.map +0 -1
  227. package/dist/chat-providers/dto/http/set-active-chat-provider.dto.d.ts +0 -3
  228. package/dist/chat-providers/dto/http/set-active-chat-provider.dto.js.map +0 -1
  229. package/dist/chat-providers/dto/http/update-chat-provider.dto.d.ts +0 -5
  230. package/dist/chat-providers/dto/http/update-chat-provider.dto.js +0 -49
  231. package/dist/chat-providers/dto/http/update-chat-provider.dto.js.map +0 -1
  232. package/dist/chat-providers/dto/service/chat-providers.service.types.d.ts +0 -30
  233. package/dist/chat-providers/dto/service/chat-providers.service.types.js +0 -3
  234. package/dist/chat-providers/dto/service/chat-providers.service.types.js.map +0 -1
  235. package/dist/chat-providers/enums/chat-provider-type.enum.d.ts +0 -4
  236. package/dist/chat-providers/enums/chat-provider-type.enum.js +0 -9
  237. package/dist/chat-providers/enums/chat-provider-type.enum.js.map +0 -1
  238. package/dist/chat-providers/enums/index.d.ts +0 -1
  239. package/dist/chat-providers/enums/index.js.map +0 -1
  240. package/dist/chat-providers/errors/chat-providers.errors.d.ts +0 -25
  241. package/dist/chat-providers/errors/chat-providers.errors.js +0 -48
  242. package/dist/chat-providers/errors/chat-providers.errors.js.map +0 -1
  243. package/dist/llm/llm.module.d.ts +0 -2
  244. package/dist/llm/llm.module.js +0 -23
  245. package/dist/llm/llm.module.js.map +0 -1
  246. package/dist/llm/openai-responses.service.d.ts +0 -12
  247. package/dist/llm/openai-responses.service.js +0 -69
  248. package/dist/llm/openai-responses.service.js.map +0 -1
  249. package/dist/public/assets/index-CPjOd3Sq.css +0 -1
  250. package/dist/public/assets/index-DVNE_5Yh.js +0 -73
  251. package/dist/public/beta/assets/index-BRla0F5r.js +0 -94
  252. package/dist/public/beta/assets/index-BaBvl20K.css +0 -1
  253. package/dist/threads/backends/adk.backend.d.ts +0 -18
  254. package/dist/threads/backends/adk.backend.js +0 -264
  255. package/dist/threads/backends/adk.backend.js.map +0 -1
  256. package/dist/threads/backends/chat-backend.interface.d.ts +0 -39
  257. package/dist/threads/backends/chat-backend.interface.js.map +0 -1
  258. package/dist/threads/backends/chat-backend.utils.d.ts +0 -5
  259. package/dist/threads/backends/chat-backend.utils.js.map +0 -1
  260. package/dist/threads/backends/openai.backend.d.ts +0 -19
  261. package/dist/threads/backends/openai.backend.js +0 -155
  262. package/dist/threads/backends/openai.backend.js.map +0 -1
  263. package/dist/threads/openai-mcp-server-factory.service.d.ts +0 -8
  264. package/dist/threads/openai-mcp-server-factory.service.js +0 -114
  265. 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,5 @@
1
+ import { BeforeTurnContext, ThreadAgentHook } from './thread-agent-hook.interface';
2
+ export declare class EasterEggHook implements ThreadAgentHook {
3
+ readonly name = "easter-egg";
4
+ beforeTurn(context: BeforeTurnContext): Promise<string | null>;
5
+ }
@@ -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,5 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.THREAD_AGENT_HOOKS = void 0;
4
+ exports.THREAD_AGENT_HOOKS = Symbol('THREAD_AGENT_HOOKS');
5
+ //# sourceMappingURL=thread-agent-hook.interface.js.map
@@ -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"}