@truefoundry/ai-gateway-sdk 0.0.0-stage → 0.0.2

package/dist/providers/openrouter/utils.js

@@ -0,0 +1,15 @@
+export const transformReasoningParams = (params) => {
+    const reasoning = { ...params.reasoning };
+    if (params.reasoning_effort) {
+        reasoning.effort = params.reasoning_effort;
+    }
+    return Object.keys(reasoning).length > 0 ? reasoning : null;
+};
+export const transformUsageOptions = (params) => {
+    const usage = { ...params.usage };
+    if (params.stream_options?.include_usage) {
+        usage.include = params.stream_options?.include_usage;
+    }
+    return Object.keys(usage).length > 0 ? usage : null;
+};
+//# sourceMappingURL=utils.js.map

package/dist/providers/openrouter/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/providers/openrouter/utils.ts"],"names":[],"mappings":"AAqBA,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,MAAwB,EAAE,EAAE;IACnE,MAAM,SAAS,GAA6B,EAAE,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;IACpE,IAAI,MAAM,CAAC,gBAAgB,EAAE,CAAC;QAC5B,SAAS,CAAC,MAAM,GAAG,MAAM,CAAC,gBAAgB,CAAC;IAC7C,CAAC;IACD,OAAO,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC;AAC9D,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,MAAwB,EAAE,EAAE;IAChE,MAAM,KAAK,GAAyB,EAAE,GAAG,MAAM,CAAC,KAAK,EAAE,CAAC;IACxD,IAAI,MAAM,CAAC,cAAc,EAAE,aAAa,EAAE,CAAC;QACzC,KAAK,CAAC,OAAO,GAAG,MAAM,CAAC,cAAc,EAAE,aAAa,CAAC;IACvD,CAAC;IACD,OAAO,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;AACtD,CAAC,CAAC"}

package/dist/providers/palm/api.d.ts

@@ -0,0 +1,3 @@
+import { ProviderAPIConfig } from '../types.js';
+export declare const PalmApiConfig: ProviderAPIConfig;
+export default PalmApiConfig;

package/dist/providers/palm/api.js

@@ -0,0 +1,26 @@
+export const PalmApiConfig = {
+    getBaseURL: () => 'https://generativelanguage.googleapis.com/v1beta3',
+    headers: () => {
+        return { 'Content-Type': 'application/json' };
+    },
+    getEndpoint: ({ providerOptions, endpoint, request }) => {
+        const manifest = providerOptions.providerAccount.manifest;
+        const apiKey = manifest.auth_data?.api_key;
+        const model = request.body?.model;
+        switch (endpoint) {
+            case 'complete': {
+                return `/models/${model}:generateText?key=${apiKey}`;
+            }
+            case 'chatComplete': {
+                return `/models/${model}:generateMessage?key=${apiKey}`;
+            }
+            case 'embed': {
+                return `/models/${model}:embedText?key=${apiKey}`;
+            }
+            default:
+                return '';
+        }
+    },
+};
+export default PalmApiConfig;
+//# sourceMappingURL=api.js.map

package/dist/providers/palm/api.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../src/providers/palm/api.ts"],"names":[],"mappings":"AASA,MAAM,CAAC,MAAM,aAAa,GAAsB;IAC9C,UAAU,EAAE,GAAG,EAAE,CAAC,mDAAmD;IACrE,OAAO,EAAE,GAAG,EAAE;QACZ,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC;IAChD,CAAC;IACD,WAAW,EAAE,CAAC,EAAE,eAAe,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE;QACtD,MAAM,QAAQ,GAAG,eAAe,CAAC,eAAe,CAAC,QAA+B,CAAC;QACjF,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,EAAE,OAAO,CAAC;QAC3C,MAAM,KAAK,GAAI,OAAO,CAAC,IAAe,EAAE,KAAK,CAAC;QAC9C,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,UAAU,CAAC,CAAC,CAAC;gBAChB,OAAO,WAAW,KAAK,qBAAqB,MAAM,EAAE,CAAC;YACvD,CAAC;YACD,KAAK,cAAc,CAAC,CAAC,CAAC;gBACpB,OAAO,WAAW,KAAK,wBAAwB,MAAM,EAAE,CAAC;YAC1D,CAAC;YACD,KAAK,OAAO,CAAC,CAAC,CAAC;gBACb,OAAO,WAAW,KAAK,kBAAkB,MAAM,EAAE,CAAC;YACpD,CAAC;YACD;gBACE,OAAO,EAAE,CAAC;QACd,CAAC;IACH,CAAC;CACF,CAAC;AAEF,eAAe,aAAa,CAAC"}

package/dist/providers/palm/chatComplete.d.ts

@@ -0,0 +1,5 @@
+import { PalmChatCompleteResponse } from '../../types/responseBody.js';
+import { GoogleErrorResponse } from '../google/chatComplete.js';
+import { ChatCompletionResponse, ErrorResponse, ProviderConfig } from '../types.js';
+export declare const PalmChatCompleteConfig: ProviderConfig;
+export declare const PalmChatCompleteResponseTransform: (response: PalmChatCompleteResponse | GoogleErrorResponse, responseStatus: number) => ChatCompletionResponse | ErrorResponse;

package/dist/providers/palm/chatComplete.js

@@ -0,0 +1,91 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { PALM } from '../../globals.js';
+import { GoogleErrorResponseTransform } from '../google/chatComplete.js';
+import { generateInvalidProviderResponseError } from '../utils.js';
+// TODOS: this configuration does not enforce the maximum token limit for the input parameter. If you want to enforce this, you might need to add a custom validation function or a max property to the ParameterConfig interface, and then use it in the input configuration. However, this might be complex because the token count is not a simple length check, but depends on the specific tokenization method used by the model.
+export const PalmChatCompleteConfig = {
+    model: {
+        param: 'model',
+        required: true,
+        default: 'model/chat-bison-001',
+    },
+    messages: {
+        param: 'prompt',
+        default: '',
+        transform: (params) => {
+            const { examples, context, messages } = params;
+            const palmMessages = messages?.map(message => ({
+                author: message.role === 'developer' ? 'system' : message.role,
+                content: message.content,
+            }));
+            const prompt = {
+                messages: palmMessages,
+                examples,
+                context,
+            };
+            return prompt;
+        },
+    },
+    temperature: {
+        param: 'temperature',
+        default: 1,
+        min: 0,
+        max: 1,
+    },
+    top_p: {
+        param: 'topP',
+        default: 1,
+        min: 0,
+        max: 1,
+    },
+    top_k: {
+        param: 'topK',
+        default: 1,
+        min: 0,
+        max: 1,
+    },
+    n: {
+        param: 'candidateCount',
+        default: 1,
+        min: 1,
+        max: 8,
+    },
+    max_tokens: {
+        param: 'maxOutputTokens',
+        min: 1,
+    },
+    max_completion_tokens: {
+        param: 'maxOutputTokens',
+        min: 1,
+    },
+    stop: {
+        param: 'stopSequences',
+    },
+};
+export const PalmChatCompleteResponseTransform = (response, responseStatus) => {
+    if (responseStatus !== 200) {
+        const errorResponse = GoogleErrorResponseTransform(response, PALM);
+        if (errorResponse)
+            return errorResponse;
+    }
+    if ('candidates' in response) {
+        return {
+            id: Date.now().toString(),
+            object: 'chat.completion',
+            created: Math.floor(Date.now() / 1000),
+            model: 'Unknown',
+            provider: PALM,
+            choices: response.candidates?.map((generation, index) => ({
+                message: { role: 'assistant', content: generation.content },
+                index: index,
+                finish_reason: 'length',
+            })) ?? [],
+        };
+    }
+    return generateInvalidProviderResponseError(response, PALM);
+};
+//# sourceMappingURL=chatComplete.js.map

