| import { Effect } from "effect" |
| import { define } from "../internal" |
| import { ProviderV2 } from "../../provider" |
|
|
| function selectLanguage(sdk: any, modelID: string, useChat: boolean) { |
| if (useChat && sdk.chat) return sdk.chat(modelID) |
| if (sdk.responses) return sdk.responses(modelID) |
| if (sdk.messages) return sdk.messages(modelID) |
| if (sdk.chat) return sdk.chat(modelID) |
| return sdk.languageModel(modelID) |
| } |
|
|
| export const AzurePlugin = define({ |
| id: "azure", |
| effect: Effect.fn(function* (ctx) { |
| yield* ctx.catalog.transform( |
| Effect.fn(function* (evt) { |
| for (const item of evt.provider.list()) { |
| if (item.provider.api.type !== "aisdk") continue |
| if (item.provider.api.package !== "@ai-sdk/azure") continue |
| const configured = item.provider.request.body.resourceName |
| const resourceName = |
| typeof configured === "string" && configured.trim() !== "" ? configured : process.env.AZURE_RESOURCE_NAME |
| if (!resourceName) continue |
| evt.provider.update(item.provider.id, (provider) => { |
| provider.request.body.resourceName = resourceName |
| }) |
| } |
| }), |
| ) |
| yield* ctx.aisdk.sdk( |
| Effect.fn(function* (evt) { |
| if (evt.package !== "@ai-sdk/azure") return |
| if (evt.model.providerID === ProviderV2.ID.azure) { |
| if ( |
| !evt.options.resourceName && |
| !evt.options.baseURL && |
| (evt.model.api.type !== "aisdk" || !evt.model.api.url) |
| ) { |
| throw new Error( |
| "AZURE_RESOURCE_NAME is missing, set it using env var or reconnecting the azure provider and setting it", |
| ) |
| } |
| } |
| const mod = yield* Effect.promise(() => import("@ai-sdk/azure")) |
| evt.sdk = mod.createAzure(evt.options) |
| }), |
| ) |
| yield* ctx.aisdk.language( |
| Effect.fn(function* (evt) { |
| if (evt.model.providerID !== ProviderV2.ID.azure) return |
| evt.language = selectLanguage(evt.sdk, evt.model.api.id, Boolean(evt.options.useCompletionUrls)) |
| }), |
| ) |
| }), |
| }) |
|
|
| export const AzureCognitiveServicesPlugin = define({ |
| id: "azure-cognitive-services", |
| effect: Effect.fn(function* (ctx) { |
| yield* ctx.catalog.transform( |
| Effect.fn(function* (evt) { |
| const resourceName = process.env.AZURE_COGNITIVE_SERVICES_RESOURCE_NAME |
| if (!resourceName) return |
| for (const item of evt.provider.list()) { |
| if (item.provider.api.type !== "aisdk") continue |
| if (item.provider.api.package !== "@ai-sdk/openai-compatible") continue |
| if (!item.provider.id.includes("azure-cognitive-services")) continue |
| evt.provider.update(item.provider.id, (provider) => { |
| provider.request.body.baseURL = `https://${resourceName}.cognitiveservices.azure.com/openai` |
| }) |
| } |
| }), |
| ) |
| yield* ctx.aisdk.language( |
| Effect.fn(function* (evt) { |
| if (evt.model.providerID !== ProviderV2.ID.make("azure-cognitive-services")) return |
| evt.language = selectLanguage(evt.sdk, evt.model.api.id, Boolean(evt.options.useCompletionUrls)) |
| }), |
| ) |
| }), |
| }) |
|
|
