@kalenkevich/agent_007 0.0.1

package/dist/logger.js

@@ -0,0 +1,98 @@
+import * as fs from "fs";
+import * as path from "path";
+import * as util from "util";
+export class ConsoleTransport {
+    constructor(enabled) {
+        this.enabled = enabled;
+    }
+    log(level, formattedMessage, args) {
+        if (!this.enabled) {
+            return;
+        }
+        switch (level) {
+            case "DEBUG":
+                console.debug(...args);
+                break;
+            case "INFO":
+                console.info(...args);
+                break;
+            case "WARN":
+                console.warn(...args);
+                break;
+            case "ERROR":
+                console.error(...args);
+                break;
+            case "LOG":
+            default:
+                console.log(...args);
+        }
+    }
+}
+export class FileTransport {
+    constructor(enabled, logFile) {
+        this.logFile = logFile;
+        this.enabled = true;
+        this.enabled = enabled;
+    }
+    log(level, formattedMessage, args) {
+        if (!this.enabled) {
+            return;
+        }
+        try {
+            fs.appendFileSync(this.logFile, `[${level}] ${formattedMessage}\n`);
+        }
+        catch (err) {
+            console.error("Failed to write to log file:", err);
+        }
+    }
+}
+export class Logger {
+    constructor(transports) {
+        this.transports = [];
+        if (transports) {
+            this.transports = transports;
+        }
+    }
+    getTransports() {
+        if (this.transports.length) {
+            return this.transports;
+        }
+        if (process.env.DEBUG_LOGGER) {
+            const logFile = path.join(process.cwd(), "debug.log");
+            return [
+                new ConsoleTransport(process.env.DEBUG_LOGGER_CONSOLE === "true"),
+                new FileTransport(process.env.DEBUG_LOGGER === "true", logFile),
+            ];
+        }
+        return [];
+    }
+    formatMessage(level, ...args) {
+        const timestamp = new Date().toISOString();
+        const message = args
+            .map((arg) => typeof arg === "string" ? arg : util.inspect(arg, { depth: null }))
+            .join(" ");
+        return `[${timestamp}] [${level}] ${message}`;
+    }
+    log(...args) {
+        const formatted = this.formatMessage("LOG", ...args);
+        this.getTransports().forEach((t) => t.log("LOG", formatted, args));
+    }
+    debug(...args) {
+        const formatted = this.formatMessage("DEBUG", ...args);
+        this.getTransports().forEach((t) => t.log("DEBUG", formatted, args));
+    }
+    info(...args) {
+        const formatted = this.formatMessage("INFO", ...args);
+        this.getTransports().forEach((t) => t.log("INFO", formatted, args));
+    }
+    warn(...args) {
+        const formatted = this.formatMessage("WARN", ...args);
+        this.getTransports().forEach((t) => t.log("WARN", formatted, args));
+    }
+    error(...args) {
+        const formatted = this.formatMessage("ERROR", ...args);
+        this.getTransports().forEach((t) => t.log("ERROR", formatted, args));
+    }
+}
+export const logger = new Logger();
+//# sourceMappingURL=logger.js.map

