@ai-sdk/perplexity 3.1.6 → 3.1.8

package/src/perplexity-language-model-options.ts

@@ -1,124 +1,139 @@
+import {
+  lazySchema,
+  zodSchema,
+  type InferSchema,
+} from '@ai-sdk/provider-utils';
 import { z } from 'zod/v4';
 
-const userLocationSchema = z.looseObject({
-  latitude: z.number().optional(),
-  longitude: z.number().optional(),
-  country: z.string().optional(),
-  city: z.string().optional(),
-  region: z.string().optional(),
-});
-
-const webSearchFiltersSchema = z.looseObject({
-  search_domain_filter: z.array(z.string()).optional(),
-  search_recency_filter: z
-    .enum(['hour', 'day', 'week', 'month', 'year'])
-    .optional(),
-  search_after_date_filter: z.string().optional(),
-  search_before_date_filter: z.string().optional(),
-  last_updated_after_filter: z.string().optional(),
-  last_updated_before_filter: z.string().optional(),
-});
-
-const webSearchToolSchema = z.looseObject({
-  type: z.literal('web_search'),
-  filters: webSearchFiltersSchema.optional(),
-  max_results: z.number().int().positive().optional(),
-  max_tokens: z.number().optional(),
-  max_tokens_per_page: z.number().optional(),
-  search_context_size: z.enum(['low', 'medium', 'high']).optional(),
-  user_location: userLocationSchema.optional(),
-});
-
-const nativeToolSchema = z.union([
-  webSearchToolSchema,
+const userLocationSchema = () =>
   z.looseObject({
-    type: z.literal('fetch_url'),
-    max_urls: z.number().optional(),
-  }),
-  z.looseObject({ type: z.literal('people_search') }),
-  z.looseObject({ type: z.literal('finance_search') }),
-  z.looseObject({ type: z.literal('sandbox') }),
+    latitude: z.number().optional(),
+    longitude: z.number().optional(),
+    country: z.string().optional(),
+    city: z.string().optional(),
+    region: z.string().optional(),
+  });
+
+const webSearchFiltersSchema = () =>
   z.looseObject({
-    type: z.literal('mcp'),
-    server_label: z.string(),
-    server_url: z.string(),
-    allowed_tools: z.array(z.string()).optional(),
-    authorization: z.string().optional(),
-    defer_loading: z.boolean().optional(),
-    headers: z.record(z.string(), z.string()).optional(),
-  }),
+    search_domain_filter: z.array(z.string()).optional(),
+    search_recency_filter: z
+      .enum(['hour', 'day', 'week', 'month', 'year'])
+      .optional(),
+    search_after_date_filter: z.string().optional(),
+    search_before_date_filter: z.string().optional(),
+    last_updated_after_filter: z.string().optional(),
+    last_updated_before_filter: z.string().optional(),
+  });
+
+const webSearchToolSchema = () =>
   z.looseObject({
-    type: z.literal('connector'),
-    id: z.string(),
-    server_label: z.string(),
-    server_description: z.string().optional(),
-    allowed_tools: z.array(z.string()).optional(),
-  }),
-]);
+    type: z.literal('web_search'),
+    filters: webSearchFiltersSchema().optional(),
+    max_results: z.number().int().positive().optional(),
+    max_tokens: z.number().optional(),
+    max_tokens_per_page: z.number().optional(),
+    search_context_size: z.enum(['low', 'medium', 'high']).optional(),
+    user_location: userLocationSchema().optional(),
+  });
+
+const nativeToolSchema = () =>
+  z.union([
+    webSearchToolSchema(),
+    z.looseObject({
+      type: z.literal('fetch_url'),
+      max_urls: z.number().optional(),
+    }),
+    z.looseObject({ type: z.literal('people_search') }),
+    z.looseObject({ type: z.literal('finance_search') }),
+    z.looseObject({ type: z.literal('sandbox') }),
+    z.looseObject({
+      type: z.literal('mcp'),
+      server_label: z.string(),
+      server_url: z.string(),
+      allowed_tools: z.array(z.string()).optional(),
+      authorization: z.string().optional(),
+      defer_loading: z.boolean().optional(),
+      headers: z.record(z.string(), z.string()).optional(),
+    }),
+    z.looseObject({
+      type: z.literal('connector'),
+      id: z.string(),
+      server_label: z.string(),
+      server_description: z.string().optional(),
+      allowed_tools: z.array(z.string()).optional(),
+    }),
+  ]);
 