package/dist/providers/palm/chatComplete.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatComplete.js","sourceRoot":"","sources":["../../../src/providers/palm/chatComplete.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AAGrC,OAAO,EAAuB,4BAA4B,EAAE,MAAM,wBAAwB,CAAC;AAE3F,OAAO,EAAE,oCAAoC,EAAE,MAAM,UAAU,CAAC;AAEhE,saAAsa;AAEta,MAAM,CAAC,MAAM,sBAAsB,GAAmB;IACpD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,IAAI;QACd,OAAO,EAAE,sBAAsB;KAChC;IACD,QAAQ,EAAE;QACR,KAAK,EAAE,QAAQ;QACf,OAAO,EAAE,EAAE;QACX,SAAS,EAAE,CAAC,MAAc,EAAE,EAAE;YAC5B,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;YAC/C,MAAM,YAAY,GAAG,QAAQ,EAAE,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAC7C,MAAM,EAAE,OAAO,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI;gBAC9D,OAAO,EAAE,OAAO,CAAC,OAAO;aACzB,CAAC,CAAC,CAAC;YACJ,MAAM,MAAM,GAAG;gBACb,QAAQ,EAAE,YAAY;gBACtB,QAAQ;gBACR,OAAO;aACR,CAAC;YACF,OAAO,MAAM,CAAC;QAChB,CAAC;KACF;IACD,WAAW,EAAE;QACX,KAAK,EAAE,aAAa;QACpB,OAAO,EAAE,CAAC;QACV,GAAG,EAAE,CAAC;QACN,GAAG,EAAE,CAAC;KACP;IACD,KAAK,EAAE;QACL,KAAK,EAAE,MAAM;QACb,OAAO,EAAE,CAAC;QACV,GAAG,EAAE,CAAC;QACN,GAAG,EAAE,CAAC;KACP;IACD,KAAK,EAAE;QACL,KAAK,EAAE,MAAM;QACb,OAAO,EAAE,CAAC;QACV,GAAG,EAAE,CAAC;QACN,GAAG,EAAE,CAAC;KACP;IACD,CAAC,EAAE;QACD,KAAK,EAAE,gBAAgB;QACvB,OAAO,EAAE,CAAC;QACV,GAAG,EAAE,CAAC;QACN,GAAG,EAAE,CAAC;KACP;IACD,UAAU,EAAE;QACV,KAAK,EAAE,iBAAiB;QACxB,GAAG,EAAE,CAAC;KACP;IACD,qBAAqB,EAAE;QACrB,KAAK,EAAE,iBAAiB;QACxB,GAAG,EAAE,CAAC;KACP;IACD,IAAI,EAAE;QACJ,KAAK,EAAE,eAAe;KACvB;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,iCAAiC,GAGA,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE;IACzE,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;QAC3B,MAAM,aAAa,GAAG,4BAA4B,CAAC,QAA+B,EAAE,IAAI,CAAC,CAAC;QAC1F,IAAI,aAAa;YAAE,OAAO,aAAa,CAAC;IAC1C,CAAC;IAED,IAAI,YAAY,IAAI,QAAQ,EAAE,CAAC;QAC7B,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;YACzB,MAAM,EAAE,iBAAiB;YACzB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;YACtC,KAAK,EAAE,SAAS;YAChB,QAAQ,EAAE,IAAI;YACd,OAAO,EACL,QAAQ,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,UAAU,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;gBAC/C,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,UAAU,CAAC,OAAO,EAAE;gBAC3D,KAAK,EAAE,KAAK;gBACZ,aAAa,EAAE,QAAQ;aACxB,CAAC,CAAC,IAAI,EAAE;SACZ,CAAC;IACJ,CAAC;IAED,OAAO,oCAAoC,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;AAC9D,CAAC,CAAC"}

package/dist/providers/palm/complete.d.ts

@@ -0,0 +1,5 @@
+import { PalmCompleteResponse } from '../../types/responseBody.js';
+import { GoogleErrorResponse } from '../google/chatComplete.js';
+import { CompletionResponse, ErrorResponse, ProviderConfig } from '../types.js';
+export declare const PalmCompleteConfig: ProviderConfig;
+export declare const PalmCompleteResponseTransform: (response: PalmCompleteResponse | GoogleErrorResponse, responseStatus: number) => CompletionResponse | ErrorResponse;

package/dist/providers/palm/complete.js

