@taico/taico 0.4.1 → 0.5.1

package/dist/threads/agent/thread-agent.service.js

@@ -0,0 +1,286 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    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;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+var ThreadAgentService_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ThreadAgentService = exports.THREAD_AGENT_MAX_STEPS = exports.THREAD_AGENT_SLUG = void 0;
+const common_1 = require("@nestjs/common");
+const typeorm_1 = require("@nestjs/typeorm");
+const event_emitter_1 = require("@nestjs/event-emitter");
+const typeorm_2 = require("typeorm");
+const node_crypto_1 = require("node:crypto");
+const ai_1 = require("ai");
+const agents_service_1 = require("../../agents/agents.service");
+const enums_1 = require("../../ai-providers/enums");
+const enums_2 = require("../../identity-provider/enums");
+const thread_entity_1 = require("../thread.entity");
+const thread_message_entity_1 = require("../thread-message.entity");
+const threads_events_1 = require("../events/threads.events");
+const agent_context_utils_1 = require("./agent-context.utils");
+const chat_model_factory_1 = require("./chat-model.factory");
+const taico_mcp_tools_factory_1 = require("./taico-mcp-tools.factory");
+const thread_agent_log_service_1 = require("./thread-agent-log.service");
+const thread_agent_enums_1 = require("./thread-agent.enums");
+const thread_agent_prompts_1 = require("./thread-agent.prompts");
+const thread_turns_service_1 = require("./thread-turns.service");
+const thread_agent_hooks_runner_1 = require("./hooks/thread-agent-hooks.runner");
+exports.THREAD_AGENT_SLUG = 'taico';
+exports.THREAD_AGENT_MAX_STEPS = 25;
+let ThreadAgentService = ThreadAgentService_1 = class ThreadAgentService {
+    threadRepository;
+    threadMessageRepository;
+    agentsService;
+    agentLog;
+    turns;
+    chatModelFactory;
+    mcpToolsFactory;
+    hooks;
+    eventEmitter;
+    logger = new common_1.Logger(ThreadAgentService_1.name);
+    constructor(threadRepository, threadMessageRepository, agentsService, agentLog, turns, chatModelFactory, mcpToolsFactory, hooks, eventEmitter) {
+        this.threadRepository = threadRepository;
+        this.threadMessageRepository = threadMessageRepository;
+        this.agentsService = agentsService;
+        this.agentLog = agentLog;
+        this.turns = turns;
+        this.chatModelFactory = chatModelFactory;
+        this.mcpToolsFactory = mcpToolsFactory;
+        this.hooks = hooks;
+        this.eventEmitter = eventEmitter;
+    }
+    requestTurn(threadId) {
+        void this.drain(threadId).catch((error) => {
+            this.logger.error({
+                message: 'Thread agent run failed',
+                threadId,
+                error: this.describeError(error),
+            });
+        });
+    }
+    async drain(threadId) {
+        const self = await this.agentsService.getAgentBySlug({ slug: exports.THREAD_AGENT_SLUG });
+        for (;;) {
+            const turn = await this.turns.tryClaim(threadId);
+            if (!turn) {
+                return;
+            }
+            const stopHeartbeat = this.turns.startHeartbeat(turn.id);
+            try {
+                const outcome = await this.runTurn(turn, self);
+                if (outcome) {
+                    await this.hooks.runAfterTurn({ threadId, turnId: turn.id, ...outcome });
+                }
+                await this.turns.finish(turn.id, thread_agent_enums_1.ThreadTurnStatus.COMPLETED);
+            }
+            catch (error) {
+                const reason = error instanceof Error ? error.message : String(error);
+                this.logger.error({
+                    message: 'Thread agent turn failed',
+                    threadId,
+                    turnId: turn.id,
+                    error: this.describeError(error),
+                });
+                await this.turns.finish(turn.id, thread_agent_enums_1.ThreadTurnStatus.FAILED, reason);
+                await this.saveDisplayMessage({
+                    threadId,
+                    actorId: self.actorId,
+                    content: `I encountered an error while processing your message: ${reason}`,
+                    agentMessageId: null,
+                });
+                return;
+            }
+            finally {
+                stopHeartbeat();
+            }
+            if (!(await this.agentLog.hasPending(threadId, self.actorId))) {
+                return;
+            }
+        }
+    }
+    async runTurn(turn, self) {
+        const { threadId } = turn;
+        const { triggerActor, newUserMessages } = await this.agentLog.ingestPending({
+            threadId,
+            agentActorId: self.actorId,
+            turnId: turn.id,
+        });
+        if (!triggerActor) {
+            return null;
+        }
+        const thread = await this.threadRepository.findOne({ where: { id: threadId } });
+        const resolved = await this.chatModelFactory.resolve({
+            job: enums_1.AiModelJob.CONVERSATION,
+            aiProviderId: thread?.aiProviderId,
+            modelId: thread?.modelId,
+        });
+        let rows = await this.agentLog.loadContext(threadId);
+        const injected = await this.hooks.runBeforeTurn({
+            threadId,
+            turnId: turn.id,
+            newUserMessages,
+            context: rows,
+        });
+        if (injected) {
+            rows = await this.agentLog.loadContext(threadId);
+        }
+        const messages = (0, agent_context_utils_1.buildModelMessages)(rows, resolved, (row) => {
+            this.logger.warn({
+                message: 'Skipping agent log row that does not match the message schema',
+                threadId,
+                agentMessageId: row.id,
+            });
+        });
+        const mcp = await this.mcpToolsFactory.open({
+            subject: {
+                id: self.actorId,
+                slug: self.slug,
+                type: enums_2.ActorType.AGENT,
+                displayName: self.name,
+            },
+            issuedBy: triggerActor,
+        });
+        try {
+            const lastInputTokens = await this.streamTurn({
+                turn,
+                self,
+                resolved,
+                messages,
+                tools: mcp.tools,
+            });
+            return { model: resolved, lastInputTokens };
+        }
+        finally {
+            await mcp.close();
+        }
+    }
+    async streamTurn(input) {
+        const { turn, self, resolved } = input;
+        const { threadId } = turn;
+        let persisted = Promise.resolve();
+        let lastInputTokens = null;
+        const result = (0, ai_1.streamText)({
+            model: resolved.model,
+            instructions: (0, thread_agent_prompts_1.buildThreadScopedInstructions)(self.systemPrompt, threadId),
+            messages: input.messages,
+            tools: input.tools,
+            providerOptions: resolved.providerOptions,
+            stopWhen: (0, ai_1.isStepCount)(exports.THREAD_AGENT_MAX_STEPS),
+            onError: () => undefined,
+            onStepEnd: (step) => {
+                const output = {
+                    messages: step.response.messages,
+                    text: step.text,
+                    usage: step.usage,
+                };
+                lastInputTokens = step.usage?.inputTokens ?? lastInputTokens;
+                persisted = persisted.then(() => this.persistStep(turn, self, resolved, output));
+            },
+        });
+        let streamId = (0, node_crypto_1.randomUUID)();
+        for await (const part of result.stream) {
+            switch (part.type) {
+                case 'start-step':
+                    streamId = (0, node_crypto_1.randomUUID)();
+                    break;
+                case 'reasoning-delta':
+                    this.emitActivity(threadId, self.actorId, 'thinking');
+                    break;
+                case 'tool-call':
+                    this.emitActivity(threadId, self.actorId, 'tool_calling');
+                    break;
+                case 'text-delta':
+                    this.eventEmitter.emit(threads_events_1.ThreadAgentResponseDeltaEvent.INTERNAL, new threads_events_1.ThreadAgentResponseDeltaEvent({ id: self.actorId }, { threadId, streamId, delta: part.text }));
+                    break;
+                case 'error':
+                    await persisted;
+                    throw part.error instanceof Error ? part.error : new Error(String(part.error));
+            }
+        }
+        await persisted;
+        return lastInputTokens;
+    }
+    async persistStep(turn, self, resolved, step) {
+        const messages = JSON.parse(JSON.stringify(step.messages));
+        const drafts = messages.map((message) => ({
+            role: message.role,
+            message,
+            source: 'model',
+            turnId: turn.id,
+            providerType: resolved.providerType,
+            modelId: resolved.modelId,
+            usage: message.role === 'assistant'
+                ? JSON.parse(JSON.stringify(step.usage ?? null))
+                : null,
+        }));
+        const saved = await this.agentLog.append(turn.threadId, drafts);
+        const text = step.text.trim();
+        const assistantRow = saved.find((row) => row.role === thread_agent_enums_1.ThreadAgentMessageRole.ASSISTANT);
+        if (text && assistantRow) {
+            await this.saveDisplayMessage({
+                threadId: turn.threadId,
+                actorId: self.actorId,
+                content: text,
+                agentMessageId: assistantRow.id,
+            });
+        }
+    }
+    async saveDisplayMessage(input) {
+        try {
+            const saved = await this.threadMessageRepository.save(this.threadMessageRepository.create({
+                threadId: input.threadId,
+                content: input.content,
+                createdByActorId: input.actorId,
+                agentMessageId: input.agentMessageId,
+            }));
+            const withRelations = await this.threadMessageRepository.findOne({
+                where: { id: saved.id },
+                relations: ['createdByActor'],
+            });
+            if (!withRelations) {
+                return;
+            }
+            this.eventEmitter.emit(threads_events_1.MessageCreatedEvent.INTERNAL, new threads_events_1.MessageCreatedEvent({ id: input.actorId }, withRelations));
+        }
+        catch (error) {
+            this.logger.error({
+                message: 'Failed to persist agent message',
+                threadId: input.threadId,
+                error: this.describeError(error),
+            });
+        }
+    }
+    emitActivity(threadId, actorId, kind) {
+        this.eventEmitter.emit(threads_events_1.ThreadAgentActivityEvent.INTERNAL, new threads_events_1.ThreadAgentActivityEvent({ id: actorId }, { threadId, kind }));
+    }
+    describeError(error) {
+        return error instanceof Error
+            ? { message: error.message, stack: error.stack, name: error.name }
+            : String(error);
+    }
+};
+exports.ThreadAgentService = ThreadAgentService;
+exports.ThreadAgentService = ThreadAgentService = ThreadAgentService_1 = __decorate([
+    (0, common_1.Injectable)(),
+    __param(0, (0, typeorm_1.InjectRepository)(thread_entity_1.ThreadEntity)),
+    __param(1, (0, typeorm_1.InjectRepository)(thread_message_entity_1.ThreadMessageEntity)),
+    __metadata("design:paramtypes", [typeorm_2.Repository,
+        typeorm_2.Repository,
+        agents_service_1.AgentsService,
+        thread_agent_log_service_1.ThreadAgentLogService,
+        thread_turns_service_1.ThreadTurnsService,
+        chat_model_factory_1.ChatModelFactory,
+        taico_mcp_tools_factory_1.TaicoMcpToolsFactory,
+        thread_agent_hooks_runner_1.ThreadAgentHooksRunner,
+        event_emitter_1.EventEmitter2])
+], ThreadAgentService);
+//# sourceMappingURL=thread-agent.service.js.map

