@taico/taico 0.4.1 → 0.5.0

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>;
+}

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

@@ -0,0 +1,52 @@
+"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.ThreadTurnsService = void 0;
+const common_1 = require("@nestjs/common");
+const typeorm_1 = require("typeorm");
+const thread_agent_enums_1 = require("./thread-agent.enums");
+const HEARTBEAT_INTERVAL_MS = 10_000;
+const STALE_AFTER_SECONDS = 60;
+let ThreadTurnsService = class ThreadTurnsService {
+    dataSource;
+    constructor(dataSource) {
+        this.dataSource = dataSource;
+    }
+    async tryClaim(threadId) {
+        await this.dataSource.query(`UPDATE thread_turns
+          SET status = $2, error = 'Turn stopped heartbeating', finished_at = now()
+        WHERE thread_id = $1 AND status = $3
+          AND heartbeat_at < now() - make_interval(secs => $4)`, [threadId, thread_agent_enums_1.ThreadTurnStatus.FAILED, thread_agent_enums_1.ThreadTurnStatus.RUNNING, STALE_AFTER_SECONDS]);
+        const rows = await this.dataSource.query(`INSERT INTO thread_turns (thread_id, status) VALUES ($1, $2)
+       ON CONFLICT (thread_id) WHERE status = 'running' DO NOTHING
+       RETURNING id`, [threadId, thread_agent_enums_1.ThreadTurnStatus.RUNNING]);
+        return rows.length > 0 ? { id: rows[0].id, threadId } : null;
+    }
+    startHeartbeat(turnId) {
+        const timer = setInterval(() => {
+            void this.dataSource
+                .query(`UPDATE thread_turns SET heartbeat_at = now() WHERE id = $1`, [turnId])
+                .catch(() => undefined);
+        }, HEARTBEAT_INTERVAL_MS);
+        timer.unref();
+        return () => clearInterval(timer);
+    }
+    async finish(turnId, status, error) {
+        await this.dataSource.query(`UPDATE thread_turns SET status = $2, error = $3, finished_at = now()
+        WHERE id = $1 AND status = $4`, [turnId, status, error ?? null, thread_agent_enums_1.ThreadTurnStatus.RUNNING]);
+    }
+};
+exports.ThreadTurnsService = ThreadTurnsService;
+exports.ThreadTurnsService = ThreadTurnsService = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [typeorm_1.DataSource])
+], ThreadTurnsService);
+//# sourceMappingURL=thread-turns.service.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"thread-turns.service.js","sourceRoot":"","sources":["../../../src/threads/agent/thread-turns.service.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,2CAA4C;AAC5C,qCAAqC;AACrC,6DAAwD;AAExD,MAAM,qBAAqB,GAAG,MAAM,CAAC;AACrC,MAAM,mBAAmB,GAAG,EAAE,CAAC;AAaxB,IAAM,kBAAkB,GAAxB,MAAM,kBAAkB;IACA;IAA7B,YAA6B,UAAsB;QAAtB,eAAU,GAAV,UAAU,CAAY;IAAG,CAAC;IAGvD,KAAK,CAAC,QAAQ,CAAC,QAAgB;QAC7B,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CACzB;;;+DAGyD,EACzD,CAAC,QAAQ,EAAE,qCAAgB,CAAC,MAAM,EAAE,qCAAgB,CAAC,OAAO,EAAE,mBAAmB,CAAC,CACnF,CAAC;QAEF,MAAM,IAAI,GAA0B,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAC7D;;oBAEc,EACd,CAAC,QAAQ,EAAE,qCAAgB,CAAC,OAAO,CAAC,CACrC,CAAC;QAEF,OAAO,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;IAC/D,CAAC;IAGD,cAAc,CAAC,MAAc;QAC3B,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE;YAC7B,KAAK,IAAI,CAAC,UAAU;iBACjB,KAAK,CAAC,4DAA4D,EAAE,CAAC,MAAM,CAAC,CAAC;iBAC7E,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC5B,CAAC,EAAE,qBAAqB,CAAC,CAAC;QAC1B,KAAK,CAAC,KAAK,EAAE,CAAC;QAEd,OAAO,GAAG,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,MAAM,CACV,MAAc,EACd,MAA4D,EAC5D,KAAc;QAEd,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CACzB;sCACgC,EAChC,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,IAAI,IAAI,EAAE,qCAAgB,CAAC,OAAO,CAAC,CAC1D,CAAC;IACJ,CAAC;CACF,CAAA;AA9CY,gDAAkB;6BAAlB,kBAAkB;IAD9B,IAAA,mBAAU,GAAE;qCAE8B,oBAAU;GADxC,kBAAkB,CA8C9B"}