@@ -0,0 +1,82 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { PALM } from '../../globals.js';
+import { GoogleErrorResponseTransform } from '../google/chatComplete.js';
+import { generateInvalidProviderResponseError } from '../utils.js';
+// TODOS: this configuration does not enforce the maximum token limit for the input parameter. If you want to enforce this, you might need to add a custom validation function or a max property to the ParameterConfig interface, and then use it in the input configuration. However, this might be complex because the token count is not a simple length check, but depends on the specific tokenization method used by the model.
+export const PalmCompleteConfig = {
+    model: {
+        param: 'model',
+        required: true,
+        default: 'model/text-bison-001',
+    },
+    prompt: {
+        param: 'prompt',
+        default: '',
+        transform: (params) => {
+            const { prompt: text } = params;
+            const prompt = {
+                text,
+            };
+            return prompt;
+        },
+    },
+    temperature: {
+        param: 'temperature',
+        default: 1,
+        min: 0,
+        max: 1,
+    },
+    top_p: {
+        param: 'topP',
+        default: 1,
+        min: 0,
+        max: 1,
+    },
+    top_k: {
+        param: 'topK',
+        default: 1,
+        min: 0,
+        max: 1,
+    },
+    n: {
+        param: 'candidateCount',
+        default: 1,
+        min: 1,
+        max: 8,
+    },
+    max_tokens: {
+        param: 'maxOutputTokens',
+        min: 1,
+    },
+    stop: {
+        param: 'stopSequences',
+    },
+};
+export const PalmCompleteResponseTransform = (response, responseStatus) => {
+    if (responseStatus !== 200) {
+        const errorResponse = GoogleErrorResponseTransform(response, PALM);
+        if (errorResponse)
+            return errorResponse;
+    }
+    if ('candidates' in response) {
+        return {
+            id: Date.now().toString(),
+            object: 'completion',
+            created: Math.floor(Date.now() / 1000),
+            model: 'Unknown',
+            provider: PALM,
+            choices: response.candidates?.map((generation, index) => ({
+                text: generation.output,
+                index: index,
+                logprobs: null,
+                finish_reason: 'length',
+            })) ?? [],
+        };
+    }
+    return generateInvalidProviderResponseError(response, PALM);
+};
+//# sourceMappingURL=complete.js.map

package/dist/providers/palm/complete.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"complete.js","sourceRoot":"","sources":["../../../src/providers/palm/complete.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AAGrC,OAAO,EAAuB,4BAA4B,EAAE,MAAM,wBAAwB,CAAC;AAE3F,OAAO,EAAE,oCAAoC,EAAE,MAAM,UAAU,CAAC;AAEhE,saAAsa;AAEta,MAAM,CAAC,MAAM,kBAAkB,GAAmB;IAChD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,IAAI;QACd,OAAO,EAAE,sBAAsB;KAChC;IACD,MAAM,EAAE;QACN,KAAK,EAAE,QAAQ;QACf,OAAO,EAAE,EAAE;QACX,SAAS,EAAE,CAAC,MAAc,EAAE,EAAE;YAC5B,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,CAAC;YAChC,MAAM,MAAM,GAAG;gBACb,IAAI;aACL,CAAC;YACF,OAAO,MAAM,CAAC;QAChB,CAAC;KACF;IACD,WAAW,EAAE;QACX,KAAK,EAAE,aAAa;QACpB,OAAO,EAAE,CAAC;QACV,GAAG,EAAE,CAAC;QACN,GAAG,EAAE,CAAC;KACP;IACD,KAAK,EAAE;QACL,KAAK,EAAE,MAAM;QACb,OAAO,EAAE,CAAC;QACV,GAAG,EAAE,CAAC;QACN,GAAG,EAAE,CAAC;KACP;IACD,KAAK,EAAE;QACL,KAAK,EAAE,MAAM;QACb,OAAO,EAAE,CAAC;QACV,GAAG,EAAE,CAAC;QACN,GAAG,EAAE,CAAC;KACP;IACD,CAAC,EAAE;QACD,KAAK,EAAE,gBAAgB;QACvB,OAAO,EAAE,CAAC;QACV,GAAG,EAAE,CAAC;QACN,GAAG,EAAE,CAAC;KACP;IACD,UAAU,EAAE;QACV,KAAK,EAAE,iBAAiB;QACxB,GAAG,EAAE,CAAC;KACP;IACD,IAAI,EAAE;QACJ,KAAK,EAAE,eAAe;KACvB;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,6BAA6B,GAGA,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE;IACrE,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;QAC3B,MAAM,aAAa,GAAG,4BAA4B,CAAC,QAA+B,EAAE,IAAI,CAAC,CAAC;QAC1F,IAAI,aAAa;YAAE,OAAO,aAAa,CAAC;IAC1C,CAAC;IAED,IAAI,YAAY,IAAI,QAAQ,EAAE,CAAC;QAC7B,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;YACzB,MAAM,EAAE,YAAY;YACpB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;YACtC,KAAK,EAAE,SAAS;YAChB,QAAQ,EAAE,IAAI;YACd,OAAO,EACL,QAAQ,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,UAAU,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;gBAC/C,IAAI,EAAE,UAAU,CAAC,MAAM;gBACvB,KAAK,EAAE,KAAK;gBACZ,QAAQ,EAAE,IAAI;gBACd,aAAa,EAAE,QAAQ;aACxB,CAAC,CAAC,IAAI,EAAE;SACZ,CAAC;IACJ,CAAC;IAED,OAAO,oCAAoC,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;AAC9D,CAAC,CAAC"}

package/dist/providers/palm/embed.d.ts

@@ -0,0 +1,12 @@
+import { EmbedResponse } from '../../types/embedRequestBody.js';
+import { GoogleErrorResponse } from '../google/chatComplete.js';
+import { ErrorResponse, ProviderConfig } from '../types.js';
+export declare const PalmEmbedConfig: ProviderConfig;
+interface embedding {
+    value: number[];
+}
+interface PalmEmbedResponse {
+    embedding: embedding;
+}
+export declare const PalmEmbedResponseTransform: (response: PalmEmbedResponse | GoogleErrorResponse, responseStatus: number, _responseHeaders: Headers, _strictOpenAiCompliance: boolean, gatewayRequestUrl: string, gatewayRequest: Params) => EmbedResponse | ErrorResponse;
+export {};

package/dist/providers/palm/embed.js

@@ -0,0 +1,46 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { PALM } from '../../globals.js';
+import { GoogleErrorResponseTransform } from '../google/chatComplete.js';
+import { generateInvalidProviderResponseError } from '../utils.js';
+export const PalmEmbedConfig = {
+    input: {
+        param: 'text',
+        required: true,
+    },
+    model: {
+        param: 'model',
+        default: 'models/embedding-gecko-001',
+    },
+};
+export const PalmEmbedResponseTransform = (response, responseStatus, _responseHeaders, _strictOpenAiCompliance, _gatewayRequestUrl, gatewayRequest) => {
+    if (responseStatus !== 200) {
+        const errorResponse = GoogleErrorResponseTransform(response, PALM);
+        if (errorResponse)
+            return errorResponse;
+    }
+    const model = gatewayRequest.model || '';
+    if ('embedding' in response) {
+        return {
+            object: 'list',
+            data: [
+                {
+                    object: 'embedding',
+                    embedding: response.embedding.value,
+                    index: 0,
+                },
+            ],
+            model,
+            usage: {
+                prompt_tokens: -1,
+                total_tokens: -1,
+            },
+            provider: PALM,
+        };
+    }
+    return generateInvalidProviderResponseError(response, PALM);
+};
+//# sourceMappingURL=embed.js.map

