@ai-sdk/perplexity 0.0.6 → 0.0.8

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/CHANGELOG.md CHANGED
@@ -1,5 +1,19 @@
1
1
  # @ai-sdk/perplexity
2
2
 
3
+ ## 0.0.8
4
+
5
+ ### Patch Changes
6
+
7
+ - Updated dependencies [3ff4ef8]
8
+ - @ai-sdk/provider-utils@2.1.7
9
+ - @ai-sdk/openai-compatible@0.1.9
10
+
11
+ ## 0.0.7
12
+
13
+ ### Patch Changes
14
+
15
+ - 18713a5: feat (provider/perplexity): add support for return_images
16
+
3
17
  ## 0.0.6
4
18
 
5
19
  ### Patch Changes
package/dist/index.js CHANGED
@@ -34,61 +34,76 @@ var import_zod2 = require("zod");
34
34
  // src/perplexity-metadata-extractor.ts
35
35
  var import_provider_utils = require("@ai-sdk/provider-utils");
36
36
  var import_zod = require("zod");
37
- var buildPerplexityMetadata = (citations, usage) => {
38
- var _a, _b;
39
- return citations || usage ? {
40
- perplexity: {
41
- ...citations && { citations },
42
- ...usage && {
43
- usage: {
44
- citationTokens: (_a = usage.citation_tokens) != null ? _a : NaN,
45
- numSearchQueries: (_b = usage.num_search_queries) != null ? _b : NaN
46
- }
47
- }
48
- }
49
- } : void 0;
50
- };
51
37
  var perplexityMetadataExtractor = {
52
38
  extractMetadata: ({ parsedBody }) => {
53
- var _a, _b;
39
+ var _a, _b, _c;
54
40
  const parsed = (0, import_provider_utils.safeValidateTypes)({
55
41
  value: parsedBody,
56
42
  schema: perplexityResponseSchema
57
43
  });
58
44
  return !parsed.success ? void 0 : buildPerplexityMetadata(
59
45
  (_a = parsed.value.citations) != null ? _a : void 0,
60
- (_b = parsed.value.usage) != null ? _b : void 0
46
+ (_b = parsed.value.images) != null ? _b : void 0,
47
+ (_c = parsed.value.usage) != null ? _c : void 0
61
48
  );
62
49
  },
63
50
  createStreamExtractor: () => {
64
51
  let citations;
52
+ let images;
65
53
  let usage;
66
54
  return {
67
55
  processChunk: (chunk) => {
56
+ var _a, _b, _c;
68
57
  const parsed = (0, import_provider_utils.safeValidateTypes)({
69
58
  value: chunk,
70
59
  schema: perplexityStreamChunkSchema
71
60
  });
72
61
  if (parsed.success) {
73
- if (parsed.value.citations) {
74
- citations = parsed.value.citations;
75
- }
76
- if (parsed.value.usage) {
77
- usage = parsed.value.usage;
78
- }
62
+ citations = (_a = parsed.value.citations) != null ? _a : citations;
63
+ images = (_b = parsed.value.images) != null ? _b : images;
64
+ usage = (_c = parsed.value.usage) != null ? _c : usage;
79
65
  }
80
66
  },
81
- buildMetadata: () => buildPerplexityMetadata(citations, usage)
67
+ buildMetadata: () => buildPerplexityMetadata(citations, images, usage)
82
68
  };
83
69
  }
84
70
  };
71
+ var buildPerplexityMetadata = (citations, images, usage) => {
72
+ var _a, _b;
73
+ return citations || images || usage ? {
74
+ perplexity: {
75
+ ...citations && { citations },
76
+ ...images && {
77
+ images: images.map((image) => ({
78
+ imageUrl: image.image_url,
79
+ originUrl: image.origin_url,
80
+ height: image.height,
81
+ width: image.width
82
+ }))
83
+ },
84
+ ...usage && {
85
+ usage: {
86
+ citationTokens: (_a = usage.citation_tokens) != null ? _a : NaN,
87
+ numSearchQueries: (_b = usage.num_search_queries) != null ? _b : NaN
88
+ }
89
+ }
90
+ }
91
+ } : void 0;
92
+ };
85
93
  var perplexityCitationSchema = import_zod.z.array(import_zod.z.string());
94
+ var perplexityImageSchema = import_zod.z.object({
95
+ image_url: import_zod.z.string(),
96
+ origin_url: import_zod.z.string(),
97
+ height: import_zod.z.number(),
98
+ width: import_zod.z.number()
99
+ });
86
100
  var perplexityUsageSchema = import_zod.z.object({
87
101
  citation_tokens: import_zod.z.number().nullish(),
88
102
  num_search_queries: import_zod.z.number().nullish()
89
103
  });
90
104
  var perplexityResponseSchema = import_zod.z.object({
91
105
  citations: perplexityCitationSchema.nullish(),
106
+ images: import_zod.z.array(perplexityImageSchema).nullish(),
92
107
  usage: perplexityUsageSchema.nullish()
93
108
  });
94
109
  var perplexityStreamChunkSchema = import_zod.z.object({
@@ -102,6 +117,7 @@ var perplexityStreamChunkSchema = import_zod.z.object({
102
117
  })
103
118
  ).nullish(),
104
119
  citations: perplexityCitationSchema.nullish(),
120
+ images: import_zod.z.array(perplexityImageSchema).nullish(),
105
121
  usage: perplexityUsageSchema.nullish()
106
122
  });
107
123
 
