| import os from "os" |
| import { InstallationVersion } from "../../installation/version" |
| import { Effect } from "effect" |
| import { define } from "../internal" |
| import { ProviderV2 } from "../../provider" |
|
|
| export const GitLabPlugin = define({ |
| id: "gitlab", |
| effect: Effect.fn(function* (ctx) { |
| yield* ctx.aisdk.sdk( |
| Effect.fn(function* (evt) { |
| if (evt.package !== "gitlab-ai-provider") return |
| const mod = yield* Effect.promise(() => import("gitlab-ai-provider")) |
| evt.sdk = mod.createGitLab({ |
| ...evt.options, |
| instanceUrl: |
| typeof evt.options.instanceUrl === "string" |
| ? evt.options.instanceUrl |
| : (process.env.GITLAB_INSTANCE_URL ?? "https://gitlab.com"), |
| apiKey: typeof evt.options.apiKey === "string" ? evt.options.apiKey : process.env.GITLAB_TOKEN, |
| aiGatewayHeaders: { |
| "User-Agent": `opencode/${InstallationVersion} gitlab-ai-provider/${mod.VERSION} (${os.platform()} ${os.release()}; ${os.arch()})`, |
| "anthropic-beta": "context-1m-2025-08-07", |
| ...evt.options.aiGatewayHeaders, |
| }, |
| featureFlags: { |
| duo_agent_platform_agentic_chat: true, |
| duo_agent_platform: true, |
| ...evt.options.featureFlags, |
| }, |
| }) |
| }), |
| ) |
| yield* ctx.aisdk.language( |
| Effect.fn(function* (evt) { |
| if (evt.model.providerID !== ProviderV2.ID.gitlab) return |
| const featureFlags = |
| typeof evt.options.featureFlags === "object" && evt.options.featureFlags ? evt.options.featureFlags : {} |
| if (evt.model.api.id.startsWith("duo-workflow-")) { |
| const gitlab = yield* Effect.promise(() => import("gitlab-ai-provider")).pipe(Effect.orDie) |
| const workflowRef = |
| typeof evt.model.request.body.workflowRef === "string" ? evt.model.request.body.workflowRef : undefined |
| const workflowDefinition = |
| typeof evt.model.request.body.workflowDefinition === "string" |
| ? evt.model.request.body.workflowDefinition |
| : undefined |
| const language = evt.sdk.workflowChat( |
| gitlab.isWorkflowModel(evt.model.api.id) ? evt.model.api.id : "duo-workflow", |
| { |
| featureFlags, |
| workflowDefinition, |
| }, |
| ) |
| if (workflowRef) language.selectedModelRef = workflowRef |
| evt.language = language |
| return |
| } |
| evt.language = evt.sdk.agenticChat(evt.model.api.id, { |
| aiGatewayHeaders: evt.options.aiGatewayHeaders, |
| featureFlags, |
| }) |
| }), |
| ) |
| }), |
| }) |
|
|
