gitlab-ai-provider 5.0.0 → 5.1.0

package/dist/index.js

@@ -41,6 +41,7 @@ __export(index_exports, {
   GitLabDirectAccessClient: () => GitLabDirectAccessClient,
   GitLabError: () => GitLabError,
   GitLabModelCache: () => GitLabModelCache,
+  GitLabModelConfigRegistry: () => GitLabModelConfigRegistry,
   GitLabModelDiscovery: () => GitLabModelDiscovery,
   GitLabOAuthManager: () => GitLabOAuthManager,
   GitLabOpenAILanguageModel: () => GitLabOpenAILanguageModel,
@@ -69,7 +70,8 @@ __export(index_exports, {
   getWorkflowModelRef: () => getWorkflowModelRef,
   gitlab: () => gitlab,
   isResponsesApiModel: () => isResponsesApiModel,
-  isWorkflowModel: () => isWorkflowModel
+  isWorkflowModel: () => isWorkflowModel,
+  parseModelsYml: () => parseModelsYml
 });
 module.exports = __toCommonJS(index_exports);
 
@@ -357,7 +359,8 @@ var GitLabAnthropicLanguageModel = class {
     const messages = [];
     for (const message of prompt) {
       if (message.role === "system") {
-        systemMessage = message.content;
+        systemMessage = systemMessage ? `${systemMessage}
+${message.content}` : message.content;
         continue;
       }
       if (message.role === "user") {
@@ -1573,7 +1576,7 @@ var GitLabOpenAILanguageModel = class {
 var import_isomorphic_ws = __toESM(require("isomorphic-ws"));
 
 // src/version.ts
-var VERSION = true ? "3.6.0" : "0.0.0-dev";
+var VERSION = true ? "5.0.0" : "0.0.0-dev";
 
 // src/gitlab-workflow-types.ts
 var WorkflowType = /* @__PURE__ */ ((WorkflowType2) => {
@@ -4242,6 +4245,170 @@ function createGitLab(options = {}) {
   return provider;
 }
 var gitlab = createGitLab();
+
+// src/gitlab-model-config.ts
+var fs3 = __toESM(require("fs"));
+var path4 = __toESM(require("path"));
+var os3 = __toESM(require("os"));
+var DEFAULT_MODELS_YML_URL = "https://gitlab.com/gitlab-org/modelops/applied-ml/code-suggestions/ai-assist/-/raw/main/ai_gateway/model_selection/models.yml";
+var DEFAULT_TTL_MS = 24 * 60 * 60 * 1e3;
+var DEFAULT_CONTEXT = 2e5;
+var DEFAULT_OUTPUT = 64e3;
+function getCacheFilePath2() {
+  const cacheHome = process.env.XDG_CACHE_HOME || path4.join(os3.homedir(), ".cache");
+  return path4.join(cacheHome, "opencode", "gitlab-model-configs.json");
+}
+function readCacheFile() {
+  try {
+    const filePath = getCacheFilePath2();
+    if (!fs3.existsSync(filePath)) return null;
+    const raw = fs3.readFileSync(filePath, "utf-8");
+    return JSON.parse(raw);
+  } catch {
+    return null;
+  }
+}
+function writeCacheFile(configs) {
+  try {
+    const filePath = getCacheFilePath2();
+    const dir = path4.dirname(filePath);
+    fs3.mkdirSync(dir, { recursive: true, mode: 448 });
+    const data = {
+      configs: Object.fromEntries(configs),
+      updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+    };
+    fs3.writeFileSync(filePath, JSON.stringify(data, null, 2), { mode: 384 });
+  } catch {
+  }
+}
+function loadCacheFile(ttlMs) {
+  const cached = readCacheFile();
+  if (!cached) return null;
+  const writtenAt = new Date(cached.updatedAt).getTime();
+  const age = Date.now() - writtenAt;
+  if (age > ttlMs) return null;
+  return { configs: new Map(Object.entries(cached.configs)), writtenAt };
+}
+var GitLabModelConfigRegistry = class {
+  url;
+  ttlMs;
+  fetchFn;
+  memCache = null;
+  memExpiresAt = 0;
+  pending = null;
+  constructor(options) {
+    this.url = options?.url ?? DEFAULT_MODELS_YML_URL;
+    this.ttlMs = options?.ttlMs ?? DEFAULT_TTL_MS;
+    this.fetchFn = options?.fetch ?? fetch;
+  }
+  /**
+   * Get model configs, fetching and caching as needed.
+   * Returns a Map keyed by `gitlab_identifier` (the discovery `ref`).
+   */
+  async getConfigs() {
+    if (this.memCache && Date.now() < this.memExpiresAt) {
+      return this.memCache;
+    }
+    const fileCached = loadCacheFile(this.ttlMs);
+    if (fileCached) {
+      this.memCache = fileCached.configs;
+      this.memExpiresAt = fileCached.writtenAt + this.ttlMs;
+      return fileCached.configs;
+    }
+    if (this.pending) return this.pending;
+    this.pending = this.fetchConfigs();
+    try {
+      return await this.pending;
+    } finally {
+      this.pending = null;
+    }
+  }
+  /**
+   * Look up config for a single model ref.
+   * Returns defaults if the ref is not found or fetch fails.
+   */
+  async getConfig(ref) {
+    const configs = await this.getConfigs();
+    return configs.get(ref) ?? { context: DEFAULT_CONTEXT, output: DEFAULT_OUTPUT };
+  }
+  /** Invalidate both in-memory and file caches. */
+  invalidateCache() {
+    this.memCache = null;
+    this.memExpiresAt = 0;
+    try {
+      const filePath = getCacheFilePath2();
+      if (fs3.existsSync(filePath)) {
+        fs3.unlinkSync(filePath);
+      }
+    } catch {
+    }
+  }
+  async fetchConfigs() {
+    try {
+      const res = await this.fetchFn(this.url);
+      if (!res.ok) throw new Error(`HTTP ${res.status}`);
+      const text = await res.text();
+      const configs = parseModelsYml(text);
+      this.memCache = configs;
+      this.memExpiresAt = Date.now() + this.ttlMs;
+      writeCacheFile(configs);
+      return configs;
+    } catch {
+      return this.memCache ?? loadCacheFile(Infinity)?.configs ?? /* @__PURE__ */ new Map();
+    }
+  }
+};
+function parseModelsYml(text) {
+  const configs = /* @__PURE__ */ new Map();
+  let currentIdentifier = null;
+  let currentContext = 0;
+  let currentOutput = 0;
+  let inParams = false;
+  let paramsIndent = -1;
+  for (const line of text.split("\n")) {
+    const idMatch = line.match(/^\s*gitlab_identifier:\s*"?([^"#\s]+)"?/);
+    if (idMatch) {
+      if (currentIdentifier) {
+        configs.set(currentIdentifier, {
+          context: currentContext || DEFAULT_CONTEXT,
+          output: currentOutput || DEFAULT_OUTPUT
+        });
+      }
+      currentIdentifier = idMatch[1];
+      currentContext = 0;
+      currentOutput = 0;
+      inParams = false;
+      paramsIndent = -1;
+    }
+    const ctxMatch = line.match(/^\s*max_context_tokens:\s*([0-9_]+)/);
+    if (ctxMatch && currentIdentifier) {
+      currentContext = parseInt(ctxMatch[1].replace(/_/g, ""), 10);
+    }
+    if (/^\s*params:\s*$/.test(line)) {
+      inParams = true;
+      paramsIndent = line.match(/^(\s*)/)?.[1].length ?? 0;
+    } else if (inParams && /^\s*\S+:/.test(line)) {
+      const indent = line.match(/^(\s*)/)?.[1].length ?? 0;
+      if (indent <= paramsIndent) {
+        inParams = false;
+        paramsIndent = -1;
+      }
+    }
+    if (inParams) {
+      const maxTokensMatch = line.match(/^\s*max_tokens:\s*([0-9_]+)/);
+      if (maxTokensMatch && currentIdentifier) {
+        currentOutput = parseInt(maxTokensMatch[1].replace(/_/g, ""), 10);
+      }
+    }
+  }
+  if (currentIdentifier) {
+    configs.set(currentIdentifier, {
+      context: currentContext || DEFAULT_CONTEXT,
+      output: currentOutput || DEFAULT_OUTPUT
+    });
+  }
+  return configs;
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AGENT_PRIVILEGES,
@@ -4256,6 +4423,7 @@ var gitlab = createGitLab();
   GitLabDirectAccessClient,
   GitLabError,
   GitLabModelCache,
+  GitLabModelConfigRegistry,
   GitLabModelDiscovery,
   GitLabOAuthManager,
   GitLabOpenAILanguageModel,
@@ -4284,6 +4452,7 @@ var gitlab = createGitLab();
   getWorkflowModelRef,
   gitlab,
   isResponsesApiModel,
-  isWorkflowModel
+  isWorkflowModel,
+  parseModelsYml
 });
 //# sourceMappingURL=index.js.map