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

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

@@ -0,0 +1,29 @@
+import { ErrorResponse, ProviderConfig } from '../types.js';
+export declare const PerplexityDecisionConfig: ProviderConfig;
+interface PerplexityDecisionUsage {
+    input_tokens?: number;
+    output_tokens?: number;
+}
+export interface PerplexityDecisionResponse {
+    model?: string;
+    answers?: Record<string, unknown>;
+    usage?: PerplexityDecisionUsage;
+}
+interface PerplexityDecisionErrorResponse {
+    error?: {
+        message?: string;
+        type?: string;
+        code?: string | number | null;
+    };
+    message?: string;
+    detail?: unknown;
+}
+export interface DecisionResponse extends PerplexityDecisionResponse {
+    usage?: PerplexityDecisionUsage & {
+        prompt_tokens?: number;
+        completion_tokens?: number;
+        total_tokens?: number;
+    };
+}
+export declare const PerplexityDecisionResponseTransform: (response: PerplexityDecisionResponse | PerplexityDecisionErrorResponse, responseStatus: number) => DecisionResponse | ErrorResponse;
+export {};

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

@@ -0,0 +1,47 @@
+import { PERPLEXITY_AI } from '../../globals.js';
+import { generateErrorResponse, generateInvalidProviderResponseError } from '../utils.js';
+export const PerplexityDecisionConfig = {
+    model: {
+        param: 'model',
+        required: true,
+    },
+    state: {
+        param: 'state',
+        required: true,
+    },
+    questions: {
+        param: 'questions',
+        required: true,
+    },
+};
+export const PerplexityDecisionResponseTransform = (response, responseStatus) => {
+    if (responseStatus !== 200) {
+        const errorResponse = response;
+        const message = errorResponse.error?.message ??
+            errorResponse.message ??
+            (typeof errorResponse.detail === 'string' ? errorResponse.detail : undefined) ??
+            'Perplexity request failed';
+        const errorCode = errorResponse.error?.code;
+        return generateErrorResponse({
+            message,
+            type: errorResponse.error?.type ?? null,
+            code: errorCode != null ? String(errorCode) : String(responseStatus),
+        }, PERPLEXITY_AI);
+    }
+    const decisionResponse = response;
+    if (!decisionResponse.answers) {
+        return generateInvalidProviderResponseError(response, PERPLEXITY_AI);
+    }
+    const inputTokens = decisionResponse.usage?.input_tokens ?? 0;
+    const outputTokens = decisionResponse.usage?.output_tokens ?? 0;
+    return {
+        ...decisionResponse,
+        usage: {
+            ...decisionResponse.usage,
+            prompt_tokens: inputTokens,
+            completion_tokens: outputTokens,
+            total_tokens: inputTokens + outputTokens,
+        },
+    };
+};
+//# sourceMappingURL=decision.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"decision.js","sourceRoot":"","sources":["../../../src/providers/perplexity-ai/decision.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAE9C,OAAO,EAAE,qBAAqB,EAAE,oCAAoC,EAAE,MAAM,UAAU,CAAC;AAEvF,MAAM,CAAC,MAAM,wBAAwB,GAAmB;IACtD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,IAAI;KACf;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,IAAI;KACf;IACD,SAAS,EAAE;QACT,KAAK,EAAE,WAAW;QAClB,QAAQ,EAAE,IAAI;KACf;CACF,CAAC;AA2BF,MAAM,CAAC,MAAM,mCAAmC,GAAG,CACjD,QAAsE,EACtE,cAAsB,EACY,EAAE;IACpC,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;QAC3B,MAAM,aAAa,GAAG,QAA2C,CAAC;QAClE,MAAM,OAAO,GACX,aAAa,CAAC,KAAK,EAAE,OAAO;YAC5B,aAAa,CAAC,OAAO;YACrB,CAAC,OAAO,aAAa,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;YAC7E,2BAA2B,CAAC;QAC9B,MAAM,SAAS,GAAG,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC;QAC5C,OAAO,qBAAqB,CAC1B;YACE,OAAO;YACP,IAAI,EAAE,aAAa,CAAC,KAAK,EAAE,IAAI,IAAI,IAAI;YACvC,IAAI,EAAE,SAAS,IAAI,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,cAAc,CAAC;SACrE,EACD,aAAa,CACd,CAAC;IACJ,CAAC;IAED,MAAM,gBAAgB,GAAG,QAAsC,CAAC;IAChE,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC;QAC9B,OAAO,oCAAoC,CAAC,QAAmC,EAAE,aAAa,CAAC,CAAC;IAClG,CAAC;IAED,MAAM,WAAW,GAAG,gBAAgB,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC,CAAC;IAC9D,MAAM,YAAY,GAAG,gBAAgB,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;IAEhE,OAAO;QACL,GAAG,gBAAgB;QACnB,KAAK,EAAE;YACL,GAAG,gBAAgB,CAAC,KAAK;YACzB,aAAa,EAAE,WAAW;YAC1B,iBAAiB,EAAE,YAAY;YAC/B,YAAY,EAAE,WAAW,GAAG,YAAY;SACzC;KACF,CAAC;AACJ,CAAC,CAAC"}

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

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

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