package/dist/providers/palm/embed.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embed.js","sourceRoot":"","sources":["../../../src/providers/palm/embed.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AAErC,OAAO,EAAuB,4BAA4B,EAAE,MAAM,wBAAwB,CAAC;AAE3F,OAAO,EAAE,oCAAoC,EAAE,MAAM,UAAU,CAAC;AAEhE,MAAM,CAAC,MAAM,eAAe,GAAmB;IAC7C,KAAK,EAAE;QACL,KAAK,EAAE,MAAM;QACb,QAAQ,EAAE,IAAI;KACf;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,OAAO,EAAE,4BAA4B;KACtC;CACF,CAAC;AAUF,MAAM,CAAC,MAAM,0BAA0B,GAOF,CACnC,QAAQ,EACR,cAAc,EACd,gBAAgB,EAChB,uBAAuB,EACvB,kBAAkB,EAClB,cAAc,EACd,EAAE;IACF,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;QAC3B,MAAM,aAAa,GAAG,4BAA4B,CAAC,QAA+B,EAAE,IAAI,CAAC,CAAC;QAC1F,IAAI,aAAa;YAAE,OAAO,aAAa,CAAC;IAC1C,CAAC;IAED,MAAM,KAAK,GAAI,cAAc,CAAC,KAAgB,IAAI,EAAE,CAAC;IACrD,IAAI,WAAW,IAAI,QAAQ,EAAE,CAAC;QAC5B,OAAO;YACL,MAAM,EAAE,MAAM;YACd,IAAI,EAAE;gBACJ;oBACE,MAAM,EAAE,WAAW;oBACnB,SAAS,EAAE,QAAQ,CAAC,SAAS,CAAC,KAAK;oBACnC,KAAK,EAAE,CAAC;iBACT;aACF;YACD,KAAK;YACL,KAAK,EAAE;gBACL,aAAa,EAAE,CAAC,CAAC;gBACjB,YAAY,EAAE,CAAC,CAAC;aACjB;YACD,QAAQ,EAAE,IAAI;SACf,CAAC;IACJ,CAAC;IAED,OAAO,oCAAoC,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;AAC9D,CAAC,CAAC"}

package/dist/providers/palm/index.d.ts

@@ -0,0 +1,8 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { ProviderConfigs } from '../types.js';
+declare const PalmAIConfig: ProviderConfigs;
+export default PalmAIConfig;

package/dist/providers/palm/index.js

@@ -0,0 +1,17 @@
+import PalmApiConfig from './api.js';
+import { PalmChatCompleteConfig, PalmChatCompleteResponseTransform } from './chatComplete.js';
+import { PalmCompleteConfig, PalmCompleteResponseTransform } from './complete.js';
+import { PalmEmbedConfig, PalmEmbedResponseTransform } from './embed.js';
+const PalmAIConfig = {
+    complete: PalmCompleteConfig,
+    embed: PalmEmbedConfig,
+    api: PalmApiConfig,
+    chatComplete: PalmChatCompleteConfig,
+    responseTransforms: {
+        complete: PalmCompleteResponseTransform,
+        chatComplete: PalmChatCompleteResponseTransform,
+        embed: PalmEmbedResponseTransform,
+    },
+};
+export default PalmAIConfig;
+//# sourceMappingURL=index.js.map

package/dist/providers/palm/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/providers/palm/index.ts"],"names":[],"mappings":"AAMA,OAAO,aAAa,MAAM,OAAO,CAAC;AAClC,OAAO,EAAE,sBAAsB,EAAE,iCAAiC,EAAE,MAAM,gBAAgB,CAAC;AAC3F,OAAO,EAAE,kBAAkB,EAAE,6BAA6B,EAAE,MAAM,YAAY,CAAC;AAC/E,OAAO,EAAE,eAAe,EAAE,0BAA0B,EAAE,MAAM,SAAS,CAAC;AAEtE,MAAM,YAAY,GAAoB;IACpC,QAAQ,EAAE,kBAAkB;IAC5B,KAAK,EAAE,eAAe;IACtB,GAAG,EAAE,aAAa;IAClB,YAAY,EAAE,sBAAsB;IACpC,kBAAkB,EAAE;QAClB,QAAQ,EAAE,6BAA6B;QACvC,YAAY,EAAE,iCAAiC;QAC/C,KAAK,EAAE,0BAA0B;KAClC;CACF,CAAC;AAEF,eAAe,YAAY,CAAC"}

package/dist/providers/perplexity-ai/api.d.ts

@@ -0,0 +1,3 @@
+import { ProviderAPIConfig } from '../types.js';
+declare const PerplexityAIApiConfig: ProviderAPIConfig;
+export default PerplexityAIApiConfig;

package/dist/providers/perplexity-ai/api.js

@@ -0,0 +1,20 @@
+const PerplexityAIApiConfig = {
+    getBaseURL: () => 'https://api.perplexity.ai',
+    headers: ({ providerOptions }) => {
+        const manifest = providerOptions.providerAccount.manifest;
+        return { Authorization: `Bearer ${manifest.auth_data?.api_key}` };
+    },
+    getEndpoint: ({ endpoint }) => {
+        switch (endpoint) {
+            case 'chatComplete':
+            case 'createModelResponse':
+                return '/v1/agent';
+            case 'decision':
+                return '/v1/decisions';
+            default:
+                return '';
+        }
+    },
+};
+export default PerplexityAIApiConfig;
+//# sourceMappingURL=api.js.map

package/dist/providers/perplexity-ai/api.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../src/providers/perplexity-ai/api.ts"],"names":[],"mappings":"AAQA,MAAM,qBAAqB,GAAsB;IAC/C,UAAU,EAAE,GAAG,EAAE,CAAC,2BAA2B;IAC7C,OAAO,EAAE,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE;QAC/B,MAAM,QAAQ,GAAG,eAAe,CAAC,eAAe,CAAC,QAAuC,CAAC;QACzF,OAAO,EAAE,aAAa,EAAE,UAAU,QAAQ,CAAC,SAAS,EAAE,OAAO,EAAE,EAAE,CAAC;IACpE,CAAC;IACD,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE;QAC5B,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,cAAc,CAAC;YACpB,KAAK,qBAAqB;gBACxB,OAAO,WAAW,CAAC;YACrB,KAAK,UAAU;gBACb,OAAO,eAAe,CAAC;YACzB;gBACE,OAAO,EAAE,CAAC;QACd,CAAC;IACH,CAAC;CACF,CAAC;AAEF,eAAe,qBAAqB,CAAC"}