package/dist/logger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.js","sourceRoot":"","sources":["../src/logger.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAO7B,MAAM,OAAO,gBAAgB;IAG3B,YAAY,OAAgB;QAC1B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,GAAG,CAAC,KAAa,EAAE,gBAAwB,EAAE,IAAW;QACtD,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,OAAO;QACT,CAAC;QAED,QAAQ,KAAK,EAAE,CAAC;YACd,KAAK,OAAO;gBACV,OAAO,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC;gBACvB,MAAM;YACR,KAAK,MAAM;gBACT,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;gBACtB,MAAM;YACR,KAAK,MAAM;gBACT,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;gBACtB,MAAM;YACR,KAAK,OAAO;gBACV,OAAO,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC;gBACvB,MAAM;YACR,KAAK,KAAK,CAAC;YACX;gBACE,OAAO,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;CACF;AAED,MAAM,OAAO,aAAa;IAGxB,YACE,OAAgB,EACR,OAAe;QAAf,YAAO,GAAP,OAAO,CAAQ;QAJzB,YAAO,GAAY,IAAI,CAAC;QAMtB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,GAAG,CAAC,KAAa,EAAE,gBAAwB,EAAE,IAAW;QACtD,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,EAAE,CAAC,cAAc,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,KAAK,gBAAgB,IAAI,CAAC,CAAC;QACtE,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,8BAA8B,EAAE,GAAG,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;CACF;AAED,MAAM,OAAO,MAAM;IAGjB,YAAY,UAA2B;QAF/B,eAAU,GAAmB,EAAE,CAAC;QAGtC,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC/B,CAAC;IACH,CAAC;IAGD,aAAa;QACX,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;YAC3B,OAAO,IAAI,CAAC,UAAU,CAAC;QACzB,CAAC;QAED,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,CAAC;YAC7B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,WAAW,CAAC,CAAC;YAEtD,OAAO;gBACL,IAAI,gBAAgB,CAAC,OAAO,CAAC,GAAG,CAAC,oBAAoB,KAAK,MAAM,CAAC;gBACjE,IAAI,aAAa,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,KAAK,MAAM,EAAE,OAAO,CAAC;aAChE,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAEO,aAAa,CAAC,KAAa,EAAE,GAAG,IAAW;QACjD,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAC3C,MAAM,OAAO,GAAG,IAAI;aACjB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CACX,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CACnE;aACA,IAAI,CAAC,GAAG,CAAC,CAAC;QACb,OAAO,IAAI,SAAS,MAAM,KAAK,KAAK,OAAO,EAAE,CAAC;IAChD,CAAC;IAED,GAAG,CAAC,GAAG,IAAW;QAChB,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,GAAG,IAAI,CAAC,CAAC;QACrD,IAAI,CAAC,aAAa,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC;IACrE,CAAC;IAED,KAAK,CAAC,GAAG,IAAW;QAClB,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;QACvD,IAAI,CAAC,aAAa,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC;IACvE,CAAC;IAED,IAAI,CAAC,GAAG,IAAW;QACjB,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,CAAC;QACtD,IAAI,CAAC,aAAa,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC;IACtE,CAAC;IAED,IAAI,CAAC,GAAG,IAAW;QACjB,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,CAAC;QACtD,IAAI,CAAC,aAAa,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC;IACtE,CAAC;IAED,KAAK,CAAC,GAAG,IAAW;QAClB,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;QACvD,IAAI,CAAC,aAAa,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC;IACvE,CAAC;CACF;AAED,MAAM,CAAC,MAAM,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC"}

package/dist/model/adaptive_model.d.ts

@@ -0,0 +1,16 @@
+import type { LlmModel, LlmModelConfig } from "./model.js";
+import type { LlmRequest } from "./request.js";
+import type { LlmResponse } from "./response.js";
+import type { ModelConfig } from "../config/config.js";
+export declare class AdaptiveLlmModel implements LlmModel {
+    private currentModel;
+    private fallbackConfigs?;
+    readonly modelName: string;
+    constructor(config: {
+        main: ModelConfig;
+        fallback?: ModelConfig[];
+    });
+    setModel(modelConfig: ModelConfig): void;
+    run(request: LlmRequest, config?: LlmModelConfig): AsyncGenerator<LlmResponse, void, unknown>;
+    countTokens(request: LlmRequest): Promise<number>;
+}

package/dist/model/adaptive_model.js

@@ -0,0 +1,57 @@
+import { resolveLlmModel } from "./registry.js";
+import { logger } from "../logger.js";
+export class AdaptiveLlmModel {
+    constructor(config) {
+        this.modelName = "adaptive";
+        this.setModel(config.main);
+        this.fallbackConfigs = config.fallback;
+    }
+    setModel(modelConfig) {
+        const ModelClass = resolveLlmModel(modelConfig.modelName);
+        this.currentModel = new ModelClass(modelConfig);
+    }
+    async *run(request, config) {
+        if (!this.currentModel) {
+            throw new Error("Model not initialized");
+        }
+        try {
+            yield* this.currentModel.run(request, config);
+        }
+        catch (e) {
+            logger.error("[AdaptiveLlmModel] error:", e);
+            if (this.fallbackConfigs && this.fallbackConfigs.length > 0) {
+                for (const fallbackConfig of this.fallbackConfigs) {
+                    logger.info(`[AdaptiveLlmModel] Attempting fallback to ${fallbackConfig.modelName}`);
+                    try {
+                        const FallbackModelClass = resolveLlmModel(fallbackConfig.modelName);
+                        this.currentModel = new FallbackModelClass(fallbackConfig);
+                        yield* this.currentModel.run(request, config);
+                        return; // Success, exit the generator
+                    }
+                    catch (fallbackError) {
+                        logger.error(`[AdaptiveLlmModel] fallback to ${fallbackConfig.modelName} failed:`, fallbackError);
+                        // Continue loop to try next fallback
+                    }
+                }
+                // If all fallbacks failed
+                yield {
+                    errorCode: "ADAPTIVE_MODEL_ALL_FALLBACKS_FAILED",
+                    errorMessage: "All fallback models failed.",
+                };
+            }
+            else {
+                yield {
+                    errorCode: "ADAPTIVE_MODEL_ERROR",
+                    errorMessage: e instanceof Error ? e.message : String(e),
+                };
+            }
+        }
+    }
+    async countTokens(request) {
+        if (!this.currentModel) {
+            throw new Error("Model not initialized");
+        }
+        return this.currentModel.countTokens(request);
+    }
+}
+//# sourceMappingURL=adaptive_model.js.map