@@ -0,0 +1,23 @@
+import PerplexityAIApiConfig from './api.js';
+import { PerplexityChatCompleteViaResponsesResponseTransform, PerplexityChatCompleteViaResponsesStreamChunkTransform, transformPerplexityChatCompletionsToResponsesRequest, } from './chatCompleteViaResponses.js';
+import { PerplexityCreateModelResponseConfig, PerplexityCreateModelResponseStreamChunkTransform, PerplexityCreateModelResponseTransform, transformPerplexityCreateModelResponseRequest, } from './createModelResponse.js';
+import { PerplexityDecisionConfig, PerplexityDecisionResponseTransform } from './decision.js';
+const PerplexityAIConfig = {
+    api: PerplexityAIApiConfig,
+    chatComplete: PerplexityCreateModelResponseConfig,
+    createModelResponse: PerplexityCreateModelResponseConfig,
+    decision: PerplexityDecisionConfig,
+    requestTransforms: {
+        chatComplete: transformPerplexityChatCompletionsToResponsesRequest,
+        createModelResponse: transformPerplexityCreateModelResponseRequest,
+    },
+    responseTransforms: {
+        chatComplete: PerplexityChatCompleteViaResponsesResponseTransform,
+        'stream-chatComplete': PerplexityChatCompleteViaResponsesStreamChunkTransform,
+        createModelResponse: PerplexityCreateModelResponseTransform,
+        decision: PerplexityDecisionResponseTransform,
+        'stream-createModelResponse': PerplexityCreateModelResponseStreamChunkTransform,
+    },
+};
+export default PerplexityAIConfig;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/providers/perplexity-ai/index.ts"],"names":[],"mappings":"AACA,OAAO,qBAAqB,MAAM,OAAO,CAAC;AAC1C,OAAO,EACL,mDAAmD,EACnD,sDAAsD,EACtD,oDAAoD,GACrD,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,mCAAmC,EACnC,iDAAiD,EACjD,sCAAsC,EACtC,6CAA6C,GAC9C,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,wBAAwB,EAAE,mCAAmC,EAAE,MAAM,YAAY,CAAC;AAE3F,MAAM,kBAAkB,GAAoB;IAC1C,GAAG,EAAE,qBAAqB;IAC1B,YAAY,EAAE,mCAAmC;IACjD,mBAAmB,EAAE,mCAAmC;IACxD,QAAQ,EAAE,wBAAwB;IAClC,iBAAiB,EAAE;QACjB,YAAY,EAAE,oDAAoD;QAClE,mBAAmB,EAAE,6CAA6C;KACnE;IACD,kBAAkB,EAAE;QAClB,YAAY,EAAE,mDAAmD;QACjE,qBAAqB,EAAE,sDAAsD;QAC7E,mBAAmB,EAAE,sCAAsC;QAC3D,QAAQ,EAAE,mCAAmC;QAC7C,4BAA4B,EAAE,iDAAiD;KAChF;CACF,CAAC;AAEF,eAAe,kBAAkB,CAAC"}

package/dist/providers/requestHandlerApiArgs.d.ts

@@ -0,0 +1,12 @@
+/**
+ * Build ProviderApiBaseArgs from a requestHandler's SDK inputs (GATEWAY-3351).
+ * Package-internal; not part of the 0.x public export surface.
+ */
+import type { SdkRequestBody } from '../types/contracts.js';
+import type { Endpoint, ProviderApiBaseArgs, RequestHandlerArgs } from './types.js';
+export declare function providerApiArgsFromHandler<T = SdkRequestBody>(args: RequestHandlerArgs<T>, overrides?: {
+    endpoint?: Endpoint;
+    method?: string;
+    body?: SdkRequestBody;
+    path?: string;
+}): ProviderApiBaseArgs;

package/dist/providers/requestHandlerApiArgs.js

@@ -0,0 +1,19 @@
+export function providerApiArgsFromHandler(args, overrides) {
+    const endpoint = overrides?.endpoint ?? args.request.endpoint;
+    const request = {
+        ...args.request,
+        endpoint,
+        method: overrides?.method ?? args.request.method,
+        body: (overrides?.body ?? args.request.body),
+        path: overrides?.path ?? args.request.path,
+    };
+    return {
+        request,
+        providerOptions: args.providerOptions,
+        callConfig: args.callConfig,
+        headerOptions: args.headerOptions,
+        session: args.session,
+        endpoint,
+    };
+}
+//# sourceMappingURL=requestHandlerApiArgs.js.map

package/dist/providers/requestHandlerApiArgs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"requestHandlerApiArgs.js","sourceRoot":"","sources":["../../src/providers/requestHandlerApiArgs.ts"],"names":[],"mappings":"AAOA,MAAM,UAAU,0BAA0B,CACxC,IAA2B,EAC3B,SAA0F;IAE1F,MAAM,QAAQ,GAAG,SAAS,EAAE,QAAQ,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC;IAC9D,MAAM,OAAO,GAAG;QACd,GAAG,IAAI,CAAC,OAAO;QACf,QAAQ;QACR,MAAM,EAAE,SAAS,EAAE,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM;QAChD,IAAI,EAAE,CAAC,SAAS,EAAE,IAAI,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,CAAmB;QAC9D,IAAI,EAAE,SAAS,EAAE,IAAI,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI;KAC3C,CAAC;IACF,OAAO;QACL,OAAO;QACP,eAAe,EAAE,IAAI,CAAC,eAAe;QACrC,UAAU,EAAE,IAAI,CAAC,UAAU;QAC3B,aAAa,EAAE,IAAI,CAAC,aAAa;QACjC,OAAO,EAAE,IAAI,CAAC,OAAO;QACrB,QAAQ;KACT,CAAC;AACJ,CAAC"}

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

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

package/dist/providers/sagemaker/api.js

