gitlab-ai-provider 5.1.2 → 5.2.0

package/CHANGELOG.md

@@ -2,6 +2,11 @@
 
 All notable changes to this project will be documented in this file. See [Conventional Commits](https://conventionalcommits.org) for commit guidelines.
 
+## 5.2.0 (2026-03-18)
+
+- fix: update dists ([57692b5](https://gitlab.com/vglafirov/gitlab-ai-provider/commit/57692b5))
+- feat: add discoverWorkflowModels() for high-level workflow model discovery ([b61ed5a](https://gitlab.com/vglafirov/gitlab-ai-provider/commit/b61ed5a))
+
 ## <small>5.1.2 (2026-03-17)</small>
 
 - fix: regenerate dists ([091d3ae](https://gitlab.com/vglafirov/gitlab-ai-provider/commit/091d3ae))

package/dist/index.d.mts

@@ -1573,4 +1573,41 @@ declare class GitLabModelConfigRegistry {
  */
 declare function parseModelsYml(text: string): Map<string, ModelConfig>;
 
-export { AGENT_PRIVILEGES, type ActionResponsePayload, type AdditionalContext, type AiChatAvailableModels, type AiModel, BUNDLED_CLIENT_ID, CLIENT_VERSION, type ClientEvent, DEFAULT_AGENT_PRIVILEGES, DEFAULT_AI_GATEWAY_URL, DEFAULT_CLIENT_CAPABILITIES, DEFAULT_WORKFLOW_DEFINITION, type DirectAccessToken, type DiscoveredModels, GITLAB_COM_URL, type GenerateTokenResponse, type GitLabAgenticOptions, type GitLabAnthropicConfig, GitLabAnthropicLanguageModel, GitLabDirectAccessClient, type GitLabDirectAccessConfig, GitLabError, type GitLabErrorOptions, GitLabModelCache, GitLabModelConfigRegistry, GitLabModelDiscovery, GitLabOAuthManager, type GitLabOAuthTokenResponse, type GitLabOAuthTokens, type GitLabOpenAIConfig, GitLabOpenAILanguageModel, type GitLabProject, GitLabProjectCache, GitLabProjectDetector, type GitLabProjectDetectorConfig, type GitLabProvider, type GitLabProviderSettings, GitLabWorkflowClient, type GitLabWorkflowClientConfig, GitLabWorkflowLanguageModel, type GitLabWorkflowLanguageModelConfig, type GitLabWorkflowOptions, GitLabWorkflowTokenClient, MODEL_ID_TO_ANTHROPIC_MODEL, MODEL_MAPPINGS, type McpToolDefinition, type ModelCacheEntry, type ModelConfig, type ModelConfigRegistryOptions, type ModelDiscoveryConfig, type ModelMapping, type ModelProvider, type NewCheckpoint, OAUTH_SCOPES, OPENCODE_GITLAB_AUTH_CLIENT_ID, type OpenAIApiType, type OpenCodeAuth, type OpenCodeAuthApi, type OpenCodeAuthOAuth, type RunMcpTool, type StartRequest, TOKEN_EXPIRY_SKEW_MS, VERSION, WORKFLOW_ENVIRONMENT, WS_HEARTBEAT_INTERVAL_MS, WS_KEEPALIVE_PING_INTERVAL_MS, type WorkflowAction, type WorkflowClientEvent, type WorkflowStatus, type WorkflowToolExecutor, WorkflowType, type WorkflowWebSocketOptions, createGitLab, getAnthropicModelForModelId, getModelMapping, getOpenAIApiType, getOpenAIModelForModelId, getProviderForModelId, getValidModelsForProvider, getWorkflowModelRef, gitlab, isResponsesApiModel, isWorkflowModel, parseModelsYml };
+/**
+ * High-level workflow model discovery.
+ *
+ * Combines project detection, GraphQL model discovery, model config lookup,
+ * and MODEL_MAPPINGS to produce a list of available workflow models with
+ * their token limits. Consumers map these into their own internal model
+ * representations.
+ */
+
+interface DiscoveredWorkflowModel {
+    /** Model ID suitable for use with `sdk.workflowChat()` (e.g. "duo-workflow-sonnet-4-6") */
+    id: string;
+    /** Upstream model ref (e.g. "claude_sonnet_4_6") */
+    ref: string;
+    /** Human-readable name (e.g. "Claude Sonnet 4.6") */
+    name: string;
+    /** Max input context tokens */
+    context: number;
+    /** Max output tokens */
+    output: number;
+    /** Whether this model was admin-pinned */
+    pinned: boolean;
+}
+interface WorkflowDiscoveryConfig {
+    instanceUrl: string;
+    getHeaders: () => Record<string, string>;
+    fetch?: typeof fetch;
+}
+interface WorkflowDiscoveryOptions {
+    workingDirectory: string;
+}
+interface WorkflowDiscoveryResult {
+    models: DiscoveredWorkflowModel[];
+    project: GitLabProject | null;
+}
+declare function discoverWorkflowModels(config: WorkflowDiscoveryConfig, options: WorkflowDiscoveryOptions): Promise<WorkflowDiscoveryResult>;
+
+export { AGENT_PRIVILEGES, type ActionResponsePayload, type AdditionalContext, type AiChatAvailableModels, type AiModel, BUNDLED_CLIENT_ID, CLIENT_VERSION, type ClientEvent, DEFAULT_AGENT_PRIVILEGES, DEFAULT_AI_GATEWAY_URL, DEFAULT_CLIENT_CAPABILITIES, DEFAULT_WORKFLOW_DEFINITION, type DirectAccessToken, type DiscoveredModels, type DiscoveredWorkflowModel, GITLAB_COM_URL, type GenerateTokenResponse, type GitLabAgenticOptions, type GitLabAnthropicConfig, GitLabAnthropicLanguageModel, GitLabDirectAccessClient, type GitLabDirectAccessConfig, GitLabError, type GitLabErrorOptions, GitLabModelCache, GitLabModelConfigRegistry, GitLabModelDiscovery, GitLabOAuthManager, type GitLabOAuthTokenResponse, type GitLabOAuthTokens, type GitLabOpenAIConfig, GitLabOpenAILanguageModel, type GitLabProject, GitLabProjectCache, GitLabProjectDetector, type GitLabProjectDetectorConfig, type GitLabProvider, type GitLabProviderSettings, GitLabWorkflowClient, type GitLabWorkflowClientConfig, GitLabWorkflowLanguageModel, type GitLabWorkflowLanguageModelConfig, type GitLabWorkflowOptions, GitLabWorkflowTokenClient, MODEL_ID_TO_ANTHROPIC_MODEL, MODEL_MAPPINGS, type McpToolDefinition, type ModelCacheEntry, type ModelConfig, type ModelConfigRegistryOptions, type ModelDiscoveryConfig, type ModelMapping, type ModelProvider, type NewCheckpoint, OAUTH_SCOPES, OPENCODE_GITLAB_AUTH_CLIENT_ID, type OpenAIApiType, type OpenCodeAuth, type OpenCodeAuthApi, type OpenCodeAuthOAuth, type RunMcpTool, type StartRequest, TOKEN_EXPIRY_SKEW_MS, VERSION, WORKFLOW_ENVIRONMENT, WS_HEARTBEAT_INTERVAL_MS, WS_KEEPALIVE_PING_INTERVAL_MS, type WorkflowAction, type WorkflowClientEvent, type WorkflowDiscoveryConfig, type WorkflowDiscoveryOptions, type WorkflowDiscoveryResult, type WorkflowStatus, type WorkflowToolExecutor, WorkflowType, type WorkflowWebSocketOptions, createGitLab, discoverWorkflowModels, getAnthropicModelForModelId, getModelMapping, getOpenAIApiType, getOpenAIModelForModelId, getProviderForModelId, getValidModelsForProvider, getWorkflowModelRef, gitlab, isResponsesApiModel, isWorkflowModel, parseModelsYml };

package/dist/index.d.ts

@@ -1573,4 +1573,41 @@ declare class GitLabModelConfigRegistry {
  */
 declare function parseModelsYml(text: string): Map<string, ModelConfig>;
 
-export { AGENT_PRIVILEGES, type ActionResponsePayload, type AdditionalContext, type AiChatAvailableModels, type AiModel, BUNDLED_CLIENT_ID, CLIENT_VERSION, type ClientEvent, DEFAULT_AGENT_PRIVILEGES, DEFAULT_AI_GATEWAY_URL, DEFAULT_CLIENT_CAPABILITIES, DEFAULT_WORKFLOW_DEFINITION, type DirectAccessToken, type DiscoveredModels, GITLAB_COM_URL, type GenerateTokenResponse, type GitLabAgenticOptions, type GitLabAnthropicConfig, GitLabAnthropicLanguageModel, GitLabDirectAccessClient, type GitLabDirectAccessConfig, GitLabError, type GitLabErrorOptions, GitLabModelCache, GitLabModelConfigRegistry, GitLabModelDiscovery, GitLabOAuthManager, type GitLabOAuthTokenResponse, type GitLabOAuthTokens, type GitLabOpenAIConfig, GitLabOpenAILanguageModel, type GitLabProject, GitLabProjectCache, GitLabProjectDetector, type GitLabProjectDetectorConfig, type GitLabProvider, type GitLabProviderSettings, GitLabWorkflowClient, type GitLabWorkflowClientConfig, GitLabWorkflowLanguageModel, type GitLabWorkflowLanguageModelConfig, type GitLabWorkflowOptions, GitLabWorkflowTokenClient, MODEL_ID_TO_ANTHROPIC_MODEL, MODEL_MAPPINGS, type McpToolDefinition, type ModelCacheEntry, type ModelConfig, type ModelConfigRegistryOptions, type ModelDiscoveryConfig, type ModelMapping, type ModelProvider, type NewCheckpoint, OAUTH_SCOPES, OPENCODE_GITLAB_AUTH_CLIENT_ID, type OpenAIApiType, type OpenCodeAuth, type OpenCodeAuthApi, type OpenCodeAuthOAuth, type RunMcpTool, type StartRequest, TOKEN_EXPIRY_SKEW_MS, VERSION, WORKFLOW_ENVIRONMENT, WS_HEARTBEAT_INTERVAL_MS, WS_KEEPALIVE_PING_INTERVAL_MS, type WorkflowAction, type WorkflowClientEvent, type WorkflowStatus, type WorkflowToolExecutor, WorkflowType, type WorkflowWebSocketOptions, createGitLab, getAnthropicModelForModelId, getModelMapping, getOpenAIApiType, getOpenAIModelForModelId, getProviderForModelId, getValidModelsForProvider, getWorkflowModelRef, gitlab, isResponsesApiModel, isWorkflowModel, parseModelsYml };
+/**
+ * High-level workflow model discovery.
+ *
+ * Combines project detection, GraphQL model discovery, model config lookup,
+ * and MODEL_MAPPINGS to produce a list of available workflow models with
+ * their token limits. Consumers map these into their own internal model
+ * representations.
+ */
+
+interface DiscoveredWorkflowModel {
+    /** Model ID suitable for use with `sdk.workflowChat()` (e.g. "duo-workflow-sonnet-4-6") */
+    id: string;
+    /** Upstream model ref (e.g. "claude_sonnet_4_6") */
+    ref: string;
+    /** Human-readable name (e.g. "Claude Sonnet 4.6") */
+    name: string;
+    /** Max input context tokens */
+    context: number;
+    /** Max output tokens */
+    output: number;
+    /** Whether this model was admin-pinned */
+    pinned: boolean;
+}
+interface WorkflowDiscoveryConfig {
+    instanceUrl: string;
+    getHeaders: () => Record<string, string>;
+    fetch?: typeof fetch;
+}
+interface WorkflowDiscoveryOptions {
+    workingDirectory: string;
+}
+interface WorkflowDiscoveryResult {
+    models: DiscoveredWorkflowModel[];
+    project: GitLabProject | null;
+}
+declare function discoverWorkflowModels(config: WorkflowDiscoveryConfig, options: WorkflowDiscoveryOptions): Promise<WorkflowDiscoveryResult>;
+
+export { AGENT_PRIVILEGES, type ActionResponsePayload, type AdditionalContext, type AiChatAvailableModels, type AiModel, BUNDLED_CLIENT_ID, CLIENT_VERSION, type ClientEvent, DEFAULT_AGENT_PRIVILEGES, DEFAULT_AI_GATEWAY_URL, DEFAULT_CLIENT_CAPABILITIES, DEFAULT_WORKFLOW_DEFINITION, type DirectAccessToken, type DiscoveredModels, type DiscoveredWorkflowModel, GITLAB_COM_URL, type GenerateTokenResponse, type GitLabAgenticOptions, type GitLabAnthropicConfig, GitLabAnthropicLanguageModel, GitLabDirectAccessClient, type GitLabDirectAccessConfig, GitLabError, type GitLabErrorOptions, GitLabModelCache, GitLabModelConfigRegistry, GitLabModelDiscovery, GitLabOAuthManager, type GitLabOAuthTokenResponse, type GitLabOAuthTokens, type GitLabOpenAIConfig, GitLabOpenAILanguageModel, type GitLabProject, GitLabProjectCache, GitLabProjectDetector, type GitLabProjectDetectorConfig, type GitLabProvider, type GitLabProviderSettings, GitLabWorkflowClient, type GitLabWorkflowClientConfig, GitLabWorkflowLanguageModel, type GitLabWorkflowLanguageModelConfig, type GitLabWorkflowOptions, GitLabWorkflowTokenClient, MODEL_ID_TO_ANTHROPIC_MODEL, MODEL_MAPPINGS, type McpToolDefinition, type ModelCacheEntry, type ModelConfig, type ModelConfigRegistryOptions, type ModelDiscoveryConfig, type ModelMapping, type ModelProvider, type NewCheckpoint, OAUTH_SCOPES, OPENCODE_GITLAB_AUTH_CLIENT_ID, type OpenAIApiType, type OpenCodeAuth, type OpenCodeAuthApi, type OpenCodeAuthOAuth, type RunMcpTool, type StartRequest, TOKEN_EXPIRY_SKEW_MS, VERSION, WORKFLOW_ENVIRONMENT, WS_HEARTBEAT_INTERVAL_MS, WS_KEEPALIVE_PING_INTERVAL_MS, type WorkflowAction, type WorkflowClientEvent, type WorkflowDiscoveryConfig, type WorkflowDiscoveryOptions, type WorkflowDiscoveryResult, type WorkflowStatus, type WorkflowToolExecutor, WorkflowType, type WorkflowWebSocketOptions, createGitLab, discoverWorkflowModels, getAnthropicModelForModelId, getModelMapping, getOpenAIApiType, getOpenAIModelForModelId, getProviderForModelId, getValidModelsForProvider, getWorkflowModelRef, gitlab, isResponsesApiModel, isWorkflowModel, parseModelsYml };

package/dist/index.js

@@ -61,6 +61,7 @@ __export(index_exports, {
   WS_KEEPALIVE_PING_INTERVAL_MS: () => WS_KEEPALIVE_PING_INTERVAL_MS,
   WorkflowType: () => WorkflowType,
   createGitLab: () => createGitLab,
+  discoverWorkflowModels: () => discoverWorkflowModels,
   getAnthropicModelForModelId: () => getAnthropicModelForModelId,
   getModelMapping: () => getModelMapping,
   getOpenAIApiType: () => getOpenAIApiType,
@@ -1576,7 +1577,7 @@ var GitLabOpenAILanguageModel = class {
 var import_isomorphic_ws = __toESM(require("isomorphic-ws"));
 
 // src/version.ts
-var VERSION = true ? "5.1.1" : "0.0.0-dev";
+var VERSION = true ? "5.2.0" : "0.0.0-dev";
 
 // src/gitlab-workflow-types.ts
 var WorkflowType = /* @__PURE__ */ ((WorkflowType2) => {
@@ -4493,6 +4494,60 @@ function parseModelsYml(text) {
   }
   return configs;
 }
+
+// src/gitlab-workflow-discovery.ts
+var configRegistry = new GitLabModelConfigRegistry();
+async function discoverWorkflowModels(config, options) {
+  const detector = new GitLabProjectDetector({
+    instanceUrl: config.instanceUrl,
+    getHeaders: config.getHeaders,
+    fetch: config.fetch
+  });
+  let project = null;
+  try {
+    project = await detector.detectProject(options.workingDirectory);
+  } catch {
+    return { models: [], project: null };
+  }
+  const namespaceId = project?.namespaceId;
+  if (!namespaceId) return { models: [], project };
+  const discovery = new GitLabModelDiscovery({
+    instanceUrl: config.instanceUrl,
+    getHeaders: config.getHeaders,
+    fetch: config.fetch
+  });
+  const [discovered, configs] = await Promise.all([
+    discovery.discover(`gid://gitlab/Group/${namespaceId}`),
+    configRegistry.getConfigs()
+  ]);
+  const refToID = /* @__PURE__ */ new Map();
+  for (const [id, mapping] of Object.entries(MODEL_MAPPINGS)) {
+    if (mapping.provider === "workflow" && id !== "duo-workflow" && id !== "duo-workflow-default") {
+      refToID.set(mapping.model, id);
+    }
+  }
+  const all = discovered.pinnedModel ? [discovered.pinnedModel] : [
+    ...discovered.selectableModels ?? [],
+    ...discovered.defaultModel ? [discovered.defaultModel] : []
+  ];
+  const seen = /* @__PURE__ */ new Set();
+  const models = [];
+  for (const model of all) {
+    if (!model.ref || seen.has(model.ref)) continue;
+    seen.add(model.ref);
+    const id = refToID.get(model.ref) ?? `duo-workflow-${model.ref.replace(/[/_]/g, "-")}`;
+    const limits = configs.get(model.ref);
+    models.push({
+      id,
+      ref: model.ref,
+      name: model.name,
+      context: limits?.context ?? 2e5,
+      output: limits?.output ?? 64e3,
+      pinned: !!discovered.pinnedModel && discovered.pinnedModel.ref === model.ref
+    });
+  }
+  return { models, project };
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AGENT_PRIVILEGES,
@@ -4527,6 +4582,7 @@ function parseModelsYml(text) {
   WS_KEEPALIVE_PING_INTERVAL_MS,
   WorkflowType,
   createGitLab,
+  discoverWorkflowModels,
   getAnthropicModelForModelId,
   getModelMapping,
   getOpenAIApiType,