package/dist/model/adaptive_model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"adaptive_model.js","sourceRoot":"","sources":["../../src/model/adaptive_model.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAChD,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAEtC,MAAM,OAAO,gBAAgB;IAK3B,YAAY,MAAuD;QAF1D,cAAS,GAAW,UAAU,CAAC;QAGtC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC3B,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,QAAQ,CAAC;IACzC,CAAC;IAED,QAAQ,CAAC,WAAwB;QAC/B,MAAM,UAAU,GAAG,eAAe,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;QAC1D,IAAI,CAAC,YAAY,GAAG,IAAI,UAAU,CAAC,WAAW,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,CAAC,GAAG,CACR,OAAmB,EACnB,MAAuB;QAEvB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QAED,IAAI,CAAC;YACH,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAChD,CAAC;QAAC,OAAO,CAAU,EAAE,CAAC;YACpB,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE,CAAC,CAAC,CAAC;YAE7C,IAAI,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC5D,KAAK,MAAM,cAAc,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;oBAClD,MAAM,CAAC,IAAI,CACT,6CAA6C,cAAc,CAAC,SAAS,EAAE,CACxE,CAAC;oBACF,IAAI,CAAC;wBACH,MAAM,kBAAkB,GAAG,eAAe,CACxC,cAAc,CAAC,SAAS,CACzB,CAAC;wBAEF,IAAI,CAAC,YAAY,GAAG,IAAI,kBAAkB,CAAC,cAAc,CAAC,CAAC;wBAC3D,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;wBAC9C,OAAO,CAAC,8BAA8B;oBACxC,CAAC;oBAAC,OAAO,aAAsB,EAAE,CAAC;wBAChC,MAAM,CAAC,KAAK,CACV,kCAAkC,cAAc,CAAC,SAAS,UAAU,EACpE,aAAa,CACd,CAAC;wBACF,qCAAqC;oBACvC,CAAC;gBACH,CAAC;gBACD,0BAA0B;gBAC1B,MAAM;oBACJ,SAAS,EAAE,qCAAqC;oBAChD,YAAY,EAAE,6BAA6B;iBAC5C,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,MAAM;oBACJ,SAAS,EAAE,sBAAsB;oBACjC,YAAY,EAAE,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;iBACzD,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAAmB;QACnC,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QACD,OAAO,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAChD,CAAC;CACF"}

package/dist/model/google/gemini_model.d.ts

@@ -0,0 +1,15 @@
+import type { LlmRequest } from "../request.js";
+import type { LlmResponse } from "../response.js";
+import type { ModelConfig } from "../../config/config.js";
+export interface RunConfig {
+    stream?: boolean;
+    abortSignal?: AbortSignal;
+}
+export declare class Gemini {
+    readonly modelName: string;
+    private readonly client;
+    constructor(config: ModelConfig);
+    run(request: LlmRequest, config?: RunConfig): AsyncGenerator<LlmResponse, void, unknown>;
+    countTokens(request: LlmRequest): Promise<number>;
+    private runStream;
+}

package/dist/model/google/gemini_model.js

