ai-gateway-provider
Version:
AI Gateway Provider for AI-SDK
106 lines (89 loc) • 3.96 kB
JavaScript
// src/providers/amazon-bedrock.ts
import { createAmazonBedrock as createAmazonBedrockOriginal } from "@ai-sdk/amazon-bedrock";
// src/auth.ts
var CF_TEMP_TOKEN = "CF_TEMP_TOKEN";
function authWrapper(func) {
return (config) => {
if (!config) {
return func({ apiKey: CF_TEMP_TOKEN });
}
if (config.apiKey === void 0) {
config.apiKey = CF_TEMP_TOKEN;
}
return func(config);
};
}
// src/providers/amazon-bedrock.ts
var createAmazonBedrock = (...args) => authWrapper(createAmazonBedrockOriginal)(...args);
// src/providers/google-vertex.ts
import { createVertex as createVertexOriginal } from "@ai-sdk/google-vertex";
var createVertex = (...rest) => {
if (!rest || !rest[0]) {
return createVertexOriginal({ googleAuthOptions: { apiKey: CF_TEMP_TOKEN } });
}
if (rest[0].googleAuthOptions && rest[0].googleAuthOptions.apiKey === void 0 && rest[0].googleAuthOptions.credentials === void 0) {
rest[0].googleAuthOptions.apiKey = CF_TEMP_TOKEN;
}
return createVertexOriginal(...rest);
};
// src/providers/openai.ts
import { createOpenAI as createOpenAIOriginal } from "@ai-sdk/openai";
var createOpenAI = (...args) => authWrapper(createOpenAIOriginal)(...args);
// src/providers/xai.ts
import { createXai as createXaiOriginal } from "@ai-sdk/xai";
var createXai = (...args) => authWrapper(createXaiOriginal)(...args);
// src/providers/google.ts
import { createGoogleGenerativeAI as createGoogleGenerativeAIOriginal } from "@ai-sdk/google";
var createGoogleGenerativeAI = (...args) => authWrapper(createGoogleGenerativeAIOriginal)(...args);
// src/providers/groq.ts
import { createGroq as createGroqOriginal } from "@ai-sdk/groq";
var createGroq = (...args) => authWrapper(createGroqOriginal)(...args);
// src/providers/mistral.ts
import { createMistral as createMistralOriginal } from "@ai-sdk/mistral";
var createMistral = (...args) => authWrapper(createMistralOriginal)(...args);
// src/providers/perplexity.ts
import { createPerplexity as createPerplexityOriginal } from "@ai-sdk/perplexity";
var createPerplexity = (...args) => authWrapper(createPerplexityOriginal)(...args);
// src/providers/anthropic.ts
import { createAnthropic as createAnthropicOriginal } from "@ai-sdk/anthropic";
var createAnthropic = (...args) => authWrapper(createAnthropicOriginal)(...args);
// src/providers/azure.ts
import { createAzure as createAzureOriginal } from "@ai-sdk/azure";
var createAzure = (...args) => authWrapper(createAzureOriginal)(...args);
// src/providers/cerebras.ts
import { createCerebras as createCerebrasOriginal } from "@ai-sdk/cerebras";
var createCerebras = (...args) => authWrapper(createCerebrasOriginal)(...args);
// src/providers/cohere.ts
import { createCohere as createCohereOriginal } from "@ai-sdk/cohere";
var createCohere = (...args) => authWrapper(createCohereOriginal)(...args);
// src/providers/deepgram.ts
import { createDeepgram as createDeepgramOriginal } from "@ai-sdk/deepgram";
var createDeepgram = (...args) => authWrapper(createDeepgramOriginal)(...args);
// src/providers/deepseek.ts
import { createDeepSeek as createDeepSeekOriginal } from "@ai-sdk/deepseek";
var createDeepSeek = (...args) => authWrapper(createDeepSeekOriginal)(...args);
// src/providers/elevenlabs.ts
import { createElevenLabs as createElevenLabsOriginal } from "@ai-sdk/elevenlabs";
var createElevenLabs = (...args) => authWrapper(createElevenLabsOriginal)(...args);
// src/providers/fireworks.ts
import { createFireworks as createFireworksOriginal } from "@ai-sdk/fireworks";
var createFireworks = (...args) => authWrapper(createFireworksOriginal)(...args);
export {
createAmazonBedrock,
createAnthropic,
createAzure,
createCerebras,
createCohere,
createDeepSeek,
createDeepgram,
createElevenLabs,
createFireworks,
createGoogleGenerativeAI,
createGroq,
createMistral,
createOpenAI,
createPerplexity,
createVertex,
createXai
};
//# sourceMappingURL=index.mjs.map