gitlab-ai-provider 5.2.0 → 5.2.1

package/dist/index.mjs

@@ -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.
@@ -4426,29 +4447,8 @@ function parseModelsYml(text) {
 
 // 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()
-  ]);
+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") {
@@ -4477,6 +4477,34 @@ async function discoverWorkflowModels(config, options) {
   }
   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,