@@ -0,0 +1,86 @@
+import { SAGEMAKER } from '../../globals.js';
+import { sdkRequestUrl } from '../../utils/sdkRequestUrl.js';
+import { generateAWSHeaders, getCredentialsFromAssumedRole, getCredentialsFromRegion } from '../bedrock/utils.js';
+function sagemakerRegion(account, model) {
+    return model?.region ?? account.region;
+}
+function sagemakerModelName(model, headerModelName) {
+    // Authorized model_id wins so a caller cannot invoke a different endpoint with account creds.
+    return model?.model_id || headerModelName || '';
+}
+const SagemakerAPIConfig = {
+    getBaseURL: ({ providerOptions }) => {
+        const account = providerOptions.providerAccount.manifest;
+        const model = providerOptions.model?.manifest;
+        return `https://runtime.sagemaker.${sagemakerRegion(account, model)}.amazonaws.com`;
+    },
+    headers: async ({ providerOptions, headerOptions, session, transformedRequestBody, transformedRequestUrl }) => {
+        const headers = {
+            'content-type': 'application/json',
+        };
+        const account = providerOptions.providerAccount.manifest;
+        const model = providerOptions.model?.manifest;
+        const region = sagemakerRegion(account, model);
+        const auth = account.auth_data;
+        let awsAccessKeyId = '';
+        let awsSecretAccessKey = '';
+        let awsSessionToken = '';
+        if (auth?.type === 'access-key-based') {
+            awsAccessKeyId = auth.access_key_id || '';
+            awsSecretAccessKey = auth.secret_access_key || '';
+        }
+        else if (auth?.type === 'assumed-role-based') {
+            const credentials = await getCredentialsFromAssumedRole(session, auth.assumed_role_arn || '', region || '', auth.external_id, SAGEMAKER);
+            awsAccessKeyId = credentials.accessKeyId;
+            awsSecretAccessKey = credentials.secretAccessKey;
+            awsSessionToken = credentials.sessionToken;
+        }
+        else {
+            const credentials = await getCredentialsFromRegion(region || '');
+            awsAccessKeyId = credentials.accessKeyId;
+            awsSecretAccessKey = credentials.secretAccessKey;
+            awsSessionToken = credentials.sessionToken;
+        }
+        const awsHeaders = await generateAWSHeaders(transformedRequestBody, headers, transformedRequestUrl || '', 'POST', 'sagemaker', region || '', awsAccessKeyId, awsSecretAccessKey, awsSessionToken);
+        const { amznSagemakerCustomAttributes, amznSagemakerTargetModel, amznSagemakerTargetVariant, amznSagemakerTargetContainerHostname, amznSagemakerInferenceId, amznSagemakerEnableExplanations, amznSagemakerInferenceComponent, amznSagemakerSessionId, } = headerOptions.awsSagemaker;
+        if (amznSagemakerCustomAttributes) {
+            awsHeaders['x-amzn-sagemaker-custom-attributes'] = amznSagemakerCustomAttributes;
+        }
+        if (amznSagemakerTargetModel) {
+            awsHeaders['x-amzn-sagemaker-target-model'] = amznSagemakerTargetModel;
+        }
+        if (amznSagemakerTargetVariant) {
+            awsHeaders['x-amzn-sagemaker-target-variant'] = amznSagemakerTargetVariant;
+        }
+        if (amznSagemakerTargetContainerHostname) {
+            awsHeaders['x-amzn-sagemaker-target-container-hostname'] = amznSagemakerTargetContainerHostname;
+        }
+        if (amznSagemakerInferenceId) {
+            awsHeaders['x-amzn-sagemaker-inference-id'] = amznSagemakerInferenceId;
+        }
+        if (amznSagemakerEnableExplanations) {
+            awsHeaders['x-amzn-sagemaker-enable-explanations'] = amznSagemakerEnableExplanations;
+        }
+        if (amznSagemakerInferenceComponent) {
+            awsHeaders['x-amzn-sagemaker-inference-component'] = amznSagemakerInferenceComponent;
+        }
+        if (amznSagemakerSessionId) {
+            awsHeaders['x-amzn-sagemaker-session-id'] = amznSagemakerSessionId;
+        }
+        return awsHeaders;
+    },
+    getEndpoint: ({ endpoint, request, providerOptions, headerOptions }) => {
+        const model = providerOptions.model?.manifest;
+        const modelName = sagemakerModelName(model, headerOptions.awsSagemaker.amznSagemakerModelName);
+        const invocationPath = `/endpoints/${modelName}/invocations`;
+        if (endpoint === 'chatComplete') {
+            return invocationPath;
+        }
+        if (endpoint === 'embed') {
+            return invocationPath;
+        }
+        return sdkRequestUrl(request).split('/v1')[1];
+    },
+};
+export default SagemakerAPIConfig;
+//# sourceMappingURL=api.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../src/providers/sagemaker/api.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAC1D,OAAO,EAAE,kBAAkB,EAAE,6BAA6B,EAAE,wBAAwB,EAAE,MAAM,kBAAkB,CAAC;AAG/G,SAAS,eAAe,CAAC,OAAoC,EAAE,KAAiC;IAC9F,OAAO,KAAK,EAAE,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC;AACzC,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAiC,EAAE,eAAmC;IAChG,8FAA8F;IAC9F,OAAO,KAAK,EAAE,QAAQ,IAAI,eAAe,IAAI,EAAE,CAAC;AAClD,CAAC;AAED,MAAM,kBAAkB,GAAsB;IAC5C,UAAU,EAAE,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE;QAClC,MAAM,OAAO,GAAG,eAAe,CAAC,eAAe,CAAC,QAAuC,CAAC;QACxF,MAAM,KAAK,GAAG,eAAe,CAAC,KAAK,EAAE,QAAsC,CAAC;QAC5E,OAAO,6BAA6B,eAAe,CAAC,OAAO,EAAE,KAAK,CAAC,gBAAgB,CAAC;IACtF,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,EAAE,eAAe,EAAE,aAAa,EAAE,OAAO,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,EAAE,EAAE;QAC5G,MAAM,OAAO,GAA2B;YACtC,cAAc,EAAE,kBAAkB;SACnC,CAAC;QAEF,MAAM,OAAO,GAAG,eAAe,CAAC,eAAe,CAAC,QAAuC,CAAC;QACxF,MAAM,KAAK,GAAG,eAAe,CAAC,KAAK,EAAE,QAAsC,CAAC;QAC5E,MAAM,MAAM,GAAG,eAAe,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAC/C,MAAM,IAAI,GAAG,OAAO,CAAC,SAAS,CAAC;QAE/B,IAAI,cAAc,GAAG,EAAE,CAAC;QACxB,IAAI,kBAAkB,GAAG,EAAE,CAAC;QAC5B,IAAI,eAAe,GAAG,EAAE,CAAC;QAEzB,IAAI,IAAI,EAAE,IAAI,KAAK,kBAAkB,EAAE,CAAC;YACtC,cAAc,GAAG,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC;YAC1C,kBAAkB,GAAG,IAAI,CAAC,iBAAiB,IAAI,EAAE,CAAC;QACpD,CAAC;aAAM,IAAI,IAAI,EAAE,IAAI,KAAK,oBAAoB,EAAE,CAAC;YAC/C,MAAM,WAAW,GAAG,MAAM,6BAA6B,CACrD,OAAO,EACP,IAAI,CAAC,gBAAgB,IAAI,EAAE,EAC3B,MAAM,IAAI,EAAE,EACZ,IAAI,CAAC,WAAW,EAChB,SAAS,CACV,CAAC;YACF,cAAc,GAAG,WAAW,CAAC,WAAqB,CAAC;YACnD,kBAAkB,GAAG,WAAW,CAAC,eAAyB,CAAC;YAC3D,eAAe,GAAG,WAAW,CAAC,YAAsB,CAAC;QACvD,CAAC;aAAM,CAAC;YACN,MAAM,WAAW,GAAG,MAAM,wBAAwB,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;YACjE,cAAc,GAAG,WAAW,CAAC,WAAqB,CAAC;YACnD,kBAAkB,GAAG,WAAW,CAAC,eAAyB,CAAC;YAC3D,eAAe,GAAG,WAAW,CAAC,YAAsB,CAAC;QACvD,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,kBAAkB,CACzC,sBAAsB,EACtB,OAAO,EACP,qBAAqB,IAAI,EAAE,EAC3B,MAAM,EACN,WAAW,EACX,MAAM,IAAI,EAAE,EACZ,cAAc,EACd,kBAAkB,EAClB,eAAe,CAChB,CAAC;QAEF,MAAM,EACJ,6BAA6B,EAC7B,wBAAwB,EACxB,0BAA0B,EAC1B,oCAAoC,EACpC,wBAAwB,EACxB,+BAA+B,EAC/B,+BAA+B,EAC/B,sBAAsB,GACvB,GAAG,aAAa,CAAC,YAAY,CAAC;QAE/B,IAAI,6BAA6B,EAAE,CAAC;YAClC,UAAU,CAAC,oCAAoC,CAAC,GAAG,6BAA6B,CAAC;QACnF,CAAC;QAED,IAAI,wBAAwB,EAAE,CAAC;YAC7B,UAAU,CAAC,+BAA+B,CAAC,GAAG,wBAAwB,CAAC;QACzE,CAAC;QAED,IAAI,0BAA0B,EAAE,CAAC;YAC/B,UAAU,CAAC,iCAAiC,CAAC,GAAG,0BAA0B,CAAC;QAC7E,CAAC;QAED,IAAI,oCAAoC,EAAE,CAAC;YACzC,UAAU,CAAC,4CAA4C,CAAC,GAAG,oCAAoC,CAAC;QAClG,CAAC;QAED,IAAI,wBAAwB,EAAE,CAAC;YAC7B,UAAU,CAAC,+BAA+B,CAAC,GAAG,wBAAwB,CAAC;QACzE,CAAC;QAED,IAAI,+BAA+B,EAAE,CAAC;YACpC,UAAU,CAAC,sCAAsC,CAAC,GAAG,+BAA+B,CAAC;QACvF,CAAC;QAED,IAAI,+BAA+B,EAAE,CAAC;YACpC,UAAU,CAAC,sCAAsC,CAAC,GAAG,+BAA+B,CAAC;QACvF,CAAC;QAED,IAAI,sBAAsB,EAAE,CAAC;YAC3B,UAAU,CAAC,6BAA6B,CAAC,GAAG,sBAAsB,CAAC;QACrE,CAAC;QACD,OAAO,UAAU,CAAC;IACpB,CAAC;IACD,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,eAAe,EAAE,aAAa,EAAE,EAAE,EAAE;QACrE,MAAM,KAAK,GAAG,eAAe,CAAC,KAAK,EAAE,QAAsC,CAAC;QAC5E,MAAM,SAAS,GAAG,kBAAkB,CAAC,KAAK,EAAE,aAAa,CAAC,YAAY,CAAC,sBAAsB,CAAC,CAAC;QAC/F,MAAM,cAAc,GAAG,cAAc,SAAS,cAAc,CAAC;QAC7D,IAAI,QAAQ,KAAK,cAAc,EAAE,CAAC;YAChC,OAAO,cAAc,CAAC;QACxB,CAAC;QACD,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;YACzB,OAAO,cAAc,CAAC;QACxB,CAAC;QACD,OAAO,aAAa,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC;CACF,CAAC;AAEF,eAAe,kBAAkB,CAAC"}

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