package/dist/index.js.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"sources":["../src/index.ts","../src/perplexity-provider.ts","../src/perplexity-metadata-extractor.ts"],"sourcesContent":["export { createPerplexity, perplexity } from './perplexity-provider';\nexport type {\n PerplexityErrorData,\n PerplexityProvider,\n PerplexityProviderSettings,\n} from './perplexity-provider';\n","import {\n LanguageModelV1,\n NoSuchModelError,\n ProviderV1,\n} from '@ai-sdk/provider';\nimport { OpenAICompatibleChatLanguageModel } from '@ai-sdk/openai-compatible';\nimport {\n FetchFunction,\n loadApiKey,\n withoutTrailingSlash,\n} from '@ai-sdk/provider-utils';\nimport {\n PerplexityChatModelId,\n PerplexityChatSettings,\n} from './perplexity-chat-settings';\nimport { z } from 'zod';\nimport { ProviderErrorStructure } from '@ai-sdk/openai-compatible';\nimport { perplexityMetadataExtractor } from './perplexity-metadata-extractor';\n\n// Add error schema and structure\nconst perplexityErrorSchema = z.object({\n code: z.string(),\n error: z.string(),\n});\n\nexport type PerplexityErrorData = z.infer<typeof perplexityErrorSchema>;\n\nconst perplexityErrorStructure: ProviderErrorStructure<PerplexityErrorData> = {\n errorSchema: perplexityErrorSchema,\n errorToMessage: data => data.error,\n};\n\nexport interface PerplexityProvider extends ProviderV1 {\n /**\nCreates an Perplexity chat model for text generation.\n */\n (\n modelId: PerplexityChatModelId,\n settings?: PerplexityChatSettings,\n ): LanguageModelV1;\n\n /**\nCreates an Perplexity language model for text generation.\n */\n languageModel(\n modelId: PerplexityChatModelId,\n settings?: PerplexityChatSettings,\n ): LanguageModelV1;\n\n /**\nCreates an Perplexity chat model for text generation.\n */\n chat: (\n modelId: PerplexityChatModelId,\n settings?: PerplexityChatSettings,\n ) => LanguageModelV1;\n}\n\nexport interface PerplexityProviderSettings {\n /**\nBase URL for the perplexity API calls.\n */\n baseURL?: string;\n\n /**\nAPI key for authenticating requests.\n */\n apiKey?: string;\n\n /**\nCustom headers to include in the requests.\n */\n headers?: Record<string, string>;\n\n /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n */\n fetch?: FetchFunction;\n}\n\nexport function createPerplexity(\n options: PerplexityProviderSettings = {},\n): PerplexityProvider {\n const baseURL = withoutTrailingSlash(\n options.baseURL ?? 'https://api.perplexity.ai',\n );\n const getHeaders = () => ({\n Authorization: `Bearer ${loadApiKey({\n apiKey: options.apiKey,\n environmentVariableName: 'PERPLEXITY_API_KEY',\n description: 'Perplexity',\n })}`,\n ...options.headers,\n });\n\n const createLanguageModel = (\n modelId: PerplexityChatModelId,\n settings: PerplexityChatSettings = {},\n ) => {\n return new OpenAICompatibleChatLanguageModel(modelId, settings, {\n provider: 'perplexity.chat',\n url: ({ path }) => `${baseURL}${path}`,\n headers: getHeaders,\n fetch: options.fetch,\n defaultObjectGenerationMode: 'json',\n errorStructure: perplexityErrorStructure,\n metadataExtractor: perplexityMetadataExtractor,\n supportsStructuredOutputs: true,\n });\n };\n\n const provider = (\n modelId: PerplexityChatModelId,\n settings?: PerplexityChatSettings,\n ) => createLanguageModel(modelId, settings);\n\n provider.languageModel = createLanguageModel;\n provider.chat = createLanguageModel;\n provider.textEmbeddingModel = (modelId: string) => {\n throw new NoSuchModelError({ modelId, modelType: 'textEmbeddingModel' });\n };\n\n return provider as PerplexityProvider;\n}\n\nexport const perplexity = createPerplexity();\n","import { MetadataExtractor } from '@ai-sdk/openai-compatible';\nimport { safeValidateTypes } from '@ai-sdk/provider-utils';\nimport { z } from 'zod';\n\nconst buildPerplexityMetadata = (\n citations: string[] | undefined,\n usage: z.infer<typeof perplexityUsageSchema> | undefined,\n) => {\n return citations || usage\n ? {\n perplexity: {\n ...(citations && { citations }),\n ...(usage && {\n usage: {\n citationTokens: usage.citation_tokens ?? NaN,\n numSearchQueries: usage.num_search_queries ?? NaN,\n },\n }),\n },\n }\n : undefined;\n};\n\nexport const perplexityMetadataExtractor: MetadataExtractor = {\n extractMetadata: ({ parsedBody }: { parsedBody: unknown }) => {\n const parsed = safeValidateTypes({\n value: parsedBody,\n schema: perplexityResponseSchema,\n });\n\n return !parsed.success\n ? undefined\n : buildPerplexityMetadata(\n parsed.value.citations ?? undefined,\n parsed.value.usage ?? undefined,\n );\n },\n\n createStreamExtractor: () => {\n let citations: string[] | undefined;\n let usage: z.infer<typeof perplexityUsageSchema> | undefined;\n\n return {\n processChunk: (chunk: unknown) => {\n const parsed = safeValidateTypes({\n value: chunk,\n schema: perplexityStreamChunkSchema,\n });\n\n if (parsed.success) {\n // Update citations and usage with latest data from each chunk\n if (parsed.value.citations) {\n citations = parsed.value.citations;\n }\n if (parsed.value.usage) {\n usage = parsed.value.usage;\n }\n }\n },\n buildMetadata: () => buildPerplexityMetadata(citations, usage),\n };\n },\n};\n\n// Schema for citations\nconst perplexityCitationSchema = z.array(z.string());\n\nconst perplexityUsageSchema = z.object({\n citation_tokens: z.number().nullish(),\n num_search_queries: z.number().nullish(),\n});\n\n// Update response schema to include usage\nconst perplexityResponseSchema = z.object({\n citations: perplexityCitationSchema.nullish(),\n usage: perplexityUsageSchema.nullish(),\n});\n\n// Update stream chunk schema to match example format\nconst perplexityStreamChunkSchema = z.object({\n choices: z\n .array(\n z.object({\n finish_reason: z.string().nullish(),\n delta: z\n .object({\n role: z.string(),\n content: z.string(),\n })\n .nullish(),\n }),\n )\n .nullish(),\n citations: perplexityCitationSchema.nullish(),\n usage: perplexityUsageSchema.nullish(),\n});\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,sBAIO;AACP,+BAAkD;AAClD,IAAAA,yBAIO;AAKP,IAAAC,cAAkB;;;ACdlB,4BAAkC;AAClC,iBAAkB;AAElB,IAAM,0BAA0B,CAC9B,WACA,UACG;AAPL;AAQE,SAAO,aAAa,QAChB;AAAA,IACE,YAAY;AAAA,MACV,GAAI,aAAa,EAAE,UAAU;AAAA,MAC7B,GAAI,SAAS;AAAA,QACX,OAAO;AAAA,UACL,iBAAgB,WAAM,oBAAN,YAAyB;AAAA,UACzC,mBAAkB,WAAM,uBAAN,YAA4B;AAAA,QAChD;AAAA,MACF;AAAA,IACF;AAAA,EACF,IACA;AACN;AAEO,IAAM,8BAAiD;AAAA,EAC5D,iBAAiB,CAAC,EAAE,WAAW,MAA+B;AAxBhE;AAyBI,UAAM,aAAS,yCAAkB;AAAA,MAC/B,OAAO;AAAA,MACP,QAAQ;AAAA,IACV,CAAC;AAED,WAAO,CAAC,OAAO,UACX,SACA;AAAA,OACE,YAAO,MAAM,cAAb,YAA0B;AAAA,OAC1B,YAAO,MAAM,UAAb,YAAsB;AAAA,IACxB;AAAA,EACN;AAAA,EAEA,uBAAuB,MAAM;AAC3B,QAAI;AACJ,QAAI;AAEJ,WAAO;AAAA,MACL,cAAc,CAAC,UAAmB;AAChC,cAAM,aAAS,yCAAkB;AAAA,UAC/B,OAAO;AAAA,UACP,QAAQ;AAAA,QACV,CAAC;AAED,YAAI,OAAO,SAAS;AAElB,cAAI,OAAO,MAAM,WAAW;AAC1B,wBAAY,OAAO,MAAM;AAAA,UAC3B;AACA,cAAI,OAAO,MAAM,OAAO;AACtB,oBAAQ,OAAO,MAAM;AAAA,UACvB;AAAA,QACF;AAAA,MACF;AAAA,MACA,eAAe,MAAM,wBAAwB,WAAW,KAAK;AAAA,IAC/D;AAAA,EACF;AACF;AAGA,IAAM,2BAA2B,aAAE,MAAM,aAAE,OAAO,CAAC;AAEnD,IAAM,wBAAwB,aAAE,OAAO;AAAA,EACrC,iBAAiB,aAAE,OAAO,EAAE,QAAQ;AAAA,EACpC,oBAAoB,aAAE,OAAO,EAAE,QAAQ;AACzC,CAAC;AAGD,IAAM,2BAA2B,aAAE,OAAO;AAAA,EACxC,WAAW,yBAAyB,QAAQ;AAAA,EAC5C,OAAO,sBAAsB,QAAQ;AACvC,CAAC;AAGD,IAAM,8BAA8B,aAAE,OAAO;AAAA,EAC3C,SAAS,aACN;AAAA,IACC,aAAE,OAAO;AAAA,MACP,eAAe,aAAE,OAAO,EAAE,QAAQ;AAAA,MAClC,OAAO,aACJ,OAAO;AAAA,QACN,MAAM,aAAE,OAAO;AAAA,QACf,SAAS,aAAE,OAAO;AAAA,MACpB,CAAC,EACA,QAAQ;AAAA,IACb,CAAC;AAAA,EACH,EACC,QAAQ;AAAA,EACX,WAAW,yBAAyB,QAAQ;AAAA,EAC5C,OAAO,sBAAsB,QAAQ;AACvC,CAAC;;;AD3ED,IAAM,wBAAwB,cAAE,OAAO;AAAA,EACrC,MAAM,cAAE,OAAO;AAAA,EACf,OAAO,cAAE,OAAO;AAClB,CAAC;AAID,IAAM,2BAAwE;AAAA,EAC5E,aAAa;AAAA,EACb,gBAAgB,UAAQ,KAAK;AAC/B;AAmDO,SAAS,iBACd,UAAsC,CAAC,GACnB;AAnFtB;AAoFE,QAAM,cAAU;AAAA,KACd,aAAQ,YAAR,YAAmB;AAAA,EACrB;AACA,QAAM,aAAa,OAAO;AAAA,IACxB,eAAe,cAAU,mCAAW;AAAA,MAClC,QAAQ,QAAQ;AAAA,MAChB,yBAAyB;AAAA,MACzB,aAAa;AAAA,IACf,CAAC,CAAC;AAAA,IACF,GAAG,QAAQ;AAAA,EACb;AAEA,QAAM,sBAAsB,CAC1B,SACA,WAAmC,CAAC,MACjC;AACH,WAAO,IAAI,2DAAkC,SAAS,UAAU;AAAA,MAC9D,UAAU;AAAA,MACV,KAAK,CAAC,EAAE,KAAK,MAAM,GAAG,OAAO,GAAG,IAAI;AAAA,MACpC,SAAS;AAAA,MACT,OAAO,QAAQ;AAAA,MACf,6BAA6B;AAAA,MAC7B,gBAAgB;AAAA,MAChB,mBAAmB;AAAA,MACnB,2BAA2B;AAAA,IAC7B,CAAC;AAAA,EACH;AAEA,QAAM,WAAW,CACf,SACA,aACG,oBAAoB,SAAS,QAAQ;AAE1C,WAAS,gBAAgB;AACzB,WAAS,OAAO;AAChB,WAAS,qBAAqB,CAAC,YAAoB;AACjD,UAAM,IAAI,iCAAiB,EAAE,SAAS,WAAW,qBAAqB,CAAC;AAAA,EACzE;AAEA,SAAO;AACT;AAEO,IAAM,aAAa,iBAAiB;","names":["import_provider_utils","import_zod"]}
1
+ {"version":3,"sources":["../src/index.ts","../src/perplexity-provider.ts","../src/perplexity-metadata-extractor.ts"],"sourcesContent":["export { createPerplexity, perplexity } from './perplexity-provider';\nexport type {\n PerplexityErrorData,\n PerplexityProvider,\n PerplexityProviderSettings,\n} from './perplexity-provider';\n","import {\n LanguageModelV1,\n NoSuchModelError,\n ProviderV1,\n} from '@ai-sdk/provider';\nimport { OpenAICompatibleChatLanguageModel } from '@ai-sdk/openai-compatible';\nimport {\n FetchFunction,\n loadApiKey,\n withoutTrailingSlash,\n} from '@ai-sdk/provider-utils';\nimport {\n PerplexityChatModelId,\n PerplexityChatSettings,\n} from './perplexity-chat-settings';\nimport { z } from 'zod';\nimport { ProviderErrorStructure } from '@ai-sdk/openai-compatible';\nimport { perplexityMetadataExtractor } from './perplexity-metadata-extractor';\n\n// Add error schema and structure\nconst perplexityErrorSchema = z.object({\n code: z.string(),\n error: z.string(),\n});\n\nexport type PerplexityErrorData = z.infer<typeof perplexityErrorSchema>;\n\nconst perplexityErrorStructure: ProviderErrorStructure<PerplexityErrorData> = {\n errorSchema: perplexityErrorSchema,\n errorToMessage: data => data.error,\n};\n\nexport interface PerplexityProvider extends ProviderV1 {\n /**\nCreates an Perplexity chat model for text generation.\n */\n (\n modelId: PerplexityChatModelId,\n settings?: PerplexityChatSettings,\n ): LanguageModelV1;\n\n /**\nCreates an Perplexity language model for text generation.\n */\n languageModel(\n modelId: PerplexityChatModelId,\n settings?: PerplexityChatSettings,\n ): LanguageModelV1;\n\n /**\nCreates an Perplexity chat model for text generation.\n */\n chat: (\n modelId: PerplexityChatModelId,\n settings?: PerplexityChatSettings,\n ) => LanguageModelV1;\n}\n\nexport interface PerplexityProviderSettings {\n /**\nBase URL for the perplexity API calls.\n */\n baseURL?: string;\n\n /**\nAPI key for authenticating requests.\n */\n apiKey?: string;\n\n /**\nCustom headers to include in the requests.\n */\n headers?: Record<string, string>;\n\n /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n */\n fetch?: FetchFunction;\n}\n\nexport function createPerplexity(\n options: PerplexityProviderSettings = {},\n): PerplexityProvider {\n const baseURL = withoutTrailingSlash(\n options.baseURL ?? 'https://api.perplexity.ai',\n );\n const getHeaders = () => ({\n Authorization: `Bearer ${loadApiKey({\n apiKey: options.apiKey,\n environmentVariableName: 'PERPLEXITY_API_KEY',\n description: 'Perplexity',\n })}`,\n ...options.headers,\n });\n\n const createLanguageModel = (\n modelId: PerplexityChatModelId,\n settings: PerplexityChatSettings = {},\n ) => {\n return new OpenAICompatibleChatLanguageModel(modelId, settings, {\n provider: 'perplexity.chat',\n url: ({ path }) => `${baseURL}${path}`,\n headers: getHeaders,\n fetch: options.fetch,\n defaultObjectGenerationMode: 'json',\n errorStructure: perplexityErrorStructure,\n metadataExtractor: perplexityMetadataExtractor,\n supportsStructuredOutputs: true,\n });\n };\n\n const provider = (\n modelId: PerplexityChatModelId,\n settings?: PerplexityChatSettings,\n ) => createLanguageModel(modelId, settings);\n\n provider.languageModel = createLanguageModel;\n provider.chat = createLanguageModel;\n provider.textEmbeddingModel = (modelId: string) => {\n throw new NoSuchModelError({ modelId, modelType: 'textEmbeddingModel' });\n };\n\n return provider;\n}\n\nexport const perplexity = createPerplexity();\n","import { MetadataExtractor } from '@ai-sdk/openai-compatible';\nimport { safeValidateTypes } from '@ai-sdk/provider-utils';\nimport { z } from 'zod';\n\nexport const perplexityMetadataExtractor: MetadataExtractor = {\n extractMetadata: ({ parsedBody }: { parsedBody: unknown }) => {\n const parsed = safeValidateTypes({\n value: parsedBody,\n schema: perplexityResponseSchema,\n });\n\n return !parsed.success\n ? undefined\n : buildPerplexityMetadata(\n parsed.value.citations ?? undefined,\n parsed.value.images ?? undefined,\n parsed.value.usage ?? undefined,\n );\n },\n\n createStreamExtractor: () => {\n let citations: string[] | undefined;\n let images: PerplexityImageData[] | undefined;\n let usage: PerplexityUsageData | undefined;\n\n return {\n processChunk: (chunk: unknown) => {\n const parsed = safeValidateTypes({\n value: chunk,\n schema: perplexityStreamChunkSchema,\n });\n\n if (parsed.success) {\n citations = parsed.value.citations ?? citations;\n images = parsed.value.images ?? images;\n usage = parsed.value.usage ?? usage;\n }\n },\n buildMetadata: () => buildPerplexityMetadata(citations, images, usage),\n };\n },\n};\n\nconst buildPerplexityMetadata = (\n citations: string[] | undefined,\n images: PerplexityImageData[] | undefined,\n usage: PerplexityUsageData | undefined,\n) => {\n return citations || images || usage\n ? {\n perplexity: {\n ...(citations && { citations }),\n ...(images && {\n images: images.map(image => ({\n imageUrl: image.image_url,\n originUrl: image.origin_url,\n height: image.height,\n width: image.width,\n })),\n }),\n ...(usage && {\n usage: {\n citationTokens: usage.citation_tokens ?? NaN,\n numSearchQueries: usage.num_search_queries ?? NaN,\n },\n }),\n },\n }\n : undefined;\n};\n\nconst perplexityCitationSchema = z.array(z.string());\n\nconst perplexityImageSchema = z.object({\n image_url: z.string(),\n origin_url: z.string(),\n height: z.number(),\n width: z.number(),\n});\n\ntype PerplexityImageData = z.infer<typeof perplexityImageSchema>;\n\nconst perplexityUsageSchema = z.object({\n citation_tokens: z.number().nullish(),\n num_search_queries: z.number().nullish(),\n});\n\ntype PerplexityUsageData = z.infer<typeof perplexityUsageSchema>;\n\nconst perplexityResponseSchema = z.object({\n citations: perplexityCitationSchema.nullish(),\n images: z.array(perplexityImageSchema).nullish(),\n usage: perplexityUsageSchema.nullish(),\n});\n\nconst perplexityStreamChunkSchema = z.object({\n choices: z\n .array(\n z.object({\n finish_reason: z.string().nullish(),\n delta: z\n .object({\n role: z.string(),\n content: z.string(),\n })\n .nullish(),\n }),\n )\n .nullish(),\n citations: perplexityCitationSchema.nullish(),\n images: z.array(perplexityImageSchema).nullish(),\n usage: perplexityUsageSchema.nullish(),\n});\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,sBAIO;AACP,+BAAkD;AAClD,IAAAA,yBAIO;AAKP,IAAAC,cAAkB;;;ACdlB,4BAAkC;AAClC,iBAAkB;AAEX,IAAM,8BAAiD;AAAA,EAC5D,iBAAiB,CAAC,EAAE,WAAW,MAA+B;AALhE;AAMI,UAAM,aAAS,yCAAkB;AAAA,MAC/B,OAAO;AAAA,MACP,QAAQ;AAAA,IACV,CAAC;AAED,WAAO,CAAC,OAAO,UACX,SACA;AAAA,OACE,YAAO,MAAM,cAAb,YAA0B;AAAA,OAC1B,YAAO,MAAM,WAAb,YAAuB;AAAA,OACvB,YAAO,MAAM,UAAb,YAAsB;AAAA,IACxB;AAAA,EACN;AAAA,EAEA,uBAAuB,MAAM;AAC3B,QAAI;AACJ,QAAI;AACJ,QAAI;AAEJ,WAAO;AAAA,MACL,cAAc,CAAC,UAAmB;AA1BxC;AA2BQ,cAAM,aAAS,yCAAkB;AAAA,UAC/B,OAAO;AAAA,UACP,QAAQ;AAAA,QACV,CAAC;AAED,YAAI,OAAO,SAAS;AAClB,uBAAY,YAAO,MAAM,cAAb,YAA0B;AACtC,oBAAS,YAAO,MAAM,WAAb,YAAuB;AAChC,mBAAQ,YAAO,MAAM,UAAb,YAAsB;AAAA,QAChC;AAAA,MACF;AAAA,MACA,eAAe,MAAM,wBAAwB,WAAW,QAAQ,KAAK;AAAA,IACvE;AAAA,EACF;AACF;AAEA,IAAM,0BAA0B,CAC9B,WACA,QACA,UACG;AA/CL;AAgDE,SAAO,aAAa,UAAU,QAC1B;AAAA,IACE,YAAY;AAAA,MACV,GAAI,aAAa,EAAE,UAAU;AAAA,MAC7B,GAAI,UAAU;AAAA,QACZ,QAAQ,OAAO,IAAI,YAAU;AAAA,UAC3B,UAAU,MAAM;AAAA,UAChB,WAAW,MAAM;AAAA,UACjB,QAAQ,MAAM;AAAA,UACd,OAAO,MAAM;AAAA,QACf,EAAE;AAAA,MACJ;AAAA,MACA,GAAI,SAAS;AAAA,QACX,OAAO;AAAA,UACL,iBAAgB,WAAM,oBAAN,YAAyB;AAAA,UACzC,mBAAkB,WAAM,uBAAN,YAA4B;AAAA,QAChD;AAAA,MACF;AAAA,IACF;AAAA,EACF,IACA;AACN;AAEA,IAAM,2BAA2B,aAAE,MAAM,aAAE,OAAO,CAAC;AAEnD,IAAM,wBAAwB,aAAE,OAAO;AAAA,EACrC,WAAW,aAAE,OAAO;AAAA,EACpB,YAAY,aAAE,OAAO;AAAA,EACrB,QAAQ,aAAE,OAAO;AAAA,EACjB,OAAO,aAAE,OAAO;AAClB,CAAC;AAID,IAAM,wBAAwB,aAAE,OAAO;AAAA,EACrC,iBAAiB,aAAE,OAAO,EAAE,QAAQ;AAAA,EACpC,oBAAoB,aAAE,OAAO,EAAE,QAAQ;AACzC,CAAC;AAID,IAAM,2BAA2B,aAAE,OAAO;AAAA,EACxC,WAAW,yBAAyB,QAAQ;AAAA,EAC5C,QAAQ,aAAE,MAAM,qBAAqB,EAAE,QAAQ;AAAA,EAC/C,OAAO,sBAAsB,QAAQ;AACvC,CAAC;AAED,IAAM,8BAA8B,aAAE,OAAO;AAAA,EAC3C,SAAS,aACN;AAAA,IACC,aAAE,OAAO;AAAA,MACP,eAAe,aAAE,OAAO,EAAE,QAAQ;AAAA,MAClC,OAAO,aACJ,OAAO;AAAA,QACN,MAAM,aAAE,OAAO;AAAA,QACf,SAAS,aAAE,OAAO;AAAA,MACpB,CAAC,EACA,QAAQ;AAAA,IACb,CAAC;AAAA,EACH,EACC,QAAQ;AAAA,EACX,WAAW,yBAAyB,QAAQ;AAAA,EAC5C,QAAQ,aAAE,MAAM,qBAAqB,EAAE,QAAQ;AAAA,EAC/C,OAAO,sBAAsB,QAAQ;AACvC,CAAC;;;AD5FD,IAAM,wBAAwB,cAAE,OAAO;AAAA,EACrC,MAAM,cAAE,OAAO;AAAA,EACf,OAAO,cAAE,OAAO;AAClB,CAAC;AAID,IAAM,2BAAwE;AAAA,EAC5E,aAAa;AAAA,EACb,gBAAgB,UAAQ,KAAK;AAC/B;AAmDO,SAAS,iBACd,UAAsC,CAAC,GACnB;AAnFtB;AAoFE,QAAM,cAAU;AAAA,KACd,aAAQ,YAAR,YAAmB;AAAA,EACrB;AACA,QAAM,aAAa,OAAO;AAAA,IACxB,eAAe,cAAU,mCAAW;AAAA,MAClC,QAAQ,QAAQ;AAAA,MAChB,yBAAyB;AAAA,MACzB,aAAa;AAAA,IACf,CAAC,CAAC;AAAA,IACF,GAAG,QAAQ;AAAA,EACb;AAEA,QAAM,sBAAsB,CAC1B,SACA,WAAmC,CAAC,MACjC;AACH,WAAO,IAAI,2DAAkC,SAAS,UAAU;AAAA,MAC9D,UAAU;AAAA,MACV,KAAK,CAAC,EAAE,KAAK,MAAM,GAAG,OAAO,GAAG,IAAI;AAAA,MACpC,SAAS;AAAA,MACT,OAAO,QAAQ;AAAA,MACf,6BAA6B;AAAA,MAC7B,gBAAgB;AAAA,MAChB,mBAAmB;AAAA,MACnB,2BAA2B;AAAA,IAC7B,CAAC;AAAA,EACH;AAEA,QAAM,WAAW,CACf,SACA,aACG,oBAAoB,SAAS,QAAQ;AAE1C,WAAS,gBAAgB;AACzB,WAAS,OAAO;AAChB,WAAS,qBAAqB,CAAC,YAAoB;AACjD,UAAM,IAAI,iCAAiB,EAAE,SAAS,WAAW,qBAAqB,CAAC;AAAA,EACzE;AAEA,SAAO;AACT;AAEO,IAAM,aAAa,iBAAiB;","names":["import_provider_utils","import_zod"]}
package/dist/index.mjs CHANGED
@@ -12,61 +12,76 @@ import { z as z2 } from "zod";
12
12
  // src/perplexity-metadata-extractor.ts
13
13
  import { safeValidateTypes } from "@ai-sdk/provider-utils";
14
14
  import { z } from "zod";
15
- var buildPerplexityMetadata = (citations, usage) => {
16
- var _a, _b;
17
- return citations || usage ? {
18
- perplexity: {
19
- ...citations && { citations },
20
- ...usage && {
21
- usage: {
22
- citationTokens: (_a = usage.citation_tokens) != null ? _a : NaN,
23
- numSearchQueries: (_b = usage.num_search_queries) != null ? _b : NaN
24
- }
25
- }
26
- }
27
- } : void 0;
28
- };
29
15
  var perplexityMetadataExtractor = {
30
16
  extractMetadata: ({ parsedBody }) => {
31
- var _a, _b;
17
+ var _a, _b, _c;
32
18
  const parsed = safeValidateTypes({
33
19
  value: parsedBody,
34
20
  schema: perplexityResponseSchema
35
21
  });
36
22
  return !parsed.success ? void 0 : buildPerplexityMetadata(
37
23
  (_a = parsed.value.citations) != null ? _a : void 0,
38
- (_b = parsed.value.usage) != null ? _b : void 0
24
+ (_b = parsed.value.images) != null ? _b : void 0,
25
+ (_c = parsed.value.usage) != null ? _c : void 0
39
26
  );
40
27
  },
41
28
  createStreamExtractor: () => {
42
29
  let citations;
30
+ let images;
43
31
  let usage;
44
32
  return {
45
33
  processChunk: (chunk) => {
34
+ var _a, _b, _c;
46
35
  const parsed = safeValidateTypes({
47
36
  value: chunk,
48
37
  schema: perplexityStreamChunkSchema
49
38
  });
50
39
  if (parsed.success) {
51
- if (parsed.value.citations) {
52
- citations = parsed.value.citations;
53
- }
54
- if (parsed.value.usage) {
55
- usage = parsed.value.usage;
56
- }
40
+ citations = (_a = parsed.value.citations) != null ? _a : citations;
41
+ images = (_b = parsed.value.images) != null ? _b : images;
42
+ usage = (_c = parsed.value.usage) != null ? _c : usage;
57
43
  }
58
44
  },
59
- buildMetadata: () => buildPerplexityMetadata(citations, usage)
45
+ buildMetadata: () => buildPerplexityMetadata(citations, images, usage)
60
46
  };
61
47
  }
62
48
  };
49
+ var buildPerplexityMetadata = (citations, images, usage) => {
50
+ var _a, _b;
51
+ return citations || images || usage ? {
52
+ perplexity: {
53
+ ...citations && { citations },
54
+ ...images && {
55
+ images: images.map((image) => ({
56
+ imageUrl: image.image_url,
57
+ originUrl: image.origin_url,
58
+ height: image.height,
59
+ width: image.width
60
+ }))
61
+ },
62
+ ...usage && {
63
+ usage: {
64
+ citationTokens: (_a = usage.citation_tokens) != null ? _a : NaN,
65
+ numSearchQueries: (_b = usage.num_search_queries) != null ? _b : NaN
66
+ }
67
+ }
68
+ }
69
+ } : void 0;
70
+ };
63
71
  var perplexityCitationSchema = z.array(z.string());
72
+ var perplexityImageSchema = z.object({
73
+ image_url: z.string(),
74
+ origin_url: z.string(),
75
+ height: z.number(),
76
+ width: z.number()
77
+ });
64
78
  var perplexityUsageSchema = z.object({
65
79
  citation_tokens: z.number().nullish(),
66
80
  num_search_queries: z.number().nullish()
67
81
  });
68
82
  var perplexityResponseSchema = z.object({
69
83
  citations: perplexityCitationSchema.nullish(),
84
+ images: z.array(perplexityImageSchema).nullish(),
70
85
  usage: perplexityUsageSchema.nullish()
71
86
  });
72
87
  var perplexityStreamChunkSchema = z.object({
@@ -80,6 +95,7 @@ var perplexityStreamChunkSchema = z.object({
80
95
  })
81
96
  ).nullish(),
82
97
  citations: perplexityCitationSchema.nullish(),
98
+ images: z.array(perplexityImageSchema).nullish(),
83
99
  usage: perplexityUsageSchema.nullish()
84
100
  });