package/dist/threads/agent/thread-agent.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"thread-agent.service.js","sourceRoot":"","sources":["../../../src/threads/agent/thread-agent.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CAAoD;AACpD,6CAAmD;AACnD,yDAAsD;AACtD,qCAAqC;AACrC,6CAAyC;AACzC,2BAA2D;AAC3D,gEAA0D;AAC1D,oDAAoD;AAEpD,yDAAwD;AACxD,oDAAgD;AAChD,oEAA+D;AAC/D,6DAKkC;AAClC,+DAA2D;AAC3D,6DAA2E;AAC3E,uEAAiE;AACjE,yEAAsF;AACtF,6DAAgF;AAChF,iEAAuE;AACvE,iEAAyE;AACzE,iFAA2E;AAE9D,QAAA,iBAAiB,GAAG,OAAO,CAAC;AAC5B,QAAA,sBAAsB,GAAG,EAAE,CAAC;AAmBlC,IAAM,kBAAkB,0BAAxB,MAAM,kBAAkB;IAKV;IAEA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IAbF,MAAM,GAAG,IAAI,eAAM,CAAC,oBAAkB,CAAC,IAAI,CAAC,CAAC;IAE9D,YAEmB,gBAA0C,EAE1C,uBAAwD,EACxD,aAA4B,EAC5B,QAA+B,EAC/B,KAAyB,EACzB,gBAAkC,EAClC,eAAqC,EACrC,KAA6B,EAC7B,YAA2B;QAT3B,qBAAgB,GAAhB,gBAAgB,CAA0B;QAE1C,4BAAuB,GAAvB,uBAAuB,CAAiC;QACxD,kBAAa,GAAb,aAAa,CAAe;QAC5B,aAAQ,GAAR,QAAQ,CAAuB;QAC/B,UAAK,GAAL,KAAK,CAAoB;QACzB,qBAAgB,GAAhB,gBAAgB,CAAkB;QAClC,oBAAe,GAAf,eAAe,CAAsB;QACrC,UAAK,GAAL,KAAK,CAAwB;QAC7B,iBAAY,GAAZ,YAAY,CAAe;IAC3C,CAAC;IAGJ,WAAW,CAAC,QAAgB;QAC1B,KAAK,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACxC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBAChB,OAAO,EAAE,yBAAyB;gBAClC,QAAQ;gBACR,KAAK,EAAE,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC;aACjC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAMD,KAAK,CAAC,KAAK,CAAC,QAAgB;QAC1B,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC,EAAE,IAAI,EAAE,yBAAiB,EAAE,CAAC,CAAC;QAElF,SAAS,CAAC;YACR,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACjD,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,OAAO;YACT,CAAC;YAED,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACzD,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBAC/C,IAAI,OAAO,EAAE,CAAC;oBAEZ,MAAM,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;gBAC3E,CAAC;gBACD,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,qCAAgB,CAAC,SAAS,CAAC,CAAC;YAC/D,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,MAAM,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACtE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;oBAChB,OAAO,EAAE,0BAA0B;oBACnC,QAAQ;oBACR,MAAM,EAAE,IAAI,CAAC,EAAE;oBACf,KAAK,EAAE,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC;iBACjC,CAAC,CAAC;gBACH,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,EAAE,qCAAgB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBAClE,MAAM,IAAI,CAAC,kBAAkB,CAAC;oBAC5B,QAAQ;oBACR,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,OAAO,EAAE,yDAAyD,MAAM,EAAE;oBAC1E,cAAc,EAAE,IAAI;iBACrB,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;oBAAS,CAAC;gBACT,aAAa,EAAE,CAAC;YAClB,CAAC;YAID,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;gBAC9D,OAAO;YACT,CAAC;QACH,CAAC;IACH,CAAC;IAGO,KAAK,CAAC,OAAO,CAAC,IAAiB,EAAE,IAAiB;QACxD,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC;QAE1B,MAAM,EAAE,YAAY,EAAE,eAAe,EAAE,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC;YAC1E,QAAQ;YACR,YAAY,EAAE,IAAI,CAAC,OAAO;YAC1B,MAAM,EAAE,IAAI,CAAC,EAAE;SAChB,CAAC,CAAC;QACH,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;QAChF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC;YACnD,GAAG,EAAE,kBAAU,CAAC,YAAY;YAC5B,YAAY,EAAE,MAAM,EAAE,YAAY;YAClC,OAAO,EAAE,MAAM,EAAE,OAAO;SACzB,CAAC,CAAC;QAEH,IAAI,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QACrD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC;YAC9C,QAAQ;YACR,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,eAAe;YACf,OAAO,EAAE,IAAI;SACd,CAAC,CAAC;QACH,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QACnD,CAAC;QACD,MAAM,QAAQ,GAAG,IAAA,wCAAkB,EAAC,IAAI,EAAE,QAAQ,EAAE,CAAC,GAAG,EAAE,EAAE;YAC1D,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;gBACf,OAAO,EAAE,+DAA+D;gBACxE,QAAQ;gBACR,cAAc,EAAE,GAAG,CAAC,EAAE;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC;YAC1C,OAAO,EAAE;gBACP,EAAE,EAAE,IAAI,CAAC,OAAO;gBAChB,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,IAAI,EAAE,iBAAS,CAAC,KAAK;gBACrB,WAAW,EAAE,IAAI,CAAC,IAAI;aACvB;YACD,QAAQ,EAAE,YAAY;SACvB,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC;gBAC5C,IAAI;gBACJ,IAAI;gBACJ,QAAQ;gBACR,QAAQ;gBACR,KAAK,EAAE,GAAG,CAAC,KAAK;aACjB,CAAC,CAAC;YACH,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,eAAe,EAAE,CAAC;QAC9C,CAAC;gBAAS,CAAC;YACT,MAAM,GAAG,CAAC,KAAK,EAAE,CAAC;QACpB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,KAMxB;QACC,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;QACvC,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC;QAG1B,IAAI,SAAS,GAAkB,OAAO,CAAC,OAAO,EAAE,CAAC;QACjD,IAAI,eAAe,GAAkB,IAAI,CAAC;QAE1C,MAAM,MAAM,GAAG,IAAA,eAAU,EAAC;YACxB,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,YAAY,EAAE,IAAA,oDAA6B,EAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,CAAC;YACxE,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,eAAe,EAAE,QAAQ,CAAC,eAAe;YACzC,QAAQ,EAAE,IAAA,gBAAW,EAAC,8BAAsB,CAAC;YAE7C,OAAO,EAAE,GAAG,EAAE,CAAC,SAAS;YACxB,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE;gBAClB,MAAM,MAAM,GAAe;oBACzB,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ;oBAChC,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,KAAK,EAAE,IAAI,CAAC,KAAK;iBAClB,CAAC;gBACF,eAAe,GAAG,IAAI,CAAC,KAAK,EAAE,WAAW,IAAI,eAAe,CAAC;gBAC7D,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC;YACnF,CAAC;SACF,CAAC,CAAC;QAEH,IAAI,QAAQ,GAAG,IAAA,wBAAU,GAAE,CAAC;QAE5B,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YACvC,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;gBAClB,KAAK,YAAY;oBACf,QAAQ,GAAG,IAAA,wBAAU,GAAE,CAAC;oBACxB,MAAM;gBACR,KAAK,iBAAiB;oBACpB,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;oBACtD,MAAM;gBACR,KAAK,WAAW;oBACd,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,IAAI,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;oBAC1D,MAAM;gBACR,KAAK,YAAY;oBACf,IAAI,CAAC,YAAY,CAAC,IAAI,CACpB,8CAA6B,CAAC,QAAQ,EACtC,IAAI,8CAA6B,CAC/B,EAAE,EAAE,EAAE,IAAI,CAAC,OAAO,EAAE,EACpB,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,CAAC,IAAI,EAAE,CACzC,CACF,CAAC;oBACF,MAAM;gBACR,KAAK,OAAO;oBACV,MAAM,SAAS,CAAC;oBAChB,MAAM,IAAI,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;YACnF,CAAC;QACH,CAAC;QAED,MAAM,SAAS,CAAC;QAChB,OAAO,eAAe,CAAC;IACzB,CAAC;IAGO,KAAK,CAAC,WAAW,CACvB,IAAiB,EACjB,IAAiB,EACjB,QAA2B,EAC3B,IAAgB;QAGhB,MAAM,QAAQ,GAAmB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;QAE3E,MAAM,MAAM,GAAG,QAAQ,CAAC,GAAG,CACzB,CAAC,OAAO,EAAqB,EAAE,CAAC,CAAC;YAC/B,IAAI,EAAE,OAAO,CAAC,IAA8B;YAC5C,OAAO;YACP,MAAM,EAAE,OAAO;YACf,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,YAAY,EAAE,QAAQ,CAAC,YAAY;YACnC,OAAO,EAAE,QAAQ,CAAC,OAAO;YACzB,KAAK,EACH,OAAO,CAAC,IAAI,KAAK,WAAW;gBAC1B,CAAC,CAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,CAAoC;gBACpF,CAAC,CAAC,IAAI;SACX,CAAC,CACH,CAAC;QAEF,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAEhE,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QAC9B,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,2CAAsB,CAAC,SAAS,CAAC,CAAC;QACxF,IAAI,IAAI,IAAI,YAAY,EAAE,CAAC;YACzB,MAAM,IAAI,CAAC,kBAAkB,CAAC;gBAC5B,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,OAAO,EAAE,IAAI;gBACb,cAAc,EAAE,YAAY,CAAC,EAAE;aAChC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAAC,KAKhC;QACC,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC,IAAI,CACnD,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC;gBAClC,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,gBAAgB,EAAE,KAAK,CAAC,OAAO;gBAC/B,cAAc,EAAE,KAAK,CAAC,cAAc;aACrC,CAAC,CACH,CAAC;YAEF,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC,OAAO,CAAC;gBAC/D,KAAK,EAAE,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE;gBACvB,SAAS,EAAE,CAAC,gBAAgB,CAAC;aAC9B,CAAC,CAAC;YACH,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,OAAO;YACT,CAAC;YAED,IAAI,CAAC,YAAY,CAAC,IAAI,CACpB,oCAAmB,CAAC,QAAQ,EAC5B,IAAI,oCAAmB,CAAC,EAAE,EAAE,EAAE,KAAK,CAAC,OAAO,EAAE,EAAE,aAAa,CAAC,CAC9D,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBAChB,OAAO,EAAE,iCAAiC;gBAC1C,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,KAAK,EAAE,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC;aACjC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,YAAY,CAClB,QAAgB,EAChB,OAAe,EACf,IAA6B;QAE7B,IAAI,CAAC,YAAY,CAAC,IAAI,CACpB,yCAAwB,CAAC,QAAQ,EACjC,IAAI,yCAAwB,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAClE,CAAC;IACJ,CAAC;IAEO,aAAa,CAAC,KAAc;QAClC,OAAO,KAAK,YAAY,KAAK;YAC3B,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE;YAClE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACpB,CAAC;CACF,CAAA;AAzSY,gDAAkB;6BAAlB,kBAAkB;IAD9B,IAAA,mBAAU,GAAE;IAKR,WAAA,IAAA,0BAAgB,EAAC,4BAAY,CAAC,CAAA;IAE9B,WAAA,IAAA,0BAAgB,EAAC,2CAAmB,CAAC,CAAA;qCADH,oBAAU;QAEH,oBAAU;QACpB,8BAAa;QAClB,gDAAqB;QACxB,yCAAkB;QACP,qCAAgB;QACjB,8CAAoB;QAC9B,kDAAsB;QACf,6BAAa;GAdnC,kBAAkB,CAyS9B"}

