gitlab-ai-provider 5.1.2 → 5.2.1

package/dist/index.mjs

@@ -1506,7 +1506,7 @@ var GitLabOpenAILanguageModel = class {
 import WebSocket from "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) => {
@@ -2675,6 +2675,7 @@ import * as fs from "fs";
 import * as path2 from "path";
 import * as os from "os";
 import * as crypto from "crypto";
+var DISCOVERY_TTL_MS = 10 * 60 * 1e3;
 function getCacheFilePath() {
   const cacheHome = process.env.XDG_CACHE_HOME || path2.join(os.homedir(), ".cache");
   return path2.join(cacheHome, "opencode", "gitlab-workflow-model-cache.json");
@@ -2724,13 +2725,23 @@ var GitLabModelCache = class {
     data[this.key] = entry;
     this.writeAll(data);
   }
+  /**
+   * Returns true if the discovery data is missing or older than DISCOVERY_TTL_MS.
+   */
+  isDiscoveryExpired() {
+    const entry = this.load();
+    if (!entry?.discovery || !entry.updatedAt) return true;
+    return Date.now() - new Date(entry.updatedAt).getTime() > DISCOVERY_TTL_MS;
+  }
   /**
    * Update only the discovery portion of the cache, preserving selection.
+   * Optionally persists the associated GitLab project.
    */
-  saveDiscovery(discovery) {
+  saveDiscovery(discovery, project) {
     const existing = this.load();
     this.save({
       discovery,
+      project: project !== void 0 ? project : existing?.project ?? null,
       selectedModelRef: existing?.selectedModelRef ?? null,
       selectedModelName: existing?.selectedModelName ?? null,
       updatedAt: (/* @__PURE__ */ new Date()).toISOString()
@@ -2738,13 +2749,22 @@ var GitLabModelCache = class {
   }
   /**
    * Update only the selected model, preserving the discovery data.
+   * If name is null but ref is provided, attempts to resolve the name
+   * from the cached discovery data.
    */
   saveSelection(ref, name) {
     const existing = this.load();
+    let resolved = name;
+    if (!resolved && ref && existing?.discovery) {
+      const d = existing.discovery;
+      const match = d.pinnedModel?.ref === ref ? d.pinnedModel : d.selectableModels.find((m) => m.ref === ref) ?? (d.defaultModel?.ref === ref ? d.defaultModel : null);
+      resolved = match?.name ?? null;
+    }
     this.save({
       discovery: existing?.discovery ?? null,
+      project: existing?.project ?? null,
       selectedModelRef: ref,
-      selectedModelName: name,
+      selectedModelName: resolved,
       updatedAt: (/* @__PURE__ */ new Date()).toISOString()
     });
   }
@@ -2886,7 +2906,8 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
    */
   set selectedModelRef(ref) {
     this._selectedModelRef = ref ?? void 0;
-    this.modelCache.saveSelection(ref, this._selectedModelName ?? null);
+    this._selectedModelName = void 0;
+    this.modelCache.saveSelection(ref, null);
   }
   /**
    * Get the cached selected model display name.
@@ -4423,6 +4444,67 @@ function parseModelsYml(text) {
   }
   return configs;
 }
+
+// src/gitlab-workflow-discovery.ts
+var configRegistry = new GitLabModelConfigRegistry();
+async function buildModels(discovered, project) {
+  const configs = await 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 };
+}
+async function discoverWorkflowModels(config, options) {
+  const cache = new GitLabModelCache(options.workingDirectory, config.instanceUrl);
+  if (!cache.isDiscoveryExpired()) {
+    const entry = cache.load();
+    return buildModels(entry.discovery, entry.project);
+  }
+  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 = await discovery.discover(`gid://gitlab/Group/${namespaceId}`);
+  cache.saveDiscovery(discovered, project);
+  return buildModels(discovered, project);
+}
 export {
   AGENT_PRIVILEGES,
   BUNDLED_CLIENT_ID,
@@ -4456,6 +4538,7 @@ export {
   WS_KEEPALIVE_PING_INTERVAL_MS,
   WorkflowType,
   createGitLab,
+  discoverWorkflowModels,
   getAnthropicModelForModelId,
   getModelMapping,
   getOpenAIApiType,