@@ -0,0 +1,131 @@
+import { GoogleGenAI, ThinkingLevel, } from "@google/genai";
+import { contentToGenAIContent } from "./gen_ai_convert_utils.js";
+import { StreamingResponseAggregator } from "./gemini_streaming_utils.js";
+import { createLlmResponse } from "./gemini_response_utils.js";
+import { logger } from "../../logger.js";
+export class Gemini {
+    constructor(config) {
+        this.modelName = config.modelName;
+        this.client = new GoogleGenAI(config);
+    }
+    async *run(request, config) {
+        logger.debug("[Gemini Model] Running model with request:", JSON.stringify(request, null, 2));
+        if (config?.stream) {
+            yield* this.runStream(request, config);
+            return;
+        }
+        try {
+            const response = await this.client.models.generateContent(toGenAiRequest({
+                model: this.modelName,
+                request,
+                config,
+            }));
+            logger.debug("[Gemini Model] response received:", JSON.stringify(response, null, 2));
+            yield createLlmResponse(response);
+        }
+        catch (e) {
+            logger.error("[Gemini Model] error:", e);
+            yield {
+                errorCode: "GEMINI_ERROR",
+                errorMessage: extractErrorMessage(e),
+            };
+        }
+    }
+    async countTokens(request) {
+        const response = await this.client.models.countTokens({
+            model: this.modelName,
+            contents: request.contents.map((c) => contentToGenAIContent(c)),
+            config: {
+                tools: request.tools
+                    ? [{ functionDeclarations: request.tools }]
+                    : undefined,
+                systemInstruction: request.systemInstructions,
+            },
+        });
+        logger.debug("[Gemini Model] token count:", response.totalTokens);
+        return response.totalTokens || 0;
+    }
+    async *runStream(request, config) {
+        try {
+            const aggregator = new StreamingResponseAggregator();
+            const stream = await this.client.models.generateContentStream(toGenAiRequest({
+                model: this.modelName,
+                request,
+                config,
+            }));
+            for await (const response of stream) {
+                for await (const llmResponse of aggregator.processResponse(response)) {
+                    logger.debug("[Gemini Model] yielding streaming response", JSON.stringify(llmResponse, null, 2));
+                    yield llmResponse;
+                }
+            }
+            const finalResponse = aggregator.close();
+            if (finalResponse) {
+                logger.debug("[Gemini Model] yielding final streaming response", JSON.stringify(finalResponse, null, 2));
+                yield finalResponse;
+            }
+        }
+        catch (e) {
+            logger.error("[Gemini Model] stream error:", e);
+            yield {
+                errorCode: "GEMINI_STREAM_ERROR",
+                errorMessage: extractErrorMessage(e),
+            };
+        }
+        logger.debug("[Gemini Model] stream finished");
+    }
+}
+function toGenAiRequest({ model, request, config = {}, }) {
+    return {
+        model,
+        contents: request.contents.map((c) => contentToGenAIContent(c)),
+        config: {
+            ...(config || {}),
+            tools: request.tools
+                ? [{ functionDeclarations: request.tools }]
+                : undefined,
+            thinkingConfig: request.thinkingConfig?.enabled ? {
+                includeThoughts: request.thinkingConfig.enabled,
+                thinkingLevel: toThinkingLevelConfig(request.thinkingConfig.level || "auto"),
+            } : undefined,
+            systemInstruction: request.systemInstructions,
+        },
+    };
+}
+function extractErrorMessage(e) {
+    if (e instanceof Error) {
+        try {
+            const parsed = JSON.parse(e.message);
+            if (parsed.error && parsed.error.message) {
+                try {
+                    const innerParsed = JSON.parse(parsed.error.message);
+                    if (innerParsed.error && innerParsed.error.message) {
+                        return innerParsed.error.message;
+                    }
+                }
+                catch {
+                    return parsed.error.message;
+                }
+                return parsed.error.message;
+            }
+        }
+        catch {
+            return e.message;
+        }
+        return e.message;
+    }
+    return String(e);
+}
+function toThinkingLevelConfig(thinkingLevel) {
+    switch (thinkingLevel) {
+        case "low":
+            return ThinkingLevel.LOW;
+        case "medium":
+            return ThinkingLevel.MEDIUM;
+        case "high":
+            return ThinkingLevel.HIGH;
+        case "auto":
+            return ThinkingLevel.HIGH;
+    }
+}
+//# sourceMappingURL=gemini_model.js.map