package/dist/threads/agent/thread-agent.service.spec.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/threads/agent/thread-agent.service.spec.js

@@ -0,0 +1,184 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const ai_1 = require("ai");
+const test_1 = require("ai/test");
+const threads_events_1 = require("../events/threads.events");
+const thread_agent_service_1 = require("./thread-agent.service");
+const thread_agent_enums_1 = require("./thread-agent.enums");
+jest.mock('src/agents/agents.service', () => ({ AgentsService: class {
+    } }));
+jest.mock('./chat-model.factory', () => ({ ChatModelFactory: class {
+    } }));
+jest.mock('./taico-mcp-tools.factory', () => ({ TaicoMcpToolsFactory: class {
+    } }));
+jest.mock('./thread-agent-log.service', () => ({ ThreadAgentLogService: class {
+    } }));
+jest.mock('./hooks/thread-agent-hooks.runner', () => ({ ThreadAgentHooksRunner: class {
+    } }));
+const THREAD_ID = 'thread-1';
+const AGENT_ACTOR_ID = 'agent-actor';
+const usage = {
+    inputTokens: { total: 3, noCache: 3, cacheRead: undefined, cacheWrite: undefined },
+    outputTokens: { total: 5, text: 5, reasoning: undefined },
+};
+const finish = (unified) => ({
+    type: 'finish',
+    finishReason: { unified, raw: undefined },
+    usage,
+});
+const textStep = (text) => [
+    { type: 'text-start', id: 't' },
+    { type: 'text-delta', id: 't', delta: text },
+    { type: 'text-end', id: 't' },
+    finish('stop'),
+];
+const toolCallStep = [
+    { type: 'text-start', id: 't' },
+    { type: 'text-delta', id: 't', delta: 'Let me check.' },
+    { type: 'text-end', id: 't' },
+    { type: 'tool-call', toolCallId: 'c1', toolName: 'tasks__fetch', input: '{"id":"1"}' },
+    finish('tool-calls'),
+];
+function createHarness(steps) {
+    let call = 0;
+    const prompts = [];
+    const model = new test_1.MockLanguageModelV4({
+        doStream: async (options) => {
+            prompts.push(options.prompt);
+            const chunks = steps[call++];
+            if (!chunks) {
+                throw new Error('model unavailable');
+            }
+            return { stream: (0, ai_1.simulateReadableStream)({ chunks: chunks }) };
+        },
+    });
+    const appended = [];
+    const displayMessages = [];
+    const emitted = [];
+    const finished = [];
+    const agentLog = {
+        ingestPending: jest.fn().mockResolvedValue({
+            triggerActor: { id: 'user-1', slug: 'fran', type: 'human', displayName: 'Fran' },
+            newUserMessages: ['hello'],
+        }),
+        hasPending: jest.fn().mockResolvedValue(false),
+        loadContext: jest.fn().mockResolvedValue([
+            {
+                id: 'm0',
+                source: 'user',
+                message: { role: 'user', content: '[Fran @fran] says:\nhello' },
+                formatVersion: 1,
+                providerType: null,
+                modelId: null,
+            },
+        ]),
+        append: jest.fn(async (_threadId, drafts) => drafts.map((draft) => {
+            const row = { id: `agent-msg-${appended.length + 1}`, ...draft };
+            appended.push(row);
+            return row;
+        })),
+    };
+    const turns = {
+        tryClaim: jest.fn().mockResolvedValue({ id: 'turn-1', threadId: THREAD_ID }),
+        startHeartbeat: jest.fn().mockReturnValue(() => undefined),
+        finish: jest.fn(async (_id, status, error) => {
+            finished.push({ status, error });
+        }),
+    };
+    const threadMessageRepository = {
+        create: (input) => input,
+        save: jest.fn(async (input) => {
+            const saved = { id: `display-${displayMessages.length + 1}`, ...input };
+            displayMessages.push(saved);
+            return saved;
+        }),
+        findOne: jest.fn(async ({ where }) => displayMessages.find((message) => message.id === where.id)),
+    };
+    const mcpClose = jest.fn().mockResolvedValue(undefined);
+    const hooks = {
+        runBeforeTurn: jest.fn().mockResolvedValue(false),
+        runAfterTurn: jest.fn().mockResolvedValue(undefined),
+    };
+    const service = new thread_agent_service_1.ThreadAgentService({ findOne: jest.fn().mockResolvedValue({ id: THREAD_ID, aiProviderId: null, modelId: null }) }, threadMessageRepository, {
+        getAgentBySlug: jest.fn().mockResolvedValue({
+            actorId: AGENT_ACTOR_ID,
+            slug: 'taico',
+            name: 'Taico',
+            systemPrompt: 'You are Taico.',
+            modelId: 'mock-model-id',
+        }),
+    }, agentLog, turns, {
+        resolve: jest.fn().mockResolvedValue({ model, providerType: 'mock', modelId: 'mock-model-id' }),
+    }, {
+        open: jest.fn().mockResolvedValue({
+            tools: {
+                tasks__fetch: {
+                    inputSchema: (0, ai_1.jsonSchema)({ type: 'object', properties: { id: { type: 'string' } } }),
+                    execute: async () => ({ name: 'Ship it' }),
+                },
+            },
+            close: mcpClose,
+        }),
+    }, hooks, { emit: (event, payload) => emitted.push({ event, payload }) });
+    return { hooks, service, agentLog, turns, appended, displayMessages, emitted, finished, prompts, mcpClose };
+}
+describe('ThreadAgentService', () => {
+    it('persists every step and links each visible message to its agent message', async () => {
+        const h = createHarness([toolCallStep, textStep('All done.')]);
+        await h.service.drain(THREAD_ID);
+        expect(h.appended.map((row) => row.role)).toEqual([
+            thread_agent_enums_1.ThreadAgentMessageRole.ASSISTANT,
+            thread_agent_enums_1.ThreadAgentMessageRole.TOOL,
+            thread_agent_enums_1.ThreadAgentMessageRole.ASSISTANT,
+        ]);
+        expect(h.appended.every((row) => row.providerType === 'mock')).toBe(true);
+        expect(h.displayMessages).toEqual([
+            expect.objectContaining({ content: 'Let me check.', agentMessageId: 'agent-msg-1' }),
+            expect.objectContaining({ content: 'All done.', agentMessageId: 'agent-msg-3' }),
+        ]);
+        expect(h.finished).toEqual([{ status: thread_agent_enums_1.ThreadTurnStatus.COMPLETED, error: undefined }]);
+        expect(h.mcpClose).toHaveBeenCalledTimes(1);
+        expect(h.hooks.runBeforeTurn).toHaveBeenCalledWith(expect.objectContaining({ threadId: THREAD_ID, newUserMessages: ['hello'] }));
+        expect(h.hooks.runAfterTurn).toHaveBeenCalledWith(expect.objectContaining({ threadId: THREAD_ID, turnId: 'turn-1', lastInputTokens: 3 }));
+        const deltas = h.emitted.filter((e) => e.event === threads_events_1.ThreadAgentResponseDeltaEvent.INTERNAL);
+        expect(deltas).toHaveLength(2);
+        expect(h.emitted.filter((e) => e.event === threads_events_1.MessageCreatedEvent.INTERNAL)).toHaveLength(2);
+    });
+    it('sends the stored conversation to the model', async () => {
+        const h = createHarness([textStep('Hi.')]);
+        await h.service.drain(THREAD_ID);
+        expect(h.prompts[0]).toEqual([
+            expect.objectContaining({ role: 'system' }),
+            { role: 'user', content: [{ type: 'text', text: '[Fran @fran] says:\nhello' }] },
+        ]);
+    });
+    it('does nothing when another turn is running on the thread', async () => {
+        const h = createHarness([textStep('Hi.')]);
+        h.turns.tryClaim.mockResolvedValue(null);
+        await h.service.drain(THREAD_ID);
+        expect(h.agentLog.ingestPending).not.toHaveBeenCalled();
+        expect(h.displayMessages).toEqual([]);
+    });
+    it('runs another turn when a message arrived while the first one was running', async () => {
+        const h = createHarness([textStep('First.'), textStep('Second.')]);
+        h.agentLog.hasPending.mockResolvedValueOnce(true).mockResolvedValue(false);
+        await h.service.drain(THREAD_ID);
+        expect(h.turns.tryClaim).toHaveBeenCalledTimes(2);
+        expect(h.displayMessages.map((m) => m.content)).toEqual(['First.', 'Second.']);
+    });
+    it('fails the turn and tells the user when the model call fails', async () => {
+        const h = createHarness([]);
+        await h.service.drain(THREAD_ID);
+        expect(h.finished).toEqual([
+            { status: thread_agent_enums_1.ThreadTurnStatus.FAILED, error: expect.stringContaining('model unavailable') },
+        ]);
+        expect(h.displayMessages).toEqual([
+            expect.objectContaining({
+                content: expect.stringContaining('I encountered an error'),
+                agentMessageId: null,
+            }),
+        ]);
+        expect(h.mcpClose).toHaveBeenCalledTimes(1);
+    });
+});
+//# sourceMappingURL=thread-agent.service.spec.js.map