85
101
 
@@ -1 +1 @@
1
- {"version":3,"sources":["../src/perplexity-provider.ts","../src/perplexity-metadata-extractor.ts"],"sourcesContent":["import {\n LanguageModelV1,\n NoSuchModelError,\n ProviderV1,\n} from '@ai-sdk/provider';\nimport { OpenAICompatibleChatLanguageModel } from '@ai-sdk/openai-compatible';\nimport {\n FetchFunction,\n loadApiKey,\n withoutTrailingSlash,\n} from '@ai-sdk/provider-utils';\nimport {\n PerplexityChatModelId,\n PerplexityChatSettings,\n} from './perplexity-chat-settings';\nimport { z } from 'zod';\nimport { ProviderErrorStructure } from '@ai-sdk/openai-compatible';\nimport { perplexityMetadataExtractor } from './perplexity-metadata-extractor';\n\n// Add error schema and structure\nconst perplexityErrorSchema = z.object({\n code: z.string(),\n error: z.string(),\n});\n\nexport type PerplexityErrorData = z.infer<typeof perplexityErrorSchema>;\n\nconst perplexityErrorStructure: ProviderErrorStructure<PerplexityErrorData> = {\n errorSchema: perplexityErrorSchema,\n errorToMessage: data => data.error,\n};\n\nexport interface PerplexityProvider extends ProviderV1 {\n /**\nCreates an Perplexity chat model for text generation.\n */\n (\n modelId: PerplexityChatModelId,\n settings?: PerplexityChatSettings,\n ): LanguageModelV1;\n\n /**\nCreates an Perplexity language model for text generation.\n */\n languageModel(\n modelId: PerplexityChatModelId,\n settings?: PerplexityChatSettings,\n ): LanguageModelV1;\n\n /**\nCreates an Perplexity chat model for text generation.\n */\n chat: (\n modelId: PerplexityChatModelId,\n settings?: PerplexityChatSettings,\n ) => LanguageModelV1;\n}\n\nexport interface PerplexityProviderSettings {\n /**\nBase URL for the perplexity API calls.\n */\n baseURL?: string;\n\n /**\nAPI key for authenticating requests.\n */\n apiKey?: string;\n\n /**\nCustom headers to include in the requests.\n */\n headers?: Record<string, string>;\n\n /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n */\n fetch?: FetchFunction;\n}\n\nexport function createPerplexity(\n options: PerplexityProviderSettings = {},\n): PerplexityProvider {\n const baseURL = withoutTrailingSlash(\n options.baseURL ?? 'https://api.perplexity.ai',\n );\n const getHeaders = () => ({\n Authorization: `Bearer ${loadApiKey({\n apiKey: options.apiKey,\n environmentVariableName: 'PERPLEXITY_API_KEY',\n description: 'Perplexity',\n })}`,\n ...options.headers,\n });\n\n const createLanguageModel = (\n modelId: PerplexityChatModelId,\n settings: PerplexityChatSettings = {},\n ) => {\n return new OpenAICompatibleChatLanguageModel(modelId, settings, {\n provider: 'perplexity.chat',\n url: ({ path }) => `${baseURL}${path}`,\n headers: getHeaders,\n fetch: options.fetch,\n defaultObjectGenerationMode: 'json',\n errorStructure: perplexityErrorStructure,\n metadataExtractor: perplexityMetadataExtractor,\n supportsStructuredOutputs: true,\n });\n };\n\n const provider = (\n modelId: PerplexityChatModelId,\n settings?: PerplexityChatSettings,\n ) => createLanguageModel(modelId, settings);\n\n provider.languageModel = createLanguageModel;\n provider.chat = createLanguageModel;\n provider.textEmbeddingModel = (modelId: string) => {\n throw new NoSuchModelError({ modelId, modelType: 'textEmbeddingModel' });\n };\n\n return provider as PerplexityProvider;\n}\n\nexport const perplexity = createPerplexity();\n","import { MetadataExtractor } from '@ai-sdk/openai-compatible';\nimport { safeValidateTypes } from '@ai-sdk/provider-utils';\nimport { z } from 'zod';\n\nconst buildPerplexityMetadata = (\n citations: string[] | undefined,\n usage: z.infer<typeof perplexityUsageSchema> | undefined,\n) => {\n return citations || usage\n ? {\n perplexity: {\n ...(citations && { citations }),\n ...(usage && {\n usage: {\n citationTokens: usage.citation_tokens ?? NaN,\n numSearchQueries: usage.num_search_queries ?? NaN,\n },\n }),\n },\n }\n : undefined;\n};\n\nexport const perplexityMetadataExtractor: MetadataExtractor = {\n extractMetadata: ({ parsedBody }: { parsedBody: unknown }) => {\n const parsed = safeValidateTypes({\n value: parsedBody,\n schema: perplexityResponseSchema,\n });\n\n return !parsed.success\n ? undefined\n : buildPerplexityMetadata(\n parsed.value.citations ?? undefined,\n parsed.value.usage ?? undefined,\n );\n },\n\n createStreamExtractor: () => {\n let citations: string[] | undefined;\n let usage: z.infer<typeof perplexityUsageSchema> | undefined;\n\n return {\n processChunk: (chunk: unknown) => {\n const parsed = safeValidateTypes({\n value: chunk,\n schema: perplexityStreamChunkSchema,\n });\n\n if (parsed.success) {\n // Update citations and usage with latest data from each chunk\n if (parsed.value.citations) {\n citations = parsed.value.citations;\n }\n if (parsed.value.usage) {\n usage = parsed.value.usage;\n }\n }\n },\n buildMetadata: () => buildPerplexityMetadata(citations, usage),\n };\n },\n};\n\n// Schema for citations\nconst perplexityCitationSchema = z.array(z.string());\n\nconst perplexityUsageSchema = z.object({\n citation_tokens: z.number().nullish(),\n num_search_queries: z.number().nullish(),\n});\n\n// Update response schema to include usage\nconst perplexityResponseSchema = z.object({\n citations: perplexityCitationSchema.nullish(),\n usage: perplexityUsageSchema.nullish(),\n});\n\n// Update stream chunk schema to match example format\nconst perplexityStreamChunkSchema = z.object({\n choices: z\n .array(\n z.object({\n finish_reason: z.string().nullish(),\n delta: z\n .object({\n role: z.string(),\n content: z.string(),\n })\n .nullish(),\n }),\n )\n .nullish(),\n citations: perplexityCitationSchema.nullish(),\n usage: perplexityUsageSchema.nullish(),\n});\n"],"mappings":";AAAA;AAAA,EAEE;AAAA,OAEK;AACP,SAAS,yCAAyC;AAClD;AAAA,EAEE;AAAA,EACA;AAAA,OACK;AAKP,SAAS,KAAAA,UAAS;;;ACdlB,SAAS,yBAAyB;AAClC,SAAS,SAAS;AAElB,IAAM,0BAA0B,CAC9B,WACA,UACG;AAPL;AAQE,SAAO,aAAa,QAChB;AAAA,IACE,YAAY;AAAA,MACV,GAAI,aAAa,EAAE,UAAU;AAAA,MAC7B,GAAI,SAAS;AAAA,QACX,OAAO;AAAA,UACL,iBAAgB,WAAM,oBAAN,YAAyB;AAAA,UACzC,mBAAkB,WAAM,uBAAN,YAA4B;AAAA,QAChD;AAAA,MACF;AAAA,IACF;AAAA,EACF,IACA;AACN;AAEO,IAAM,8BAAiD;AAAA,EAC5D,iBAAiB,CAAC,EAAE,WAAW,MAA+B;AAxBhE;AAyBI,UAAM,SAAS,kBAAkB;AAAA,MAC/B,OAAO;AAAA,MACP,QAAQ;AAAA,IACV,CAAC;AAED,WAAO,CAAC,OAAO,UACX,SACA;AAAA,OACE,YAAO,MAAM,cAAb,YAA0B;AAAA,OAC1B,YAAO,MAAM,UAAb,YAAsB;AAAA,IACxB;AAAA,EACN;AAAA,EAEA,uBAAuB,MAAM;AAC3B,QAAI;AACJ,QAAI;AAEJ,WAAO;AAAA,MACL,cAAc,CAAC,UAAmB;AAChC,cAAM,SAAS,kBAAkB;AAAA,UAC/B,OAAO;AAAA,UACP,QAAQ;AAAA,QACV,CAAC;AAED,YAAI,OAAO,SAAS;AAElB,cAAI,OAAO,MAAM,WAAW;AAC1B,wBAAY,OAAO,MAAM;AAAA,UAC3B;AACA,cAAI,OAAO,MAAM,OAAO;AACtB,oBAAQ,OAAO,MAAM;AAAA,UACvB;AAAA,QACF;AAAA,MACF;AAAA,MACA,eAAe,MAAM,wBAAwB,WAAW,KAAK;AAAA,IAC/D;AAAA,EACF;AACF;AAGA,IAAM,2BAA2B,EAAE,MAAM,EAAE,OAAO,CAAC;AAEnD,IAAM,wBAAwB,EAAE,OAAO;AAAA,EACrC,iBAAiB,EAAE,OAAO,EAAE,QAAQ;AAAA,EACpC,oBAAoB,EAAE,OAAO,EAAE,QAAQ;AACzC,CAAC;AAGD,IAAM,2BAA2B,EAAE,OAAO;AAAA,EACxC,WAAW,yBAAyB,QAAQ;AAAA,EAC5C,OAAO,sBAAsB,QAAQ;AACvC,CAAC;AAGD,IAAM,8BAA8B,EAAE,OAAO;AAAA,EAC3C,SAAS,EACN;AAAA,IACC,EAAE,OAAO;AAAA,MACP,eAAe,EAAE,OAAO,EAAE,QAAQ;AAAA,MAClC,OAAO,EACJ,OAAO;AAAA,QACN,MAAM,EAAE,OAAO;AAAA,QACf,SAAS,EAAE,OAAO;AAAA,MACpB,CAAC,EACA,QAAQ;AAAA,IACb,CAAC;AAAA,EACH,EACC,QAAQ;AAAA,EACX,WAAW,yBAAyB,QAAQ;AAAA,EAC5C,OAAO,sBAAsB,QAAQ;AACvC,CAAC;;;AD3ED,IAAM,wBAAwBC,GAAE,OAAO;AAAA,EACrC,MAAMA,GAAE,OAAO;AAAA,EACf,OAAOA,GAAE,OAAO;AAClB,CAAC;AAID,IAAM,2BAAwE;AAAA,EAC5E,aAAa;AAAA,EACb,gBAAgB,UAAQ,KAAK;AAC/B;AAmDO,SAAS,iBACd,UAAsC,CAAC,GACnB;AAnFtB;AAoFE,QAAM,UAAU;AAAA,KACd,aAAQ,YAAR,YAAmB;AAAA,EACrB;AACA,QAAM,aAAa,OAAO;AAAA,IACxB,eAAe,UAAU,WAAW;AAAA,MAClC,QAAQ,QAAQ;AAAA,MAChB,yBAAyB;AAAA,MACzB,aAAa;AAAA,IACf,CAAC,CAAC;AAAA,IACF,GAAG,QAAQ;AAAA,EACb;AAEA,QAAM,sBAAsB,CAC1B,SACA,WAAmC,CAAC,MACjC;AACH,WAAO,IAAI,kCAAkC,SAAS,UAAU;AAAA,MAC9D,UAAU;AAAA,MACV,KAAK,CAAC,EAAE,KAAK,MAAM,GAAG,OAAO,GAAG,IAAI;AAAA,MACpC,SAAS;AAAA,MACT,OAAO,QAAQ;AAAA,MACf,6BAA6B;AAAA,MAC7B,gBAAgB;AAAA,MAChB,mBAAmB;AAAA,MACnB,2BAA2B;AAAA,IAC7B,CAAC;AAAA,EACH;AAEA,QAAM,WAAW,CACf,SACA,aACG,oBAAoB,SAAS,QAAQ;AAE1C,WAAS,gBAAgB;AACzB,WAAS,OAAO;AAChB,WAAS,qBAAqB,CAAC,YAAoB;AACjD,UAAM,IAAI,iBAAiB,EAAE,SAAS,WAAW,qBAAqB,CAAC;AAAA,EACzE;AAEA,SAAO;AACT;AAEO,IAAM,aAAa,iBAAiB;","names":["z","z"]}
1
+ {"version":3,"sources":["../src/perplexity-provider.ts","../src/perplexity-metadata-extractor.ts"],"sourcesContent":["import {\n LanguageModelV1,\n NoSuchModelError,\n ProviderV1,\n} from '@ai-sdk/provider';\nimport { OpenAICompatibleChatLanguageModel } from '@ai-sdk/openai-compatible';\nimport {\n FetchFunction,\n loadApiKey,\n withoutTrailingSlash,\n} from '@ai-sdk/provider-utils';\nimport {\n PerplexityChatModelId,\n PerplexityChatSettings,\n} from './perplexity-chat-settings';\nimport { z } from 'zod';\nimport { ProviderErrorStructure } from '@ai-sdk/openai-compatible';\nimport { perplexityMetadataExtractor } from './perplexity-metadata-extractor';\n\n// Add error schema and structure\nconst perplexityErrorSchema = z.object({\n code: z.string(),\n error: z.string(),\n});\n\nexport type PerplexityErrorData = z.infer<typeof perplexityErrorSchema>;\n\nconst perplexityErrorStructure: ProviderErrorStructure<PerplexityErrorData> = {\n errorSchema: perplexityErrorSchema,\n errorToMessage: data => data.error,\n};\n\nexport interface PerplexityProvider extends ProviderV1 {\n /**\nCreates an Perplexity chat model for text generation.\n */\n (\n modelId: PerplexityChatModelId,\n settings?: PerplexityChatSettings,\n ): LanguageModelV1;\n\n /**\nCreates an Perplexity language model for text generation.\n */\n languageModel(\n modelId: PerplexityChatModelId,\n settings?: PerplexityChatSettings,\n ): LanguageModelV1;\n\n /**\nCreates an Perplexity chat model for text generation.\n */\n chat: (\n modelId: PerplexityChatModelId,\n settings?: PerplexityChatSettings,\n ) => LanguageModelV1;\n}\n\nexport interface PerplexityProviderSettings {\n /**\nBase URL for the perplexity API calls.\n */\n baseURL?: string;\n\n /**\nAPI key for authenticating requests.\n */\n apiKey?: string;\n\n /**\nCustom headers to include in the requests.\n */\n headers?: Record<string, string>;\n\n /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n */\n fetch?: FetchFunction;\n}\n\nexport function createPerplexity(\n options: PerplexityProviderSettings = {},\n): PerplexityProvider {\n const baseURL = withoutTrailingSlash(\n options.baseURL ?? 'https://api.perplexity.ai',\n );\n const getHeaders = () => ({\n Authorization: `Bearer ${loadApiKey({\n apiKey: options.apiKey,\n environmentVariableName: 'PERPLEXITY_API_KEY',\n description: 'Perplexity',\n })}`,\n ...options.headers,\n });\n\n const createLanguageModel = (\n modelId: PerplexityChatModelId,\n settings: PerplexityChatSettings = {},\n ) => {\n return new OpenAICompatibleChatLanguageModel(modelId, settings, {\n provider: 'perplexity.chat',\n url: ({ path }) => `${baseURL}${path}`,\n headers: getHeaders,\n fetch: options.fetch,\n defaultObjectGenerationMode: 'json',\n errorStructure: perplexityErrorStructure,\n metadataExtractor: perplexityMetadataExtractor,\n supportsStructuredOutputs: true,\n });\n };\n\n const provider = (\n modelId: PerplexityChatModelId,\n settings?: PerplexityChatSettings,\n ) => createLanguageModel(modelId, settings);\n\n provider.languageModel = createLanguageModel;\n provider.chat = createLanguageModel;\n provider.textEmbeddingModel = (modelId: string) => {\n throw new NoSuchModelError({ modelId, modelType: 'textEmbeddingModel' });\n };\n\n return provider;\n}\n\nexport const perplexity = createPerplexity();\n","import { MetadataExtractor } from '@ai-sdk/openai-compatible';\nimport { safeValidateTypes } from '@ai-sdk/provider-utils';\nimport { z } from 'zod';\n\nexport const perplexityMetadataExtractor: MetadataExtractor = {\n extractMetadata: ({ parsedBody }: { parsedBody: unknown }) => {\n const parsed = safeValidateTypes({\n value: parsedBody,\n schema: perplexityResponseSchema,\n });\n\n return !parsed.success\n ? undefined\n : buildPerplexityMetadata(\n parsed.value.citations ?? undefined,\n parsed.value.images ?? undefined,\n parsed.value.usage ?? undefined,\n );\n },\n\n createStreamExtractor: () => {\n let citations: string[] | undefined;\n let images: PerplexityImageData[] | undefined;\n let usage: PerplexityUsageData | undefined;\n\n return {\n processChunk: (chunk: unknown) => {\n const parsed = safeValidateTypes({\n value: chunk,\n schema: perplexityStreamChunkSchema,\n });\n\n if (parsed.success) {\n citations = parsed.value.citations ?? citations;\n images = parsed.value.images ?? images;\n usage = parsed.value.usage ?? usage;\n }\n },\n buildMetadata: () => buildPerplexityMetadata(citations, images, usage),\n };\n },\n};\n\nconst buildPerplexityMetadata = (\n citations: string[] | undefined,\n images: PerplexityImageData[] | undefined,\n usage: PerplexityUsageData | undefined,\n) => {\n return citations || images || usage\n ? {\n perplexity: {\n ...(citations && { citations }),\n ...(images && {\n images: images.map(image => ({\n imageUrl: image.image_url,\n originUrl: image.origin_url,\n height: image.height,\n width: image.width,\n })),\n }),\n ...(usage && {\n usage: {\n citationTokens: usage.citation_tokens ?? NaN,\n numSearchQueries: usage.num_search_queries ?? NaN,\n },\n }),\n },\n }\n : undefined;\n};\n\nconst perplexityCitationSchema = z.array(z.string());\n\nconst perplexityImageSchema = z.object({\n image_url: z.string(),\n origin_url: z.string(),\n height: z.number(),\n width: z.number(),\n});\n\ntype PerplexityImageData = z.infer<typeof perplexityImageSchema>;\n\nconst perplexityUsageSchema = z.object({\n citation_tokens: z.number().nullish(),\n num_search_queries: z.number().nullish(),\n});\n\ntype PerplexityUsageData = z.infer<typeof perplexityUsageSchema>;\n\nconst perplexityResponseSchema = z.object({\n citations: perplexityCitationSchema.nullish(),\n images: z.array(perplexityImageSchema).nullish(),\n usage: perplexityUsageSchema.nullish(),\n});\n\nconst perplexityStreamChunkSchema = z.object({\n choices: z\n .array(\n z.object({\n finish_reason: z.string().nullish(),\n delta: z\n .object({\n role: z.string(),\n content: z.string(),\n })\n .nullish(),\n }),\n )\n .nullish(),\n citations: perplexityCitationSchema.nullish(),\n images: z.array(perplexityImageSchema).nullish(),\n usage: perplexityUsageSchema.nullish(),\n});\n"],"mappings":";AAAA;AAAA,EAEE;AAAA,OAEK;AACP,SAAS,yCAAyC;AAClD;AAAA,EAEE;AAAA,EACA;AAAA,OACK;AAKP,SAAS,KAAAA,UAAS;;;ACdlB,SAAS,yBAAyB;AAClC,SAAS,SAAS;AAEX,IAAM,8BAAiD;AAAA,EAC5D,iBAAiB,CAAC,EAAE,WAAW,MAA+B;AALhE;AAMI,UAAM,SAAS,kBAAkB;AAAA,MAC/B,OAAO;AAAA,MACP,QAAQ;AAAA,IACV,CAAC;AAED,WAAO,CAAC,OAAO,UACX,SACA;AAAA,OACE,YAAO,MAAM,cAAb,YAA0B;AAAA,OAC1B,YAAO,MAAM,WAAb,YAAuB;AAAA,OACvB,YAAO,MAAM,UAAb,YAAsB;AAAA,IACxB;AAAA,EACN;AAAA,EAEA,uBAAuB,MAAM;AAC3B,QAAI;AACJ,QAAI;AACJ,QAAI;AAEJ,WAAO;AAAA,MACL,cAAc,CAAC,UAAmB;AA1BxC;AA2BQ,cAAM,SAAS,kBAAkB;AAAA,UAC/B,OAAO;AAAA,UACP,QAAQ;AAAA,QACV,CAAC;AAED,YAAI,OAAO,SAAS;AAClB,uBAAY,YAAO,MAAM,cAAb,YAA0B;AACtC,oBAAS,YAAO,MAAM,WAAb,YAAuB;AAChC,mBAAQ,YAAO,MAAM,UAAb,YAAsB;AAAA,QAChC;AAAA,MACF;AAAA,MACA,eAAe,MAAM,wBAAwB,WAAW,QAAQ,KAAK;AAAA,IACvE;AAAA,EACF;AACF;AAEA,IAAM,0BAA0B,CAC9B,WACA,QACA,UACG;AA/CL;AAgDE,SAAO,aAAa,UAAU,QAC1B;AAAA,IACE,YAAY;AAAA,MACV,GAAI,aAAa,EAAE,UAAU;AAAA,MAC7B,GAAI,UAAU;AAAA,QACZ,QAAQ,OAAO,IAAI,YAAU;AAAA,UAC3B,UAAU,MAAM;AAAA,UAChB,WAAW,MAAM;AAAA,UACjB,QAAQ,MAAM;AAAA,UACd,OAAO,MAAM;AAAA,QACf,EAAE;AAAA,MACJ;AAAA,MACA,GAAI,SAAS;AAAA,QACX,OAAO;AAAA,UACL,iBAAgB,WAAM,oBAAN,YAAyB;AAAA,UACzC,mBAAkB,WAAM,uBAAN,YAA4B;AAAA,QAChD;AAAA,MACF;AAAA,IACF;AAAA,EACF,IACA;AACN;AAEA,IAAM,2BAA2B,EAAE,MAAM,EAAE,OAAO,CAAC;AAEnD,IAAM,wBAAwB,EAAE,OAAO;AAAA,EACrC,WAAW,EAAE,OAAO;AAAA,EACpB,YAAY,EAAE,OAAO;AAAA,EACrB,QAAQ,EAAE,OAAO;AAAA,EACjB,OAAO,EAAE,OAAO;AAClB,CAAC;AAID,IAAM,wBAAwB,EAAE,OAAO;AAAA,EACrC,iBAAiB,EAAE,OAAO,EAAE,QAAQ;AAAA,EACpC,oBAAoB,EAAE,OAAO,EAAE,QAAQ;AACzC,CAAC;AAID,IAAM,2BAA2B,EAAE,OAAO;AAAA,EACxC,WAAW,yBAAyB,QAAQ;AAAA,EAC5C,QAAQ,EAAE,MAAM,qBAAqB,EAAE,QAAQ;AAAA,EAC/C,OAAO,sBAAsB,QAAQ;AACvC,CAAC;AAED,IAAM,8BAA8B,EAAE,OAAO;AAAA,EAC3C,SAAS,EACN;AAAA,IACC,EAAE,OAAO;AAAA,MACP,eAAe,EAAE,OAAO,EAAE,QAAQ;AAAA,MAClC,OAAO,EACJ,OAAO;AAAA,QACN,MAAM,EAAE,OAAO;AAAA,QACf,SAAS,EAAE,OAAO;AAAA,MACpB,CAAC,EACA,QAAQ;AAAA,IACb,CAAC;AAAA,EACH,EACC,QAAQ;AAAA,EACX,WAAW,yBAAyB,QAAQ;AAAA,EAC5C,QAAQ,EAAE,MAAM,qBAAqB,EAAE,QAAQ;AAAA,EAC/C,OAAO,sBAAsB,QAAQ;AACvC,CAAC;;;AD5FD,IAAM,wBAAwBC,GAAE,OAAO;AAAA,EACrC,MAAMA,GAAE,OAAO;AAAA,EACf,OAAOA,GAAE,OAAO;AAClB,CAAC;AAID,IAAM,2BAAwE;AAAA,EAC5E,aAAa;AAAA,EACb,gBAAgB,UAAQ,KAAK;AAC/B;AAmDO,SAAS,iBACd,UAAsC,CAAC,GACnB;AAnFtB;AAoFE,QAAM,UAAU;AAAA,KACd,aAAQ,YAAR,YAAmB;AAAA,EACrB;AACA,QAAM,aAAa,OAAO;AAAA,IACxB,eAAe,UAAU,WAAW;AAAA,MAClC,QAAQ,QAAQ;AAAA,MAChB,yBAAyB;AAAA,MACzB,aAAa;AAAA,IACf,CAAC,CAAC;AAAA,IACF,GAAG,QAAQ;AAAA,EACb;AAEA,QAAM,sBAAsB,CAC1B,SACA,WAAmC,CAAC,MACjC;AACH,WAAO,IAAI,kCAAkC,SAAS,UAAU;AAAA,MAC9D,UAAU;AAAA,MACV,KAAK,CAAC,EAAE,KAAK,MAAM,GAAG,OAAO,GAAG,IAAI;AAAA,MACpC,SAAS;AAAA,MACT,OAAO,QAAQ;AAAA,MACf,6BAA6B;AAAA,MAC7B,gBAAgB;AAAA,MAChB,mBAAmB;AAAA,MACnB,2BAA2B;AAAA,IAC7B,CAAC;AAAA,EACH;AAEA,QAAM,WAAW,CACf,SACA,aACG,oBAAoB,SAAS,QAAQ;AAE1C,WAAS,gBAAgB;AACzB,WAAS,OAAO;AAChB,WAAS,qBAAqB,CAAC,YAAoB;AACjD,UAAM,IAAI,iBAAiB,EAAE,SAAS,WAAW,qBAAqB,CAAC;AAAA,EACzE;AAEA,SAAO;AACT;AAEO,IAAM,aAAa,iBAAiB;","names":["z","z"]}
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@ai-sdk/perplexity",
3
- "version": "0.0.6",
3
+ "version": "0.0.8",
4
4
  "license": "Apache-2.0",
5
5
  "sideEffects": false,
6
6
  "main": "./dist/index.js",
@@ -19,9 +19,9 @@
19
19
  }
20
20
  },
21
21
  "dependencies": {
22
- "@ai-sdk/openai-compatible": "0.1.8",
22
+ "@ai-sdk/openai-compatible": "0.1.9",
23
23
  "@ai-sdk/provider": "1.0.7",
24
- "@ai-sdk/provider-utils": "2.1.6"
24
+ "@ai-sdk/provider-utils": "2.1.7"
25
25
  },
26
26
  "devDependencies": {
27
27
  "@types/node": "^18",