Something went wrong. Try again.
opencode v2 mirror
Something went wrong. Try again.
3.0 kB · 100 lines
TypeScript
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101import type { LanguageModelV3 } from "@ai-sdk/provider"import { type FetchFunction, withoutTrailingSlash, withUserAgentSuffix } from "@ai-sdk/provider-utils"import { OpenAICompatibleChatLanguageModel } from "./chat/openai-compatible-chat-language-model.js"import { OpenAIResponsesLanguageModel } from "./responses/openai-responses-language-model.js"
// Import the version or define itconst VERSION = "0.1.0"
export type OpenaiCompatibleModelId = string
export interface OpenaiCompatibleProviderSettings { /** * API key for authenticating requests. */ apiKey?: string
/** * Base URL for the OpenAI Compatible API calls. */ baseURL?: string
/** * Name of the provider. */ name?: string
/** * Custom headers to include in the requests. */ headers?: Record<string, string>
/** * Custom fetch implementation. */ fetch?: FetchFunction}
export interface OpenaiCompatibleProvider { (modelId: OpenaiCompatibleModelId): LanguageModelV3 chat(modelId: OpenaiCompatibleModelId): LanguageModelV3 responses(modelId: OpenaiCompatibleModelId): LanguageModelV3 languageModel(modelId: OpenaiCompatibleModelId): LanguageModelV3
// embeddingModel(modelId: any): EmbeddingModelV2
// imageModel(modelId: any): ImageModelV2}
/** * Create an OpenAI Compatible provider instance. */export function createOpenaiCompatible(options: OpenaiCompatibleProviderSettings = {}): OpenaiCompatibleProvider { const baseURL = withoutTrailingSlash(options.baseURL ?? "https://api.openai.com/v1")
if (!baseURL) { throw new Error("baseURL is required") }
// Merge headers: defaults first, then user overrides const headers = { // Default OpenAI Compatible headers (can be overridden by user) ...(options.apiKey && { Authorization: `Bearer ${options.apiKey}` }), ...options.headers, }
const getHeaders = () => withUserAgentSuffix(headers, `ai-sdk/openai-compatible/${VERSION}`)
const createChatModel = (modelId: OpenaiCompatibleModelId) => { return new OpenAICompatibleChatLanguageModel(modelId, { provider: `${options.name ?? "openai-compatible"}.chat`, headers: getHeaders, url: ({ path }) => `${baseURL}${path}`, fetch: options.fetch, }) }
const createResponsesModel = (modelId: OpenaiCompatibleModelId) => { return new OpenAIResponsesLanguageModel(modelId, { provider: `${options.name ?? "openai-compatible"}.responses`, headers: getHeaders, url: ({ path }) => `${baseURL}${path}`, fetch: options.fetch, }) }
const createLanguageModel = (modelId: OpenaiCompatibleModelId) => createChatModel(modelId)
const provider = function (modelId: OpenaiCompatibleModelId) { return createChatModel(modelId) }
provider.languageModel = createLanguageModel provider.chat = createChatModel provider.responses = createResponsesModel
return provider as OpenaiCompatibleProvider}
// Default OpenAI Compatible provider instanceexport const openaiCompatible = createOpenaiCompatible()