package/dist/threads/agent/thread-agent.service.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"thread-agent.service.spec.js","sourceRoot":"","sources":["../../../src/threads/agent/thread-agent.service.spec.ts"],"names":[],"mappings":";;AAAA,2BAAwD;AACxD,kCAA8C;AAC9C,6DAGkC;AAClC,iEAA4D;AAC5D,6DAAgF;AAIhF,IAAI,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,aAAa,EAAE;KAAQ,EAAE,CAAC,CAAC,CAAC;AAC5E,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,gBAAgB,EAAE;KAAQ,EAAE,CAAC,CAAC,CAAC;AAC1E,IAAI,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,oBAAoB,EAAE;KAAQ,EAAE,CAAC,CAAC,CAAC;AACnF,IAAI,CAAC,IAAI,CAAC,4BAA4B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,qBAAqB,EAAE;KAAQ,EAAE,CAAC,CAAC,CAAC;AACrF,IAAI,CAAC,IAAI,CAAC,mCAAmC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,sBAAsB,EAAE;KAAQ,EAAE,CAAC,CAAC,CAAC;AAE7F,MAAM,SAAS,GAAG,UAAU,CAAC;AAC7B,MAAM,cAAc,GAAG,aAAa,CAAC;AAErC,MAAM,KAAK,GAAG;IACZ,WAAW,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,UAAU,EAAE,SAAS,EAAE;IAClF,YAAY,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE;CAC1D,CAAC;AAEF,MAAM,MAAM,GAAG,CAAC,OAA8B,EAAE,EAAE,CAAC,CAAC;IAClD,IAAI,EAAE,QAAiB;IACvB,YAAY,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,SAAS,EAAE;IACzC,KAAK;CACN,CAAC,CAAC;AAEH,MAAM,QAAQ,GAAG,CAAC,IAAY,EAAE,EAAE,CAAC;IACjC,EAAE,IAAI,EAAE,YAAqB,EAAE,EAAE,EAAE,GAAG,EAAE;IACxC,EAAE,IAAI,EAAE,YAAqB,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE;IACrD,EAAE,IAAI,EAAE,UAAmB,EAAE,EAAE,EAAE,GAAG,EAAE;IACtC,MAAM,CAAC,MAAM,CAAC;CACf,CAAC;AAEF,MAAM,YAAY,GAAG;IACnB,EAAE,IAAI,EAAE,YAAqB,EAAE,EAAE,EAAE,GAAG,EAAE;IACxC,EAAE,IAAI,EAAE,YAAqB,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,eAAe,EAAE;IAChE,EAAE,IAAI,EAAE,UAAmB,EAAE,EAAE,EAAE,GAAG,EAAE;IACtC,EAAE,IAAI,EAAE,WAAoB,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE,cAAc,EAAE,KAAK,EAAE,YAAY,EAAE;IAC/F,MAAM,CAAC,YAAY,CAAC;CACrB,CAAC;AAEF,SAAS,aAAa,CAAC,KAAkB;IACvC,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,MAAM,OAAO,GAAc,EAAE,CAAC;IAC9B,MAAM,KAAK,GAAG,IAAI,0BAAmB,CAAC;QACpC,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YAC1B,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAC7B,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;YAC7B,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;YACvC,CAAC;YACD,OAAO,EAAE,MAAM,EAAE,IAAA,2BAAsB,EAAC,EAAE,MAAM,EAAE,MAAiB,EAAE,CAAC,EAAE,CAAC;QAC3E,CAAC;KACF,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAiF,EAAE,CAAC;IAClG,MAAM,eAAe,GAA0E,EAAE,CAAC;IAClG,MAAM,OAAO,GAA+C,EAAE,CAAC;IAC/D,MAAM,QAAQ,GAAwD,EAAE,CAAC;IAEzE,MAAM,QAAQ,GAAG;QACf,aAAa,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YACzC,YAAY,EAAE,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE;YAChF,eAAe,EAAE,CAAC,OAAO,CAAC;SAC3B,CAAC;QACF,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,KAAK,CAAC;QAC9C,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YACvC;gBACE,EAAE,EAAE,IAAI;gBACR,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,2BAA2B,EAAE;gBAC/D,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,IAAI;gBAClB,OAAO,EAAE,IAAI;aACd;SACF,CAAC;QACF,MAAM,EAAE,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,SAAiB,EAAE,MAAsC,EAAE,EAAE,CAClF,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YACnB,MAAM,GAAG,GAAG,EAAE,EAAE,EAAE,aAAa,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,EAAE,GAAG,KAAK,EAA+B,CAAC;YAC9F,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACnB,OAAO,GAAG,CAAC;QACb,CAAC,CAAC,CACH;KACF,CAAC;IAEF,MAAM,KAAK,GAAG;QACZ,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC;QAC5E,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;QAC1D,MAAM,EAAE,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,GAAW,EAAE,MAAwB,EAAE,KAAc,EAAE,EAAE;YAC9E,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QACnC,CAAC,CAAC;KACH,CAAC;IAEF,MAAM,uBAAuB,GAAG;QAC9B,MAAM,EAAE,CAAC,KAA8B,EAAE,EAAE,CAAC,KAAK;QACjD,IAAI,EAAE,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,KAAyD,EAAE,EAAE;YAChF,MAAM,KAAK,GAAG,EAAE,EAAE,EAAE,WAAW,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,EAAE,GAAG,KAAK,EAAE,CAAC;YACxE,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC5B,OAAO,KAAK,CAAC;QACf,CAAC,CAAC;QACF,OAAO,EAAE,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,EAAE,KAAK,EAA6B,EAAE,EAAE,CAC9D,eAAe,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,KAAK,KAAK,CAAC,EAAE,CAAC,CAC3D;KACF,CAAC;IAEF,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;IACxD,MAAM,KAAK,GAAG;QACZ,aAAa,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,KAAK,CAAC;QACjD,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;KACrD,CAAC;IAEF,MAAM,OAAO,GAAG,IAAI,yCAAkB,CACpC,EAAE,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,YAAY,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,EAAW,EACvG,uBAAgC,EAChC;QACE,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YAC1C,OAAO,EAAE,cAAc;YACvB,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,OAAO;YACb,YAAY,EAAE,gBAAgB;YAC9B,OAAO,EAAE,eAAe;SACzB,CAAC;KACM,EACV,QAAiB,EACjB,KAAc,EACd;QACE,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC;KACvF,EACV;QACE,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YAChC,KAAK,EAAE;gBACL,YAAY,EAAE;oBACZ,WAAW,EAAE,IAAA,eAAU,EAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;oBACnF,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;iBAC3C;aACF;YACD,KAAK,EAAE,QAAQ;SAChB,CAAC;KACM,EACV,KAAc,EACd,EAAE,IAAI,EAAE,CAAC,KAAa,EAAE,OAAgB,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,EAAW,CACzF,CAAC;IAEF,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,eAAe,EAAE,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;AAC9G,CAAC;AAED,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACvF,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,YAAY,EAAE,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;QAE/D,MAAM,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAEjC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;YAChD,2CAAsB,CAAC,SAAS;YAChC,2CAAsB,CAAC,IAAI;YAC3B,2CAAsB,CAAC,SAAS;SACjC,CAAC,CAAC;QACH,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,YAAY,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1E,MAAM,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC;YAChC,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,eAAe,EAAE,cAAc,EAAE,aAAa,EAAE,CAAC;YACpF,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,aAAa,EAAE,CAAC;SACjF,CAAC,CAAC;QACH,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,EAAE,qCAAgB,CAAC,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;QACvF,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC5C,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAChD,MAAM,CAAC,gBAAgB,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,eAAe,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAC7E,CAAC;QACF,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,gBAAgB,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,CAAC,EAAE,CAAC,CACvF,CAAC;QAEF,MAAM,MAAM,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,8CAA6B,CAAC,QAAQ,CAAC,CAAC;QAC3F,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CACJ,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,oCAAmB,CAAC,QAAQ,CAAC,CAClE,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACpB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAE3C,MAAM,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAEjC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAC3B,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;YAC3C,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,2BAA2B,EAAE,CAAC,EAAE;SACjF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC3C,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAEzC,MAAM,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAEjC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACxD,MAAM,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACxF,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QACnE,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAE3E,MAAM,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAEjC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAClD,MAAM,CAAC,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC;IACjF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,CAAC,GAAG,aAAa,CAAC,EAAE,CAAC,CAAC;QAE5B,MAAM,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAEjC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACzB,EAAE,MAAM,EAAE,qCAAgB,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,EAAE;SACzF,CAAC,CAAC;QACH,MAAM,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC;YAChC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC,wBAAwB,CAAC;gBAC1D,cAAc,EAAE,IAAI;aACrB,CAAC;SACH,CAAC,CAAC;QACH,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/threads/agent/thread-turn.entity.d.ts