package/dist/model/google/gemini_model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini_model.js","sourceRoot":"","sources":["../../../src/model/google/gemini_model.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,WAAW,EAEX,aAAa,GACd,MAAM,eAAe,CAAC;AAIvB,OAAO,EAAE,qBAAqB,EAAE,MAAM,2BAA2B,CAAC;AAClE,OAAO,EAAE,2BAA2B,EAAE,MAAM,6BAA6B,CAAC;AAC1E,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAC/D,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AAOzC,MAAM,OAAO,MAAM;IAIjB,YAAY,MAAmB;QAC7B,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QAClC,IAAI,CAAC,MAAM,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,CAAC,GAAG,CACR,OAAmB,EACnB,MAAkB;QAElB,MAAM,CAAC,KAAK,CACV,4CAA4C,EAC5C,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CACjC,CAAC;QAEF,IAAI,MAAM,EAAE,MAAM,EAAE,CAAC;YACnB,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YACvC,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,eAAe,CACvD,cAAc,CAAC;gBACb,KAAK,EAAE,IAAI,CAAC,SAAS;gBACrB,OAAO;gBACP,MAAM;aACP,CAAC,CACH,CAAC;YAEF,MAAM,CAAC,KAAK,CACV,mCAAmC,EACnC,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAClC,CAAC;YAEF,MAAM,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QACpC,CAAC;QAAC,OAAO,CAAU,EAAE,CAAC;YACpB,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC;YACzC,MAAM;gBACJ,SAAS,EAAE,cAAc;gBACzB,YAAY,EAAE,mBAAmB,CAAC,CAAC,CAAC;aACrC,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAAmB;QACnC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC;YACpD,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC/D,MAAM,EAAE;gBACN,KAAK,EAAE,OAAO,CAAC,KAAK;oBAClB,CAAC,CAAC,CAAC,EAAE,oBAAoB,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC;oBAC3C,CAAC,CAAC,SAAS;gBACb,iBAAiB,EAAE,OAAO,CAAC,kBAAkB;aAC9C;SACF,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,6BAA6B,EAAE,QAAQ,CAAC,WAAW,CAAC,CAAC;QAElE,OAAO,QAAQ,CAAC,WAAW,IAAI,CAAC,CAAC;IACnC,CAAC;IAEO,KAAK,CAAC,CAAC,SAAS,CACtB,OAAmB,EACnB,MAAkB;QAElB,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,2BAA2B,EAAE,CAAC;YACrD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,qBAAqB,CAC3D,cAAc,CAAC;gBACb,KAAK,EAAE,IAAI,CAAC,SAAS;gBACrB,OAAO;gBACP,MAAM;aACP,CAAC,CACH,CAAC;YAEF,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,MAAM,EAAE,CAAC;gBACpC,IAAI,KAAK,EAAE,MAAM,WAAW,IAAI,UAAU,CAAC,eAAe,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACrE,MAAM,CAAC,KAAK,CACV,4CAA4C,EAC5C,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC,CACrC,CAAC;oBAEF,MAAM,WAAW,CAAC;gBACpB,CAAC;YACH,CAAC;YAED,MAAM,aAAa,GAAG,UAAU,CAAC,KAAK,EAAE,CAAC;YACzC,IAAI,aAAa,EAAE,CAAC;gBAClB,MAAM,CAAC,KAAK,CACV,kDAAkD,EAClD,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,CACvC,CAAC;gBAEF,MAAM,aAAa,CAAC;YACtB,CAAC;QACH,CAAC;QAAC,OAAO,CAAU,EAAE,CAAC;YACpB,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE,CAAC,CAAC,CAAC;YAChD,MAAM;gBACJ,SAAS,EAAE,qBAAqB;gBAChC,YAAY,EAAE,mBAAmB,CAAC,CAAC,CAAC;aACrC,CAAC;QACJ,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACjD,CAAC;CACF;AAOD,SAAS,cAAc,CAAC,EACtB,KAAK,EACL,OAAO,EACP,MAAM,GAAG,EAAE,GACU;IACrB,OAAO;QACL,KAAK;QACL,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC/D,MAAM,EAAE;YACN,GAAG,CAAC,MAAM,IAAI,EAAE,CAAC;YACjB,KAAK,EAAE,OAAO,CAAC,KAAK;gBAClB,CAAC,CAAC,CAAC,EAAE,oBAAoB,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC;gBAC3C,CAAC,CAAC,SAAS;YACb,cAAc,EAAE,OAAO,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,CAAC;gBAChD,eAAe,EAAE,OAAO,CAAC,cAAc,CAAC,OAAO;gBAC/C,aAAa,EAAE,qBAAqB,CAClC,OAAO,CAAC,cAAc,CAAC,KAAK,IAAI,MAAM,CACvC;aACF,CAAC,CAAC,CAAC,SAAS;YACb,iBAAiB,EAAE,OAAO,CAAC,kBAAkB;SAC9C;KACF,CAAC;AACJ,CAAC;AAED,SAAS,mBAAmB,CAAC,CAAU;IACrC,IAAI,CAAC,YAAY,KAAK,EAAE,CAAC;QACvB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;YACrC,IAAI,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;gBACzC,IAAI,CAAC;oBACH,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBACrD,IAAI,WAAW,CAAC,KAAK,IAAI,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;wBACnD,OAAO,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC;oBACnC,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,OAAO,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC;gBAC9B,CAAC;gBACD,OAAO,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC;YAC9B,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,CAAC,OAAO,CAAC;QACnB,CAAC;QACD,OAAO,CAAC,CAAC,OAAO,CAAC;IACnB,CAAC;IACD,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC;AACnB,CAAC;AAED,SAAS,qBAAqB,CAC5B,aAAiD;IAEjD,QAAQ,aAAa,EAAE,CAAC;QACtB,KAAK,KAAK;YACR,OAAO,aAAa,CAAC,GAAG,CAAC;QAC3B,KAAK,QAAQ;YACX,OAAO,aAAa,CAAC,MAAM,CAAC;QAC9B,KAAK,MAAM;YACT,OAAO,aAAa,CAAC,IAAI,CAAC;QAC5B,KAAK,MAAM;YACT,OAAO,aAAa,CAAC,IAAI,CAAC;IAC9B,CAAC;AACH,CAAC"}

