@ai-sdk/perplexity 3.0.8 → 3.0.9

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @ai-sdk/perplexity
 
+## 3.0.9
+
+### Patch Changes
+
+- 8dc54db: chore: add src folders to package bundle
+
 ## 3.0.8
 
 ### Patch Changes

package/dist/index.js

@@ -479,7 +479,7 @@ var errorToMessage = (data) => {
 };
 
 // src/version.ts
-var VERSION = true ? "3.0.8" : "0.0.0-test";
+var VERSION = true ? "3.0.9" : "0.0.0-test";
 
 // src/perplexity-provider.ts
 function createPerplexity(options = {}) {

package/dist/index.mjs

@@ -466,7 +466,7 @@ var errorToMessage = (data) => {
 };
 
 // src/version.ts
-var VERSION = true ? "3.0.8" : "0.0.0-test";
+var VERSION = true ? "3.0.9" : "0.0.0-test";
 
 // src/perplexity-provider.ts
 function createPerplexity(options = {}) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/perplexity",
-  "version": "3.0.8",
+  "version": "3.0.9",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -8,6 +8,7 @@
   "types": "./dist/index.d.ts",
   "files": [
     "dist/**/*",
+    "src",
     "CHANGELOG.md",
     "README.md"
   ],
@@ -28,7 +29,7 @@
     "tsup": "^8",
     "typescript": "5.8.3",
     "zod": "3.25.76",
-    "@ai-sdk/test-server": "1.0.1",
+    "@ai-sdk/test-server": "1.0.2",
     "@vercel/ai-tsconfig": "0.0.0"
   },
   "peerDependencies": {

package/src/__snapshots__/convert-to-perplexity-messages.test.ts.snap

@@ -0,0 +1,48 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`convertToPerplexityMessages > assistant messages > should convert an assistant message with text content 1`] = `
+[
+  {
+    "content": "Assistant reply",
+    "role": "assistant",
+  },
+]
+`;
+
+exports[`convertToPerplexityMessages > system messages > should convert a system message with text content 1`] = `
+[
+  {
+    "content": "System initialization",
+    "role": "system",
+  },
+]
+`;
+
+exports[`convertToPerplexityMessages > user messages > should convert a user message with image parts 1`] = `
+[
+  {
+    "content": [
+      {
+        "text": "Hello ",
+        "type": "text",
+      },
+      {
+        "image_url": {
+          "url": "data:image/png;base64,AAECAw==",
+        },
+        "type": "image_url",
+      },
+    ],
+    "role": "user",
+  },
+]
+`;
+
+exports[`convertToPerplexityMessages > user messages > should convert a user message with text parts 1`] = `
+[
+  {
+    "content": "Hello World",
+    "role": "user",
+  },
+]
+`;

package/src/convert-perplexity-usage.ts

@@ -0,0 +1,48 @@
+import { LanguageModelV3Usage } from '@ai-sdk/provider';
+
+export function convertPerplexityUsage(
+  usage:
+    | {
+        prompt_tokens?: number | null | undefined;
+        completion_tokens?: number | null | undefined;
+        reasoning_tokens?: number | null | undefined;
+      }
+    | undefined
+    | null,
+): LanguageModelV3Usage {
+  if (usage == null) {
+    return {
+      inputTokens: {
+        total: undefined,
+        noCache: undefined,
+        cacheRead: undefined,
+        cacheWrite: undefined,
+      },
+      outputTokens: {
+        total: undefined,
+        text: undefined,
+        reasoning: undefined,
+      },
+      raw: undefined,
+    };
+  }
+
+  const promptTokens = usage.prompt_tokens ?? 0;
+  const completionTokens = usage.completion_tokens ?? 0;
+  const reasoningTokens = usage.reasoning_tokens ?? 0;
+
+  return {
+    inputTokens: {
+      total: promptTokens,
+      noCache: promptTokens,
+      cacheRead: undefined,
+      cacheWrite: undefined,
+    },
+    outputTokens: {
+      total: completionTokens,
+      text: completionTokens - reasoningTokens,
+      reasoning: reasoningTokens,
+    },
+    raw: usage,
+  };
+}

package/src/convert-to-perplexity-messages.test.ts

@@ -0,0 +1,85 @@
+import { convertToPerplexityMessages } from './convert-to-perplexity-messages';
+import { UnsupportedFunctionalityError } from '@ai-sdk/provider';
+import { describe, it, expect } from 'vitest';
+
+describe('convertToPerplexityMessages', () => {
+  describe('system messages', () => {
+    it('should convert a system message with text content', () => {
+      expect(
+        convertToPerplexityMessages([
+          {
+            role: 'system',
+            content: 'System initialization',
+          },
+        ]),
+      ).toMatchSnapshot();
+    });
+  });
+
+  describe('user messages', () => {
+    it('should convert a user message with text parts', () => {
+      expect(
+        convertToPerplexityMessages([
+          {
+            role: 'user',
+            content: [
+              { type: 'text', text: 'Hello ' },
+              { type: 'text', text: 'World' },
+            ],
+          },
+        ]),
+      ).toMatchSnapshot();
+    });
+
+    it('should convert a user message with image parts', () => {
+      expect(
+        convertToPerplexityMessages([
+          {
+            role: 'user',
+            content: [
+              { type: 'text', text: 'Hello ' },
+              {
+                type: 'file',
+                data: new Uint8Array([0, 1, 2, 3]),
+                mediaType: 'image/png',
+              },
+            ],
+          },
+        ]),
+      ).toMatchSnapshot();
+    });
+  });
+
+  describe('assistant messages', () => {
+    it('should convert an assistant message with text content', () => {
+      expect(
+        convertToPerplexityMessages([
+          {
+            role: 'assistant',
+            content: [{ type: 'text', text: 'Assistant reply' }],
+          },
+        ]),
+      ).toMatchSnapshot();
+    });
+  });
+
+  describe('tool messages', () => {
+    it('should throw an error for tool messages', () => {
+      expect(() => {
+        convertToPerplexityMessages([
+          {
+            role: 'tool',
+            content: [
+              {
+                type: 'tool-result',
+                toolCallId: 'dummy-tool-call-id',
+                toolName: 'dummy-tool-name',
+                output: { type: 'text', value: 'This should fail' },
+              },
+            ],
+          },
+        ]);
+      }).toThrow(UnsupportedFunctionalityError);
+    });
+  });
+});

