@ketd/gemini-cli-sdk 0.3.3 → 0.3.5

package/dist/index.cjs

@@ -77,6 +77,101 @@ var JsonInputMessageType = /* @__PURE__ */ ((JsonInputMessageType2) => {
   return JsonInputMessageType2;
 })(JsonInputMessageType || {});
 
+// src/logger.ts
+var LogLevel = /* @__PURE__ */ ((LogLevel2) => {
+  LogLevel2[LogLevel2["DEBUG"] = 0] = "DEBUG";
+  LogLevel2[LogLevel2["INFO"] = 1] = "INFO";
+  LogLevel2[LogLevel2["WARN"] = 2] = "WARN";
+  LogLevel2[LogLevel2["ERROR"] = 3] = "ERROR";
+  LogLevel2[LogLevel2["SILENT"] = 4] = "SILENT";
+  return LogLevel2;
+})(LogLevel || {});
+var consoleLogger = {
+  debug: (message, ...args) => console.log(message, ...args),
+  info: (message, ...args) => console.log(message, ...args),
+  warn: (message, ...args) => console.warn(message, ...args),
+  error: (message, ...args) => console.error(message, ...args)
+};
+var silentLogger = {
+  debug: () => {
+  },
+  info: () => {
+  },
+  warn: () => {
+  },
+  error: () => {
+  }
+};
+var SDKLogger = class {
+  logger;
+  level;
+  prefix;
+  constructor(options = {}) {
+    this.logger = options.logger || consoleLogger;
+    this.level = options.level ?? 1 /* INFO */;
+    this.prefix = options.prefix || "";
+  }
+  /**
+   * Update logger configuration
+   */
+  configure(options) {
+    if (options.logger !== void 0) {
+      this.logger = options.logger;
+    }
+    if (options.level !== void 0) {
+      this.level = options.level;
+    }
+    if (options.prefix !== void 0) {
+      this.prefix = options.prefix;
+    }
+  }
+  /**
+   * Set log level
+   */
+  setLevel(level) {
+    this.level = level;
+  }
+  /**
+   * Set custom logger implementation
+   */
+  setLogger(logger) {
+    this.logger = logger;
+  }
+  formatMessage(message) {
+    return this.prefix ? `${this.prefix} ${message}` : message;
+  }
+  debug(message, ...args) {
+    if (this.level <= 0 /* DEBUG */) {
+      this.logger.debug(this.formatMessage(message), ...args);
+    }
+  }
+  info(message, ...args) {
+    if (this.level <= 1 /* INFO */) {
+      this.logger.info(this.formatMessage(message), ...args);
+    }
+  }
+  warn(message, ...args) {
+    if (this.level <= 2 /* WARN */) {
+      this.logger.warn(this.formatMessage(message), ...args);
+    }
+  }
+  error(message, ...args) {
+    if (this.level <= 3 /* ERROR */) {
+      this.logger.error(this.formatMessage(message), ...args);
+    }
+  }
+};
+var sdkLogger = new SDKLogger({
+  prefix: "[GeminiSDK]",
+  level: 1 /* INFO */
+});
+function createLogger(component, options = {}) {
+  return new SDKLogger({
+    ...options,
+    prefix: options.prefix || `[${component}]`
+  });
+}
+
 // src/query.ts
 function buildCliArgs(options, prompt) {
   const args = [];
@@ -108,7 +203,7 @@ function buildCliArgs(options, prompt) {
   args.push(prompt);
   return args;
 }
-function buildEnv(options) {
+function buildEnv(options, logger) {
   const env = {
     ...process.env,
     ...options.env
@@ -119,25 +214,19 @@ function buildEnv(options) {
     if (useVertexAI) {
       env.GOOGLE_API_KEY = options.apiKey;
       env.GOOGLE_GENAI_USE_VERTEXAI = "true";
-      if (options.debug) {
-        console.log("[SDK] Vertex AI mode: Setting GOOGLE_API_KEY:", options.apiKey.substring(0, 10) + "...");
-      }
+      logger.debug("Vertex AI mode: Setting GOOGLE_API_KEY:", options.apiKey.substring(0, 10) + "...");
     } else {
       env.GEMINI_API_KEY = options.apiKey;
-      if (options.debug) {
-        console.log("[SDK] Standard mode: Setting GEMINI_API_KEY:", options.apiKey.substring(0, 10) + "...");
-      }
+      logger.debug("Standard mode: Setting GEMINI_API_KEY:", options.apiKey.substring(0, 10) + "...");
     }
   }
   if (!useVertexAI && env.GOOGLE_API_KEY) {
     delete env.GOOGLE_API_KEY;
-    if (options.debug) {
-      console.log("[SDK] Removed GOOGLE_API_KEY from environment (not using Vertex AI)");
-    }
+    logger.debug("Removed GOOGLE_API_KEY from environment (not using Vertex AI)");
   }
   if (options.debug) {
     env.DEBUG = "1";
-    console.log("[SDK] Environment variables set:", {
+    logger.debug("Environment variables set:", {
       GEMINI_API_KEY: env.GEMINI_API_KEY ? "***" : void 0,
       GOOGLE_API_KEY: env.GOOGLE_API_KEY ? "***" : void 0,
       GOOGLE_GENAI_USE_VERTEXAI: env.GOOGLE_GENAI_USE_VERTEXAI,
@@ -147,6 +236,10 @@ function buildEnv(options) {
   return env;
 }
 async function* query(prompt, options) {
+  const logger = createLogger("GeminiSDK", {
+    logger: options.logger,
+    level: options.debug ? 0 /* DEBUG */ : 1 /* INFO */
+  });
   if (!options.pathToGeminiCLI) {
     throw new GeminiSDKError("pathToGeminiCLI is required");
   }
@@ -156,7 +249,7 @@ async function* query(prompt, options) {
     );
   }
   const args = buildCliArgs(options, prompt);
-  const env = buildEnv(options);
+  const env = buildEnv(options, logger);
   const cwd = options.cwd || process.cwd();
   const nodeExecutable = options.pathToNode || "node";
   let geminiProcess;
@@ -172,9 +265,7 @@ async function* query(prompt, options) {
   const stderrChunks = [];
   geminiProcess.stderr?.on("data", (data) => {
     stderrChunks.push(data);
-    if (options.debug) {
-      console.error("[Gemini CLI stderr]:", data.toString());
-    }
+    logger.error("stderr:", data.toString());
   });
   let timeoutId;
   if (options.timeout) {
@@ -194,10 +285,8 @@ async function* query(prompt, options) {
         hasYieldedEvents = true;
         yield event;
       } catch (parseError) {
-        if (options.debug) {
-          console.error("[Gemini SDK] Failed to parse JSON line:", line);
-          console.error("[Gemini SDK] Parse error:", parseError);
-        }
+        logger.debug("Failed to parse JSON line:", line);
+        logger.debug("Parse error:", parseError);
       }
     }
   } catch (error) {
@@ -400,6 +489,10 @@ var GeminiStreamClient = class extends events.EventEmitter {
   constructor(options) {
     super();
     this.options = options;
+    this.logger = createLogger("GeminiStreamClient", {
+      logger: options.logger,
+      level: options.debug ? 0 /* DEBUG */ : 1 /* INFO */
+    });
     if (!options.pathToGeminiCLI) {
       throw new GeminiSDKError("pathToGeminiCLI is required");
     }
@@ -414,6 +507,7 @@ var GeminiStreamClient = class extends events.EventEmitter {
   initEvent = null;
   initTimeout = null;
   tempSettingsPath = null;
+  logger;
   /**
    * Start the Gemini CLI process
    */
@@ -457,9 +551,11 @@ var GeminiStreamClient = class extends events.EventEmitter {
     }
     if (this.process.stderr) {
       this.process.stderr.on("data", (chunk) => {
-        if (this.options.debug) {
-          console.error("[GeminiStreamClient] stderr:", chunk.toString());
+        const message = chunk.toString().trim();
+        if (!message || message.includes("Flushing log events")) {
+          return;
         }
+        this.logger.error("stderr:", message);
       });
     }
     this.emit("started");
@@ -526,6 +622,34 @@ var GeminiStreamClient = class extends events.EventEmitter {
     };
     this.writeMessage(message);
   }
+  /**
+   * Resume session from a session file
+   *
+   * This sends a control message to the CLI to:
+   * 1. Load history from the specified session file
+   * 2. Update the ChatRecordingService to use the session file
+   *
+   * Used when a warm pool adapter is assigned to a real session that has history.
+   *
+   * @param sessionFilePath - Path to the session file to resume from
+   */
+  async resumeSession(sessionFilePath) {
+    if (!this.isReady()) {
+      throw new GeminiSDKError("Client not ready. Call start() first.");
+    }
+    if (!sessionFilePath) {
+      throw new GeminiSDKError("sessionFilePath is required");
+    }
+    const message = {
+      type: "control" /* CONTROL */,
+      control: {
+        subtype: "resume_session",
+        sessionFilePath
+      },
+      session_id: this.options.sessionId
+    };
+    this.writeMessage(message);
+  }
   /**
    * Stop the CLI process
    */
@@ -567,11 +691,9 @@ var GeminiStreamClient = class extends events.EventEmitter {
     if (this.tempSettingsPath) {
       try {
         fs__namespace.unlinkSync(this.tempSettingsPath);
-        if (this.options.debug) {
-          console.log("[GeminiStreamClient] Cleaned up temp settings:", this.tempSettingsPath);
-        }
+        this.logger.debug("Cleaned up temp settings:", this.tempSettingsPath);
       } catch (error) {
-        console.error("[GeminiStreamClient] Failed to clean up temp settings:", error);
+        this.logger.error("Failed to clean up temp settings:", error);
       }
       this.tempSettingsPath = null;
     }
@@ -616,9 +738,7 @@ var GeminiStreamClient = class extends events.EventEmitter {
     }
     if (!fs__namespace.existsSync(geminiConfigDir)) {
       fs__namespace.mkdirSync(geminiConfigDir, { recursive: true });
-      if (this.options.debug) {
-        console.log("[GeminiStreamClient] Created config directory:", geminiConfigDir);
-      }
+      this.logger.debug("Created config directory:", geminiConfigDir);
     }
     this.tempSettingsPath = path2__namespace.join(geminiConfigDir, "settings.json");
     const settings = {};
@@ -633,10 +753,8 @@ var GeminiStreamClient = class extends events.EventEmitter {
     }
     try {
       fs__namespace.writeFileSync(this.tempSettingsPath, JSON.stringify(settings, null, 2), "utf-8");
-      if (this.options.debug) {
-        console.log("[GeminiStreamClient] Wrote settings to:", this.tempSettingsPath);
-        console.log("[GeminiStreamClient] Settings content:", JSON.stringify(settings, null, 2));
-      }
+      this.logger.debug("Wrote settings to:", this.tempSettingsPath);
+      this.logger.debug("Settings content:", JSON.stringify(settings, null, 2));
     } catch (error) {
       throw new GeminiSDKError(`Failed to write settings file: ${error}`);
     }
@@ -654,9 +772,7 @@ var GeminiStreamClient = class extends events.EventEmitter {
     ];
     if (this.options.resumeSessionFilePath) {
       args.push("--resume-from-file", this.options.resumeSessionFilePath);
-      if (this.options.debug) {
-        console.log("[GeminiStreamClient] Resuming from session file:", this.options.resumeSessionFilePath);
-      }
+      this.logger.debug("Resuming from session file:", this.options.resumeSessionFilePath);
     }
     if (this.options.model) {
       args.push("--model", this.options.model);
@@ -679,28 +795,20 @@ var GeminiStreamClient = class extends events.EventEmitter {
     };
     if (this.options.apiKey) {
       const useVertexAI = this.options.env?.GOOGLE_GENAI_USE_VERTEXAI === "true";
-      if (this.options.debug) {
-        console.log("[GeminiStreamClient] buildEnv() - API Key prefix:", this.options.apiKey.substring(0, 3));
-        console.log("[GeminiStreamClient] buildEnv() - GOOGLE_GENAI_USE_VERTEXAI:", this.options.env?.GOOGLE_GENAI_USE_VERTEXAI);
-        console.log("[GeminiStreamClient] buildEnv() - useVertexAI:", useVertexAI);
-      }
+      this.logger.debug("buildEnv() - API Key prefix:", this.options.apiKey.substring(0, 3));
+      this.logger.debug("buildEnv() - GOOGLE_GENAI_USE_VERTEXAI:", this.options.env?.GOOGLE_GENAI_USE_VERTEXAI);
+      this.logger.debug("buildEnv() - useVertexAI:", useVertexAI);
       if (useVertexAI) {
         env.GOOGLE_API_KEY = this.options.apiKey;
-        if (this.options.debug) {
-          console.log("[GeminiStreamClient] buildEnv() - Setting GOOGLE_API_KEY for Vertex AI");
-        }
+        this.logger.debug("buildEnv() - Setting GOOGLE_API_KEY for Vertex AI");
       } else {
         env.GEMINI_API_KEY = this.options.apiKey;
-        if (this.options.debug) {
-          console.log("[GeminiStreamClient] buildEnv() - Setting GEMINI_API_KEY for AI Studio");
-        }
+        this.logger.debug("buildEnv() - Setting GEMINI_API_KEY for AI Studio");
       }
     }
-    if (this.options.debug) {
-      console.log("[GeminiStreamClient] buildEnv() - Final env has GOOGLE_API_KEY:", !!env.GOOGLE_API_KEY);
-      console.log("[GeminiStreamClient] buildEnv() - Final env has GEMINI_API_KEY:", !!env.GEMINI_API_KEY);
-      console.log("[GeminiStreamClient] buildEnv() - Final env GOOGLE_GENAI_USE_VERTEXAI:", env.GOOGLE_GENAI_USE_VERTEXAI);
-    }
+    this.logger.debug("buildEnv() - Final env has GOOGLE_API_KEY:", !!env.GOOGLE_API_KEY);
+    this.logger.debug("buildEnv() - Final env has GEMINI_API_KEY:", !!env.GEMINI_API_KEY);
+    this.logger.debug("buildEnv() - Final env GOOGLE_GENAI_USE_VERTEXAI:", env.GOOGLE_GENAI_USE_VERTEXAI);
     return env;
   }
   /**
@@ -711,19 +819,15 @@ var GeminiStreamClient = class extends events.EventEmitter {
       throw new GeminiSDKError("stdin stream not available");
     }
     const json = JSON.stringify(message);
-    if (this.options.debug) {
-      console.log("[GeminiStreamClient] Writing message to stdin:", json.substring(0, 100));
-    }
+    this.logger.debug("Writing message to stdin:", json.substring(0, 100));
     const success = this.stdinStream.write(json + "\n", (error) => {
       if (error) {
-        console.error("[GeminiStreamClient] Write error:", error);
-      } else if (this.options.debug) {
-        console.log("[GeminiStreamClient] Write callback: message flushed to stdin");
+        this.logger.error("Write error:", error);
+      } else {
+        this.logger.debug("Write callback: message flushed to stdin");
       }
     });
-    if (this.options.debug) {
-      console.log("[GeminiStreamClient] Write success:", success, "Stream writable:", this.stdinStream.writable);
-    }
+    this.logger.debug("Write success:", success, "Stream writable:", this.stdinStream.writable);
   }
   /**
    * Start reading JSONL events from stdout
@@ -738,23 +842,17 @@ var GeminiStreamClient = class extends events.EventEmitter {
         return;
       }
       if (trimmed.startsWith("[")) {
-        if (this.options.debug) {
-          console.log("[GeminiStreamClient] Skipping debug output:", trimmed.substring(0, 100));
-        }
         return;
       }
       try {
         const event = JSON.parse(trimmed);
         this.handleEvent(event);
       } catch (error) {
-        console.error("[GeminiStreamClient] Failed to parse JSON:", trimmed);
-        console.error("[GeminiStreamClient] Error:", error);
+        this.logger.error("Failed to parse JSON:", trimmed);
+        this.logger.error("Error:", error);
       }
     });
     this.readlineInterface.on("close", () => {
-      if (this.options.debug) {
-        console.log("[GeminiStreamClient] readline interface closed");
-      }
     });
   }
   /**
@@ -796,9 +894,7 @@ var GeminiStreamClient = class extends events.EventEmitter {
    * Handle process exit
    */
   handleProcessExit(code, signal) {
-    if (this.options.debug) {
-      console.log("[GeminiStreamClient] Process exited:", { code, signal });
-    }
+    this.logger.debug("Process exited:", { code, signal });
     if (code !== 0 && code !== null) {
       this.status = "error" /* ERROR */;
       this.emit("error", new GeminiSDKError(`Process exited with code ${code}`, code));
@@ -817,7 +913,7 @@ var GeminiStreamClient = class extends events.EventEmitter {
    * Handle process error
    */
   handleProcessError(error) {
-    console.error("[GeminiStreamClient] Process error:", error);
+    this.logger.error("Process error:", error);
     this.status = "error" /* ERROR */;
     this.emit("error", error);
   }
@@ -876,7 +972,7 @@ function validateModel(model) {
     return defaultModel;
   }
   if (!model.startsWith("gemini-")) {
-    console.warn(`Warning: Model name "${model}" does not start with "gemini-"`);
+    sdkLogger.warn(`Warning: Model name "${model}" does not start with "gemini-"`);
   }
   return model;
 }
@@ -901,12 +997,17 @@ exports.GeminiSDKError = GeminiSDKError;
 exports.GeminiStreamClient = GeminiStreamClient;
 exports.JsonInputMessageType = JsonInputMessageType;
 exports.JsonStreamEventType = JsonStreamEventType;
+exports.LogLevel = LogLevel;
 exports.ProcessStatus = ProcessStatus;
+exports.SDKLogger = SDKLogger;
+exports.createLogger = createLogger;
 exports.findGeminiCLI = findGeminiCLI;
 exports.formatDuration = formatDuration;
 exports.formatTokens = formatTokens;
 exports.getApiKey = getApiKey;
 exports.query = query;
+exports.sdkLogger = sdkLogger;
+exports.silentLogger = silentLogger;
 exports.validateApiKey = validateApiKey;
 exports.validateModel = validateModel;
 //# sourceMappingURL=index.cjs.map