package/dist/model/google/gemini_response_utils.d.ts

@@ -0,0 +1,3 @@
+import { GenerateContentResponse } from "@google/genai";
+import type { LlmResponse } from "../response.js";
+export declare function createLlmResponse(response: GenerateContentResponse): LlmResponse;

package/dist/model/google/gemini_response_utils.js

@@ -0,0 +1,36 @@
+import { GenerateContentResponse, FinishReason } from "@google/genai";
+import { genAIContentToContent } from "./gen_ai_convert_utils.js";
+export function createLlmResponse(response) {
+    const candidate = response.candidates?.[0];
+    const llmResponse = {};
+    if (response.usageMetadata) {
+        llmResponse.usageMetadata = {
+            inputTokens: response.usageMetadata.promptTokenCount,
+            outputTokens: response.usageMetadata.candidatesTokenCount,
+            cachedTokens: response.usageMetadata.cachedContentTokenCount,
+        };
+    }
+    if (candidate) {
+        if (candidate.content) {
+            llmResponse.content = genAIContentToContent(candidate.content);
+        }
+        if (candidate.finishReason) {
+            llmResponse.finishReason = candidate.finishReason;
+            if (candidate.finishReason !== FinishReason.STOP) {
+                llmResponse.errorCode = candidate.finishReason;
+                llmResponse.errorMessage = candidate.finishMessage;
+            }
+        }
+    }
+    if (candidate) {
+        const cand = candidate;
+        if (cand.groundingMetadata) {
+            llmResponse.groundingMetadata = cand.groundingMetadata;
+        }
+        if (cand.citationMetadata) {
+            llmResponse.citationMetadata = cand.citationMetadata;
+        }
+    }
+    return llmResponse;
+}
+//# sourceMappingURL=gemini_response_utils.js.map

package/dist/model/google/gemini_response_utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini_response_utils.js","sourceRoot":"","sources":["../../../src/model/google/gemini_response_utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAEtE,OAAO,EAAE,qBAAqB,EAAE,MAAM,2BAA2B,CAAC;AAElE,MAAM,UAAU,iBAAiB,CAC/B,QAAiC;IAEjC,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;IAE3C,MAAM,WAAW,GAAgB,EAAE,CAAC;IAEpC,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;QAC3B,WAAW,CAAC,aAAa,GAAG;YAC1B,WAAW,EAAE,QAAQ,CAAC,aAAa,CAAC,gBAAgB;YACpD,YAAY,EAAE,QAAQ,CAAC,aAAa,CAAC,oBAAoB;YACzD,YAAY,EAAE,QAAQ,CAAC,aAAa,CAAC,uBAAuB;SAC7D,CAAC;IACJ,CAAC;IAED,IAAI,SAAS,EAAE,CAAC;QACd,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;YACtB,WAAW,CAAC,OAAO,GAAG,qBAAqB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACjE,CAAC;QAED,IAAI,SAAS,CAAC,YAAY,EAAE,CAAC;YAC3B,WAAW,CAAC,YAAY,GAAG,SAAS,CAAC,YAAY,CAAC;YAElD,IAAI,SAAS,CAAC,YAAY,KAAK,YAAY,CAAC,IAAI,EAAE,CAAC;gBACjD,WAAW,CAAC,SAAS,GAAG,SAAS,CAAC,YAAY,CAAC;gBAC/C,WAAW,CAAC,YAAY,GAAI,SAAiB,CAAC,aAAa,CAAC;YAC9D,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,IAAI,GAAG,SAAgB,CAAC;QAC9B,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC1B,WAAmB,CAAC,iBAAiB,GAAG,IAAI,CAAC,iBAAiB,CAAC;QAClE,CAAC;QACD,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACzB,WAAmB,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;QAChE,CAAC;IACH,CAAC;IAED,OAAO,WAAW,CAAC;AACrB,CAAC"}

package/dist/model/google/gemini_streaming_utils.d.ts