@@ -0,0 +1,10 @@
+import { ChatCompletionResponse, ErrorResponse, ProviderConfig, SageMakerErrorResponse } from '../types.js';
+export declare const SagemakerChatCompleteConfig: ProviderConfig;
+interface SagemakerChatCompleteResponse extends ChatCompletionResponse {
+}
+/**
+ * Converts SageMaker/LMI NDJSON stream to OpenAI-style SSE chunks.
+ */
+export declare const SagemakerChatCompleteStreamChunkTransform: (responseChunk: string, fallbackId: string, _streamState: Record<string, unknown>, _strictOpenAiCompliance: boolean, _gatewayRequest: unknown) => string | undefined;
+export declare const SagemakerResponseTransform: (response: SagemakerChatCompleteResponse | SageMakerErrorResponse, responseStatus: number) => ChatCompletionResponse | ErrorResponse;
+export {};

package/dist/providers/sagemaker/chatComplete.js

@@ -0,0 +1,160 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { SAGEMAKER } from '../../globals.js';
+import { sagemakerResponseErrorHandler } 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 SagemakerChatCompleteConfig = {
+    model: {
+        param: 'model',
+        removeFromParams: true,
+    },
+    messages: {
+        param: 'messages',
+        default: '',
+    },
+    functions: {
+        param: 'functions',
+    },
+    function_call: {
+        param: 'function_call',
+    },
+    max_tokens: {
+        param: 'max_tokens',
+        min: 0,
+    },
+    max_completion_tokens: {
+        param: 'max_completion_tokens',
+        min: 0,
+    },
+    temperature: {
+        param: 'temperature',
+        default: 1,
+        min: 0,
+        max: 2,
+    },
+    top_p: {
+        param: 'top_p',
+        default: 1,
+        min: 0,
+        max: 1,
+    },
+    n: {
+        param: 'n',
+        default: 1,
+    },
+    logprobs: {
+        param: 'logprobs',
+        default: false,
+    },
+    top_logprobs: {
+        param: 'top_logprobs',
+    },
+    stream: {
+        param: 'stream',
+        default: false,
+    },
+    stop: {
+        param: 'stop',
+    },
+    presence_penalty: {
+        param: 'presence_penalty',
+        min: -2,
+        max: 2,
+    },
+    frequency_penalty: {
+        param: 'frequency_penalty',
+        min: -2,
+        max: 2,
+    },
+    logit_bias: {
+        param: 'logit_bias',
+    },
+    user: {
+        param: 'user',
+    },
+    tools: {
+        param: 'tools',
+    },
+    tool_choice: {
+        param: 'tool_choice',
+    },
+    response_format: {
+        param: 'response_format',
+    },
+    store: {
+        param: 'store',
+    },
+    metadata: {
+        param: 'metadata',
+    },
+    modalities: {
+        param: 'modalities',
+    },
+    audio: {
+        param: 'audio',
+    },
+    seed: {
+        param: 'seed',
+    },
+    prediction: {
+        param: 'prediction',
+    },
+    reasoning_effort: {
+        param: 'reasoning_effort',
+    },
+    stream_options: {
+        param: 'stream_options',
+        required: (params) => params.stream === true,
+        default: { include_usage: true },
+        transform: (params) => {
+            const existingStreamOptions = params.stream_options || {};
+            return {
+                ...existingStreamOptions,
+                include_usage: true,
+            };
+        },
+    },
+};
+/**
+ * Converts SageMaker/LMI NDJSON stream to OpenAI-style SSE chunks.
+ */
+export const SagemakerChatCompleteStreamChunkTransform = (responseChunk, fallbackId, _streamState, _strictOpenAiCompliance, _gatewayRequest) => {
+    let chunk = responseChunk.trim();
+    chunk = chunk.replace(/^data:\s*/, '').trim();
+    if (!chunk)
+        return undefined;
+    if (chunk === '[DONE]')
+        return `data: [DONE]\n\n`;
+    let parsed;
+    try {
+        parsed = JSON.parse(chunk);
+    }
+    catch {
+        return undefined;
+    }
+    const choices = (parsed.choices ?? []).map(c => ({
+        index: c.index ?? 0,
+        delta: c.delta ?? {},
+        finish_reason: c.finish_reason ?? null,
+    }));
+    const openAiChunk = {
+        id: parsed.id ?? fallbackId,
+        object: parsed.object ?? 'chat.completion.chunk',
+        created: parsed.created ?? Math.floor(Date.now() / 1000),
+        model: parsed.model ?? '',
+        provider: SAGEMAKER,
+        choices,
+        ...(parsed.usage && { usage: parsed.usage }),
+    };
+    return `data: ${JSON.stringify(openAiChunk)}\n\n`;
+};
+export const SagemakerResponseTransform = (response, responseStatus) => {
+    if (responseStatus !== 200 && 'ErrorCode' in response) {
+        return sagemakerResponseErrorHandler(responseStatus, response);
+    }
+    return { ...response, provider: SAGEMAKER };
+};
+//# sourceMappingURL=chatComplete.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"chatComplete.js","sourceRoot":"","sources":["../../../src/providers/sagemaker/chatComplete.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAG1C,OAAO,EAAE,6BAA6B,EAAE,MAAM,SAAS,CAAC;AAExD,saAAsa;AAEta,MAAM,CAAC,MAAM,2BAA2B,GAAmB;IACzD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,gBAAgB,EAAE,IAAI;KACvB;IACD,QAAQ,EAAE;QACR,KAAK,EAAE,UAAU;QACjB,OAAO,EAAE,EAAE;KACZ;IACD,SAAS,EAAE;QACT,KAAK,EAAE,WAAW;KACnB;IACD,aAAa,EAAE;QACb,KAAK,EAAE,eAAe;KACvB;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;QACnB,GAAG,EAAE,CAAC;KACP;IACD,qBAAqB,EAAE;QACrB,KAAK,EAAE,uBAAuB;QAC9B,GAAG,EAAE,CAAC;KACP;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,OAAO;QACd,OAAO,EAAE,CAAC;QACV,GAAG,EAAE,CAAC;QACN,GAAG,EAAE,CAAC;KACP;IACD,CAAC,EAAE;QACD,KAAK,EAAE,GAAG;QACV,OAAO,EAAE,CAAC;KACX;IACD,QAAQ,EAAE;QACR,KAAK,EAAE,UAAU;QACjB,OAAO,EAAE,KAAK;KACf;IACD,YAAY,EAAE;QACZ,KAAK,EAAE,cAAc;KACtB;IACD,MAAM,EAAE;QACN,KAAK,EAAE,QAAQ;QACf,OAAO,EAAE,KAAK;KACf;IACD,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM;KACd;IACD,gBAAgB,EAAE;QAChB,KAAK,EAAE,kBAAkB;QACzB,GAAG,EAAE,CAAC,CAAC;QACP,GAAG,EAAE,CAAC;KACP;IACD,iBAAiB,EAAE;QACjB,KAAK,EAAE,mBAAmB;QAC1B,GAAG,EAAE,CAAC,CAAC;QACP,GAAG,EAAE,CAAC;KACP;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;KACpB;IACD,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM;KACd;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;KACf;IACD,WAAW,EAAE;QACX,KAAK,EAAE,aAAa;KACrB;IACD,eAAe,EAAE;QACf,KAAK,EAAE,iBAAiB;KACzB;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;KACf;IACD,QAAQ,EAAE;QACR,KAAK,EAAE,UAAU;KAClB;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;KACpB;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;KACf;IACD,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM;KACd;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;KACpB;IACD,gBAAgB,EAAE;QAChB,KAAK,EAAE,kBAAkB;KAC1B;IACD,cAAc,EAAE;QACd,KAAK,EAAE,gBAAgB;QACvB,QAAQ,EAAE,CAAC,MAAc,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,KAAK,IAAI;QACpD,OAAO,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;QAChC,SAAS,EAAE,CAAC,MAAW,EAAE,EAAE;YACzB,MAAM,qBAAqB,GAAG,MAAM,CAAC,cAAc,IAAI,EAAE,CAAC;YAC1D,OAAO;gBACL,GAAG,qBAAqB;gBACxB,aAAa,EAAE,IAAI;aACpB,CAAC;QACJ,CAAC;KACF;CACF,CAAC;AAoCF;;GAEG;AACH,MAAM,CAAC,MAAM,yCAAyC,GAAG,CACvD,aAAqB,EACrB,UAAkB,EAClB,YAAqC,EACrC,uBAAgC,EAChC,eAAwB,EACJ,EAAE;IACtB,IAAI,KAAK,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC;IACjC,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IAC9C,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAC7B,IAAI,KAAK,KAAK,QAAQ;QAAE,OAAO,kBAAkB,CAAC;IAElD,IAAI,MAA4B,CAAC;IACjC,IAAI,CAAC;QACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAyB,CAAC;IACrD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,OAAO,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC/C,KAAK,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC;QACnB,KAAK,EAAE,CAAC,CAAC,KAAK,IAAI,EAAE;QACpB,aAAa,EAAE,CAAC,CAAC,aAAa,IAAI,IAAI;KACvC,CAAC,CAAC,CAAC;IAEJ,MAAM,WAAW,GAAG;QAClB,EAAE,EAAE,MAAM,CAAC,EAAE,IAAI,UAAU;QAC3B,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,uBAAuB;QAChD,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;QACxD,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,EAAE;QACzB,QAAQ,EAAE,SAAS;QACnB,OAAO;QACP,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;KAC7C,CAAC;IAEF,OAAO,SAAS,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,MAAM,CAAC;AACpD,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,0BAA0B,GAGO,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE;IACzE,IAAI,cAAc,KAAK,GAAG,IAAI,WAAW,IAAI,QAAQ,EAAE,CAAC;QACtD,OAAO,6BAA6B,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;IACjE,CAAC;IAED,OAAO,EAAE,GAAI,QAAmC,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC;AAC1E,CAAC,CAAC"}

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

