@taico/taico 0.4.0 → 0.5.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/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 +27 -21
- package/dist/app-init/app-init.runner.js.map +1 -1
- package/dist/app-init/mcp/internal-worker-auth.mcp.d.ts +2 -0
- package/dist/app-init/mcp/internal-worker-auth.mcp.js +13 -1
- package/dist/app-init/mcp/internal-worker-auth.mcp.js.map +1 -1
- package/dist/app.module.js +15 -3
- package/dist/app.module.js.map +1 -1
- package/dist/auth/core/constants/internal-auth-target.constant.d.ts +1 -1
- package/dist/auth/core/constants/internal-auth-target.constant.js +6 -1
- package/dist/auth/core/constants/internal-auth-target.constant.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/discovery/discovery.service.js +1 -0
- package/dist/discovery/discovery.service.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/executions/active/active-task-execution.service.js +4 -0
- package/dist/executions/active/active-task-execution.service.js.map +1 -1
- package/dist/executions/events/active-executions-changed.event.d.ts +5 -0
- package/dist/executions/events/active-executions-changed.event.js +12 -0
- package/dist/executions/events/active-executions-changed.event.js.map +1 -0
- package/dist/executions/executions.gateway.d.ts +18 -0
- package/dist/executions/executions.gateway.js +85 -0
- package/dist/executions/executions.gateway.js.map +1 -0
- package/dist/executions/executions.module.js +2 -0
- package/dist/executions/executions.module.js.map +1 -1
- package/dist/executions/staleness/stale-active-task-execution-pruner.service.js +2 -0
- package/dist/executions/staleness/stale-active-task-execution-pruner.service.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-BUmnqjpv.js +117 -0
- package/dist/public/assets/index-DOm5BmlC.css +1 -0
- package/dist/public/beta/assets/{index-BRla0F5r.js → index-n0qUSWPJ.js} +30 -30
- package/dist/public/beta/index.html +1 -1
- 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/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 @@
|
|
|
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"}
|
|
@@ -0,0 +1,41 @@
|
|
|
1
|
+
import { DataSource } from 'typeorm';
|
|
2
|
+
import { ModelMessage } from 'ai';
|
|
3
|
+
import { ActorType } from "../../identity-provider/enums";
|
|
4
|
+
import { AgentContextRow } from './agent-context.utils';
|
|
5
|
+
import { ThreadAgentMessageEntity } from './thread-agent-message.entity';
|
|
6
|
+
import { ThreadAgentMessageKind, ThreadAgentMessageRole } from './thread-agent.enums';
|
|
7
|
+
export type AgentMessageDraft = {
|
|
8
|
+
role: ThreadAgentMessageRole;
|
|
9
|
+
message: ModelMessage;
|
|
10
|
+
source: string;
|
|
11
|
+
kind?: ThreadAgentMessageKind;
|
|
12
|
+
turnId?: string | null;
|
|
13
|
+
providerType?: string | null;
|
|
14
|
+
modelId?: string | null;
|
|
15
|
+
usage?: Record<string, unknown> | null;
|
|
16
|
+
createdAt?: Date;
|
|
17
|
+
};
|
|
18
|
+
export type TriggerActor = {
|
|
19
|
+
id: string;
|
|
20
|
+
slug: string;
|
|
21
|
+
type: ActorType;
|
|
22
|
+
displayName: string;
|
|
23
|
+
};
|
|
24
|
+
export type IngestPendingResult = {
|
|
25
|
+
triggerActor: TriggerActor | null;
|
|
26
|
+
newUserMessages: string[];
|
|
27
|
+
};
|
|
28
|
+
export declare class ThreadAgentLogService {
|
|
29
|
+
private readonly dataSource;
|
|
30
|
+
constructor(dataSource: DataSource);
|
|
31
|
+
append(threadId: string, drafts: AgentMessageDraft[]): Promise<ThreadAgentMessageEntity[]>;
|
|
32
|
+
ingestPending(input: {
|
|
33
|
+
threadId: string;
|
|
34
|
+
agentActorId: string;
|
|
35
|
+
turnId: string;
|
|
36
|
+
}): Promise<IngestPendingResult>;
|
|
37
|
+
hasPending(threadId: string, agentActorId: string): Promise<boolean>;
|
|
38
|
+
loadContext(threadId: string): Promise<AgentContextRow[]>;
|
|
39
|
+
private lockHead;
|
|
40
|
+
private appendWithManager;
|
|
41
|
+
}
|
|
@@ -0,0 +1,157 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
3
|
+
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
4
|
+
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
5
|
+
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
|
|
6
|
+
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
7
|
+
};
|
|
8
|
+
var __metadata = (this && this.__metadata) || function (k, v) {
|
|
9
|
+
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
|
|
10
|
+
};
|
|
11
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
12
|
+
exports.ThreadAgentLogService = void 0;
|
|
13
|
+
const common_1 = require("@nestjs/common");
|
|
14
|
+
const typeorm_1 = require("typeorm");
|
|
15
|
+
const node_crypto_1 = require("node:crypto");
|
|
16
|
+
const threads_errors_1 = require("../errors/threads.errors");
|
|
17
|
+
const thread_agent_prompts_1 = require("./thread-agent.prompts");
|
|
18
|
+
const thread_agent_message_entity_1 = require("./thread-agent-message.entity");
|
|
19
|
+
const thread_agent_enums_1 = require("./thread-agent.enums");
|
|
20
|
+
let ThreadAgentLogService = class ThreadAgentLogService {
|
|
21
|
+
dataSource;
|
|
22
|
+
constructor(dataSource) {
|
|
23
|
+
this.dataSource = dataSource;
|
|
24
|
+
}
|
|
25
|
+
async append(threadId, drafts) {
|
|
26
|
+
return this.dataSource.transaction((manager) => this.appendWithManager(manager, threadId, drafts));
|
|
27
|
+
}
|
|
28
|
+
async ingestPending(input) {
|
|
29
|
+
return this.dataSource.transaction(async (manager) => {
|
|
30
|
+
const headId = await this.lockHead(manager, input.threadId);
|
|
31
|
+
const isBackfill = headId === null;
|
|
32
|
+
const unlinked = await manager.query(`SELECT tm.id, tm.content, tm.created_at, a.id AS actor_id, a.slug AS actor_slug,
|
|
33
|
+
a.type AS actor_type, a.display_name AS actor_display_name
|
|
34
|
+
FROM thread_messages tm
|
|
35
|
+
JOIN actors a ON a.id = tm.created_by_actor_id
|
|
36
|
+
WHERE tm.thread_id = $1 AND tm.agent_message_id IS NULL
|
|
37
|
+
ORDER BY tm.created_at ASC, tm.id ASC`, [input.threadId]);
|
|
38
|
+
const pending = isBackfill
|
|
39
|
+
? unlinked
|
|
40
|
+
: unlinked.filter((row) => row.actor_id !== input.agentActorId);
|
|
41
|
+
if (pending.length === 0) {
|
|
42
|
+
return { triggerActor: null, newUserMessages: [] };
|
|
43
|
+
}
|
|
44
|
+
const drafts = pending.map((row) => {
|
|
45
|
+
const fromAgent = row.actor_id === input.agentActorId;
|
|
46
|
+
return fromAgent
|
|
47
|
+
? {
|
|
48
|
+
role: thread_agent_enums_1.ThreadAgentMessageRole.ASSISTANT,
|
|
49
|
+
message: { role: 'assistant', content: row.content },
|
|
50
|
+
source: 'backfill',
|
|
51
|
+
createdAt: row.created_at,
|
|
52
|
+
}
|
|
53
|
+
: {
|
|
54
|
+
role: thread_agent_enums_1.ThreadAgentMessageRole.USER,
|
|
55
|
+
message: {
|
|
56
|
+
role: 'user',
|
|
57
|
+
content: (0, thread_agent_prompts_1.formatMessage)(row.content, {
|
|
58
|
+
displayName: row.actor_display_name,
|
|
59
|
+
slug: row.actor_slug,
|
|
60
|
+
}),
|
|
61
|
+
},
|
|
62
|
+
source: 'user',
|
|
63
|
+
turnId: input.turnId,
|
|
64
|
+
createdAt: row.created_at,
|
|
65
|
+
};
|
|
66
|
+
});
|
|
67
|
+
const saved = await this.appendWithManager(manager, input.threadId, drafts);
|
|
68
|
+
for (const [index, row] of pending.entries()) {
|
|
69
|
+
await manager.query(`UPDATE thread_messages SET agent_message_id = $1 WHERE id = $2`, [saved[index].id, row.id]);
|
|
70
|
+
}
|
|
71
|
+
const last = pending[pending.length - 1];
|
|
72
|
+
if (last.actor_id === input.agentActorId) {
|
|
73
|
+
return { triggerActor: null, newUserMessages: [] };
|
|
74
|
+
}
|
|
75
|
+
const lastAgentIndex = pending
|
|
76
|
+
.map((row) => row.actor_id)
|
|
77
|
+
.lastIndexOf(input.agentActorId);
|
|
78
|
+
return {
|
|
79
|
+
triggerActor: {
|
|
80
|
+
id: last.actor_id,
|
|
81
|
+
slug: last.actor_slug,
|
|
82
|
+
type: last.actor_type,
|
|
83
|
+
displayName: last.actor_display_name,
|
|
84
|
+
},
|
|
85
|
+
newUserMessages: pending.slice(lastAgentIndex + 1).map((row) => row.content),
|
|
86
|
+
};
|
|
87
|
+
});
|
|
88
|
+
}
|
|
89
|
+
async hasPending(threadId, agentActorId) {
|
|
90
|
+
const rows = await this.dataSource.query(`SELECT 1 FROM thread_messages
|
|
91
|
+
WHERE thread_id = $1 AND agent_message_id IS NULL AND created_by_actor_id <> $2
|
|
92
|
+
LIMIT 1`, [threadId, agentActorId]);
|
|
93
|
+
return rows.length > 0;
|
|
94
|
+
}
|
|
95
|
+
async loadContext(threadId) {
|
|
96
|
+
const rows = await this.dataSource.query(`WITH RECURSIVE chain AS (
|
|
97
|
+
SELECT m.id, m.parent_id, m.kind, m.source, m.message, m.format_version, m.provider_type, m.model_id, 0 AS depth
|
|
98
|
+
FROM thread_agent_messages m
|
|
99
|
+
JOIN threads t ON t.head_agent_message_id = m.id
|
|
100
|
+
WHERE t.id = $1
|
|
101
|
+
UNION ALL
|
|
102
|
+
SELECT p.id, p.parent_id, p.kind, p.source, p.message, p.format_version, p.provider_type, p.model_id, c.depth + 1
|
|
103
|
+
FROM thread_agent_messages p
|
|
104
|
+
JOIN chain c ON p.id = c.parent_id
|
|
105
|
+
WHERE c.kind <> $2
|
|
106
|
+
)
|
|
107
|
+
SELECT id, source, message, format_version, provider_type, model_id FROM chain ORDER BY depth DESC`, [threadId, thread_agent_enums_1.ThreadAgentMessageKind.SUMMARY]);
|
|
108
|
+
return rows.map((row) => ({
|
|
109
|
+
id: row.id,
|
|
110
|
+
source: row.source,
|
|
111
|
+
message: row.message,
|
|
112
|
+
formatVersion: row.format_version,
|
|
113
|
+
providerType: row.provider_type,
|
|
114
|
+
modelId: row.model_id,
|
|
115
|
+
}));
|
|
116
|
+
}
|
|
117
|
+
async lockHead(manager, threadId) {
|
|
118
|
+
const rows = await manager.query(`SELECT head_agent_message_id FROM threads WHERE id = $1 FOR UPDATE`, [threadId]);
|
|
119
|
+
if (rows.length === 0) {
|
|
120
|
+
throw new threads_errors_1.ThreadNotFoundError(threadId);
|
|
121
|
+
}
|
|
122
|
+
return rows[0].head_agent_message_id;
|
|
123
|
+
}
|
|
124
|
+
async appendWithManager(manager, threadId, drafts) {
|
|
125
|
+
let parentId = await this.lockHead(manager, threadId);
|
|
126
|
+
const rows = drafts.map((draft) => {
|
|
127
|
+
const row = manager.create(thread_agent_message_entity_1.ThreadAgentMessageEntity, {
|
|
128
|
+
id: (0, node_crypto_1.randomUUID)(),
|
|
129
|
+
threadId,
|
|
130
|
+
parentId,
|
|
131
|
+
role: draft.role,
|
|
132
|
+
kind: draft.kind ?? thread_agent_enums_1.ThreadAgentMessageKind.MESSAGE,
|
|
133
|
+
source: draft.source,
|
|
134
|
+
message: draft.message,
|
|
135
|
+
turnId: draft.turnId ?? null,
|
|
136
|
+
providerType: draft.providerType ?? null,
|
|
137
|
+
modelId: draft.modelId ?? null,
|
|
138
|
+
usage: draft.usage ?? null,
|
|
139
|
+
...(draft.createdAt ? { createdAt: draft.createdAt } : {}),
|
|
140
|
+
});
|
|
141
|
+
parentId = row.id;
|
|
142
|
+
return row;
|
|
143
|
+
});
|
|
144
|
+
if (rows.length === 0) {
|
|
145
|
+
return rows;
|
|
146
|
+
}
|
|
147
|
+
await manager.insert(thread_agent_message_entity_1.ThreadAgentMessageEntity, rows);
|
|
148
|
+
await manager.query(`UPDATE threads SET head_agent_message_id = $1 WHERE id = $2`, [parentId, threadId]);
|
|
149
|
+
return rows;
|
|
150
|
+
}
|
|
151
|
+
};
|
|
152
|
+
exports.ThreadAgentLogService = ThreadAgentLogService;
|
|
153
|
+
exports.ThreadAgentLogService = ThreadAgentLogService = __decorate([
|
|
154
|
+
(0, common_1.Injectable)(),
|
|
155
|
+
__metadata("design:paramtypes", [typeorm_1.DataSource])
|
|
156
|
+
], ThreadAgentLogService);
|
|
157
|
+
//# sourceMappingURL=thread-agent-log.service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"thread-agent-log.service.js","sourceRoot":"","sources":["../../../src/threads/agent/thread-agent-log.service.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,2CAA4C;AAC5C,qCAAoD;AAEpD,6CAAyC;AAGzC,6DAA+D;AAC/D,iEAAuD;AAEvD,+EAAyE;AACzE,6DAG8B;AA2CvB,IAAM,qBAAqB,GAA3B,MAAM,qBAAqB;IACH;IAA7B,YAA6B,UAAsB;QAAtB,eAAU,GAAV,UAAU,CAAY;IAAG,CAAC;IAGvD,KAAK,CAAC,MAAM,CACV,QAAgB,EAChB,MAA2B;QAE3B,OAAO,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,OAAO,EAAE,EAAE,CAC7C,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,CAClD,CAAC;IACJ,CAAC;IASD,KAAK,CAAC,aAAa,CAAC,KAInB;QACC,OAAO,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;YACnD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC5D,MAAM,UAAU,GAAG,MAAM,KAAK,IAAI,CAAC;YAEnC,MAAM,QAAQ,GAAwB,MAAM,OAAO,CAAC,KAAK,CACvD;;;;;gDAKwC,EACxC,CAAC,KAAK,CAAC,QAAQ,CAAC,CACjB,CAAC;YAGF,MAAM,OAAO,GAAG,UAAU;gBACxB,CAAC,CAAC,QAAQ;gBACV,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,KAAK,KAAK,CAAC,YAAY,CAAC,CAAC;YAElE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC;YACrD,CAAC;YAED,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,EAAqB,EAAE;gBACpD,MAAM,SAAS,GAAG,GAAG,CAAC,QAAQ,KAAK,KAAK,CAAC,YAAY,CAAC;gBACtD,OAAO,SAAS;oBACd,CAAC,CAAC;wBACE,IAAI,EAAE,2CAAsB,CAAC,SAAS;wBACtC,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE;wBACpD,MAAM,EAAE,UAAU;wBAClB,SAAS,EAAE,GAAG,CAAC,UAAU;qBAC1B;oBACH,CAAC,CAAC;wBACE,IAAI,EAAE,2CAAsB,CAAC,IAAI;wBACjC,OAAO,EAAE;4BACP,IAAI,EAAE,MAAM;4BACZ,OAAO,EAAE,IAAA,oCAAa,EAAC,GAAG,CAAC,OAAO,EAAE;gCAClC,WAAW,EAAE,GAAG,CAAC,kBAAkB;gCACnC,IAAI,EAAE,GAAG,CAAC,UAAU;6BACrB,CAAC;yBACH;wBACD,MAAM,EAAE,MAAM;wBACd,MAAM,EAAE,KAAK,CAAC,MAAM;wBACpB,SAAS,EAAE,GAAG,CAAC,UAAU;qBAC1B,CAAC;YACR,CAAC,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAE5E,KAAK,MAAM,CAAC,KAAK,EAAE,GAAG,CAAC,IAAI,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;gBAC7C,MAAM,OAAO,CAAC,KAAK,CACjB,gEAAgE,EAChE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,CAAC,CAC1B,CAAC;YACJ,CAAC;YAED,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YACzC,IAAI,IAAI,CAAC,QAAQ,KAAK,KAAK,CAAC,YAAY,EAAE,CAAC;gBACzC,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC;YACrD,CAAC;YAED,MAAM,cAAc,GAAG,OAAO;iBAC3B,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC;iBAC1B,WAAW,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAEnC,OAAO;gBACL,YAAY,EAAE;oBACZ,EAAE,EAAE,IAAI,CAAC,QAAQ;oBACjB,IAAI,EAAE,IAAI,CAAC,UAAU;oBACrB,IAAI,EAAE,IAAI,CAAC,UAAU;oBACrB,WAAW,EAAE,IAAI,CAAC,kBAAkB;iBACrC;gBACD,eAAe,EAAE,OAAO,CAAC,KAAK,CAAC,cAAc,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC;aAC7E,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAGD,KAAK,CAAC,UAAU,CAAC,QAAgB,EAAE,YAAoB;QACrD,MAAM,IAAI,GAAc,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CACjD;;gBAEU,EACV,CAAC,QAAQ,EAAE,YAAY,CAAC,CACzB,CAAC;QACF,OAAO,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;IACzB,CAAC;IAMD,KAAK,CAAC,WAAW,CAAC,QAAgB;QAChC,MAAM,IAAI,GAOL,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAC9B;;;;;;;;;;;0GAWoG,EACpG,CAAC,QAAQ,EAAE,2CAAsB,CAAC,OAAO,CAAC,CAC3C,CAAC;QAEF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACxB,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,aAAa,EAAE,GAAG,CAAC,cAAc;YACjC,YAAY,EAAE,GAAG,CAAC,aAAa;YAC/B,OAAO,EAAE,GAAG,CAAC,QAAQ;SACtB,CAAC,CAAC,CAAC;IACN,CAAC;IAEO,KAAK,CAAC,QAAQ,CACpB,OAAsB,EACtB,QAAgB;QAEhB,MAAM,IAAI,GACR,MAAM,OAAO,CAAC,KAAK,CACjB,oEAAoE,EACpE,CAAC,QAAQ,CAAC,CACX,CAAC;QAEJ,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,MAAM,IAAI,oCAAmB,CAAC,QAAQ,CAAC,CAAC;QAC1C,CAAC;QAED,OAAO,IAAI,CAAC,CAAC,CAAC,CAAC,qBAAqB,CAAC;IACvC,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAC7B,OAAsB,EACtB,QAAgB,EAChB,MAA2B;QAE3B,IAAI,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAEtD,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YAChC,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC,sDAAwB,EAAE;gBACnD,EAAE,EAAE,IAAA,wBAAU,GAAE;gBAChB,QAAQ;gBACR,QAAQ;gBACR,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,2CAAsB,CAAC,OAAO;gBAClD,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,OAAO,EAAE,KAAK,CAAC,OAA6C;gBAC5D,MAAM,EAAE,KAAK,CAAC,MAAM,IAAI,IAAI;gBAC5B,YAAY,EAAE,KAAK,CAAC,YAAY,IAAI,IAAI;gBACxC,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,IAAI;gBAC9B,KAAK,EAAE,KAAK,CAAC,KAAK,IAAI,IAAI;gBAC1B,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC3D,CAAC,CAAC;YACH,QAAQ,GAAG,GAAG,CAAC,EAAE,CAAC;YAClB,OAAO,GAAG,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,OAAO,CAAC,MAAM,CAClB,sDAAwB,EACxB,IAA0D,CAC3D,CAAC;QAEF,MAAM,OAAO,CAAC,KAAK,CACjB,6DAA6D,EAC7D,CAAC,QAAQ,EAAE,QAAQ,CAAC,CACrB,CAAC;QAEF,OAAO,IAAI,CAAC;IACd,CAAC;CACF,CAAA;AAlNY,sDAAqB;gCAArB,qBAAqB;IADjC,IAAA,mBAAU,GAAE;qCAE8B,oBAAU;GADxC,qBAAqB,CAkNjC"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import { ThreadEntity } from '../thread.entity';
|
|
2
|
+
import { ThreadAgentMessageKind, ThreadAgentMessageRole } from './thread-agent.enums';
|
|
3
|
+
export declare class ThreadAgentMessageEntity {
|
|
4
|
+
id: string;
|
|
5
|
+
threadId: string;
|
|
6
|
+
thread?: ThreadEntity;
|
|
7
|
+
parentId: string | null;
|
|
8
|
+
parent?: ThreadAgentMessageEntity | null;
|
|
9
|
+
turnId: string | null;
|
|
10
|
+
role: ThreadAgentMessageRole;
|
|
11
|
+
kind: ThreadAgentMessageKind;
|
|
12
|
+
source: string;
|
|
13
|
+
message: Record<string, unknown>;
|
|
14
|
+
formatVersion: number;
|
|
15
|
+
providerType: string | null;
|
|
16
|
+
modelId: string | null;
|
|
17
|
+
usage: Record<string, unknown> | null;
|
|
18
|
+
createdAt: Date;
|
|
19
|
+
}
|