@@ -0,0 +1,28 @@
+import { GenerateContentResponse } from "@google/genai";
+import type { LlmResponse } from "../response.js";
+/**
+ * Aggregates partial streaming responses.
+ *
+ * It aggregates content from partial responses, and generates LlmResponses for
+ * individual (partial) model responses, as well as for aggregated content.
+ */
+export declare class StreamingResponseAggregator {
+    private usageMetadata?;
+    private response?;
+    private partsSequence;
+    private currentTextBuffer;
+    private currentTextIsThought?;
+    private finishReason?;
+    private currentFcName?;
+    private currentFcArgs;
+    private currentFcId?;
+    private currentThoughtSignature?;
+    private flushTextBufferToSequence;
+    private getValueFromPartialArg;
+    private setValueByJsonPath;
+    private flushFunctionCallToSequence;
+    private processStreamingFunctionCall;
+    private processFunctionCallPart;
+    processResponse(response: GenerateContentResponse): AsyncGenerator<LlmResponse, void, void>;
+    close(): LlmResponse | undefined;
+}

package/dist/model/google/gemini_streaming_utils.js

@@ -0,0 +1,235 @@
+import { GenerateContentResponse, FinishReason } from "@google/genai";
+import { generateClientFunctionCallId } from "../../core/functions.js";
+import { isTextContentPart, isThoughtContentPart, isFunctionCallContentPart, } from "../../content.js";
+import { createLlmResponse } from "./gemini_response_utils.js";
+/**
+ * Aggregates partial streaming responses.
+ *
+ * It aggregates content from partial responses, and generates LlmResponses for
+ * individual (partial) model responses, as well as for aggregated content.
+ */
+export class StreamingResponseAggregator {
+    constructor() {
+        // For progressive SSE streaming mode: accumulate parts in order
+        this.partsSequence = [];
+        this.currentTextBuffer = "";
+        this.currentFcArgs = {};
+    }
+    flushTextBufferToSequence() {
+        if (!this.currentTextBuffer) {
+            return;
+        }
+        if (this.currentTextIsThought) {
+            this.partsSequence.push({
+                type: "thought",
+                thought: this.currentTextBuffer,
+            });
+        }
+        else {
+            this.partsSequence.push({
+                type: "text",
+                text: this.currentTextBuffer,
+            });
+        }
+        this.currentTextBuffer = "";
+        this.currentTextIsThought = undefined;
+    }
+    getValueFromPartialArg(partialArg, jsonPath) {
+        let value = null;
+        let hasValue = false;
+        const stringValue = partialArg.stringValue;
+        const numberValue = partialArg.numberValue;
+        const boolValue = partialArg.boolValue;
+        const nullValue = partialArg.nullValue;
+        if (stringValue !== undefined) {
+            const stringChunk = stringValue;
+            hasValue = true;
+            const pathWithoutPrefix = jsonPath.startsWith("$.")
+                ? jsonPath.slice(2)
+                : jsonPath;
+            const pathParts = pathWithoutPrefix.split(".");
+            let existingValue = this.currentFcArgs;
+            for (const part of pathParts) {
+                if (existingValue &&
+                    typeof existingValue === "object" &&
+                    part in existingValue) {
+                    existingValue = existingValue[part];
+                }
+                else {
+                    break;
+                }
+            }
+            if (typeof existingValue === "string") {
+                value = existingValue + stringChunk;
+            }
+            else {
+                value = stringChunk;
+            }
+        }
+        else if (numberValue !== undefined) {
+            value = numberValue;
+            hasValue = true;
+        }
+        else if (boolValue !== undefined) {
+            value = boolValue;
+            hasValue = true;
+        }
+        else if (nullValue !== undefined) {
+            value = null;
+            hasValue = true;
+        }
+        return [value, hasValue];
+    }
+    setValueByJsonPath(jsonPath, value) {
+        const path = jsonPath.startsWith("$.") ? jsonPath.slice(2) : jsonPath;
+        const pathParts = path.split(".");
+        let current = this.currentFcArgs;
+        for (let i = 0; i < pathParts.length - 1; i++) {
+            const part = pathParts[i];
+            if (!(part in current)) {
+                current[part] = {};
+            }
+            current = current[part];
+        }
+        current[pathParts[pathParts.length - 1]] = value;
+    }
+    flushFunctionCallToSequence() {
+        if (this.currentFcName) {
+            const fcPart = {
+                type: "function_call",
+                name: this.currentFcName,
+                args: JSON.parse(JSON.stringify(this.currentFcArgs)),
+                id: this.currentFcId ?? generateClientFunctionCallId(),
+            };
+            if (this.currentThoughtSignature) {
+                fcPart.thoughtSignature = this.currentThoughtSignature.toString();
+            }
+            this.partsSequence.push(fcPart);
+            this.currentFcName = undefined;
+            this.currentFcArgs = {};
+            this.currentFcId = undefined;
+            this.currentThoughtSignature = undefined;
+        }
+    }
+    processStreamingFunctionCall(fc) {
+        if (fc.name) {
+            this.currentFcName = fc.name;
+        }
+        if (fc.id) {
+            this.currentFcId = fc.id;
+        }
+        for (const partialArg of fc.partialArgs || []) {
+            const jsonPath = partialArg.jsonPath;
+            if (!jsonPath) {
+                continue;
+            }
+            const [value, hasValue] = this.getValueFromPartialArg(partialArg, jsonPath);
+            if (hasValue) {
+                this.setValueByJsonPath(jsonPath, value);
+            }
+        }
+        if (!fc.willContinue) {
+            this.flushTextBufferToSequence();
+            this.flushFunctionCallToSequence();
+        }
+    }
+    processFunctionCallPart(part) {
+        const fc = part;
+        if (fc.partialArgs || fc.willContinue) {
+            if (!fc.id && !this.currentFcId) {
+                fc.id = generateClientFunctionCallId();
+            }
+            if (part.thoughtSignature && !this.currentThoughtSignature) {
+                this.currentThoughtSignature = part.thoughtSignature;
+            }
+            this.processStreamingFunctionCall(fc);
+        }
+        else {
+            if (fc.name) {
+                if (!fc.id) {
+                    fc.id = generateClientFunctionCallId();
+                }
+                this.flushTextBufferToSequence();
+                this.partsSequence.push(part);
+            }
+        }
+    }
+    async *processResponse(response) {
+        this.response = response;
+        const llmResponse = createLlmResponse(response);
+        this.usageMetadata = llmResponse.usageMetadata;
+        // if (llmResponse.groundingMetadata) {
+        //   this.groundingMetadata = llmResponse.groundingMetadata;
+        // }
+        // if (llmResponse.citationMetadata) {
+        //   this.citationMetadata = llmResponse.citationMetadata;
+        // }
+        if (llmResponse.finishReason) {
+            this.finishReason = llmResponse.finishReason;
+        }
+        if (llmResponse.content && llmResponse.content.parts) {
+            for (const part of llmResponse.content.parts) {
+                if (isTextContentPart(part)) {
+                    const isThought = false;
+                    if (this.currentTextBuffer &&
+                        isThought !== this.currentTextIsThought) {
+                        this.flushTextBufferToSequence();
+                    }
+                    if (!this.currentTextBuffer) {
+                        this.currentTextIsThought = isThought;
+                    }
+                    this.currentTextBuffer += part.text;
+                }
+                else if (isThoughtContentPart(part)) {
+                    const isThought = true;
+                    if (this.currentTextBuffer &&
+                        isThought !== this.currentTextIsThought) {
+                        this.flushTextBufferToSequence();
+                    }
+                    if (!this.currentTextBuffer) {
+                        this.currentTextIsThought = isThought;
+                    }
+                    this.currentTextBuffer += part.thought;
+                }
+                else if (isFunctionCallContentPart(part)) {
+                    this.processFunctionCallPart(part);
+                }
+                else {
+                    this.flushTextBufferToSequence();
+                    this.partsSequence.push(part);
+                }
+            }
+        }
+        llmResponse.partial = true;
+        yield llmResponse;
+    }
+    close() {
+        if (!this.response?.candidates || this.response.candidates.length === 0) {
+            return;
+        }
+        this.flushTextBufferToSequence();
+        this.flushFunctionCallToSequence();
+        const finalParts = this.partsSequence;
+        if (finalParts.length === 0) {
+            return;
+        }
+        const candidate = this.response.candidates[0];
+        const finishReason = this.finishReason ?? candidate.finishReason;
+        return {
+            content: {
+                role: "agent",
+                parts: finalParts,
+            },
+            // groundingMetadata: this.groundingMetadata,
+            // citationMetadata: this.citationMetadata,
+            errorCode: finishReason === FinishReason.STOP ? undefined : finishReason,
+            errorMessage: finishReason === FinishReason.STOP
+                ? undefined
+                : candidate.finishMessage,
+            usageMetadata: this.usageMetadata,
+            finishReason: finishReason,
+            partial: false,
+        };
+    }
+}
+//# sourceMappingURL=gemini_streaming_utils.js.map