@google/gemini-cli-a2a-server 0.33.0-preview.3 → 0.33.0-preview.4

package/dist/a2a-server.mjs

@@ -214180,8 +214180,8 @@ var GIT_COMMIT_INFO, CLI_VERSION;
 var init_git_commit = __esm({
   "packages/core/dist/src/generated/git-commit.js"() {
     "use strict";
-    GIT_COMMIT_INFO = "03a8fc311";
-    CLI_VERSION = "0.33.0-preview.3";
+    GIT_COMMIT_INFO = "7e6e40c81";
+    CLI_VERSION = "0.33.0-preview.4";
   }
 });
 
@@ -291439,7 +291439,7 @@ function getVersion() {
   }
   versionPromise = (async () => {
     const pkgJson = await getPackageJson(__dirname3);
-    return "0.33.0-preview.3";
+    return "0.33.0-preview.4";
   })();
   return versionPromise;
 }
@@ -292834,14 +292834,16 @@ var init_recordingContentGenerator = __esm({
 });
 
 // packages/core/dist/src/core/contentGenerator.js
-async function createContentGeneratorConfig(config3, authType, apiKey) {
+async function createContentGeneratorConfig(config3, authType, apiKey, baseUrl, customHeaders) {
   const geminiApiKey = apiKey || process.env["GEMINI_API_KEY"] || await loadApiKey() || void 0;
   const googleApiKey = process.env["GOOGLE_API_KEY"] || void 0;
   const googleCloudProject = process.env["GOOGLE_CLOUD_PROJECT"] || process.env["GOOGLE_CLOUD_PROJECT_ID"] || void 0;
   const googleCloudLocation = process.env["GOOGLE_CLOUD_LOCATION"] || void 0;
   const contentGeneratorConfig = {
     authType,
-    proxy: config3?.getProxy()
+    proxy: config3?.getProxy(),
+    baseUrl,
+    customHeaders
   };
   if (authType === AuthType2.LOGIN_WITH_GOOGLE || authType === AuthType2.COMPUTE_ADC) {
     return contentGeneratorConfig;
@@ -292882,8 +292884,11 @@ async function createContentGenerator(config3, gcConfig, sessionId2) {
       const httpOptions = { headers: baseHeaders };
       return new LoggingContentGenerator(await createCodeAssistContentGenerator(httpOptions, config3.authType, gcConfig, sessionId2), gcConfig);
     }
-    if (config3.authType === AuthType2.USE_GEMINI || config3.authType === AuthType2.USE_VERTEX_AI) {
+    if (config3.authType === AuthType2.USE_GEMINI || config3.authType === AuthType2.USE_VERTEX_AI || config3.authType === AuthType2.GATEWAY) {
       let headers = { ...baseHeaders };
+      if (config3.customHeaders) {
+        headers = { ...headers, ...config3.customHeaders };
+      }
       if (gcConfig?.getUsageStatisticsEnabled()) {
         const installationManager2 = new InstallationManager();
         const installationId = installationManager2.getInstallationId();
@@ -292893,6 +292898,9 @@ async function createContentGenerator(config3, gcConfig, sessionId2) {
         };
       }
       const httpOptions = { headers };
+      if (config3.baseUrl) {
+        httpOptions.baseUrl = config3.baseUrl;
+      }
       const googleGenAI = new GoogleGenAI({
         apiKey: config3.apiKey === "" ? void 0 : config3.apiKey,
         vertexai: config3.vertexai,
@@ -292927,6 +292935,7 @@ var init_contentGenerator = __esm({
       AuthType3["USE_VERTEX_AI"] = "vertex-ai";
       AuthType3["LEGACY_CLOUD_SHELL"] = "cloud-shell";
       AuthType3["COMPUTE_ADC"] = "compute-default-credentials";
+      AuthType3["GATEWAY"] = "gateway";
     })(AuthType2 || (AuthType2 = {}));
   }
 });
@@ -419377,7 +419386,7 @@ var init_config4 = __esm({
       getContentGenerator() {
         return this.contentGenerator;
       }
-      async refreshAuth(authMethod, apiKey) {
+      async refreshAuth(authMethod, apiKey, baseUrl, customHeaders) {
         this.modelAvailabilityService.reset();
         if (this.contentGeneratorConfig?.authType === AuthType2.USE_GEMINI && authMethod !== AuthType2.USE_GEMINI) {
           this.geminiClient.stripThoughtsFromHistory();
@@ -419386,7 +419395,7 @@ var init_config4 = __esm({
         if (this.contentGeneratorConfig) {
           this.contentGeneratorConfig.authType = void 0;
         }
-        const newContentGeneratorConfig = await createContentGeneratorConfig(this, authMethod, apiKey);
+        const newContentGeneratorConfig = await createContentGeneratorConfig(this, authMethod, apiKey, baseUrl, customHeaders);
         this.contentGenerator = await createContentGenerator(newContentGeneratorConfig, this, this.getSessionId());
         this.contentGeneratorConfig = newContentGeneratorConfig;
         this.baseLlmClient = new BaseLlmClient(this.contentGenerator, this);