@@ -0,0 +1,10 @@
+import { EmbedParams, EmbedResponse } from '../../types/embedRequestBody.js';
+import { ErrorResponse, ProviderConfig } from '../types.js';
+export declare const SagemakerEmbedConfig: ProviderConfig;
+/**
+ * Must match the arity and argument order of `handleResponseTransformWithGenericError` in
+ * `providers/utils.ts`: (body, status, responseHeaders, strictOpenAiCompliance, gatewayRequestUrl, gatewayRequest).
+ * If the signature is shorter or reorders parameters, `gatewayRequest` is bound to the wrong value (e.g. a boolean),
+ * `gatewayRequest.input` becomes undefined, and token estimation returns 0.
+ */
+export declare const SagemakerEmbedResponseTransform: (response: number[][] | ErrorResponse, responseStatus: number, _responseHeaders: Headers, _strictOpenAiCompliance: boolean, _gatewayRequestUrl: string, gatewayRequest: EmbedParams) => EmbedResponse | ErrorResponse;

package/dist/providers/sagemaker/embed.js

@@ -0,0 +1,55 @@
+import { SAGEMAKER } from '../../globals.js';
+import { numTokensInEmbeddingInput } from '../../utils.js';
+import { OpenAIErrorResponseTransform } from '../openai/utils.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 SagemakerEmbedConfig = {
+    input: {
+        param: 'inputs',
+        required: true,
+    },
+    model: {
+        param: 'model',
+    },
+    user: {
+        param: 'user',
+    },
+    encoding_format: {
+        param: 'encoding_format',
+        required: false,
+    },
+    dimensions: {
+        param: 'dimensions',
+    },
+};
+/**
+ * Must match the arity and argument order of `handleResponseTransformWithGenericError` in
+ * `providers/utils.ts`: (body, status, responseHeaders, strictOpenAiCompliance, gatewayRequestUrl, gatewayRequest).
+ * If the signature is shorter or reorders parameters, `gatewayRequest` is bound to the wrong value (e.g. a boolean),
+ * `gatewayRequest.input` becomes undefined, and token estimation returns 0.
+ */
+export const SagemakerEmbedResponseTransform = (response, responseStatus, _responseHeaders, _strictOpenAiCompliance, _gatewayRequestUrl, gatewayRequest) => {
+    if (responseStatus !== 200) {
+        return OpenAIErrorResponseTransform(response, SAGEMAKER);
+    }
+    // Calculate total tokens (approximate based on embedding dimensions)
+    const promptTokenCount = numTokensInEmbeddingInput(gatewayRequest.input);
+    if (Array.isArray(response)) {
+        return {
+            object: 'list',
+            data: response.map((embedding, index) => ({
+                object: 'embedding',
+                embedding: embedding,
+                index: index,
+            })),
+            provider: SAGEMAKER,
+            model: gatewayRequest.model,
+            usage: {
+                prompt_tokens: promptTokenCount,
+                total_tokens: promptTokenCount,
+            },
+        };
+    }
+    return generateInvalidProviderResponseError(response, SAGEMAKER);
+};
+//# sourceMappingURL=embed.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"embed.js","sourceRoot":"","sources":["../../../src/providers/sagemaker/embed.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAE1C,OAAO,EAAE,yBAAyB,EAAE,MAAM,aAAa,CAAC;AACxD,OAAO,EAAE,4BAA4B,EAAE,MAAM,iBAAiB,CAAC;AAE/D,OAAO,EAAE,oCAAoC,EAAE,MAAM,UAAU,CAAC;AAEhE,saAAsa;AAEta,MAAM,CAAC,MAAM,oBAAoB,GAAmB;IAClD,KAAK,EAAE;QACL,KAAK,EAAE,QAAQ;QACf,QAAQ,EAAE,IAAI;KACf;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;KACf;IACD,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM;KACd;IACD,eAAe,EAAE;QACf,KAAK,EAAE,iBAAiB;QACxB,QAAQ,EAAE,KAAK;KAChB;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;KACpB;CACF,CAAC;AAEF;;;;;GAKG;AACH,MAAM,CAAC,MAAM,+BAA+B,GAOP,CACnC,QAAQ,EACR,cAAc,EACd,gBAAgB,EAChB,uBAAuB,EACvB,kBAAkB,EAClB,cAAc,EACd,EAAE;IACF,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;QAC3B,OAAO,4BAA4B,CAAC,QAAyB,EAAE,SAAS,CAAC,CAAC;IAC5E,CAAC;IAED,qEAAqE;IACrE,MAAM,gBAAgB,GAAG,yBAAyB,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;IAEzE,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC5B,OAAO;YACL,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,SAAmB,EAAE,KAAa,EAAE,EAAE,CAAC,CAAC;gBAC1D,MAAM,EAAE,WAAW;gBACnB,SAAS,EAAE,SAAS;gBACpB,KAAK,EAAE,KAAK;aACb,CAAC,CAAC;YACH,QAAQ,EAAE,SAAS;YACnB,KAAK,EAAE,cAAc,CAAC,KAAK;YAC3B,KAAK,EAAE;gBACL,aAAa,EAAE,gBAAgB;gBAC/B,YAAY,EAAE,gBAAgB;aAC/B;SACF,CAAC;IACJ,CAAC;IAED,OAAO,oCAAoC,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;AACnE,CAAC,CAAC"}

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

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