package/src/convert-to-perplexity-messages.ts

@@ -0,0 +1,107 @@
+import {
+  LanguageModelV3Prompt,
+  UnsupportedFunctionalityError,
+} from '@ai-sdk/provider';
+import {
+  PerplexityMessageContent,
+  PerplexityPrompt,
+} from './perplexity-language-model-prompt';
+import { convertUint8ArrayToBase64 } from '@ai-sdk/provider-utils';
+
+export function convertToPerplexityMessages(
+  prompt: LanguageModelV3Prompt,
+): PerplexityPrompt {
+  const messages: PerplexityPrompt = [];
+
+  for (const { role, content } of prompt) {
+    switch (role) {
+      case 'system': {
+        messages.push({ role: 'system', content });
+        break;
+      }
+
+      case 'user':
+      case 'assistant': {
+        const hasMultipartContent = content.some(
+          part =>
+            (part.type === 'file' && part.mediaType.startsWith('image/')) ||
+            (part.type === 'file' && part.mediaType === 'application/pdf'),
+        );
+
+        const messageContent = content
+          .map((part, index) => {
+            switch (part.type) {
+              case 'text': {
+                return {
+                  type: 'text',
+                  text: part.text,
+                };
+              }
+              case 'file': {
+                if (part.mediaType === 'application/pdf') {
+                  return part.data instanceof URL
+                    ? {
+                        type: 'file_url',
+                        file_url: {
+                          url: part.data.toString(),
+                        },
+                        file_name: part.filename,
+                      }
+                    : {
+                        type: 'file_url',
+                        file_url: {
+                          url:
+                            typeof part.data === 'string'
+                              ? part.data
+                              : convertUint8ArrayToBase64(part.data),
+                        },
+                        file_name: part.filename || `document-${index}.pdf`,
+                      };
+                } else if (part.mediaType.startsWith('image/')) {
+                  return part.data instanceof URL
+                    ? {
+                        type: 'image_url',
+                        image_url: {
+                          url: part.data.toString(),
+                        },
+                      }
+                    : {
+                        type: 'image_url',
+                        image_url: {
+                          url: `data:${part.mediaType ?? 'image/jpeg'};base64,${
+                            typeof part.data === 'string'
+                              ? part.data
+                              : convertUint8ArrayToBase64(part.data)
+                          }`,
+                        },
+                      };
+                }
+              }
+            }
+          })
+          .filter(Boolean) as PerplexityMessageContent[];
+        messages.push({
+          role,
+          content: hasMultipartContent
+            ? messageContent
+            : messageContent
+                .filter(part => part.type === 'text')
+                .map(part => part.text)
+                .join(''),
+        });
+        break;
+      }
+      case 'tool': {
+        throw new UnsupportedFunctionalityError({
+          functionality: 'Tool messages',
+        });
+      }
+      default: {
+        const _exhaustiveCheck: never = role;
+        throw new Error(`Unsupported role: ${_exhaustiveCheck}`);
+      }
+    }
+  }
+
+  return messages;
+}

package/src/index.ts

@@ -0,0 +1,6 @@
+export { createPerplexity, perplexity } from './perplexity-provider';
+export type {
+  PerplexityProvider,
+  PerplexityProviderSettings,
+} from './perplexity-provider';
+export { VERSION } from './version';

package/src/map-perplexity-finish-reason.ts

@@ -0,0 +1,13 @@
+import { LanguageModelV3FinishReason } from '@ai-sdk/provider';
+
+export function mapPerplexityFinishReason(
+  finishReason: string | null | undefined,
+): LanguageModelV3FinishReason['unified'] {
+  switch (finishReason) {
+    case 'stop':
+    case 'length':
+      return finishReason;
+    default:
+      return 'other';
+  }
+}

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

@@ -0,0 +1,8 @@
+// https://docs.perplexity.ai/models/model-cards
+export type PerplexityLanguageModelId =
+  | 'sonar-deep-research'
+  | 'sonar-reasoning-pro'
+  | 'sonar-reasoning'
+  | 'sonar-pro'
+  | 'sonar'
+  | (string & {});

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

@@ -0,0 +1,25 @@
+export type PerplexityPrompt = Array<PerplexityMessage>;
+
+export type PerplexityMessage = {
+  role: 'system' | 'user' | 'assistant';
+  content: string | PerplexityMessageContent[];
+};
+
+export type PerplexityMessageContent =
+  | {
+      type: 'text';
+      text: string;
+    }
+  | {
+      type: 'image_url';
+      image_url: {
+        url: string;
+      };
+    }
+  | {
+      type: 'file_url';
+      file_url: {
+        url: string;
+      };
+      file_name?: string;
+    };