@@ -0,0 +1,12 @@
+import { ThreadEntity } from '../thread.entity';
+import { ThreadTurnStatus } from './thread-agent.enums';
+export declare class ThreadTurnEntity {
+    id: string;
+    threadId: string;
+    thread?: ThreadEntity;
+    status: ThreadTurnStatus;
+    error: string | null;
+    startedAt: Date;
+    heartbeatAt: Date;
+    finishedAt: Date | null;
+}

package/dist/threads/agent/thread-turn.entity.js

@@ -0,0 +1,71 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    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;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ThreadTurnEntity = void 0;
+const typeorm_1 = require("typeorm");
+const thread_entity_1 = require("../thread.entity");
+const thread_agent_enums_1 = require("./thread-agent.enums");
+let ThreadTurnEntity = class ThreadTurnEntity {
+    id;
+    threadId;
+    thread;
+    status;
+    error;
+    startedAt;
+    heartbeatAt;
+    finishedAt;
+};
+exports.ThreadTurnEntity = ThreadTurnEntity;
+__decorate([
+    (0, typeorm_1.PrimaryGeneratedColumn)('uuid'),
+    __metadata("design:type", String)
+], ThreadTurnEntity.prototype, "id", void 0);
+__decorate([
+    (0, typeorm_1.Column)({ type: 'uuid', nullable: false, name: 'thread_id' }),
+    __metadata("design:type", String)
+], ThreadTurnEntity.prototype, "threadId", void 0);
+__decorate([
+    (0, typeorm_1.ManyToOne)(() => thread_entity_1.ThreadEntity, { onDelete: 'CASCADE' }),
+    (0, typeorm_1.JoinColumn)({ name: 'thread_id' }),
+    __metadata("design:type", thread_entity_1.ThreadEntity)
+], ThreadTurnEntity.prototype, "thread", void 0);
+__decorate([
+    (0, typeorm_1.Column)({
+        type: 'text',
+        enum: thread_agent_enums_1.ThreadTurnStatus,
+        default: thread_agent_enums_1.ThreadTurnStatus.RUNNING,
+    }),
+    __metadata("design:type", String)
+], ThreadTurnEntity.prototype, "status", void 0);
+__decorate([
+    (0, typeorm_1.Column)({ type: 'text', nullable: true }),
+    __metadata("design:type", Object)
+], ThreadTurnEntity.prototype, "error", void 0);
+__decorate([
+    (0, typeorm_1.CreateDateColumn)({ type: 'timestamptz', name: 'started_at' }),
+    __metadata("design:type", Date)
+], ThreadTurnEntity.prototype, "startedAt", void 0);
+__decorate([
+    (0, typeorm_1.Column)({ type: 'timestamptz', name: 'heartbeat_at', default: () => 'now()' }),
+    __metadata("design:type", Date)
+], ThreadTurnEntity.prototype, "heartbeatAt", void 0);
+__decorate([
+    (0, typeorm_1.Column)({ type: 'timestamptz', name: 'finished_at', nullable: true }),
+    __metadata("design:type", Object)
+], ThreadTurnEntity.prototype, "finishedAt", void 0);
+exports.ThreadTurnEntity = ThreadTurnEntity = __decorate([
+    (0, typeorm_1.Entity)({ name: 'thread_turns' }),
+    (0, typeorm_1.Index)('UQ_thread_turns_running_thread', ['threadId'], {
+        unique: true,
+        where: `"status" = 'running'`,
+    })
+], ThreadTurnEntity);
+//# sourceMappingURL=thread-turn.entity.js.map