package/dist/providers/sagemaker/index.js

@@ -0,0 +1,30 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { createModelResponseParams } from '../open-ai-base/index.js';
+import SagemakerAPIConfig from './api.js';
+import { SagemakerChatCompleteConfig, SagemakerChatCompleteStreamChunkTransform, SagemakerResponseTransform, } from './chatComplete.js';
+import { SagemakerEmbedConfig, SagemakerEmbedResponseTransform } from './embed.js';
+const SagemakerConfig = {
+    api: SagemakerAPIConfig,
+    chatComplete: SagemakerChatCompleteConfig,
+    embed: SagemakerEmbedConfig,
+    createTranscription: {},
+    createTranslation: {},
+    realtime: {},
+    cancelFinetune: {},
+    cancelBatch: {},
+    createModelResponse: createModelResponseParams([]),
+    getModelResponse: {},
+    deleteModelResponse: {},
+    listModelsResponse: {},
+    responseTransforms: {
+        chatComplete: SagemakerResponseTransform,
+        'stream-chatComplete': SagemakerChatCompleteStreamChunkTransform,
+        embed: SagemakerEmbedResponseTransform,
+    },
+};
+export default SagemakerConfig;
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/providers/sagemaker/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,yBAAyB,EAAE,MAAM,iBAAiB,CAAC;AAE5D,OAAO,kBAAkB,MAAM,OAAO,CAAC;AACvC,OAAO,EACL,2BAA2B,EAC3B,yCAAyC,EACzC,0BAA0B,GAC3B,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,oBAAoB,EAAE,+BAA+B,EAAE,MAAM,SAAS,CAAC;AAChF,MAAM,eAAe,GAAoB;IACvC,GAAG,EAAE,kBAAkB;IACvB,YAAY,EAAE,2BAA2B;IACzC,KAAK,EAAE,oBAAoB;IAC3B,mBAAmB,EAAE,EAAE;IACvB,iBAAiB,EAAE,EAAE;IACrB,QAAQ,EAAE,EAAE;IACZ,cAAc,EAAE,EAAE;IAClB,WAAW,EAAE,EAAE;IACf,mBAAmB,EAAE,yBAAyB,CAAC,EAAE,CAAC;IAClD,gBAAgB,EAAE,EAAE;IACpB,mBAAmB,EAAE,EAAE;IACvB,kBAAkB,EAAE,EAAE;IACtB,kBAAkB,EAAE;QAClB,YAAY,EAAE,0BAA0B;QACxC,qBAAqB,EAAE,yCAAyC;QAChE,KAAK,EAAE,+BAA+B;KACvC;CACF,CAAC;AAEF,eAAe,eAAe,CAAC"}