package/dist/providers/perplexity-ai/chatCompleteViaResponses.d.ts

@@ -0,0 +1,8 @@
+import { Params } from '../../types/requestBody.js';
+import { ChatCompletionResponse, ErrorResponse } from '../types.js';
+export declare const PERPLEXITY_AGENT_MODEL = "perplexity/sonar";
+export declare function transformPerplexityChatCompletionsToResponsesRequest(params: Params, requestHeaders?: Record<string, string>): Record<string, any>;
+export declare function toPerplexityRunError(response: any): ErrorResponse;
+export declare function isFailedRun(response: any, responseStatus: number): boolean;
+export declare const PerplexityChatCompleteViaResponsesResponseTransform: (response: any, responseStatus: number, _responseHeaders: Headers, _strictOpenAiCompliance: boolean, _gatewayRequestUrl: string, gatewayRequest: Params) => ChatCompletionResponse | ErrorResponse;
+export declare const PerplexityChatCompleteViaResponsesStreamChunkTransform: (chunk: string, fallbackId: string, streamState: Record<string, any>, _strictOpenAiCompliance: boolean, gatewayRequest: Params) => string | string[] | undefined;

package/dist/providers/perplexity-ai/chatCompleteViaResponses.js

@@ -0,0 +1,152 @@
+/**
+ * Serves chat completions over Perplexity's Agent API (`/v1/agent`), which replaced the
+ * Sonar chat-completions API. Mirrors `transformCursorChatCompletionsToResponsesRequest`:
+ * delegate to the shared transform, then apply a Perplexity-specific delta.
+ */
+import { TFY_DISABLE_WEB_SEARCH_HEADER } from '../../constants.js';
+import { PERPLEXITY_AI } from '../../globals.js';
+import { providerReportedCostInUsd } from '../../utils/providerReportedCost.js';
+import { parseSseDataJson } from '../../utils/sseUtils.js';
+import { transformChatCompletionsToResponsesRequest } from '../open-ai-base/chatCompleteViaResponses/requestTransform.js';
+import { createResponsesToChatCompletionsResponseTransform } from '../open-ai-base/chatCompleteViaResponses/responseTransform.js';
+import { createChatCompletionsStreamState, createResponsesToChatCompletionsStreamChunkTransform, } from '../open-ai-base/chatCompleteViaResponses/streamTransform.js';
+import { generateErrorResponse } from '../utils.js';
+export const PERPLEXITY_AGENT_MODEL = 'perplexity/sonar';
+const MODEL_PRESETS = {
+    sonar: 'fast',
+    'sonar-pro': 'low',
+    'sonar-reasoning': 'low',
+    'sonar-reasoning-pro': 'medium',
+    'sonar-deep-research': 'high',
+};
+const SEARCH_FILTER_PARAMS = [
+    'search_domain_filter',
+    'search_recency_filter',
+    'search_after_date_filter',
+    'search_before_date_filter',
+    'last_updated_after_filter',
+    'last_updated_before_filter',
+];
+function buildWebSearchTool(source) {
+    const tool = { type: 'web_search' };
+    const filters = {};
+    for (const key of SEARCH_FILTER_PARAMS) {
+        if (source[key] !== undefined)
+            filters[key] = source[key];
+    }
+    if (Object.keys(filters).length)
+        tool.filters = filters;
+    const options = source.web_search_options;
+    if (options?.search_context_size)
+        tool.search_context_size = options.search_context_size;
+    if (options?.user_location)
+        tool.user_location = options.user_location;
+    if (source.num_search_results !== undefined)
+        tool.max_results = source.num_search_results;
+    return Object.keys(tool).length > 1 ? tool : undefined;
+}
+export function transformPerplexityChatCompletionsToResponsesRequest(params, requestHeaders) {
+    const body = transformChatCompletionsToResponsesRequest(params);
+    const source = params;
+    const preset = MODEL_PRESETS[params.model ?? ''];
+    if (preset)
+        body.model = PERPLEXITY_AGENT_MODEL;
+    if (requestHeaders?.[TFY_DISABLE_WEB_SEARCH_HEADER] !== 'true') {
+        if (preset)
+            body.preset = preset;
+        const webSearch = buildWebSearchTool(source);
+        if (webSearch || !preset)
+            body.tools = [...(body.tools ?? []), webSearch ?? { type: 'web_search' }];
+    }
+    if (params.temperature !== undefined)
+        body.temperature = params.temperature;
+    if (params.top_p !== undefined)
+        body.top_p = params.top_p;
+    if (source.language_preference !== undefined)
+        body.language_preference = source.language_preference;
+    return body;
+}
+function extractSearchMetadata(output) {
+    const item = Array.isArray(output) ? output.find(entry => entry?.type === 'search_results') : undefined;
+    const results = item?.results;
+    if (!Array.isArray(results) || !results.length)
+        return {};
+    return {
+        citations: results.map((result) => result?.url).filter(Boolean),
+        search_results: results,
+    };
+}
+function withCostInUsd(chunks, cost, costBreakdown) {
+    if (typeof chunks !== 'string') {
+        return (Array.isArray(chunks) ? chunks.map(chunk => withCostInUsd(chunk, cost, costBreakdown)) : chunks);
+    }
+    const parsed = parseSseDataJson(chunks);
+    if (!parsed?.usage)
+        return chunks;
+    parsed.usage.costInUSD = cost;
+    if (costBreakdown)
+        parsed.usage.cost = costBreakdown;
+    return `data: ${JSON.stringify(parsed)}\n\n`;
+}
+export function toPerplexityRunError(response) {
+    return generateErrorResponse({
+        message: response?.error?.message ?? `Perplexity run ${response?.status}`,
+        type: response?.error?.type ?? response?.status,
+        code: response?.error?.code?.toString() ?? null,
+    }, PERPLEXITY_AI);
+}
+export function isFailedRun(response, responseStatus) {
+    return responseStatus === 200 && (response?.status === 'failed' || response?.status === 'cancelled');
+}
+const baseResponseTransform = createResponsesToChatCompletionsResponseTransform(PERPLEXITY_AI);
+const baseStreamTransform = createResponsesToChatCompletionsStreamChunkTransform(PERPLEXITY_AI);
+export const PerplexityChatCompleteViaResponsesResponseTransform = (response, responseStatus, _responseHeaders, _strictOpenAiCompliance, _gatewayRequestUrl, gatewayRequest) => {
+    if (isFailedRun(response, responseStatus))
+        return toPerplexityRunError(response);
+    const transformed = baseResponseTransform(response, responseStatus);
+    if (responseStatus !== 200 || !('choices' in transformed))
+        return transformed;
+    if (gatewayRequest?.model)
+        transformed.model = gatewayRequest.model;
+    Object.assign(transformed, extractSearchMetadata(response?.output));
+    const cost = providerReportedCostInUsd(response?.usage);
+    if (cost !== undefined && transformed.usage) {
+        transformed.usage.costInUSD = cost;
+        transformed.usage.cost = response.usage.cost;
+    }
+    return transformed;
+};
+export const PerplexityChatCompleteViaResponsesStreamChunkTransform = (chunk, fallbackId, streamState, _strictOpenAiCompliance, gatewayRequest) => {
+    // Seed the shared transform before it emits the opening role chunk. Its model normally
+    // comes from the Agent response, which uses the pinned slug for every Sonar tier.
+    if (gatewayRequest?.model) {
+        streamState.chatCompleteViaResponses ??= createChatCompletionsStreamState(fallbackId, PERPLEXITY_AI);
+        streamState.chatCompleteViaResponses.model = gatewayRequest.model;
+    }
+    const event = parseSseDataJson(chunk);
+    if (event?.type === 'response.cancelled') {
+        const failed = `data: ${JSON.stringify({ ...event, type: 'response.failed' })}`;
+        return baseStreamTransform(failed, fallbackId, streamState);
+    }
+    let transformed = baseStreamTransform(chunk, fallbackId, streamState);
+    if (event?.type !== 'response.completed')
+        return transformed;
+    const cost = providerReportedCostInUsd(event.response?.usage);
+    if (cost !== undefined)
+        transformed = withCostInUsd(transformed, cost, event.response?.usage?.cost);
+    const metadata = extractSearchMetadata(event.response?.output);
+    if (!Object.keys(metadata).length)
+        return transformed;
+    const metadataChunk = `data: ${JSON.stringify({
+        id: event.response?.id ?? fallbackId,
+        object: 'chat.completion.chunk',
+        created: event.response?.created_at ?? Math.floor(Date.now() / 1000),
+        model: gatewayRequest?.model || event.response?.model || '',
+        provider: PERPLEXITY_AI,
+        ...metadata,
+        choices: [{ index: 0, delta: {}, finish_reason: null }],
+    })}\n\n`;
+    const rest = Array.isArray(transformed) ? transformed : transformed ? [transformed] : [];
+    return [metadataChunk, ...rest];
+};
+//# sourceMappingURL=chatCompleteViaResponses.js.map