package/dist/threads/chat.service.d.ts

@@ -1,43 +1,18 @@
-import { IssuedAccessTokenService } from "../authorization-server/issued-access-token.service";
 import { AgentsService } from "../agents/agents.service";
 import { ActorEntity } from "../identity-provider/actor.entity";
-import { ChatProvidersService } from "../chat-providers/chat-providers.service";
-import { AdkBackend } from './backends/adk.backend';
-import { OpenAiBackend } from './backends/openai.backend';
-import { ChatStreamEvent } from './backends/chat-backend.interface';
-export type { ChatStreamEvent };
-export interface CreateConversationArgs {
-    threadId: string;
-}
-export interface StreamMessageToConversationArgs {
-    conversationId: string;
-    threadId: string;
-    message: string;
-    actor: ActorEntity;
-}
-export interface StreamMessageResult {
-    agentActorId: string;
-    events: AsyncIterable<ChatStreamEvent>;
-}
+import { ChatModelFactory } from './agent/chat-model.factory';
+import { TaicoMcpToolsFactory } from './agent/taico-mcp-tools.factory';
 export declare class ChatService {
     private readonly agentsService;
-    private readonly issuedAccessTokenService;
-    private readonly chatProvidersService;
-    private readonly adkBackend;
-    private readonly openAiBackend;
+    private readonly chatModelFactory;
+    private readonly mcpToolsFactory;
     private readonly logger;
-    constructor(agentsService: AgentsService, issuedAccessTokenService: IssuedAccessTokenService, chatProvidersService: ChatProvidersService, adkBackend: AdkBackend, openAiBackend: OpenAiBackend);
+    constructor(agentsService: AgentsService, chatModelFactory: ChatModelFactory, mcpToolsFactory: TaicoMcpToolsFactory);
     private getSelf;
-    private getActiveBackend;
-    ensureConversationAvailable(): Promise<void>;
-    createConversation({ threadId }: CreateConversationArgs): Promise<{
-        id: string;
-    }>;
     generateText(prompt: string): Promise<string | null>;
     runTask(args: {
         instructions: string;
         prompt: string;
         actor: ActorEntity | null;
     }): Promise<void>;
-    streamMessageToConversation(args: StreamMessageToConversationArgs): Promise<StreamMessageResult>;
 }

package/dist/threads/chat.service.js

@@ -12,89 +12,59 @@ var ChatService_1;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ChatService = void 0;
 const common_1 = require("@nestjs/common");
-const issued_access_token_service_1 = require("../authorization-server/issued-access-token.service");
+const ai_1 = require("ai");
 const agents_service_1 = require("../agents/agents.service");
 const enums_1 = require("../identity-provider/enums");