package/dist/providers/sagemaker/utils.d.ts

@@ -0,0 +1,3 @@
+import { type ErrorResponseBodyWithExtras } from '../../utils/errorHandlingUtils.js';
+import { SageMakerErrorResponse } from '../types.js';
+export declare function sagemakerResponseErrorHandler(responseStatus: number, response: SageMakerErrorResponse): ErrorResponseBodyWithExtras;

package/dist/providers/sagemaker/utils.js

@@ -0,0 +1,11 @@
+import { buildErrorBody } from '../../utils/errorHandlingUtils.js';
+export function sagemakerResponseErrorHandler(responseStatus, response) {
+    const message = response.OriginalMessage || response.Message || 'Something went wrong';
+    const code = response.OriginalStatusCode?.toString() || responseStatus.toString() || '500';
+    return buildErrorBody({
+        message,
+        statusCode: parseInt(code, 10) || 500,
+        type: null,
+    });
+}
+//# sourceMappingURL=utils.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/providers/sagemaker/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAoC,MAAM,gCAAgC,CAAC;AAGlG,MAAM,UAAU,6BAA6B,CAC3C,cAAsB,EACtB,QAAgC;IAEhC,MAAM,OAAO,GAAG,QAAQ,CAAC,eAAe,IAAI,QAAQ,CAAC,OAAO,IAAI,sBAAsB,CAAC;IACvF,MAAM,IAAI,GAAG,QAAQ,CAAC,kBAAkB,EAAE,QAAQ,EAAE,IAAI,cAAc,CAAC,QAAQ,EAAE,IAAI,KAAK,CAAC;IAC3F,OAAO,cAAc,CAAC;QACpB,OAAO;QACP,UAAU,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,GAAG;QACrC,IAAI,EAAE,IAAI;KACX,CAAC,CAAC;AACL,CAAC"}

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

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