package/dist/providers/perplexity-ai/chatCompleteViaResponses.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatCompleteViaResponses.js","sourceRoot":"","sources":["../../../src/providers/perplexity-ai/chatCompleteViaResponses.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,6BAA6B,EAAE,MAAM,iBAAiB,CAAC;AAChE,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAE9C,OAAO,EAAE,yBAAyB,EAAE,MAAM,kCAAkC,CAAC;AAC7E,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,EAAE,0CAA0C,EAAE,MAAM,2DAA2D,CAAC;AACvH,OAAO,EAAE,iDAAiD,EAAE,MAAM,4DAA4D,CAAC;AAC/H,OAAO,EACL,gCAAgC,EAChC,oDAAoD,GACrD,MAAM,0DAA0D,CAAC;AAElE,OAAO,EAAE,qBAAqB,EAAE,MAAM,UAAU,CAAC;AAEjD,MAAM,CAAC,MAAM,sBAAsB,GAAG,kBAAkB,CAAC;AAEzD,MAAM,aAAa,GAA2B;IAC5C,KAAK,EAAE,MAAM;IACb,WAAW,EAAE,KAAK;IAClB,iBAAiB,EAAE,KAAK;IACxB,qBAAqB,EAAE,QAAQ;IAC/B,qBAAqB,EAAE,MAAM;CAC9B,CAAC;AAEF,MAAM,oBAAoB,GAAG;IAC3B,sBAAsB;IACtB,uBAAuB;IACvB,0BAA0B;IAC1B,2BAA2B;IAC3B,2BAA2B;IAC3B,4BAA4B;CAC7B,CAAC;AAEF,SAAS,kBAAkB,CAAC,MAA2B;IACrD,MAAM,IAAI,GAAwB,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;IAEzD,MAAM,OAAO,GAAwB,EAAE,CAAC;IACxC,KAAK,MAAM,GAAG,IAAI,oBAAoB,EAAE,CAAC;QACvC,IAAI,MAAM,CAAC,GAAG,CAAC,KAAK,SAAS;YAAE,OAAO,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;IAC5D,CAAC;IACD,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM;QAAE,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IAExD,MAAM,OAAO,GAAG,MAAM,CAAC,kBAAkB,CAAC;IAC1C,IAAI,OAAO,EAAE,mBAAmB;QAAE,IAAI,CAAC,mBAAmB,GAAG,OAAO,CAAC,mBAAmB,CAAC;IACzF,IAAI,OAAO,EAAE,aAAa;QAAE,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC;IACvE,IAAI,MAAM,CAAC,kBAAkB,KAAK,SAAS;QAAE,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,kBAAkB,CAAC;IAE1F,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;AACzD,CAAC;AAED,MAAM,UAAU,oDAAoD,CAClE,MAAc,EACd,cAAuC;IAEvC,MAAM,IAAI,GAAG,0CAA0C,CAAC,MAAM,CAAC,CAAC;IAChE,MAAM,MAAM,GAAG,MAA6B,CAAC;IAC7C,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;IAEjD,IAAI,MAAM;QAAE,IAAI,CAAC,KAAK,GAAG,sBAAsB,CAAC;IAEhD,IAAI,cAAc,EAAE,CAAC,6BAA6B,CAAC,KAAK,MAAM,EAAE,CAAC;QAC/D,IAAI,MAAM;YAAE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACjC,MAAM,SAAS,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC7C,IAAI,SAAS,IAAI,CAAC,MAAM;YAAE,IAAI,CAAC,KAAK,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,SAAS,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;IACtG,CAAC;IAED,IAAI,MAAM,CAAC,WAAW,KAAK,SAAS;QAAE,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;IAC5E,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS;QAAE,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;IAC1D,IAAI,MAAM,CAAC,mBAAmB,KAAK,SAAS;QAAE,IAAI,CAAC,mBAAmB,GAAG,MAAM,CAAC,mBAAmB,CAAC;IAEpG,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,qBAAqB,CAAC,MAAW;IACxC,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,EAAE,IAAI,KAAK,gBAAgB,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACxG,MAAM,OAAO,GAAG,IAAI,EAAE,OAAO,CAAC;IAC9B,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM;QAAE,OAAO,EAAE,CAAC;IAC1D,OAAO;QACL,SAAS,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,MAAW,EAAE,EAAE,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC;QACpE,cAAc,EAAE,OAAO;KACxB,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAA0C,MAAS,EAAE,IAAY,EAAE,aAAuB;IAC9G,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC/B,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,KAAK,EAAE,IAAI,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAM,CAAC;IAChH,CAAC;IACD,MAAM,MAAM,GAAG,gBAAgB,CAAM,MAAM,CAAC,CAAC;IAC7C,IAAI,CAAC,MAAM,EAAE,KAAK;QAAE,OAAO,MAAM,CAAC;IAClC,MAAM,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC;IAC9B,IAAI,aAAa;QAAE,MAAM,CAAC,KAAK,CAAC,IAAI,GAAG,aAAa,CAAC;IACrD,OAAO,SAAS,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAW,CAAC;AACpD,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,QAAa;IAChD,OAAO,qBAAqB,CAC1B;QACE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,IAAI,kBAAkB,QAAQ,EAAE,MAAM,EAAE;QACzE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,IAAI,QAAQ,EAAE,MAAM;QAC/C,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,IAAI;KAChD,EACD,aAAa,CACd,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,QAAa,EAAE,cAAsB;IAC/D,OAAO,cAAc,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,MAAM,KAAK,QAAQ,IAAI,QAAQ,EAAE,MAAM,KAAK,WAAW,CAAC,CAAC;AACvG,CAAC;AAED,MAAM,qBAAqB,GAAG,iDAAiD,CAAC,aAAa,CAAC,CAAC;AAC/F,MAAM,mBAAmB,GAAG,oDAAoD,CAAC,aAAa,CAAC,CAAC;AAEhG,MAAM,CAAC,MAAM,mDAAmD,GAAG,CACjE,QAAa,EACb,cAAsB,EACtB,gBAAyB,EACzB,uBAAgC,EAChC,kBAA0B,EAC1B,cAAsB,EACkB,EAAE;IAC1C,IAAI,WAAW,CAAC,QAAQ,EAAE,cAAc,CAAC;QAAE,OAAO,oBAAoB,CAAC,QAAQ,CAAC,CAAC;IAEjF,MAAM,WAAW,GAAG,qBAAqB,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;IACpE,IAAI,cAAc,KAAK,GAAG,IAAI,CAAC,CAAC,SAAS,IAAI,WAAW,CAAC;QAAE,OAAO,WAAW,CAAC;IAE9E,IAAI,cAAc,EAAE,KAAK;QAAE,WAAW,CAAC,KAAK,GAAG,cAAc,CAAC,KAAK,CAAC;IACpE,MAAM,CAAC,MAAM,CAAC,WAAW,EAAE,qBAAqB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC;IAEpE,MAAM,IAAI,GAAG,yBAAyB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IACxD,IAAI,IAAI,KAAK,SAAS,IAAI,WAAW,CAAC,KAAK,EAAE,CAAC;QAC5C,WAAW,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC;QAClC,WAAW,CAAC,KAA4B,CAAC,IAAI,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC;IACvE,CAAC;IAED,OAAO,WAAW,CAAC;AACrB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,sDAAsD,GAAG,CACpE,KAAa,EACb,UAAkB,EAClB,WAAgC,EAChC,uBAAgC,EAChC,cAAsB,EACS,EAAE;IACjC,uFAAuF;IACvF,kFAAkF;IAClF,IAAI,cAAc,EAAE,KAAK,EAAE,CAAC;QAC1B,WAAW,CAAC,wBAAwB,KAAK,gCAAgC,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;QACrG,WAAW,CAAC,wBAAwB,CAAC,KAAK,GAAG,cAAc,CAAC,KAAK,CAAC;IACpE,CAAC;IACD,MAAM,KAAK,GAAG,gBAAgB,CAAM,KAAK,CAAC,CAAC;IAE3C,IAAI,KAAK,EAAE,IAAI,KAAK,oBAAoB,EAAE,CAAC;QACzC,MAAM,MAAM,GAAG,SAAS,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,KAAK,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,EAAE,CAAC;QAChF,OAAO,mBAAmB,CAAC,MAAM,EAAE,UAAU,EAAE,WAAW,CAAC,CAAC;IAC9D,CAAC;IAED,IAAI,WAAW,GAAG,mBAAmB,CAAC,KAAK,EAAE,UAAU,EAAE,WAAW,CAAC,CAAC;IACtE,IAAI,KAAK,EAAE,IAAI,KAAK,oBAAoB;QAAE,OAAO,WAAW,CAAC;IAE7D,MAAM,IAAI,GAAG,yBAAyB,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAC9D,IAAI,IAAI,KAAK,SAAS;QAAE,WAAW,GAAG,aAAa,CAAC,WAAW,EAAE,IAAI,EAAE,KAAK,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;IAEpG,MAAM,QAAQ,GAAG,qBAAqB,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC/D,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM;QAAE,OAAO,WAAW,CAAC;IAEtD,MAAM,aAAa,GAAG,SAAS,IAAI,CAAC,SAAS,CAAC;QAC5C,EAAE,EAAE,KAAK,CAAC,QAAQ,EAAE,EAAE,IAAI,UAAU;QACpC,MAAM,EAAE,uBAAuB;QAC/B,OAAO,EAAE,KAAK,CAAC,QAAQ,EAAE,UAAU,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;QACpE,KAAK,EAAE,cAAc,EAAE,KAAK,IAAI,KAAK,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;QAC3D,QAAQ,EAAE,aAAa;QACvB,GAAG,QAAQ;QACX,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;KACxD,CAAC,MAAM,CAAC;IAET,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACzF,OAAO,CAAC,aAAa,EAAE,GAAG,IAAI,CAAC,CAAC;AAClC,CAAC,CAAC"}