-export const perplexityLanguageModelOptions = z.looseObject({
-  /** Top-level Agent API instructions. */
-  instructions: z.string().optional(),
+export const perplexityLanguageModelOptions = lazySchema(() =>
+  zodSchema(
+    z.looseObject({
+      /** Top-level Agent API instructions. */
+      instructions: z.string().optional(),
 
-  /**
-   * Native Agent API tools. AI SDK function tools can also be supplied through
-   * the top-level `tools` option on `generateText` and `streamText`.
-   */
-  tools: z.array(nativeToolSchema).optional(),
+      /**
+       * Native Agent API tools. AI SDK function tools can also be supplied through
+       * the top-level `tools` option on `generateText` and `streamText`.
+       */
+      tools: z.array(nativeToolSchema()).optional(),
 
-  /** A fallback model list for Agent API routing. */
-  models: z.array(z.string()).optional(),
+      /** A fallback model list for Agent API routing. */
+      models: z.array(z.string()).optional(),
 
-  /** Maximum number of agentic steps. */
-  max_steps: z.number().int().positive().optional(),
+      /** Maximum number of agentic steps. */
+      max_steps: z.number().int().positive().optional(),
 
-  /** Maximum number of native tool calls. */
-  max_tool_calls: z.number().int().nonnegative().optional(),
+      /** Maximum number of native tool calls. */
+      max_tool_calls: z.number().int().nonnegative().optional(),
 
-  /** Continue a conversation from an earlier Agent API response. */
-  previous_response_id: z.string().optional(),
+      /** Continue a conversation from an earlier Agent API response. */
+      previous_response_id: z.string().optional(),
 
-  /**
-   * Whether the response can be retrieved later. Setting this to false does
-   * not disable persistence or prevent previous_response_id continuations.
-   */
-  store: z.boolean().optional(),
+      /**
+       * Whether the response can be retrieved later. Setting this to false does
+       * not disable persistence or prevent previous_response_id continuations.
+       */
+      store: z.boolean().optional(),
 
-  /** Preferred response language as an ISO 639-1 language code. */
-  language_preference: z.string().optional(),
+      /** Preferred response language as an ISO 639-1 language code. */
+      language_preference: z.string().optional(),
 
-  /** Agent API reasoning configuration. */
-  reasoning: z
-    .looseObject({
-      effort: z.enum(['minimal', 'low', 'medium', 'high', 'xhigh']).optional(),
-    })
-    .optional(),
+      /** Agent API reasoning configuration. */
+      reasoning: z
+        .looseObject({
+          effort: z
+            .enum(['minimal', 'low', 'medium', 'high', 'xhigh'])
+            .optional(),
+        })
+        .optional(),
 
-  /** Agent API skill configuration. */
-  skills: z
-    .array(
-      z.union([
-        z.looseObject({
-          type: z.literal('builtin'),
-          name: z.enum([
-            'office',
-            'office/docx',
-            'office/pdf',
-            'office/pptx',
-            'office/xlsx',
+      /** Agent API skill configuration. */
+      skills: z
+        .array(
+          z.union([
+            z.looseObject({
+              type: z.literal('builtin'),
+              name: z.enum([
+                'office',
+                'office/docx',
+                'office/pdf',
+                'office/pptx',
+                'office/xlsx',
+              ]),
+            }),
+            z.looseObject({
+              type: z.literal('inline'),
+              name: z.string(),
+              description: z.string(),
+              instructions: z.string(),
+            }),
           ]),
-        }),
-        z.looseObject({
-          type: z.literal('inline'),
-          name: z.string(),
-          description: z.string(),
-          instructions: z.string(),
-        }),
-      ]),
-    )
-    .optional(),
-});
+        )
+        .optional(),
+    }),
+  ),
+);
 
-export type PerplexityLanguageModelOptions = z.infer<
+export type PerplexityLanguageModelOptions = InferSchema<
   typeof perplexityLanguageModelOptions
 >;

package/src/perplexity-language-model.ts

@@ -11,6 +11,7 @@ import {
   type SharedV3Warning,
 } from '@ai-sdk/provider';
 import {
+  type InferSchema,
   combineHeaders,
   createEventSourceResponseHandler,
   createJsonErrorResponseHandler,
@@ -26,6 +27,7 @@ import { convertToPerplexityInput } from './convert-to-perplexity-input';
 import { mapPerplexityFinishReason } from './map-perplexity-finish-reason';
 import {
   perplexityAgentChunkSchema,
+  perplexityAgentResponseSchemaValidator,
   perplexityAgentResponseSchema,
   perplexityErrorSchema,
   perplexityErrorToMessage,
@@ -48,9 +50,13 @@ type PerplexityAgentConfig = {
   fetch?: FetchFunction;
 };
 
-type PerplexityOutputItem = z.infer<typeof perplexityOutputItemSchema>;
-type PerplexitySearchResult = z.infer<typeof perplexitySearchResultSchema>;
-type PerplexityUsage = z.infer<typeof perplexityUsageSchema>;
+type PerplexityOutputItem = z.infer<
+  ReturnType<typeof perplexityOutputItemSchema>
+>;
+type PerplexitySearchResult = z.infer<
+  ReturnType<typeof perplexitySearchResultSchema>
+>;
+type PerplexityUsage = z.infer<ReturnType<typeof perplexityUsageSchema>>;
 type PerplexityUrlSource = Extract<
   LanguageModelV3Content,
   { type: 'source'; sourceType: 'url' }
@@ -296,7 +302,7 @@ export class PerplexityLanguageModel implements LanguageModelV3 {
         errorToMessage: perplexityErrorToMessage,
       }),
       successfulResponseHandler: createJsonResponseHandler(
-        perplexityAgentResponseSchema,
+        perplexityAgentResponseSchemaValidator,
       ),
       abortSignal: options.abortSignal,
       fetch: this.config.fetch,
@@ -452,7 +458,7 @@ export class PerplexityLanguageModel implements LanguageModelV3 {
     return {
       stream: response.pipeThrough(
         new TransformStream<
-          ParseResult<z.infer<typeof perplexityAgentChunkSchema>>,
+          ParseResult<InferSchema<typeof perplexityAgentChunkSchema>>,
           LanguageModelV3StreamPart
         >({
           start(controller) {