@gitlab/gitlab-ai-provider 3.3.1 → 3.4.0

package/dist/index.mjs

@@ -181,11 +181,15 @@ var GitLabAnthropicLanguageModel = class {
   async getAnthropicClient(forceRefresh = false) {
     const tokenData = await this.directAccessClient.getDirectAccessToken(forceRefresh);
     const { "x-api-key": _removed, ...filteredHeaders } = tokenData.headers;
+    const mergedHeaders = {
+      ...filteredHeaders,
+      ...this.config.aiGatewayHeaders
+    };
     this.anthropicClient = new Anthropic({
       apiKey: null,
       authToken: tokenData.token,
       baseURL: this.directAccessClient.getAnthropicProxyUrl(),
-      defaultHeaders: filteredHeaders
+      defaultHeaders: mergedHeaders
     });
     return this.anthropicClient;
   }
@@ -686,10 +690,14 @@ var GitLabOpenAILanguageModel = class {
   async getOpenAIClient(forceRefresh = false) {
     const tokenData = await this.directAccessClient.getDirectAccessToken(forceRefresh);
     const { "x-api-key": _removed, ...filteredHeaders } = tokenData.headers;
+    const mergedHeaders = {
+      ...filteredHeaders,
+      ...this.config.aiGatewayHeaders
+    };
     this.openaiClient = new OpenAI({
       apiKey: tokenData.token,
       baseURL: this.directAccessClient.getOpenAIProxyUrl(),
-      defaultHeaders: filteredHeaders
+      defaultHeaders: mergedHeaders
     });
     return this.openaiClient;
   }
@@ -1502,11 +1510,13 @@ var GitLabOAuthManager = class {
   }
 };
 
+// src/version.ts
+var VERSION = true ? "3.3.1" : "0.0.0-dev";
+
 // src/gitlab-provider.ts
 import * as fs from "fs";
 import * as path from "path";
 import * as os from "os";
-var VERSION = "0.0.1";
 function getOpenCodeAuthPath() {
   const homeDir = os.homedir();
   const xdgDataHome = process.env.XDG_DATA_HOME;
@@ -1656,6 +1666,14 @@ function createGitLab(options = {}) {
       ...options.featureFlags,
       ...agenticOptions?.featureFlags
     };
+    const defaultAiGatewayHeaders = {
+      "User-Agent": `gitlab-ai-provider/${VERSION}`
+    };
+    const aiGatewayHeaders = {
+      ...defaultAiGatewayHeaders,
+      ...options.aiGatewayHeaders,
+      ...agenticOptions?.aiGatewayHeaders
+    };
     const baseConfig = {
       provider: `${providerName}.agentic`,
       instanceUrl,
@@ -1664,7 +1682,8 @@ function createGitLab(options = {}) {
       fetch: options.fetch,
       maxTokens: agenticOptions?.maxTokens,
       featureFlags,
-      aiGatewayUrl: options.aiGatewayUrl
+      aiGatewayUrl: options.aiGatewayUrl,
+      aiGatewayHeaders
     };
     if (mapping.provider === "openai") {
       return new GitLabOpenAILanguageModel(modelId, {
@@ -1984,6 +2003,7 @@ export {
   MODEL_MAPPINGS,
   OAUTH_SCOPES,
   TOKEN_EXPIRY_SKEW_MS,
+  VERSION,
   createGitLab,
   getAnthropicModelForModelId,
   getModelMapping,