package/dist/providers/sambanova/api.js

@@ -0,0 +1,18 @@
+const SambaNovaAPIConfig = {
+    getBaseURL: () => 'https://api.sambanova.ai'.replace(/\/$/, ''),
+    headers: ({ providerOptions }) => {
+        const manifest = providerOptions.providerAccount.manifest;
+        const apiKey = manifest.auth_data?.api_key;
+        return { Authorization: `Bearer ${apiKey}` };
+    },
+    getEndpoint: ({ endpoint }) => {
+        switch (endpoint) {
+            case 'chatComplete':
+                return '/v1/chat/completions';
+            default:
+                return '';
+        }
+    },
+};
+export default SambaNovaAPIConfig;
+//# sourceMappingURL=api.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../src/providers/sambanova/api.ts"],"names":[],"mappings":"AAGA,MAAM,kBAAkB,GAAsB;IAC5C,UAAU,EAAE,GAAG,EAAE,CAAC,0BAA0B,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC;IAC/D,OAAO,EAAE,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE;QAC/B,MAAM,QAAQ,GAAG,eAAe,CAAC,eAAe,CAAC,QAAoC,CAAC;QACtF,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,EAAE,OAAO,CAAC;QAC3C,OAAO,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE,CAAC;IAC/C,CAAC;IACD,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE;QAC5B,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,cAAc;gBACjB,OAAO,sBAAsB,CAAC;YAChC;gBACE,OAAO,EAAE,CAAC;QACd,CAAC;IACH,CAAC;CACF,CAAC;AAEF,eAAe,kBAAkB,CAAC"}

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

@@ -0,0 +1,31 @@
+export interface SambaNovaStreamChunk {
+    id: string;
+    object: string;
+    created: number;
+    model: string;
+    system_fingerprint: string;
+    choices: {
+        delta: {
+            content?: string;
+            tool_calls?: object[];
+        };
+        index: number;
+        finish_reason: string | null;
+        logprobs: object | null;
+    }[];
+    usage?: {
+        is_last_response: boolean;
+        total_tokens: number;
+        prompt_tokens: number;
+        completion_tokens: number;
+        time_to_first_token: number;
+        end_time: number;
+        start_time: number;
+        total_latency: number;
+        total_tokens_per_sec: number;
+        completion_tokens_per_sec: number;
+        completion_tokens_after_first_per_sec: number;
+        completion_tokens_after_first_per_sec_first_ten: number;
+    };
+}
+export declare const SambaNovaChatCompleteStreamChunkTransform: (response: string) => string;