Prd-5653 (#1388)

Author: anubra266

agents-cli/package.json

@@ -35,14 +35,14 @@
   "author": "Inkeep <[email protected]>",
   "license": "SEE LICENSE IN LICENSE.md",
   "dependencies": {
-    "@ai-sdk/anthropic": "3.0.0-beta.66",
-    "@ai-sdk/google": "3.0.0-beta.62",
-    "@ai-sdk/openai": "3.0.0-beta.74",
+    "@ai-sdk/anthropic": "3.0.7",
+    "@ai-sdk/google": "3.0.4",
+    "@ai-sdk/openai": "3.0.7",
     "@clack/prompts": "^0.11.0",
     "@inkeep/agents-core": "workspace:^",
     "@inkeep/agents-sdk": "workspace:^",
     "@vercel/otel": "^2.0.1",
-    "ai": "6.0.0-beta.124",
+    "ai": "6.0.14",
     "chalk": "^5.3.0",
     "cli-table3": "^0.6.3",
     "commander": "^14.0.0",
@@ -53,9 +53,9 @@
     "json-schema-to-zod": "^2.6.1",
     "keytar": "^7.9.0",
     "langfuse-vercel": "^3.38.6",
+    "open": "^10.2.0",
     "ts-morph": "^26.0.0",
     "tsx": "^4.20.5",
-    "open": "^10.2.0",
     "yaml": "^2.7.0"
   },
   "devDependencies": {

agents-docs/content/typescript-sdk/(observability)/signoz-usage.mdx

@@ -30,7 +30,7 @@ The traces explorer shows:
 
 - **Timestamp**: When each span occurred
 - **Service Name**: The service that generated the span (e.g., `inkeep-agents-run-api`)
-- **Operation Name**: Specific operations like `ai.generateObject`, `tls.connect`, `ai.toolCall`
+- **Operation Name**: Specific operations like `ai.generateText`, `tls.connect`, `ai.toolCall`
 - **Duration**: How long each operation took (in milliseconds)
 - **HTTP Method**: For HTTP operations, shows the method (POST, GET, etc.)
 - **Response Status Code**: HTTP status codes (200, 404, etc.)

agents-manage-ui/package.json

@@ -77,7 +77,7 @@
     "@sinclair/typebox": "^0.34.40",
     "@tanstack/react-query": "^5.90.12",
     "@xyflow/react": "^12.10.0",
-    "ai": "6.0.0-beta.124",
+    "ai": "6.0.14",
     "axios": "^1.11.0",
     "axios-retry": "^4.5.0",
     "better-auth": "^1.4.0",
@@ -136,8 +136,8 @@
     "vitest-canvas-mock": "^0.3.3"
   },
   "optionalDependencies": {
-    "@sentry/nextjs": "^10.32.1",
     "@posthog/react": "^1.5.2",
+    "@sentry/nextjs": "^10.32.1",
     "posthog-js": "^1.308.0"
   }
 }

agents-manage-ui/src/app/api/artifact-components/[artifactComponentId]/generate-render/route.ts

@@ -5,12 +5,12 @@
  * This route:
  * 1. Fetches the artifact component from manage-api
  * 2. Fetches the project to get base model configuration
- * 3. Uses AI SDK streamObject to generate component code and sample data
+ * 3. Uses AI SDK structured output streamText to generate component code and sample data
  * 4. Streams NDJSON response back to client
  */
 
 import { jsonSchemaToZod, ModelFactory } from '@inkeep/agents-core';
-import { streamObject } from 'ai';
+import { streamText, Output } from 'ai';
 import type { NextRequest } from 'next/server';
 import { z } from 'zod';
 import { fetchArtifactComponent } from '@/lib/api/artifact-components';
@@ -74,11 +74,12 @@ export async function POST(
       mockData: mockDataSchema.describe('Sample data matching the props schema'),
     });
 
