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

package/dist/providers/cloudera/utils.js

@@ -0,0 +1,261 @@
+/**
+ * ================================
+ * TFY Managed File
+ * ================================
+ */
+import * as crypto from 'crypto';
+import { HTTPException } from 'hono/http-exception';
+import nacl from 'tweetnacl';
+import naclUtil from 'tweetnacl-util';
+import { getCachedCredentials, storeCachedCredentials } from '../../cache/port.js';
+import { safeFetch } from '../../http/index.js';
+import { logger } from '../../logger.js';
+import { createHashedCacheKey, credentialCacheExpiration, sanitizeNatsKey } from '../../utils.js';
+/**
+ * Enum for supported signature methods
+ */
+var SignatureMethod;
+(function (SignatureMethod) {
+    SignatureMethod["RSA_V1"] = "rsav1";
+    SignatureMethod["ECDSA_V1"] = "ecdsav1";
+    SignatureMethod["ED25519_V1"] = "ed25519v1";
+})(SignatureMethod || (SignatureMethod = {}));
+/**
+ * Normalizes a PEM key by extracting base64 content and rebuilding with proper formatting
+ *
+ * What this function does:
+ * 1. If key is space-separated (no/few newlines): Extracts pure base64, rebuilds with proper PEM format
+ * 2. If key already has proper newlines: Just normalizes line endings
+ *
+ * This ensures the key is always in valid PEM format that Node.js crypto can parse.
+ */
+function normalizePemKey(key) {
+    const trimmed = key.trim();
+    // Only process PEM-formatted keys (those with BEGIN/END markers)
+    // Ed25519 keys don't have these markers and should not be normalized
+    if (!trimmed.includes('-----BEGIN')) {
+        // Not a PEM key, return as-is (e.g., Ed25519 raw base64)
+        return trimmed;
+    }
+    const lines = trimmed.split(/\r?\n/);
+    // Check if key is already properly formatted (has header, footer, and base64 lines)
+    // A properly formatted PEM key should have at least:
+    // - 1 line for BEGIN header
+    // - 1+ lines for base64 content
+    // - 1 line for END footer
+    // So minimum 3 lines, but typically more due to base64 content
+    const hasProperFormat = lines.length >= 3 && lines[0].match(/-----BEGIN[^-]+-----/) && lines[lines.length - 1].match(/-----END[^-]+-----/);
+    if (hasProperFormat) {
+        // Key is already properly formatted, just normalize line endings
+        return trimmed.replace(/\r\n/g, '\n').replace(/\r/g, '\n');
+    }
+    // Key is not properly formatted (likely space-separated), rebuild it
+    // Extract the base64 content using the same approach as Google Vertex AI
+    const base64Content = trimmed
+        .replace(/-----BEGIN[^-]+-----/g, '')
+        .replace(/-----END[^-]+-----/g, '')
+        .replace(/\s+/g, ''); // Remove all whitespace (spaces, newlines, etc.)
+    // Extract the key type from the original
+    const beginMatch = trimmed.match(/-----BEGIN[^-]+-----/);
+    const endMatch = trimmed.match(/-----END[^-]+-----/);
+    const keyType = beginMatch ? beginMatch[0] : '-----BEGIN PRIVATE KEY-----';
+    const endType = endMatch ? endMatch[0] : '-----END PRIVATE KEY-----';
+    // Rebuild PEM with proper formatting: header, base64 lines (64 chars each), footer
+    const base64Lines = [];
+    for (let i = 0; i < base64Content.length; i += 64) {
+        base64Lines.push(base64Content.substring(i, i + 64));
+    }
+    return `${keyType}\n${base64Lines.join('\n')}\n${endType}`;
+}
+function determineSignatureMethod(privateKeyPem) {
+    // Ed25519 keys are base64 encoded seed of length 44 without PEM headers
+    if (privateKeyPem.length === 44 && !privateKeyPem.includes('-----BEGIN')) {
+        return SignatureMethod.ED25519_V1;
+    }
+    // For PEM-formatted keys, check if it's EC (ECDSA) by attempting to sign
+    if (privateKeyPem.includes('-----BEGIN')) {
+        try {
+            const normalizedKey = normalizePemKey(privateKeyPem);
+            const keyObject = crypto.createPrivateKey(normalizedKey);
+            const testSigner = crypto.createSign('ecdsa-with-SHA512');
+            testSigner.update('test', 'utf8');
+            testSigner.sign(keyObject, 'base64');
+            return SignatureMethod.ECDSA_V1;
+        }
+        catch {
+            return SignatureMethod.RSA_V1;
+        }
+    }
+    return SignatureMethod.RSA_V1;
+}
+function urlSafeBase64(buffer) {
+    const buf = Buffer.isBuffer(buffer) ? buffer : Buffer.from(buffer);
+    return buf
+        .toString('base64')
+        .replace(/\+/g, '-') // Replace + with -
+        .replace(/\//g, '_'); // Replace / with _
+}
+/**
+ * Creates a signature using the appropriate algorithm for the signature method
+ */
+function createSignature(privateKeyPem, stringToSign, signatureMethod) {
+    try {
+        switch (signatureMethod) {
+            case SignatureMethod.RSA_V1:
+                return signRSA(privateKeyPem, stringToSign);
+            case SignatureMethod.ECDSA_V1:
+                return signECDSA(privateKeyPem, stringToSign);
+            case SignatureMethod.ED25519_V1:
+                return signEd25519(privateKeyPem, stringToSign);
+            default:
+                throw new Error(`Unsupported signature method: ${signatureMethod}`);
+        }
+    }
+    catch (err) {
+        throw new Error(`Signature creation failed: ${err.message}`);
+    }
+}
+/**
+ * Signs a string using RSA-SHA256
+ * Uses PKCS#1 v1.5 padding (default in Node.js crypto)
+ */
+function signRSA(privateKeyPem, stringToSign) {
+    try {
+        const normalizedKey = normalizePemKey(privateKeyPem);
+        const keyObject = crypto.createPrivateKey(normalizedKey);
+        const signer = crypto.createSign('RSA-SHA256');
+        signer.update(stringToSign, 'utf8');
+        const signature = signer.sign(keyObject);
+        return urlSafeBase64(signature);
+    }
+    catch (err) {
+        const errorMessage = err instanceof Error ? err.message : String(err);
+        logger.error(`RSA signing failed: ${errorMessage}`);
+        throw new Error(`RSA signing failed: ${errorMessage}`);
+    }
+}
+/**
+ * Signs a string using ECDSA with SHA-512
+ */
+function signECDSA(privateKeyPem, stringToSign) {
+    try {
+        const normalizedKey = normalizePemKey(privateKeyPem);
+        const keyObject = crypto.createPrivateKey(normalizedKey);
+        const signer = crypto.createSign('ecdsa-with-SHA512');
+        signer.update(stringToSign, 'utf8');
+        const signature = signer.sign(keyObject);
+        return urlSafeBase64(signature);
+    }
+    catch (err) {
+        const errorMessage = err instanceof Error ? err.message : String(err);
+        logger.error(`ECDSA signing failed: ${errorMessage}`);
+        throw new Error(`ECDSA signing failed: ${errorMessage}`);
+    }
+}
+/**
+ * Signs a string using Ed25519
+ */
+function signEd25519(privateKeyPem, stringToSign) {
+    try {
+        const seed = new Uint8Array(Buffer.from(privateKeyPem, 'base64'));
+        const keyPair = nacl.sign.keyPair.fromSeed(seed);
+        const message = naclUtil.decodeUTF8(stringToSign);
+        const signature = nacl.sign.detached(message, keyPair.secretKey);
+        return urlSafeBase64(signature);
+    }
+    catch (e) {
+        const errorMessage = e instanceof Error ? e.message : String(e);
+        logger.error(`Ed25519 signing failed: ${errorMessage}`);
+        throw new Error(`Ed25519 signing failed: ${errorMessage}`);
+    }
+}
+/**
+ * Gets canonical headers in the exact format CDP CLI uses
+ */
+function getCanonicalHeaders(headers) {
+    const contentType = headers['Content-Type'] || '';
+    const altusDate = headers['X-Altus-Date'] || '';
+    return `${contentType}\n${altusDate}`;
+}
+/**
+ * Creates a canonical string exactly matching CDP CLI's format
+ */
+function createCanonicalString(method, path, headers, signatureMethod) {
+    // Format: method + '\n' + canonicalHeaders + '\n' + path + '\n' + signatureMethod
+    const canonicalHeaders = getCanonicalHeaders(headers);
+    return `${method.toUpperCase()}\n${canonicalHeaders}\n${path}\n${signatureMethod}`;
+}
+function createAuthHeader(accessKeyId, signatureMethod, signature) {
+    const authParams = {
+        access_key_id: accessKeyId,
+        auth_method: signatureMethod,
+    };
+    const encodedParams = urlSafeBase64(Buffer.from(JSON.stringify(authParams), 'utf-8'));
+    return `${encodedParams}.${signature}`;
+}
+export async function getClouderaWorkloadAuthToken(clouderaBaseUrl, clouderaAccessKeyId, clouderaPrivateKey, clouderaWorkloadName) {
+    try {
+        // Logical key — gateway CachePort adapter prefixes tenant.
+        const hashedKey = createHashedCacheKey(clouderaAccessKeyId, clouderaPrivateKey, clouderaWorkloadName, clouderaBaseUrl);
+        const originalKey = `cloudera-workload-auth-token-${hashedKey}`;
+        const cacheKey = sanitizeNatsKey(originalKey);
+        const cachedToken = await getCachedCredentials(cacheKey);
+        if (cachedToken?.access_token) {
+            return cachedToken.access_token;
+        }
+        const fullUrl = `${clouderaBaseUrl}/iam/generateWorkloadAuthToken`;
+        // Determine signature method (RSA, ECDSA, or Ed25519) based on private key format
+        const signatureMethod = determineSignatureMethod(clouderaPrivateKey);
+        const body = JSON.stringify({ workloadName: clouderaWorkloadName });
+        const timestamp = new Date().toUTCString();
+        const headers = {
+            'Content-Type': 'application/json',
+            'X-Altus-Date': timestamp,
+        };
+        // Build canonical string for signing (must match CDP CLI format exactly)
+        const urlObj = new URL(fullUrl);
+        const canonical = createCanonicalString('POST', urlObj.pathname, headers, signatureMethod);
+        const signature = createSignature(clouderaPrivateKey, canonical, signatureMethod);
+        headers['X-Altus-Auth'] = createAuthHeader(clouderaAccessKeyId, signatureMethod, signature);
+        const response = await safeFetch(fullUrl, { ssrf: 'guard', method: 'POST', headers, body });
+        if (!response.ok) {
+            let errorText;
+            try {
+                errorText = await response.text();
+            }
+            catch {
+                errorText = 'Unable to read error response';
+            }
+            const errorMessage = `CDP token request failed with status ${response.status}: ${errorText}`;
+            logger.error(errorMessage);
+            throw new Error(errorMessage);
+        }
+        const json = (await response.json());
+        if (!json.token) {
+            throw new Error('Token not found in CDP response payload');
+        }
+        const expiration = credentialCacheExpiration(json.expireAt ? new Date(json.expireAt).getTime() : undefined);
+        await storeCachedCredentials(cacheKey, { access_token: json.token }, expiration, originalKey);
+        return json.token;
+    }
+    catch (error) {
+        // Extract error details for proper logging
+        let errorMessage = 'Unknown error';
+        let errorStack;
+        if (error instanceof Error) {
+            errorMessage = error.message;
+            errorStack = error.stack;
+        }
+        else if (typeof error === 'string') {
+            errorMessage = error;
+        }
+        else {
+            errorMessage = JSON.stringify(error);
+        }
+        logger.error(`Failed to create Cloudera workload auth token: ${errorMessage}${errorStack ? `\nStack: ${errorStack}` : ''}`);
+        throw new HTTPException(401, {
+            message: 'Failed to create Cloudera workload auth token. Please verify credentials in Cloudera model account.',
+        });
+    }
+}
+//# sourceMappingURL=utils.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/providers/cloudera/utils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,KAAK,MAAM,MAAM,QAAQ,CAAC;AACjC,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,EAAE,oBAAoB,EAAE,sBAAsB,EAAE,MAAM,kBAAkB,CAAC;AAChF,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AACvC,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AACtC,OAAO,EAAE,oBAAoB,EAAE,yBAAyB,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAQ/F;;GAEG;AACH,IAAK,eAIJ;AAJD,WAAK,eAAe;IAClB,mCAAgB,CAAA;IAChB,uCAAoB,CAAA;IACpB,2CAAwB,CAAA;AAC1B,CAAC,EAJI,eAAe,KAAf,eAAe,QAInB;AAED;;;;;;;;GAQG;AACH,SAAS,eAAe,CAAC,GAAW;IAClC,MAAM,OAAO,GAAG,GAAG,CAAC,IAAI,EAAE,CAAC;IAE3B,iEAAiE;IACjE,qEAAqE;IACrE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;QACpC,yDAAyD;QACzD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAErC,oFAAoF;IACpF,qDAAqD;IACrD,4BAA4B;IAC5B,gCAAgC;IAChC,0BAA0B;IAC1B,+DAA+D;IAC/D,MAAM,eAAe,GACnB,KAAK,CAAC,MAAM,IAAI,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,sBAAsB,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;IAErH,IAAI,eAAe,EAAE,CAAC;QACpB,iEAAiE;QACjE,OAAO,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC7D,CAAC;IAED,qEAAqE;IACrE,yEAAyE;IACzE,MAAM,aAAa,GAAG,OAAO;SAC1B,OAAO,CAAC,uBAAuB,EAAE,EAAE,CAAC;SACpC,OAAO,CAAC,qBAAqB,EAAE,EAAE,CAAC;SAClC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,iDAAiD;IAEzE,yCAAyC;IACzC,MAAM,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC;IACzD,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;IACrD,MAAM,OAAO,GAAG,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,6BAA6B,CAAC;IAC3E,MAAM,OAAO,GAAG,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,2BAA2B,CAAC;IAErE,mFAAmF;IACnF,MAAM,WAAW,GAAa,EAAE,CAAC;IACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC;QAClD,WAAW,CAAC,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;IACvD,CAAC;IAED,OAAO,GAAG,OAAO,KAAK,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,OAAO,EAAE,CAAC;AAC7D,CAAC;AAED,SAAS,wBAAwB,CAAC,aAAqB;IACrD,wEAAwE;IACxE,IAAI,aAAa,CAAC,MAAM,KAAK,EAAE,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;QACzE,OAAO,eAAe,CAAC,UAAU,CAAC;IACpC,CAAC;IAED,yEAAyE;IACzE,IAAI,aAAa,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;QACzC,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,eAAe,CAAC,aAAa,CAAC,CAAC;YACrD,MAAM,SAAS,GAAG,MAAM,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;YACzD,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC;YAC1D,UAAU,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAClC,UAAU,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YACrC,OAAO,eAAe,CAAC,QAAQ,CAAC;QAClC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,eAAe,CAAC,MAAM,CAAC;QAChC,CAAC;IACH,CAAC;IAED,OAAO,eAAe,CAAC,MAAM,CAAC;AAChC,CAAC;AAED,SAAS,aAAa,CAAC,MAA2B;IAChD,MAAM,GAAG,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACnE,OAAO,GAAG;SACP,QAAQ,CAAC,QAAQ,CAAC;SAClB,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,mBAAmB;SACvC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC,mBAAmB;AAC7C,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,aAAqB,EAAE,YAAoB,EAAE,eAAgC;IACpG,IAAI,CAAC;QACH,QAAQ,eAAe,EAAE,CAAC;YACxB,KAAK,eAAe,CAAC,MAAM;gBACzB,OAAO,OAAO,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;YAE9C,KAAK,eAAe,CAAC,QAAQ;gBAC3B,OAAO,SAAS,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;YAEhD,KAAK,eAAe,CAAC,UAAU;gBAC7B,OAAO,WAAW,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;YAElD;gBACE,MAAM,IAAI,KAAK,CAAC,iCAAiC,eAAe,EAAE,CAAC,CAAC;QACxE,CAAC;IACH,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,8BAA+B,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;IAC1E,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,SAAS,OAAO,CAAC,aAAqB,EAAE,YAAoB;IAC1D,IAAI,CAAC;QACH,MAAM,aAAa,GAAG,eAAe,CAAC,aAAa,CAAC,CAAC;QACrD,MAAM,SAAS,GAAG,MAAM,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;QAEzD,MAAM,MAAM,GAAG,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QACpC,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACzC,OAAO,aAAa,CAAC,SAAS,CAAC,CAAC;IAClC,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,YAAY,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACtE,MAAM,CAAC,KAAK,CAAC,uBAAuB,YAAY,EAAE,CAAC,CAAC;QACpD,MAAM,IAAI,KAAK,CAAC,uBAAuB,YAAY,EAAE,CAAC,CAAC;IACzD,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,SAAS,CAAC,aAAqB,EAAE,YAAoB;IAC5D,IAAI,CAAC;QACH,MAAM,aAAa,GAAG,eAAe,CAAC,aAAa,CAAC,CAAC;QACrD,MAAM,SAAS,GAAG,MAAM,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;QACzD,MAAM,MAAM,GAAG,MAAM,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QACpC,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACzC,OAAO,aAAa,CAAC,SAAS,CAAC,CAAC;IAClC,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,YAAY,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACtE,MAAM,CAAC,KAAK,CAAC,yBAAyB,YAAY,EAAE,CAAC,CAAC;QACtD,MAAM,IAAI,KAAK,CAAC,yBAAyB,YAAY,EAAE,CAAC,CAAC;IAC3D,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,aAAqB,EAAE,YAAoB;IAC9D,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC,CAAC;QAClE,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QACjD,MAAM,OAAO,GAAG,QAAQ,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QAClD,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;QACjE,OAAO,aAAa,CAAC,SAAS,CAAC,CAAC;IAClC,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,MAAM,YAAY,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QAChE,MAAM,CAAC,KAAK,CAAC,2BAA2B,YAAY,EAAE,CAAC,CAAC;QACxD,MAAM,IAAI,KAAK,CAAC,2BAA2B,YAAY,EAAE,CAAC,CAAC;IAC7D,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,mBAAmB,CAAC,OAA+B;IAC1D,MAAM,WAAW,GAAG,OAAO,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;IAClD,MAAM,SAAS,GAAG,OAAO,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;IAChD,OAAO,GAAG,WAAW,KAAK,SAAS,EAAE,CAAC;AACxC,CAAC;AAED;;GAEG;AACH,SAAS,qBAAqB,CAC5B,MAAc,EACd,IAAY,EACZ,OAA+B,EAC/B,eAAgC;IAEhC,kFAAkF;IAClF,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,OAAO,CAAC,CAAC;IACtD,OAAO,GAAG,MAAM,CAAC,WAAW,EAAE,KAAK,gBAAgB,KAAK,IAAI,KAAK,eAAe,EAAE,CAAC;AACrF,CAAC;AAED,SAAS,gBAAgB,CAAC,WAAmB,EAAE,eAAgC,EAAE,SAAiB;IAChG,MAAM,UAAU,GAAG;QACjB,aAAa,EAAE,WAAW;QAC1B,WAAW,EAAE,eAAe;KAC7B,CAAC;IACF,MAAM,aAAa,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;IACtF,OAAO,GAAG,aAAa,IAAI,SAAS,EAAE,CAAC;AACzC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAChD,eAAuB,EACvB,mBAA2B,EAC3B,kBAA0B,EAC1B,oBAA4B;IAE5B,IAAI,CAAC;QACH,2DAA2D;QAC3D,MAAM,SAAS,GAAG,oBAAoB,CACpC,mBAAmB,EACnB,kBAAkB,EAClB,oBAAoB,EACpB,eAAe,CAChB,CAAC;QACF,MAAM,WAAW,GAAG,gCAAgC,SAAS,EAAE,CAAC;QAChE,MAAM,QAAQ,GAAG,eAAe,CAAC,WAAW,CAAC,CAAC;QAE9C,MAAM,WAAW,GAAG,MAAM,oBAAoB,CAA2B,QAAQ,CAAC,CAAC;QACnF,IAAI,WAAW,EAAE,YAAY,EAAE,CAAC;YAC9B,OAAO,WAAW,CAAC,YAAY,CAAC;QAClC,CAAC;QAED,MAAM,OAAO,GAAG,GAAG,eAAe,gCAAgC,CAAC;QAEnE,kFAAkF;QAClF,MAAM,eAAe,GAAG,wBAAwB,CAAC,kBAAkB,CAAC,CAAC;QACrE,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,YAAY,EAAE,oBAAoB,EAAE,CAAC,CAAC;QACpE,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAE3C,MAAM,OAAO,GAA2B;YACtC,cAAc,EAAE,kBAAkB;YAClC,cAAc,EAAE,SAAS;SAC1B,CAAC;QAEF,yEAAyE;QACzE,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC;QAChC,MAAM,SAAS,GAAG,qBAAqB,CAAC,MAAM,EAAE,MAAM,CAAC,QAAQ,EAAE,OAAO,EAAE,eAAe,CAAC,CAAC;QAE3F,MAAM,SAAS,GAAG,eAAe,CAAC,kBAAkB,EAAE,SAAS,EAAE,eAAe,CAAC,CAAC;QAClF,OAAO,CAAC,cAAc,CAAC,GAAG,gBAAgB,CAAC,mBAAmB,EAAE,eAAe,EAAE,SAAS,CAAC,CAAC;QAE5F,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAE5F,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,IAAI,SAAiB,CAAC;YACtB,IAAI,CAAC;gBACH,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACpC,CAAC;YAAC,MAAM,CAAC;gBACP,SAAS,GAAG,+BAA+B,CAAC;YAC9C,CAAC;YACD,MAAM,YAAY,GAAG,wCAAwC,QAAQ,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAC7F,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;QAChC,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA6B,CAAC;QAEjE,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;QAC7D,CAAC;QAED,MAAM,UAAU,GAAG,yBAAyB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAC5G,MAAM,sBAAsB,CAAC,QAAQ,EAAE,EAAE,YAAY,EAAE,IAAI,CAAC,KAAK,EAAE,EAAE,UAAU,EAAE,WAAW,CAAC,CAAC;QAE9F,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,2CAA2C;QAC3C,IAAI,YAAY,GAAG,eAAe,CAAC;QACnC,IAAI,UAA8B,CAAC;QAEnC,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;YAC3B,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC;YAC7B,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC;QAC3B,CAAC;aAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YACrC,YAAY,GAAG,KAAK,CAAC;QACvB,CAAC;aAAM,CAAC;YACN,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACvC,CAAC;QAED,MAAM,CAAC,KAAK,CACV,kDAAkD,YAAY,GAAG,UAAU,CAAC,CAAC,CAAC,YAAY,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAC9G,CAAC;QACF,MAAM,IAAI,aAAa,CAAC,GAAG,EAAE;YAC3B,OAAO,EAAE,qGAAqG;SAC/G,CAAC,CAAC;IACL,CAAC;AACH,CAAC"}

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

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

package/dist/providers/cohere/api.js

@@ -0,0 +1,52 @@
+import { sdkRequestUrl } from '../../utils/sdkRequestUrl.js';
+const CohereAPIConfig = {
+    getBaseURL: () => 'https://api.cohere.ai',
+    headers: ({ providerOptions, endpoint }) => {
+        const account = providerOptions.providerAccount.manifest;
+        const headers = {
+            Authorization: `Bearer ${account.auth_data?.api_key}`,
+        };
+        if (endpoint === 'uploadFile') {
+            headers['Content-Type'] = 'multipart/form-data';
+        }
+        return headers;
+    },
+    getEndpoint: ({ endpoint, request }) => {
+        const inboundUrl = sdkRequestUrl(request);
+        switch (endpoint) {
+            case 'chatComplete':
+                return '/v2/chat';
+            case 'complete':
+                return '/v1/generate';
+            case 'embed':
+                return '/v2/embed';
+            case 'uploadFile':
+                /**
+                 * ================================
+                 * NOTE: Change the dataset name to tfy-${crypto.randomUUID()}
+                 * ================================
+                 */
+                return `/v1/datasets?name=tfy-${crypto.randomUUID()}&type=embed-input&keep_fields=custom_id,id`;
+            case 'listFiles':
+                return '/v1/datasets';
+            case 'retrieveFile':
+                return `/v1/datasets/${inboundUrl.split('/').pop()}`;
+            case 'deleteFile':
+                return `/v1/datasets/${inboundUrl.split('/').pop()}`;
+            case 'createBatch':
+                return '/v1/embed-jobs';
+            case 'listBatches':
+                return '/v1/embed-jobs';
+            case 'retrieveBatch':
+                return `/v1/embed-jobs/${inboundUrl.split('/').pop()}`;
+            case 'cancelBatch':
+                return `/v1/embed-jobs/${inboundUrl.split('batches/').pop()}`;
+            case 'rerank':
+                return '/v1/rerank';
+            default:
+                return '';
+        }
+    },
+};
+export default CohereAPIConfig;
+//# sourceMappingURL=api.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../src/providers/cohere/api.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAG1D,MAAM,eAAe,GAAsB;IACzC,UAAU,EAAE,GAAG,EAAE,CAAC,uBAAuB;IACzC,OAAO,EAAE,CAAC,EAAE,eAAe,EAAE,QAAQ,EAAE,EAAE,EAAE;QACzC,MAAM,OAAO,GAAG,eAAe,CAAC,eAAe,CAAC,QAAiC,CAAC;QAClF,MAAM,OAAO,GAA2B;YACtC,aAAa,EAAE,UAAU,OAAO,CAAC,SAAS,EAAE,OAAO,EAAE;SACtD,CAAC;QACF,IAAI,QAAQ,KAAK,YAAY,EAAE,CAAC;YAC9B,OAAO,CAAC,cAAc,CAAC,GAAG,qBAAqB,CAAC;QAClD,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IACD,WAAW,EAAE,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE;QACrC,MAAM,UAAU,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;QAC1C,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,cAAc;gBACjB,OAAO,UAAU,CAAC;YACpB,KAAK,UAAU;gBACb,OAAO,cAAc,CAAC;YACxB,KAAK,OAAO;gBACV,OAAO,WAAW,CAAC;YACrB,KAAK,YAAY;gBACf;;;;mBAIG;gBACH,OAAO,yBAAyB,MAAM,CAAC,UAAU,EAAE,4CAA4C,CAAC;YAClG,KAAK,WAAW;gBACd,OAAO,cAAc,CAAC;YACxB,KAAK,cAAc;gBACjB,OAAO,gBAAgB,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC;YACvD,KAAK,YAAY;gBACf,OAAO,gBAAgB,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC;YACvD,KAAK,aAAa;gBAChB,OAAO,gBAAgB,CAAC;YAC1B,KAAK,aAAa;gBAChB,OAAO,gBAAgB,CAAC;YAC1B,KAAK,eAAe;gBAClB,OAAO,kBAAkB,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC;YACzD,KAAK,aAAa;gBAChB,OAAO,kBAAkB,UAAU,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC;YAChE,KAAK,QAAQ;gBACX,OAAO,YAAY,CAAC;YACtB;gBACE,OAAO,EAAE,CAAC;QACd,CAAC;IACH,CAAC;CACF,CAAC;AAEF,eAAe,eAAe,CAAC"}

package/dist/providers/cohere/cancelBatch.d.ts

@@ -0,0 +1,8 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { CancelBatchResponse, ErrorResponse } from '../types.js';
+import { CohereErrorResponse } from './types.js';
+export declare const CohereCancelBatchResponseTransform: (response: Response | CohereErrorResponse, responseStatus: number, _responseHeaders: Record<string, string>, _strictOpenAiCompliance: boolean, gatewayRequestUrl: string) => CancelBatchResponse | ErrorResponse;

package/dist/providers/cohere/cancelBatch.js

@@ -0,0 +1,12 @@
+import { CohereErrorResponseTransform } from './utils.js';
+export const CohereCancelBatchResponseTransform = (response, responseStatus, _responseHeaders, _strictOpenAiCompliance, gatewayRequestUrl) => {
+    if (responseStatus !== 200 && 'message' in response) {
+        return CohereErrorResponseTransform(response);
+    }
+    return {
+        status: 'success',
+        object: 'batch',
+        id: gatewayRequestUrl?.split('/v1/batches/')?.[1]?.replace('/cancel', '') || '',
+    };
+};
+//# sourceMappingURL=cancelBatch.js.map

package/dist/providers/cohere/cancelBatch.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cancelBatch.js","sourceRoot":"","sources":["../../../src/providers/cohere/cancelBatch.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,4BAA4B,EAAE,MAAM,SAAS,CAAC;AAEvD,MAAM,CAAC,MAAM,kCAAkC,GAAG,CAChD,QAAwC,EACxC,cAAsB,EACtB,gBAAwC,EACxC,uBAAgC,EAChC,iBAAyB,EACY,EAAE;IACvC,IAAI,cAAc,KAAK,GAAG,IAAI,SAAS,IAAI,QAAQ,EAAE,CAAC;QACpD,OAAO,4BAA4B,CAAC,QAAQ,CAAC,CAAC;IAChD,CAAC;IACD,OAAO;QACL,MAAM,EAAE,SAAS;QACjB,MAAM,EAAE,OAAO;QACf,EAAE,EAAE,iBAAiB,EAAE,KAAK,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,IAAI,EAAE;KAChF,CAAC;AACJ,CAAC,CAAC"}

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

@@ -0,0 +1,6 @@
+import { Params } from '../../types/requestBody.js';
+import { ChatCompletionResponse, ErrorResponse, ProviderConfig } from '../types.js';
+import { CohereChatCompleteResponse, CohereErrorResponse, CohereStreamState } from './types.js';
+export declare const CohereChatCompleteConfig: ProviderConfig;
+export declare const CohereChatCompleteResponseTransform: (response: CohereChatCompleteResponse | CohereErrorResponse, responseStatus: number, responseHeaders: Headers, strictOpenAiCompliance: boolean, gatewayRequestUrl: string, gatewayRequest: Params) => ChatCompletionResponse | ErrorResponse;
+export declare const CohereChatCompleteStreamChunkTransform: (response: string, fallbackId: string, streamState: CohereStreamState, strictOpenAiCompliance: boolean, gatewayRequest: Params) => string;