package/dist/threads/agent/thread-turn.entity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"thread-turn.entity.js","sourceRoot":"","sources":["../../../src/threads/agent/thread-turn.entity.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,qCAQiB;AACjB,oDAAgD;AAChD,6DAAwD;AAWjD,IAAM,gBAAgB,GAAtB,MAAM,gBAAgB;IAE3B,EAAE,CAAU;IAGZ,QAAQ,CAAU;IAIlB,MAAM,CAAgB;IAOtB,MAAM,CAAoB;IAG1B,KAAK,CAAiB;IAGtB,SAAS,CAAQ;IAGjB,WAAW,CAAQ;IAGnB,UAAU,CAAe;CAC1B,CAAA;AA7BY,4CAAgB;AAE3B;IADC,IAAA,gCAAsB,EAAC,MAAM,CAAC;;4CACnB;AAGZ;IADC,IAAA,gBAAM,EAAC,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;;kDAC3C;AAIlB;IAFC,IAAA,mBAAS,EAAC,GAAG,EAAE,CAAC,4BAAY,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC;IACtD,IAAA,oBAAU,EAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;8BACzB,4BAAY;gDAAC;AAOtB;IALC,IAAA,gBAAM,EAAC;QACN,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,qCAAgB;QACtB,OAAO,EAAE,qCAAgB,CAAC,OAAO;KAClC,CAAC;;gDACwB;AAG1B;IADC,IAAA,gBAAM,EAAC,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;+CACnB;AAGtB;IADC,IAAA,0BAAgB,EAAC,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;8BAClD,IAAI;mDAAC;AAGjB;IADC,IAAA,gBAAM,EAAC,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;8BAChE,IAAI;qDAAC;AAGnB;IADC,IAAA,gBAAM,EAAC,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;oDAC5C;2BA5Bd,gBAAgB;IAL5B,IAAA,gBAAM,EAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;IAChC,IAAA,eAAK,EAAC,gCAAgC,EAAE,CAAC,UAAU,CAAC,EAAE;QACrD,MAAM,EAAE,IAAI;QACZ,KAAK,EAAE,sBAAsB;KAC9B,CAAC;GACW,gBAAgB,CA6B5B"}

package/dist/threads/agent/thread-turns.service.d.ts

@@ -0,0 +1,13 @@
+import { DataSource } from 'typeorm';
+import { ThreadTurnStatus } from './thread-agent.enums';
+export type ClaimedTurn = {
+    id: string;
+    threadId: string;
+};
+export declare class ThreadTurnsService {
+    private readonly dataSource;
+    constructor(dataSource: DataSource);
+    tryClaim(threadId: string): Promise<ClaimedTurn | null>;
+    startHeartbeat(turnId: string): () => void;
+    finish(turnId: string, status: ThreadTurnStatus.COMPLETED | ThreadTurnStatus.FAILED, error?: string): Promise<void>;
+}