@corbat-tech/coco 2.27.0 → 2.27.2

package/dist/index.js

@@ -1408,7 +1408,7 @@ function getApiKey(provider) {
     case "gemini":
       return process.env["GEMINI_API_KEY"] ?? process.env["GOOGLE_API_KEY"];
     case "vertex":
-      return void 0;
+      return process.env["VERTEX_API_KEY"] ?? process.env["GOOGLE_API_KEY"];
     case "kimi":
       return process.env["KIMI_API_KEY"] ?? process.env["MOONSHOT_API_KEY"];
     case "kimi-code":
@@ -16437,21 +16437,26 @@ var VertexProvider = class {
   config = {};
   project = "";
   location = DEFAULT_LOCATION;
+  apiKey;
   retryConfig = DEFAULT_RETRY_CONFIG;
   async initialize(config) {
     this.config = config;
     this.project = config.project ?? process.env["VERTEX_PROJECT"] ?? process.env["GOOGLE_CLOUD_PROJECT"] ?? process.env["GCLOUD_PROJECT"] ?? "";
     this.location = config.location ?? process.env["VERTEX_LOCATION"] ?? process.env["GOOGLE_CLOUD_LOCATION"] ?? DEFAULT_LOCATION;
+    this.apiKey = config.apiKey ?? process.env["VERTEX_API_KEY"] ?? process.env["GOOGLE_API_KEY"];
     if (!this.project.trim()) {
       throw new ProviderError(
         "Vertex AI project not configured. Set provider.project, VERTEX_PROJECT, or GOOGLE_CLOUD_PROJECT.",
         { provider: this.id }
       );
     }
+    if (this.apiKey?.trim()) {
+      return;
+    }
     const token = await getCachedADCToken();
     if (!token) {
       throw new ProviderError(
-        "Vertex AI ADC is not configured. Run `gcloud auth application-default login` manually, then retry.",
+        "Vertex AI authentication is not configured. Set VERTEX_API_KEY (or GOOGLE_API_KEY), or run `gcloud auth application-default login`.",
         { provider: this.id }
       );
     }
@@ -16570,10 +16575,17 @@ var VertexProvider = class {
     return `${this.getResolvedBaseUrl()}/projects/${encodeURIComponent(this.project)}/locations/${encodeURIComponent(this.location)}/publishers/google/models/${encodeURIComponent(this.getModel(model))}:${action}`;
   }
   async getHeaders() {
+    if (this.apiKey?.trim()) {
+      return {
+        "Content-Type": "application/json",
+        "x-goog-api-key": this.apiKey,
+        "x-goog-user-project": this.project
+      };
+    }
     const token = await getCachedADCToken();
     if (!token) {
       throw new ProviderError(
-        "Vertex AI ADC token is unavailable. Re-authenticate with gcloud and retry.",
+        "Vertex AI token is unavailable. Re-authenticate with gcloud or configure VERTEX_API_KEY.",
         { provider: this.id }
       );
     }
@@ -28441,9 +28453,10 @@ var SSETransport = class {
 // src/mcp/lifecycle.ts
 init_errors2();
 init_logger();
-var MCPServerManager = class {
+var MCPServerManager = class _MCPServerManager {
   connections = /* @__PURE__ */ new Map();
   logger = getLogger();
+  static STOP_TIMEOUT_MS = 5e3;
   /**
    * Create transport for a server config
    */
@@ -28530,7 +28543,15 @@ var MCPServerManager = class {
     }
     this.logger.info(`Stopping MCP server: ${name}`);
     try {
-      await connection.transport.disconnect();
+      await Promise.race([
+        connection.transport.disconnect(),
+        new Promise(
+          (_, reject) => setTimeout(
+            () => reject(new Error("MCP disconnect timeout")),
+            _MCPServerManager.STOP_TIMEOUT_MS
+          )
+        )
+      ]);
     } catch (error) {
       this.logger.error(
         `Error disconnecting server '${name}': ${error instanceof Error ? error.message : String(error)}`