package/dist/providers/perplexity-ai/createModelResponse.d.ts

@@ -0,0 +1,6 @@
+import { Params } from '../../types/requestBody.js';
+import { ProviderConfig } from '../types.js';
+export declare const PerplexityCreateModelResponseConfig: ProviderConfig;
+export declare function transformPerplexityCreateModelResponseRequest(params: Params): Record<string, any>;
+export declare const PerplexityCreateModelResponseTransform: (response: any, responseStatus: number) => import("../types.js").ErrorResponse | import("../../types/modelResponses.js").OpenAIResponse;
+export declare const PerplexityCreateModelResponseStreamChunkTransform: (chunk: string) => string;

package/dist/providers/perplexity-ai/createModelResponse.js

@@ -0,0 +1,51 @@
+import { PERPLEXITY_AI } from '../../globals.js';
+import { providerReportedCostInUsd } from '../../utils/providerReportedCost.js';
+import { parseSseDataJson } from '../../utils/sseUtils.js';
+import { createModelResponseParams, OpenAICreateModelResponseTransformer } from '../open-ai-base/index.js';
+import { isFailedRun, PERPLEXITY_AGENT_MODEL, toPerplexityRunError } from './chatCompleteViaResponses.js';
+const PERPLEXITY_ONLY_PARAMS = {
+    preset: { param: 'preset' },
+    max_steps: { param: 'max_steps' },
+    max_tool_calls: { param: 'max_tool_calls' },
+    models: { param: 'models' },
+    language_preference: { param: 'language_preference' },
+};
+export const PerplexityCreateModelResponseConfig = createModelResponseParams([], {}, PERPLEXITY_ONLY_PARAMS);
+const ALLOWED_PARAMS = new Set(Object.keys(PerplexityCreateModelResponseConfig));
+const RESPONSES_MODEL_MAP = { sonar: PERPLEXITY_AGENT_MODEL };
+export function transformPerplexityCreateModelResponseRequest(params) {
+    const source = params;
+    const body = {};
+    for (const key of Object.keys(source)) {
+        if (ALLOWED_PARAMS.has(key))
+            body[key] = source[key];
+    }
+    const mappedModel = RESPONSES_MODEL_MAP[params.model ?? ''];
+    if (mappedModel)
+        body.model = mappedModel;
+    return body;
+}
+const baseTransformer = OpenAICreateModelResponseTransformer(PERPLEXITY_AI);
+export const PerplexityCreateModelResponseTransform = (response, responseStatus) => {
+    if (isFailedRun(response, responseStatus))
+        return toPerplexityRunError(response);
+    const transformed = baseTransformer(response, responseStatus);
+    const usage = transformed.usage;
+    const cost = providerReportedCostInUsd(usage);
+    if (usage && cost !== undefined)
+        usage.costInUSD = cost;
+    return transformed;
+};
+export const PerplexityCreateModelResponseStreamChunkTransform = (chunk) => {
+    const event = parseSseDataJson(chunk);
+    if (event?.type === 'response.completed') {
+        const usage = event.response?.usage;
+        const cost = providerReportedCostInUsd(usage);
+        if (cost !== undefined) {
+            usage.costInUSD = cost;
+            chunk = chunk.replace(/^data:.*$/m, () => `data: ${JSON.stringify(event)}`);
+        }
+    }
+    return chunk.endsWith('\n\n') ? chunk : `${chunk}\n\n`;
+};
+//# sourceMappingURL=createModelResponse.js.map

