@kognitivedev/vercel-ai-provider 0.1.7 → 0.1.9

package/src/__tests__/wrap-stream-logging.test.ts

@@ -36,6 +36,110 @@ describe("wrapStream logging", () => {
         );
     });
 
+    it("should capture tool-call chunks and include them in logged conversation", async () => {
+        const mockModel = new MockLanguageModelV3({
+            doStream: async () => ({
+                stream: convertArrayToReadableStream([
+                    { type: "text-start" as const, id: "t1" },
+                    { type: "text-delta" as const, id: "t1", delta: "Let me check" },
+                    { type: "text-end" as const, id: "t1" },
+                    {
+                        type: "tool-call" as const,
+                        toolCallId: "call-1",
+                        toolName: "get_weather",
+                        input: '{"city":"London"}',
+                    },
+                    {
+                        type: "tool-result" as const,
+                        toolCallId: "call-1",
+                        toolName: "get_weather",
+                        result: { temperature: 15, unit: "celsius" },
+                    },
+                    {
+                        type: "finish" as const,
+                        finishReason: {
+                            unified: "tool-calls" as const,
+                            raw: undefined,
+                        },
+                        usage: {
+                            inputTokens: { total: 20, noCache: undefined, cacheRead: undefined, cacheWrite: undefined },
+                            outputTokens: { total: 15, text: undefined, reasoning: undefined },
+                        },
+                    },
+                ] satisfies import("@ai-sdk/provider").LanguageModelV3StreamPart[]),
+            }),
+        });
+
+        const mockProvider = () => mockModel;
+
+        const cl = createCognitiveLayer({
+            provider: mockProvider,
+            clConfig: {
+                apiKey: "test-api-key",
+                appId: "test-app",
+                projectId: "test-project",
+                processDelayMs: 0,
+                logLevel: "none",
+            },
+        });
+
+        const model = cl("mock-model", {
+            userId: "user-1",
+            projectId: "project-1",
+            sessionId: "session-1",
+        });
+
+        const result = streamText({
+            model,
+            messages: [{ role: "user", content: "What's the weather in London?" }],
+        });
+
+        // Fully consume the stream
+        await result.text;
+
+        // Wait for async logConversation to complete
+        await new Promise((r) => setTimeout(r, 100));
+
+        // Find the log call
+        const logCall = fetchCalls.find((c) => c.url.includes("/api/cognitive/log"));
+        expect(logCall).toBeDefined();
+
+        const messages = logCall!.body.messages;
+
+        // Assistant message should contain text + tool-call parts
+        const assistantMsg = messages.find((m: any) => m.role === "assistant");
+        expect(assistantMsg).toBeDefined();
+        expect(assistantMsg.content).toEqual([
+            { type: "text", text: "Let me check" },
+            {
+                type: "tool-call",
+                toolCallId: "call-1",
+                toolName: "get_weather",
+                input: '{"city":"London"}',
+            },
+        ]);
+
+        // Tool results should be in a separate tool message
+        const toolMsg = messages.find((m: any) => m.role === "tool");
+        expect(toolMsg).toBeDefined();
+        expect(toolMsg.content).toEqual([
+            {
+                type: "tool-result",
+                toolCallId: "call-1",
+                toolName: "get_weather",
+                result: { temperature: 15, unit: "celsius" },
+            },
+        ]);
+
+        // Spans should include the tool call with populated previews
+        const spans = logCall!.body.spans;
+        const toolSpan = spans?.find((s: any) => s.spanType === "tool");
+        expect(toolSpan).toBeDefined();
+        expect(toolSpan.toolName).toBe("get_weather");
+        expect(toolSpan.inputPreview).toContain("London");
+        expect(toolSpan.outputPreview).toContain("15");
+    });
+
     it("should include assistant message in logged conversation after streaming", async () => {
         const mockModel = new MockLanguageModelV3({
             doStream: async () => ({