-    // Generate using AI SDK streamObject
-    const result = streamObject({
+    const result = streamText({
       ...modelConfig,
       prompt,
-      schema: renderSchema,
+      output: Output.object({
+        schema: renderSchema,
+      }),
       temperature: 0.7,
     });
 
@@ -95,7 +96,7 @@ export async function POST(
     const stream = new ReadableStream({
       async start(controller) {
         try {
-          for await (const partialObject of result.partialObjectStream) {
+          for await (const partialObject of result.partialOutputStream) {
             // If modifying with instructions, preserve existing data
             const outputObject =
               instructions && existingData

agents-manage-ui/src/app/api/data-components/[dataComponentId]/generate-render/route.ts

@@ -5,12 +5,12 @@
  * This route:
  * 1. Fetches the data component from manage-api
  * 2. Fetches the project to get base model configuration
- * 3. Uses AI SDK streamObject to generate component code and sample data
+ * 3. Uses AI SDK structured output streamText to generate component code and sample data
  * 4. Streams NDJSON response back to client
  */
 
 import { jsonSchemaToZod, ModelFactory } from '@inkeep/agents-core';
-import { streamObject } from 'ai';
+import { streamText, Output } from 'ai';
 import type { NextRequest } from 'next/server';
 import { z } from 'zod';
 import { fetchDataComponent } from '@/lib/api/data-components';
@@ -72,11 +72,13 @@ export async function POST(
       mockData: mockDataSchema.describe('Sample data matching the props schema'),
     });
 
-    // Generate using AI SDK streamObject
-    const result = streamObject({
+    // Generate using AI SDK structured output streamText
+    const result = streamText({
       ...modelConfig,
       prompt,
-      schema: renderSchema,
+      output: Output.object({
+        schema: renderSchema,
+      }),
       temperature: 0.7,
     });
 
@@ -93,7 +95,7 @@ export async function POST(
     const stream = new ReadableStream({
       async start(controller) {
         try {
-          for await (const partialObject of result.partialObjectStream) {
+          for await (const partialObject of result.partialOutputStream) {
             // If modifying with instructions, preserve existing data
             const outputObject =
               instructions && existingData

agents-run-api/package.json

@@ -27,13 +27,13 @@
     "typecheck:watch": "tsc --noEmit --watch"
   },
   "dependencies": {
-    "@electric-sql/pglite": "^0.3.13",
-    "@ai-sdk/anthropic": "3.0.0-beta.66",
-    "@ai-sdk/gateway": "2.0.0-beta.68",
-    "@ai-sdk/google": "3.0.0-beta.62",
-    "@ai-sdk/openai": "3.0.0-beta.74",
-    "@ai-sdk/openai-compatible": "2.0.0-beta.41",
+    "@ai-sdk/anthropic": "3.0.7",
+    "@ai-sdk/gateway": "3.0.9",
+    "@ai-sdk/google": "3.0.4",
+    "@ai-sdk/openai": "3.0.7",
+    "@ai-sdk/openai-compatible": "2.0.4",
     "@alcyone-labs/modelcontextprotocol-sdk": "^1.16.0",
+    "@electric-sql/pglite": "^0.3.13",
     "@hono/otel": "^0.4.0",
     "@hono/swagger-ui": "^0.5.1",
     "@inkeep/agents-core": "workspace:^",
@@ -49,7 +49,7 @@
     "@opentelemetry/sdk-trace-base": "^2.1.0",
     "@opentelemetry/semantic-conventions": "^1.37.0",
     "@vercel/sandbox": "^0.0.24",
-    "ai": "6.0.0-beta.124",
+    "ai": "6.0.14",
     "ajv": "^8.17.1",
     "drizzle-orm": "^0.44.4",
     "fetch-to-node": "^2.1.0",

agents-run-api/src/__tests__/agents/Agent.test.ts

@@ -11,46 +11,109 @@ import { Phase1Config } from '../../agents/versions/v1/Phase1Config';
 
 // Mock the AI SDK functions
 vi.mock('ai', () => ({
-  generateText: vi.fn().mockResolvedValue({
-    text: 'Mocked response',
-    toolCalls: [],
-    finishReason: 'stop',
-    steps: [
-      {
-        content: [
-          {
-            type: 'text',
-            text: 'Mocked response',
-          },
-        ],
-        toolCalls: [
-          {
-            toolName: 'thinking_complete',
-            args: {},
-          },
-        ],
-        toolResults: [
+  generateText: vi.fn().mockImplementation(async (options: any) => {
+    // When Output.object is present in options, return "output" with the correct shape
+    if (options?.output && typeof options.output === 'object') {
+      // Try to determine the expected output structure from the schema
+      const schema = options.output.schema;
+      let outputData: any = {};
+      
+      // Check if schema expects dataComponents
+      if (schema?._def?.shape?.dataComponents) {
+        outputData = {
+          dataComponents: [
+            {
+              id: 'test-component',
+              name: 'TestComponent',
+              props: { message: 'Hello, World!' },
+            },
+          ],
+        };
+      } else if (schema?._def?.shape?.statusComponents) {
+        outputData = {
+          statusComponents: [
+            {
+              id: 'status-1',
+              type: 'text',
+              props: { text: 'Status update generated' },
+            },
+          ],
+        };
+      }
+      
+      return {
+        text: 'Mocked response',
+        toolCalls: [],
+        finishReason: 'stop',
+        output: outputData,
+        steps: [
           {
-            toolCallId: 'call_1',
-            result: 'Thinking complete',
+            content: [
+              {
+                type: 'text',
+                text: 'Mocked response',
+              },
+            ],
+            toolCalls: [
+              {
+                toolName: 'thinking_complete',
+                args: {},
+              },
+            ],
+            toolResults: [
+              {
+                toolCallId: 'call_1',
+                result: 'Thinking complete',
+              },
+            ],
           },
         ],
-      },
-    ],
-  }),
-  generateObject: vi.fn().mockResolvedValue({
-    object: {
-      dataComponents: [
+      };
+    }
+    // Otherwise, return just text as fallback
+    return {
+      text: 'Mocked response',
+      toolCalls: [],
+      finishReason: 'stop',
+      steps: [
         {
-          id: 'test-component-id',
-          name: 'TestComponent',
-          description: 'A test component',
-          props: { message: 'Hello, World!' },
+          content: [
+            {
+              type: 'text',
+              text: 'Mocked response',
+            },
+          ],
+          toolCalls: [
+            {
+              toolName: 'thinking_complete',
+              args: {},
+            },
+          ],
+          toolResults: [
+            {
+              toolCallId: 'call_1',
+              result: 'Thinking complete',
+            },
+          ],
         },
       ],
-    },
-    finishReason: 'stop',
+    };
+  }),
+  streamText: vi.fn().mockReturnValue({
+    textStream: (async function* () {
+      yield 'Mocked';
+      yield ' response';
+    })(),
+    fullStream: (async function* () {
+      yield { type: 'text-delta', textDelta: 'Mocked response' };
+    })(),
   }),
+  Output: {
+    object: vi.fn().mockImplementation((config: any) => ({
+      type: 'object',
+      ...config,
+    })),
+  },
   tool: vi.fn().mockImplementation((config) => config),
 }));
 
@@ -295,8 +358,6 @@ vi.mock('../../data/conversations.js', () => ({
     .mockResolvedValue('Mock conversation history as string'),
 }));
 
-// Import the mocked functions so we can reference them in tests
-import { generateObject, generateText } from 'ai';
 // Import the mocked module - these will automatically be mocked
 import {
   getConversationHistoryWithCompression,
@@ -1216,46 +1277,6 @@ describe('Two-Pass Generation System', () => {
     };
   });
 
-  test('should only call generateText when no data components configured', async () => {
-    const agent = new Agent({ ...mockAgentConfig, dataComponents: [] });
-    await agent.generate('Test prompt');
-
-    expect(vi.mocked(generateText)).toHaveBeenCalledTimes(1);
-    expect(vi.mocked(generateObject)).not.toHaveBeenCalled();
-  });
-
-  test('should call both generateText and generateObject when data components configured', async () => {
-    const agent = new Agent(mockAgentConfig);
-    await agent.generate('Test prompt');
-
-    expect(vi.mocked(generateText)).toHaveBeenCalledTimes(1);
-    expect(vi.mocked(generateObject)).toHaveBeenCalledTimes(1);
-  });
-
-  test('should skip generateObject when transfer detected', async () => {
-    vi.mocked(generateText).mockResolvedValueOnce({
-      text: 'Transfer needed',
-      finishReason: 'stop',
-      steps: [
-        {
-          content: [
-            {
-              type: 'text',
-              text: 'Transfer needed',
-            },
-          ],
-          toolCalls: [{ toolName: 'transfer_to_agent', args: {} }],
-        },
-      ],
-    } as any);
-
-    const agent = new Agent(mockAgentConfig);
-    await agent.generate('Test prompt');
-
-    expect(vi.mocked(generateText)).toHaveBeenCalledTimes(1);
-    expect(vi.mocked(generateObject)).not.toHaveBeenCalled();
-  });
-
   test('should return text response when no data components', async () => {
     const agent = new Agent({ ...mockAgentConfig, dataComponents: [] });
     const result = await agent.generate('Test prompt');