-const mcp_scopes_1 = require("../auth/core/scopes/mcp.scopes");
-const tasks_scopes_1 = require("../tasks/tasks.scopes");
-const context_scopes_1 = require("../context/context.scopes");
-const chat_providers_service_1 = require("../chat-providers/chat-providers.service");
-const chat_providers_errors_1 = require("../chat-providers/errors/chat-providers.errors");
-const enums_2 = require("../chat-providers/enums");
-const adk_backend_1 = require("./backends/adk.backend");
-const openai_backend_1 = require("./backends/openai.backend");
+const enums_2 = require("../ai-providers/enums");
+const ai_providers_errors_1 = require("../ai-providers/errors/ai-providers.errors");
+const chat_model_factory_1 = require("./agent/chat-model.factory");
+const taico_mcp_tools_factory_1 = require("./agent/taico-mcp-tools.factory");
+const thread_agent_service_1 = require("./agent/thread-agent.service");
 let ChatService = ChatService_1 = class ChatService {
     agentsService;
-    issuedAccessTokenService;
-    chatProvidersService;
-    adkBackend;
-    openAiBackend;
+    chatModelFactory;
+    mcpToolsFactory;
     logger = new common_1.Logger(ChatService_1.name);
-    constructor(agentsService, issuedAccessTokenService, chatProvidersService, adkBackend, openAiBackend) {
+    constructor(agentsService, chatModelFactory, mcpToolsFactory) {
         this.agentsService = agentsService;
-        this.issuedAccessTokenService = issuedAccessTokenService;
-        this.chatProvidersService = chatProvidersService;
-        this.adkBackend = adkBackend;
-        this.openAiBackend = openAiBackend;
+        this.chatModelFactory = chatModelFactory;
+        this.mcpToolsFactory = mcpToolsFactory;
     }
     async getSelf() {
-        return this.agentsService.getAgentBySlug({ slug: 'taico' });
-    }
-    async getActiveBackend() {
-        const config = await this.chatProvidersService.getActiveChatProviderConfig();
-        return config.type === enums_2.ChatProviderType.ADK ? this.adkBackend : this.openAiBackend;
-    }
-    async ensureConversationAvailable() {
-        try {
-            const backend = await this.getActiveBackend();
-            await backend.ensureAvailable();
-        }
-        catch (error) {
-            if (error instanceof chat_providers_errors_1.NoActiveChatProviderError) {
-                return;
-            }
-            throw error;
-        }
-    }
-    async createConversation({ threadId }) {
-        const backend = await this.getActiveBackend();
-        try {
-            return await backend.createConversation(threadId);
-        }
-        catch (error) {
-            this.logger.error({
-                message: 'Failed to create conversation',
-                threadId,
-                error: error instanceof Error
-                    ? { message: error.message, stack: error.stack, name: error.name }
-                    : String(error),
-            });
-            throw error;
-        }
+        return this.agentsService.getAgentBySlug({ slug: thread_agent_service_1.THREAD_AGENT_SLUG });
     }
     async generateText(prompt) {
         try {
-            const self = await this.getSelf();
-            const backend = await this.getActiveBackend();
-            return backend.generateText(prompt, self.modelId);
+            const resolved = await this.chatModelFactory.resolve({ job: enums_2.AiModelJob.QUICK_TASKS });
+            const result = await (0, ai_1.generateText)({
+                model: resolved.model,
+                prompt,
+                providerOptions: resolved.providerOptions,
+            });
+            return result.text.trim() || null;
         }
         catch (error) {
-            if (error instanceof chat_providers_errors_1.NoActiveChatProviderError) {
-                return null;
+            if (!(error instanceof ai_providers_errors_1.AiModelNotSelectedError)) {
+                this.logger.warn({
+                    message: 'Text generation failed',
+                    error: error instanceof Error
+                        ? { message: error.message, name: error.name }
+                        : String(error),
+                });
             }
-            throw error;
+            return null;
         }
     }
     async runTask(args) {
         const self = await this.getSelf();
-        const backend = await this.getActiveBackend();
+        const resolved = await this.chatModelFactory.resolve({ job: enums_2.AiModelJob.MIDDLE_MANAGER });
         const selfActor = {
             id: self.actorId,
             slug: self.slug,
             type: enums_1.ActorType.AGENT,
             displayName: self.name,
         };
-        const issuedByActor = args.actor
+        const issuedBy = args.actor
             ? {
                 id: args.actor.id,
                 slug: args.actor.slug,
@@ -102,72 +72,27 @@ let ChatService = ChatService_1 = class ChatService {
                 displayName: args.actor.displayName,
             }
             : selfActor;
-        const token = await this.issuedAccessTokenService.issueSystemToken({
-            subjectActor: selfActor,
-            issuedByActor,
-            scopes: [
-                mcp_scopes_1.McpScopes.USE.id,
-                ...tasks_scopes_1.ALL_TASKS_SCOPES.map((s) => s.id),
-                ...context_scopes_1.ALL_CONTEXT_SCOPES.map((s) => s.id),
-            ],
-        });
-        await backend.runTask({
-            instructions: args.instructions,
-            prompt: args.prompt,
-            token: token.token,
-            modelId: self.modelId,
-        });
-    }
-    async streamMessageToConversation(args) {
-        const self = await this.getSelf();
-        if (args.actor.id === self.actorId) {
-            this.logger.debug(`Actor ${args.actor.id} is the same as the agent itself. Skipping to avoid loops.`);
-            return {
-                agentActorId: self.actorId,
-                events: (async function* () { })(),
-            };
+        const mcp = await this.mcpToolsFactory.open({ subject: selfActor, issuedBy });
+        try {
+            await (0, ai_1.generateText)({
+                model: resolved.model,
+                instructions: args.instructions,
+                prompt: args.prompt,
+                tools: mcp.tools,
+                providerOptions: resolved.providerOptions,
+                stopWhen: (0, ai_1.isStepCount)(thread_agent_service_1.THREAD_AGENT_MAX_STEPS),
+            });
+        }
+        finally {
+            await mcp.close();
         }
-        const backend = await this.getActiveBackend();
-        const token = await this.issuedAccessTokenService.issueSystemToken({
-            subjectActor: {
-                id: self.actorId,
-                slug: self.slug,
-                type: enums_1.ActorType.AGENT,
-                displayName: self.name,
-            },
-            issuedByActor: {
-                id: args.actor.id,
-                slug: args.actor.slug,
-                type: args.actor.type,
-                displayName: args.actor.displayName,
-            },
-            scopes: [
-                mcp_scopes_1.McpScopes.USE.id,
-                ...tasks_scopes_1.ALL_TASKS_SCOPES.map((s) => s.id),
-                ...context_scopes_1.ALL_CONTEXT_SCOPES.map((s) => s.id),
-            ],
-        });
-        this.logger.debug(`Issued ephemeral access token for actor ${args.actor.id} with jti ${token.jti}`);
-        const events = backend.streamMessage({
-            conversationId: args.conversationId,
-            threadId: args.threadId,
-            message: args.message,
-            actor: args.actor,
-            agentName: self.name,
-            systemPrompt: self.systemPrompt,
-            modelId: self.modelId,
-            token: token.token,
-        });
-        return { agentActorId: self.actorId, events };
     }
 };
 exports.ChatService = ChatService;
 exports.ChatService = ChatService = ChatService_1 = __decorate([
     (0, common_1.Injectable)(),
     __metadata("design:paramtypes", [agents_service_1.AgentsService,
-        issued_access_token_service_1.IssuedAccessTokenService,
-        chat_providers_service_1.ChatProvidersService,
-        adk_backend_1.AdkBackend,
-        openai_backend_1.OpenAiBackend])
+        chat_model_factory_1.ChatModelFactory,
+        taico_mcp_tools_factory_1.TaicoMcpToolsFactory])
 ], ChatService);
 //# sourceMappingURL=chat.service.js.map

package/dist/threads/chat.service.js.map

@@ -1 +1 @@
-{"version":3,"file":"chat.service.js","sourceRoot":"","sources":["../../src/threads/chat.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;AAAA,2CAAoD;AACpD,qGAAgG;AAChG,6DAA0D;AAG1D,sDAAwD;AACxD,+DAA4D;AAC5D,wDAA0D;AAC1D,8DAAgE;AAChE,qFAAiF;AACjF,0FAA4F;AAC5F,mDAA4D;AAC5D,wDAAoD;AACpD,8DAA0D;AAsBnD,IAAM,WAAW,mBAAjB,MAAM,WAAW;IAIH;IACA;IACA;IACA;IACA;IAPF,MAAM,GAAG,IAAI,eAAM,CAAC,aAAW,CAAC,IAAI,CAAC,CAAC;IAEvD,YACmB,aAA4B,EAC5B,wBAAkD,EAClD,oBAA0C,EAC1C,UAAsB,EACtB,aAA4B;QAJ5B,kBAAa,GAAb,aAAa,CAAe;QAC5B,6BAAwB,GAAxB,wBAAwB,CAA0B;QAClD,yBAAoB,GAApB,oBAAoB,CAAsB;QAC1C,eAAU,GAAV,UAAU,CAAY;QACtB,kBAAa,GAAb,aAAa,CAAe;IAC5C,CAAC;IAEI,KAAK,CAAC,OAAO;QACnB,OAAO,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IAC9D,CAAC;IAEO,KAAK,CAAC,gBAAgB;QAC5B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,2BAA2B,EAAE,CAAC;QAC7E,OAAO,MAAM,CAAC,IAAI,KAAK,wBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC;IACrF,CAAC;IAEM,KAAK,CAAC,2BAA2B;QACtC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC9C,MAAM,OAAO,CAAC,eAAe,EAAE,CAAC;QAClC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,iDAAyB,EAAE,CAAC;gBAC/C,OAAO;YACT,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,kBAAkB,CAAC,EAAE,QAAQ,EAA0B;QAClE,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAE9C,IAAI,CAAC;YACH,OAAO,MAAM,OAAO,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QACpD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBAChB,OAAO,EAAE,+BAA+B;gBACxC,QAAQ;gBACR,KAAK,EAAE,KAAK,YAAY,KAAK;oBAC3B,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE;oBAClE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAClB,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,YAAY,CAAC,MAAc;QACtC,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC9C,OAAO,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACpD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,iDAAyB,EAAE,CAAC;gBAC/C,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,OAAO,CAAC,IAIpB;QACC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QAClC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAE9C,MAAM,SAAS,GAAG;YAChB,EAAE,EAAE,IAAI,CAAC,OAAO;YAChB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,iBAAS,CAAC,KAAK;YACrB,WAAW,EAAE,IAAI,CAAC,IAAI;SACvB,CAAC;QAEF,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK;YAC9B,CAAC,CAAC;gBACE,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE;gBACjB,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI;gBACrB,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI;gBACrB,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW;aACpC;YACH,CAAC,CAAC,SAAS,CAAC;QAEd,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,wBAAwB,CAAC,gBAAgB,CAAC;YACjE,YAAY,EAAE,SAAS;YACvB,aAAa;YACb,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,CAAC,OAAO,CAAC;YACpB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,2BAA2B,CACtC,IAAqC;QAErC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QAElC,IAAI,IAAI,CAAC,KAAK,CAAC,EAAE,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;YACnC,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,SAAS,IAAI,CAAC,KAAK,CAAC,EAAE,4DAA4D,CACnF,CAAC;YACF,OAAO;gBACL,YAAY,EAAE,IAAI,CAAC,OAAO;gBAC1B,MAAM,EAAE,CAAC,KAAK,SAAS,CAAC,MAAsC,CAAC,CAAC,EAAE;aACnE,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAE9C,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,wBAAwB,CAAC,gBAAgB,CAAC;YACjE,YAAY,EAAE;gBACZ,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,aAAa,EAAE;gBACb,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE;gBACjB,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI;gBACrB,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI;gBACrB,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW;aACpC;YACD,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,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,2CAA2C,IAAI,CAAC,KAAK,CAAC,EAAE,aAAa,KAAK,CAAC,GAAG,EAAE,CACjF,CAAC;QAEF,MAAM,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;YACnC,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,SAAS,EAAE,IAAI,CAAC,IAAI;YACpB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,KAAK,EAAE,KAAK,CAAC,KAAK;SACnB,CAAC,CAAC;QAEH,OAAO,EAAE,YAAY,EAAE,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;IAChD,CAAC;CACF,CAAA;AA9JY,kCAAW;sBAAX,WAAW;IADvB,IAAA,mBAAU,GAAE;qCAKuB,8BAAa;QACF,sDAAwB;QAC5B,6CAAoB;QAC9B,wBAAU;QACP,8BAAa;GARpC,WAAW,CA8JvB"}
+{"version":3,"file":"chat.service.js","sourceRoot":"","sources":["../../src/threads/chat.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;AAAA,2CAAoD;AACpD,2BAA+C;AAC/C,6DAA0D;AAG1D,sDAAwD;AACxD,iDAAoD;AACpD,oFAAsF;AACtF,mEAA8D;AAC9D,6EAAiF;AACjF,uEAGsC;AAI/B,IAAM,WAAW,mBAAjB,MAAM,WAAW;IAIH;IACA;IACA;IALF,MAAM,GAAG,IAAI,eAAM,CAAC,aAAW,CAAC,IAAI,CAAC,CAAC;IAEvD,YACmB,aAA4B,EAC5B,gBAAkC,EAClC,eAAqC;QAFrC,kBAAa,GAAb,aAAa,CAAe;QAC5B,qBAAgB,GAAhB,gBAAgB,CAAkB;QAClC,oBAAe,GAAf,eAAe,CAAsB;IACrD,CAAC;IAEI,KAAK,CAAC,OAAO;QACnB,OAAO,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC,EAAE,IAAI,EAAE,wCAAiB,EAAE,CAAC,CAAC;IACxE,CAAC;IAGM,KAAK,CAAC,YAAY,CAAC,MAAc;QACtC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,kBAAU,CAAC,WAAW,EAAE,CAAC,CAAC;YACtF,MAAM,MAAM,GAAG,MAAM,IAAA,iBAAY,EAAC;gBAChC,KAAK,EAAE,QAAQ,CAAC,KAAK;gBACrB,MAAM;gBACN,eAAe,EAAE,QAAQ,CAAC,eAAe;aAC1C,CAAC,CAAC;YACH,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,IAAI,CAAC;QACpC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,CAAC,KAAK,YAAY,6CAAuB,CAAC,EAAE,CAAC;gBAChD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;oBACf,OAAO,EAAE,wBAAwB;oBACjC,KAAK,EAAE,KAAK,YAAY,KAAK;wBAC3B,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE;wBAC9C,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iBAClB,CAAC,CAAC;YACL,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAMM,KAAK,CAAC,OAAO,CAAC,IAIpB;QACC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QAClC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,kBAAU,CAAC,cAAc,EAAE,CAAC,CAAC;QAEzF,MAAM,SAAS,GAAa;YAC1B,EAAE,EAAE,IAAI,CAAC,OAAO;YAChB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,iBAAS,CAAC,KAAK;YACrB,WAAW,EAAE,IAAI,CAAC,IAAI;SACvB,CAAC;QACF,MAAM,QAAQ,GAAa,IAAI,CAAC,KAAK;YACnC,CAAC,CAAC;gBACE,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE;gBACjB,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI;gBACrB,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI;gBACrB,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW;aACpC;YACH,CAAC,CAAC,SAAS,CAAC;QAEd,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC9E,IAAI,CAAC;YACH,MAAM,IAAA,iBAAY,EAAC;gBACjB,KAAK,EAAE,QAAQ,CAAC,KAAK;gBACrB,YAAY,EAAE,IAAI,CAAC,YAAY;gBAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,KAAK,EAAE,GAAG,CAAC,KAAK;gBAChB,eAAe,EAAE,QAAQ,CAAC,eAAe;gBACzC,QAAQ,EAAE,IAAA,gBAAW,EAAC,6CAAsB,CAAC;aAC9C,CAAC,CAAC;QACL,CAAC;gBAAS,CAAC;YACT,MAAM,GAAG,CAAC,KAAK,EAAE,CAAC;QACpB,CAAC;IACH,CAAC;CACF,CAAA;AA7EY,kCAAW;sBAAX,WAAW;IADvB,IAAA,mBAAU,GAAE;qCAKuB,8BAAa;QACV,qCAAgB;QACjB,8CAAoB;GAN7C,WAAW,CA6EvB"}