package/dist/providers/perplexity-ai/createModelResponse.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"createModelResponse.js","sourceRoot":"","sources":["../../../src/providers/perplexity-ai/createModelResponse.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAE9C,OAAO,EAAE,yBAAyB,EAAE,MAAM,kCAAkC,CAAC;AAC7E,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,EAAE,yBAAyB,EAAE,oCAAoC,EAAE,MAAM,iBAAiB,CAAC;AAElG,OAAO,EAAE,WAAW,EAAE,sBAAsB,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AAEvG,MAAM,sBAAsB,GAAmB;IAC7C,MAAM,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE;IAC3B,SAAS,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE;IACjC,cAAc,EAAE,EAAE,KAAK,EAAE,gBAAgB,EAAE;IAC3C,MAAM,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE;IAC3B,mBAAmB,EAAE,EAAE,KAAK,EAAE,qBAAqB,EAAE;CACtD,CAAC;AAEF,MAAM,CAAC,MAAM,mCAAmC,GAAmB,yBAAyB,CAC1F,EAAE,EACF,EAAE,EACF,sBAAsB,CACvB,CAAC;AAEF,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,mCAAmC,CAAC,CAAC,CAAC;AAEjF,MAAM,mBAAmB,GAA2B,EAAE,KAAK,EAAE,sBAAsB,EAAE,CAAC;AAEtF,MAAM,UAAU,6CAA6C,CAAC,MAAc;IAC1E,MAAM,MAAM,GAAG,MAA6B,CAAC;IAC7C,MAAM,IAAI,GAAwB,EAAE,CAAC;IACrC,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QACtC,IAAI,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC;YAAE,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;IACvD,CAAC;IAED,MAAM,WAAW,GAAG,mBAAmB,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;IAC5D,IAAI,WAAW;QAAE,IAAI,CAAC,KAAK,GAAG,WAAW,CAAC;IAE1C,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,eAAe,GAAG,oCAAoC,CAAC,aAAa,CAAC,CAAC;AAE5E,MAAM,CAAC,MAAM,sCAAsC,GAAG,CAAC,QAAa,EAAE,cAAsB,EAAE,EAAE;IAC9F,IAAI,WAAW,CAAC,QAAQ,EAAE,cAAc,CAAC;QAAE,OAAO,oBAAoB,CAAC,QAAQ,CAAC,CAAC;IACjF,MAAM,WAAW,GAAG,eAAe,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;IAC9D,MAAM,KAAK,GAAI,WAAkD,CAAC,KAAK,CAAC;IACxE,MAAM,IAAI,GAAG,yBAAyB,CAAC,KAAK,CAAC,CAAC;IAC9C,IAAI,KAAK,IAAI,IAAI,KAAK,SAAS;QAAE,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC;IACxD,OAAO,WAAW,CAAC;AACrB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,iDAAiD,GAAG,CAAC,KAAa,EAAU,EAAE;IACzF,MAAM,KAAK,GAAG,gBAAgB,CAAM,KAAK,CAAC,CAAC;IAC3C,IAAI,KAAK,EAAE,IAAI,KAAK,oBAAoB,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC;QACpC,MAAM,IAAI,GAAG,yBAAyB,CAAC,KAAK,CAAC,CAAC;QAC9C,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC;YACvB,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,YAAY,EAAE,GAAG,EAAE,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC9E,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,KAAK,MAAM,CAAC;AACzD,CAAC,CAAC"}