@llm4ts/core 0.1.0

package/dist/providers/GeminiApiProvider.js

@@ -0,0 +1,181 @@
+import * as Effect from "effect/Effect";
+import * as Redacted from "effect/Redacted";
+import * as Schema from "effect/Schema";
+import * as Stream from "effect/Stream";
+import { makeApiConnector } from "../Connector.js";
+import { AuthenticationError, ConfigError, ParseError } from "../Errors.js";
+import { ConnectorIds, LlmChunk, TokenUsage, ToolCall, ToolCallResponse } from "../Models.js";
+import { collect } from "../Streaming.js";
+import { parseFromText } from "../StructuredOutput.js";
+import { GeminiContent, GeminiFunctionDeclaration, GeminiGenerateContentRequest, GeminiGenerateContentRequestWithTools, GeminiGenerateContentResponse, GeminiGenerateContentResponseFull, GeminiGenerationConfig, GeminiPart, GeminiToolDef } from "./GeminiModels.js";
+const normalizeBaseUrl = (baseUrl) => baseUrl.replace(/\/+$/, "");
+export const geminiContents = (messages) => messages.map((message) => GeminiContent.make({
+    parts: [
+        GeminiPart.make({
+            text: message.content
+        })
+    ]
+}));
+const promptContents = (prompt) => [
+    GeminiContent.make({
+        parts: [
+            GeminiPart.make({
+                text: prompt
+            })
+        ]
+    })
+];
+export const buildGeminiRequest = (contents, jsonSchema) => GeminiGenerateContentRequest.make({
+    contents,
+    ...(jsonSchema === undefined
+        ? {}
+        : {
+            generationConfig: GeminiGenerationConfig.make({
+                responseMimeType: "application/json",
+                responseSchema: jsonSchema
+            })
+        })
+});
+export const buildGeminiToolRequest = (prompt, tools) => GeminiGenerateContentRequestWithTools.make({
+    contents: promptContents(prompt),
+    tools: [
+        GeminiToolDef.make({
+            functionDeclarations: tools.map((tool) => GeminiFunctionDeclaration.make({
+                name: tool.name,
+                description: tool.description,
+                parameters: tool.parameters
+            }))
+        })
+    ]
+});
+const decodeResponse = (raw, context) => Schema.decodeUnknownEffect(Schema.fromJsonString(GeminiGenerateContentResponse))(raw).pipe(Effect.mapError((error) => ParseError.make({
+    message: `${context}: ${String(error)}`,
+    raw
+})));
+const decodeToolResponse = (raw) => Schema.decodeUnknownEffect(Schema.fromJsonString(GeminiGenerateContentResponseFull))(raw).pipe(Effect.mapError((error) => ParseError.make({
+    message: `Failed to decode Gemini tool response: ${String(error)}`,
+    raw
+})));
+const usageFromResponse = (response) => {
+    const usage = response.usageMetadata;
+    return usage === undefined
+        ? undefined
+        : TokenUsage.make({
+            prompt: usage.promptTokenCount ?? 0,
+            completion: usage.candidatesTokenCount ?? 0,
+            total: usage.totalTokenCount ?? 0
+        });
+};
+const metadataFromResponse = (config, response) => {
+    const usage = response.usageMetadata;
+    return {
+        provider: "gemini-api",
+        model: config.model,
+        ...(usage?.promptTokenCount === undefined
+            ? {}
+            : { promptTokenCount: String(usage.promptTokenCount) }),
+        ...(usage?.candidatesTokenCount === undefined
+            ? {}
+            : { candidatesTokenCount: String(usage.candidatesTokenCount) }),
+        ...(usage?.totalTokenCount === undefined
+            ? {}
+            : { totalTokenCount: String(usage.totalTokenCount) })
+    };
+};
+const responseText = (response, raw) => {
+    const text = response.candidates[0]?.content.parts
+        .map((part) => part.text?.trim())
+        .find((part) => part !== undefined && part !== null && part.length > 0);
+    return text === undefined || text === null
+        ? Effect.fail(ParseError.make({
+            message: "Gemini API response has no text content in candidates. " +
+                "The response may be safety-filtered or malformed.",
+            raw
+        }))
+        : Effect.succeed(text);
+};
+export const makeGeminiApiProvider = (config, httpClient) => {
+    const requiredConfig = config.baseUrl === undefined
+        ? Effect.fail(ConfigError.make({
+            message: "Missing baseUrl for Gemini API provider"
+        }))
+        : config.apiKey === undefined
+            ? Effect.fail(AuthenticationError.make({
+                message: "Missing API key for Gemini API provider (set GEMINI_API_KEY or GOOGLE_API_KEY, or pass apiKey in the connector config)"
+            }))
+            : Effect.succeed({
+                baseUrl: normalizeBaseUrl(config.baseUrl),
+                apiKey: Redacted.value(config.apiKey)
+            });
+    const endpoint = (baseUrl, operation) => `${baseUrl}/v1beta/models/${config.model}:${operation}`;
+    const headers = (apiKey) => ({
+        "x-goog-api-key": apiKey
+    });
+    const executeRequest = (prompt, jsonSchema) => Effect.gen(function* () {
+        const { apiKey, baseUrl } = yield* requiredConfig;
+        const raw = yield* httpClient.postJson(endpoint(baseUrl, "generateContent"), JSON.stringify(buildGeminiRequest(promptContents(prompt), jsonSchema)), headers(apiKey), config.timeout);
+        const response = yield* decodeResponse(raw, "Failed to decode Gemini API response");
+        return [response, raw];
+    });
+    const executeStreamWithContents = (contents) => Stream.unwrap(Effect.map(requiredConfig, ({ apiKey, baseUrl }) => httpClient
+        .postJsonStream(endpoint(baseUrl, "streamGenerateContent"), JSON.stringify(buildGeminiRequest(contents)), headers(apiKey), config.timeout)
+        .pipe(Stream.map((line) => line.trim()), Stream.filter((line) => line.length > 0), Stream.map((line) => line.startsWith("data:") ? line.slice("data:".length).trim() : line), Stream.filter((payload) => payload.length > 0 && payload !== "[DONE]"), Stream.mapEffect((payload) => decodeResponse(payload, "Failed to decode Gemini stream chunk")), Stream.map((response) => {
+        const delta = response.candidates[0]?.content.parts.map((part) => part.text ?? "").join("") ?? "";
+        const finishReason = response.candidates[0]?.finishReason?.toLowerCase();
+        const usage = usageFromResponse(response);
+        return LlmChunk.make({
+            delta,
+            metadata: metadataFromResponse(config, response),
+            ...(finishReason === undefined || finishReason === null ? {} : { finishReason }),
+            ...(usage === undefined ? {} : { usage })
+        });
+    }), Stream.filter((chunk) => chunk.delta.length > 0 ||
+        chunk.finishReason !== undefined ||
+        chunk.usage !== undefined))));
+    const executeStructuredWithUsage = (prompt, schema, jsonSchema) => Effect.gen(function* () {
+        const [response, raw] = yield* executeRequest(prompt, jsonSchema);
+        const content = yield* responseText(response, raw);
+        const value = yield* parseFromText(content, schema, jsonSchema);
+        const result = [value, undefined, undefined];
+        return result;
+    });
+    const executeWithTools = (prompt, tools) => Effect.gen(function* () {
+        const { apiKey, baseUrl } = yield* requiredConfig;
+        const raw = yield* httpClient.postJson(endpoint(baseUrl, "generateContent"), JSON.stringify(buildGeminiToolRequest(prompt, tools)), headers(apiKey), config.timeout);
+        const response = yield* decodeToolResponse(raw);
+        const candidate = response.candidates[0];
+        if (candidate === undefined) {
+            return yield* Effect.fail(ParseError.make({
+                message: "Gemini tool response has no candidates",
+                raw
+            }));
+        }
+        const calls = candidate.content.parts.flatMap((part) => part.functionCall === undefined ? [] : [part.functionCall]);
+        const text = candidate.content.parts
+            .map((part) => part.text)
+            .find((part) => part !== undefined && part !== null);
+        return ToolCallResponse.make({
+            toolCalls: calls.map((call, index) => ToolCall.make({
+                id: `gemini_call_${index}`,
+                name: call.name,
+                arguments: JSON.stringify(call.args) ?? "{}"
+            })),
+            finishReason: candidate.finishReason ?? "STOP",
+            ...(text === undefined || text === null ? {} : { content: text })
+        });
+    });
+    const executeStream = (prompt) => executeStreamWithContents(promptContents(prompt));
+    const isAvailable = collect(executeStream("health check")).pipe(Effect.match({
+        onFailure: () => false,
+        onSuccess: () => true
+    }));
+    return makeApiConnector({
+        id: ConnectorIds.GeminiApi,
+        executeStream,
+        executeStreamWithHistory: (messages) => executeStreamWithContents(geminiContents(messages)),
+        executeWithTools,
+        executeStructuredWithUsage,
+        isAvailable
+    });
+};
+//# sourceMappingURL=GeminiApiProvider.js.map

package/dist/providers/GeminiApiProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"GeminiApiProvider.js","sourceRoot":"","sources":["../../src/providers/GeminiApiProvider.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,MAAM,eAAe,CAAA;AACvC,OAAO,KAAK,QAAQ,MAAM,iBAAiB,CAAA;AAC3C,OAAO,KAAK,MAAM,MAAM,eAAe,CAAA;AACvC,OAAO,KAAK,MAAM,MAAM,eAAe,CAAA;AACvC,OAAO,EAAE,gBAAgB,EAA0B,MAAM,iBAAiB,CAAA;AAC1E,OAAO,EAAE,mBAAmB,EAAE,WAAW,EAAE,UAAU,EAAiB,MAAM,cAAc,CAAA;AAG1F,OAAO,EACL,YAAY,EACZ,QAAQ,EACR,UAAU,EACV,QAAQ,EACR,gBAAgB,EAKjB,MAAM,cAAc,CAAA;AACrB,OAAO,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAA;AACzC,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAA;AACtD,OAAO,EACL,aAAa,EACb,yBAAyB,EACzB,4BAA4B,EAC5B,qCAAqC,EACrC,6BAA6B,EAC7B,iCAAiC,EACjC,sBAAsB,EACtB,UAAU,EACV,aAAa,EACd,MAAM,mBAAmB,CAAA;AAO1B,MAAM,gBAAgB,GAAG,CAAC,OAAe,EAAU,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;AAEjF,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,QAAgC,EAAgC,EAAE,CAC/F,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CACvB,aAAa,CAAC,IAAI,CAAC;IACjB,KAAK,EAAE;QACL,UAAU,CAAC,IAAI,CAAC;YACd,IAAI,EAAE,OAAO,CAAC,OAAO;SACtB,CAAC;KACH;CACF,CAAC,CACH,CAAA;AAEH,MAAM,cAAc,GAAG,CAAC,MAAc,EAAgC,EAAE,CAAC;IACvE,aAAa,CAAC,IAAI,CAAC;QACjB,KAAK,EAAE;YACL,UAAU,CAAC,IAAI,CAAC;gBACd,IAAI,EAAE,MAAM;aACb,CAAC;SACH;KACF,CAAC;CACH,CAAA;AAED,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAChC,QAAsC,EACtC,UAAuB,EACO,EAAE,CAChC,4BAA4B,CAAC,IAAI,CAAC;IAChC,QAAQ;IACR,GAAG,CAAC,UAAU,KAAK,SAAS;QAC1B,CAAC,CAAC,EAAE;QACJ,CAAC,CAAC;YACE,gBAAgB,EAAE,sBAAsB,CAAC,IAAI,CAAC;gBAC5C,gBAAgB,EAAE,kBAAkB;gBACpC,cAAc,EAAE,UAAU;aAC3B,CAAC;SACH,CAAC;CACP,CAAC,CAAA;AAEJ,MAAM,CAAC,MAAM,sBAAsB,GAAG,CACpC,MAAc,EACd,KAAoC,EACG,EAAE,CACzC,qCAAqC,CAAC,IAAI,CAAC;IACzC,QAAQ,EAAE,cAAc,CAAC,MAAM,CAAC;IAChC,KAAK,EAAE;QACL,aAAa,CAAC,IAAI,CAAC;YACjB,oBAAoB,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CACvC,yBAAyB,CAAC,IAAI,CAAC;gBAC7B,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;aAC5B,CAAC,CACH;SACF,CAAC;KACH;CACF,CAAC,CAAA;AAEJ,MAAM,cAAc,GAAG,CACrB,GAAW,EACX,OAAe,EAC2C,EAAE,CAC5D,MAAM,CAAC,mBAAmB,CAAC,MAAM,CAAC,cAAc,CAAC,6BAA6B,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CACxF,MAAM,CAAC,QAAQ,CAAC,CAAC,KAAK,EAAE,EAAE,CACxB,UAAU,CAAC,IAAI,CAAC;IACd,OAAO,EAAE,GAAG,OAAO,KAAK,MAAM,CAAC,KAAK,CAAC,EAAE;IACvC,GAAG;CACJ,CAAC,CACH,CACF,CAAA;AAEH,MAAM,kBAAkB,GAAG,CACzB,GAAW,EACmD,EAAE,CAChE,MAAM,CAAC,mBAAmB,CAAC,MAAM,CAAC,cAAc,CAAC,iCAAiC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAC5F,MAAM,CAAC,QAAQ,CAAC,CAAC,KAAK,EAAE,EAAE,CACxB,UAAU,CAAC,IAAI,CAAC;IACd,OAAO,EAAE,0CAA0C,MAAM,CAAC,KAAK,CAAC,EAAE;IAClE,GAAG;CACJ,CAAC,CACH,CACF,CAAA;AAEH,MAAM,iBAAiB,GAAG,CAAC,QAAuC,EAA0B,EAAE;IAC5F,MAAM,KAAK,GAAG,QAAQ,CAAC,aAAa,CAAA;IACpC,OAAO,KAAK,KAAK,SAAS;QACxB,CAAC,CAAC,SAAS;QACX,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC;YACd,MAAM,EAAE,KAAK,CAAC,gBAAgB,IAAI,CAAC;YACnC,UAAU,EAAE,KAAK,CAAC,oBAAoB,IAAI,CAAC;YAC3C,KAAK,EAAE,KAAK,CAAC,eAAe,IAAI,CAAC;SAClC,CAAC,CAAA;AACR,CAAC,CAAA;AAED,MAAM,oBAAoB,GAAG,CAC3B,MAAiB,EACjB,QAAuC,EACL,EAAE;IACpC,MAAM,KAAK,GAAG,QAAQ,CAAC,aAAa,CAAA;IACpC,OAAO;QACL,QAAQ,EAAE,YAAY;QACtB,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,GAAG,CAAC,KAAK,EAAE,gBAAgB,KAAK,SAAS;YACvC,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,EAAE,gBAAgB,EAAE,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,EAAE,CAAC;QACzD,GAAG,CAAC,KAAK,EAAE,oBAAoB,KAAK,SAAS;YAC3C,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,EAAE,oBAAoB,EAAE,MAAM,CAAC,KAAK,CAAC,oBAAoB,CAAC,EAAE,CAAC;QACjE,GAAG,CAAC,KAAK,EAAE,eAAe,KAAK,SAAS;YACtC,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,EAAE,eAAe,EAAE,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,EAAE,CAAC;KACxD,CAAA;AACH,CAAC,CAAA;AAED,MAAM,YAAY,GAAG,CACnB,QAAuC,EACvC,GAAW,EACwB,EAAE;IACrC,MAAM,IAAI,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,KAAK;SAC/C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC;SAChC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,SAAS,IAAI,IAAI,KAAK,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;IACzE,OAAO,IAAI,KAAK,SAAS,IAAI,IAAI,KAAK,IAAI;QACxC,CAAC,CAAC,MAAM,CAAC,IAAI,CACT,UAAU,CAAC,IAAI,CAAC;YACd,OAAO,EACL,yDAAyD;gBACzD,mDAAmD;YACrD,GAAG;SACJ,CAAC,CACH;QACH,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;AAC1B,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,qBAAqB,GAAG,CACnC,MAAiB,EACjB,UAA2B,EACR,EAAE;IACrB,MAAM,cAAc,GAClB,MAAM,CAAC,OAAO,KAAK,SAAS;QAC1B,CAAC,CAAC,MAAM,CAAC,IAAI,CACT,WAAW,CAAC,IAAI,CAAC;YACf,OAAO,EAAE,yCAAyC;SACnD,CAAC,CACH;QACH,CAAC,CAAC,MAAM,CAAC,MAAM,KAAK,SAAS;YAC3B,CAAC,CAAC,MAAM,CAAC,IAAI,CACT,mBAAmB,CAAC,IAAI,CAAC;gBACvB,OAAO,EACL,wHAAwH;aAC3H,CAAC,CACH;YACH,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC;gBACb,OAAO,EAAE,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC;gBACzC,MAAM,EAAE,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC;aACtC,CAAC,CAAA;IAEV,MAAM,QAAQ,GAAG,CACf,OAAe,EACf,SAAsD,EAC9C,EAAE,CAAC,GAAG,OAAO,kBAAkB,MAAM,CAAC,KAAK,IAAI,SAAS,EAAE,CAAA;IAEpE,MAAM,OAAO,GAAG,CAAC,MAAc,EAAoC,EAAE,CAAC,CAAC;QACrE,gBAAgB,EAAE,MAAM;KACzB,CAAC,CAAA;IAEF,MAAM,cAAc,GAAG,CACrB,MAAc,EACd,UAAuB,EACmE,EAAE,CAC5F,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;QAClB,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC,CAAC,cAAc,CAAA;QACjD,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,UAAU,CAAC,QAAQ,CACpC,QAAQ,CAAC,OAAO,EAAE,iBAAiB,CAAC,EACpC,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,cAAc,CAAC,MAAM,CAAC,EAAE,UAAU,CAAC,CAAC,EACtE,OAAO,CAAC,MAAM,CAAC,EACf,MAAM,CAAC,OAAO,CACf,CAAA;QACD,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,cAAc,CAAC,GAAG,EAAE,sCAAsC,CAAC,CAAA;QACnF,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAA;IACxB,CAAC,CAAC,CAAA;IAEJ,MAAM,yBAAyB,GAAG,CAChC,QAAsC,EACH,EAAE,CACrC,MAAM,CAAC,MAAM,CACX,MAAM,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,CACjD,UAAU;SACP,cAAc,CACb,QAAQ,CAAC,OAAO,EAAE,uBAAuB,CAAC,EAC1C,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,EAC5C,OAAO,CAAC,MAAM,CAAC,EACf,MAAM,CAAC,OAAO,CACf;SACA,IAAI,CACH,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,EACjC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,EACxC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAClB,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CACpE,EACD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,KAAK,QAAQ,CAAC,EACtE,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,EAAE,CAC3B,cAAc,CAAC,OAAO,EAAE,sCAAsC,CAAC,CAChE,EACD,MAAM,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;QACtB,MAAM,KAAK,GACT,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAA;QACrF,MAAM,YAAY,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,WAAW,EAAE,CAAA;QACxE,MAAM,KAAK,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAA;QACzC,OAAO,QAAQ,CAAC,IAAI,CAAC;YACnB,KAAK;YACL,QAAQ,EAAE,oBAAoB,CAAC,MAAM,EAAE,QAAQ,CAAC;YAChD,GAAG,CAAC,YAAY,KAAK,SAAS,IAAI,YAAY,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC;YAChF,GAAG,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC;SAC1C,CAAC,CAAA;IACJ,CAAC,CAAC,EACF,MAAM,CAAC,MAAM,CACX,CAAC,KAAK,EAAE,EAAE,CACR,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC;QACtB,KAAK,CAAC,YAAY,KAAK,SAAS;QAChC,KAAK,CAAC,KAAK,KAAK,SAAS,CAC5B,CACF,CACJ,CACF,CAAA;IAEH,MAAM,0BAA0B,GAAG,CACjC,MAAc,EACd,MAA4C,EAC5C,UAAsB,EAC4B,EAAE,CACpD,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;QAClB,MAAM,CAAC,QAAQ,EAAE,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,cAAc,CAAC,MAAM,EAAE,UAAU,CAAC,CAAA;QACjE,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,YAAY,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAA;QAClD,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,CAAA;QAC/D,MAAM,MAAM,GAAwB,CAAC,KAAK,EAAE,SAAS,EAAE,SAAS,CAAC,CAAA;QACjE,OAAO,MAAM,CAAA;IACf,CAAC,CAAC,CAAA;IAEJ,MAAM,gBAAgB,GAAG,CACvB,MAAc,EACd,KAAoC,EACO,EAAE,CAC7C,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;QAClB,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC,CAAC,cAAc,CAAA;QACjD,MAAM,GAAG,GAAG,KAAK,CAAC,CAAC,UAAU,CAAC,QAAQ,CACpC,QAAQ,CAAC,OAAO,EAAE,iBAAiB,CAAC,EACpC,IAAI,CAAC,SAAS,CAAC,sBAAsB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,EACrD,OAAO,CAAC,MAAM,CAAC,EACf,MAAM,CAAC,OAAO,CACf,CAAA;QACD,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAA;QAC/C,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAA;QACxC,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAC5B,OAAO,KAAK,CAAC,CAAC,MAAM,CAAC,IAAI,CACvB,UAAU,CAAC,IAAI,CAAC;gBACd,OAAO,EAAE,wCAAwC;gBACjD,GAAG;aACJ,CAAC,CACH,CAAA;QACH,CAAC;QACD,MAAM,KAAK,GAAG,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CACrD,IAAI,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAC3D,CAAA;QACD,MAAM,IAAI,GAAG,SAAS,CAAC,OAAO,CAAC,KAAK;aACjC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;aACxB,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,SAAS,IAAI,IAAI,KAAK,IAAI,CAAC,CAAA;QAEtD,OAAO,gBAAgB,CAAC,IAAI,CAAC;YAC3B,SAAS,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CACnC,QAAQ,CAAC,IAAI,CAAC;gBACZ,EAAE,EAAE,eAAe,KAAK,EAAE;gBAC1B,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI;aAC7C,CAAC,CACH;YACD,YAAY,EAAE,SAAS,CAAC,YAAY,IAAI,MAAM;YAC9C,GAAG,CAAC,IAAI,KAAK,SAAS,IAAI,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;SAClE,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEJ,MAAM,aAAa,GAAG,CAAC,MAAc,EAAqC,EAAE,CAC1E,yBAAyB,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAA;IAEnD,MAAM,WAAW,GAA2B,OAAO,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CACrF,MAAM,CAAC,KAAK,CAAC;QACX,SAAS,EAAE,GAAG,EAAE,CAAC,KAAK;QACtB,SAAS,EAAE,GAAG,EAAE,CAAC,IAAI;KACtB,CAAC,CACH,CAAA;IAED,OAAO,gBAAgB,CAAC;QACtB,EAAE,EAAE,YAAY,CAAC,SAAS;QAC1B,aAAa;QACb,wBAAwB,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,yBAAyB,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QAC3F,gBAAgB;QAChB,0BAA0B;QAC1B,WAAW;KACZ,CAAC,CAAA;AACJ,CAAC,CAAA"}

package/dist/providers/GeminiCliProvider.d.ts

@@ -0,0 +1,106 @@
+import * as Context from "effect/Context";
+import * as Effect from "effect/Effect";
+import * as Schema from "effect/Schema";
+import * as Stream from "effect/Stream";
+import { type CliConnectorShape } from "../Connector.ts";
+import type { CliConnectorConfig } from "../ConnectorConfig.ts";
+import { type LlmError } from "../Errors.ts";
+import { type LlmConfig } from "../Models.ts";
+declare const GeminiCliLogLine_base: Schema.Class<GeminiCliLogLine, Schema.TaggedStruct<"LogLine", {
+    readonly line: Schema.String;
+}>, {}>;
+export declare class GeminiCliLogLine extends GeminiCliLogLine_base {
+}
+declare const GeminiCliInit_base: Schema.Class<GeminiCliInit, Schema.TaggedStruct<"Init", {
+    readonly model: Schema.optionalKey<Schema.String>;
+    readonly sessionId: Schema.optionalKey<Schema.String>;
+}>, {}>;
+export declare class GeminiCliInit extends GeminiCliInit_base {
+}
+declare const GeminiCliMessage_base: Schema.Class<GeminiCliMessage, Schema.TaggedStruct<"Message", {
+    readonly role: Schema.optionalKey<Schema.String>;
+    readonly content: Schema.optionalKey<Schema.String>;
+    readonly delta: Schema.Boolean;
+}>, {}>;
+export declare class GeminiCliMessage extends GeminiCliMessage_base {
+}
+declare const GeminiCliToolUse_base: Schema.Class<GeminiCliToolUse, Schema.TaggedStruct<"ToolUse", {
+    readonly toolName: Schema.optionalKey<Schema.String>;
+    readonly toolId: Schema.optionalKey<Schema.String>;
+    readonly input: Schema.optionalKey<Schema.String>;
+}>, {}>;
+export declare class GeminiCliToolUse extends GeminiCliToolUse_base {
+}
+declare const GeminiCliToolResult_base: Schema.Class<GeminiCliToolResult, Schema.TaggedStruct<"ToolResult", {
+    readonly toolId: Schema.optionalKey<Schema.String>;
+    readonly status: Schema.optionalKey<Schema.String>;
+    readonly content: Schema.optionalKey<Schema.String>;
+}>, {}>;
+export declare class GeminiCliToolResult extends GeminiCliToolResult_base {
+}
+declare const GeminiCliError_base: Schema.Class<GeminiCliError, Schema.TaggedStruct<"Error", {
+    readonly message: Schema.optionalKey<Schema.String>;
+    readonly code: Schema.optionalKey<Schema.Int>;
+    readonly errorType: Schema.optionalKey<Schema.String>;
+}>, {}>;
+export declare class GeminiCliError extends GeminiCliError_base {
+}
+declare const GeminiStreamStats_base: Schema.Class<GeminiStreamStats, Schema.Struct<{
+    readonly totalTokens: Schema.optionalKey<Schema.Int>;
+    readonly inputTokens: Schema.optionalKey<Schema.Int>;
+    readonly outputTokens: Schema.optionalKey<Schema.Int>;
+    readonly cached: Schema.optionalKey<Schema.Int>;
+}>, {}>;
+export declare class GeminiStreamStats extends GeminiStreamStats_base {
+}
+declare const GeminiCliResult_base: Schema.Class<GeminiCliResult, Schema.TaggedStruct<"Result", {
+    readonly status: Schema.optionalKey<Schema.String>;
+    readonly errorMessage: Schema.optionalKey<Schema.String>;
+    readonly stats: Schema.optionalKey<typeof GeminiStreamStats>;
+}>, {}>;
+export declare class GeminiCliResult extends GeminiCliResult_base {
+}
+export declare const GeminiCliStreamEvent: Schema.Union<readonly [typeof GeminiCliLogLine, typeof GeminiCliInit, typeof GeminiCliMessage, typeof GeminiCliToolUse, typeof GeminiCliToolResult, typeof GeminiCliError, typeof GeminiCliResult]>;
+export type GeminiCliStreamEvent = typeof GeminiCliStreamEvent.Type;
+export declare const GeminiSandbox: Schema.Literals<readonly ["Docker", "Podman", "SeatbeltMacOS", "Runsc", "Lxc", "Default"]>;
+export type GeminiSandbox = typeof GeminiSandbox.Type;
+declare const GeminiCliExecutionContext_base: Schema.Class<GeminiCliExecutionContext, Schema.Struct<{
+    readonly cwd: Schema.optionalKey<Schema.String>;
+    readonly includeDirectories: Schema.withConstructorDefault<Schema.$Array<Schema.String>>;
+    readonly sandbox: Schema.optionalKey<Schema.Literals<readonly ["Docker", "Podman", "SeatbeltMacOS", "Runsc", "Lxc", "Default"]>>;
+    readonly turnLimit: Schema.optionalKey<Schema.Int>;
+    readonly readOnly: Schema.withConstructorDefault<Schema.Boolean>;
+}>, {}>;
+export declare class GeminiCliExecutionContext extends GeminiCliExecutionContext_base {
+}
+export declare const geminiCliExecutionContextFrom: (config: CliConnectorConfig) => GeminiCliExecutionContext;
+export interface GeminiCliExecutorShape {
+    readonly checkGeminiInstalled: Effect.Effect<void, LlmError>;
+    readonly runGeminiProcess: (prompt: string, config: LlmConfig, context: GeminiCliExecutionContext) => Effect.Effect<string, LlmError>;
+    readonly runGeminiProcessStream: (prompt: string, config: LlmConfig, context: GeminiCliExecutionContext) => Stream.Stream<GeminiCliStreamEvent, LlmError>;
+}
+declare const GeminiCliExecutor_base: Context.ServiceClass<GeminiCliExecutor, "@llm4ts/core/providers/GeminiCliExecutor", GeminiCliExecutorShape>;
+export declare class GeminiCliExecutor extends GeminiCliExecutor_base {
+}
+export declare const geminiSandboxEnvValue: (sandbox: GeminiSandbox) => string | undefined;
+export declare const buildGeminiArgs: (config: LlmConfig, context: GeminiCliExecutionContext, outputFormat: string) => ReadonlyArray<string>;
+export declare const geminiTurnLimitSettingsJson: (turns: number) => string;
+export declare const geminiProcessEnv: (context: GeminiCliExecutionContext, turnLimitSettingsPath?: string) => Readonly<Record<string, string>>;
+export declare const geminiQuotaDiagnostic: (stderr: ReadonlyArray<string>) => string | undefined;
+export declare const withGeminiStderrDiagnostic: (event: GeminiCliStreamEvent, stderr: ReadonlyArray<string>) => GeminiCliStreamEvent;
+export declare const validateGeminiExitCode: (exitCode: number, stderr: string, turnLimit?: number) => Effect.Effect<void, LlmError>;
+export declare const validateGeminiStreamExit: (exitCode: number, stderr: ReadonlyArray<string>, sawAssistantText: boolean, turnLimit?: number) => Effect.Effect<void, LlmError>;
+export type GeminiResponseExtraction = {
+    readonly ok: true;
+    readonly value: string;
+} | {
+    readonly ok: false;
+    readonly error: string;
+};
+export declare const isGeminiPreambleLine: (line: string) => boolean;
+export declare const isKnownGeminiStderrNoise: (line: string) => boolean;
+export declare const extractGeminiCliResponse: (output: string) => GeminiResponseExtraction;
+export declare const parseGeminiCliStreamEvent: (line: string) => GeminiCliStreamEvent;
+export declare const makeGeminiCliProvider: (config: LlmConfig, executor: GeminiCliExecutorShape, executionContext?: GeminiCliExecutionContext) => CliConnectorShape;
+export {};
+//# sourceMappingURL=GeminiCliProvider.d.ts.map

package/dist/providers/GeminiCliProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"GeminiCliProvider.d.ts","sourceRoot":"","sources":["../../src/providers/GeminiCliProvider.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,OAAO,MAAM,gBAAgB,CAAA;AACzC,OAAO,KAAK,MAAM,MAAM,eAAe,CAAA;AAGvC,OAAO,KAAK,MAAM,MAAM,eAAe,CAAA;AACvC,OAAO,KAAK,MAAM,MAAM,eAAe,CAAA;AACvC,OAAO,EAAoB,KAAK,iBAAiB,EAAE,MAAM,iBAAiB,CAAA;AAC1E,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAA;AAC/D,OAAO,EAKL,KAAK,QAAQ,EACd,MAAM,cAAc,CAAA;AAErB,OAAO,EAML,KAAK,SAAS,EAEf,MAAM,cAAc,CAAA;;;;AAYrB,qBAAa,gBAAiB,SAAQ,qBAEpC;CAAG;;;;;AAEL,qBAAa,aAAc,SAAQ,kBAGjC;CAAG;;;;;;AAEL,qBAAa,gBAAiB,SAAQ,qBAIpC;CAAG;;;;;;AAEL,qBAAa,gBAAiB,SAAQ,qBAIpC;CAAG;;;;;;AAEL,qBAAa,mBAAoB,SAAQ,wBAIvC;CAAG;;;;;;AAEL,qBAAa,cAAe,SAAQ,mBAIlC;CAAG;;;;;;;AAEL,qBAAa,iBAAkB,SAAQ,sBAKrC;CAAG;;;;;;AAEL,qBAAa,eAAgB,SAAQ,oBAInC;CAAG;AAEL,eAAO,MAAM,oBAAoB,qMAQ/B,CAAA;AACF,MAAM,MAAM,oBAAoB,GAAG,OAAO,oBAAoB,CAAC,IAAI,CAAA;AAEnE,eAAO,MAAM,aAAa,4FAOxB,CAAA;AACF,MAAM,MAAM,aAAa,GAAG,OAAO,aAAa,CAAC,IAAI,CAAA;;;;;;;;AAIrD,qBAAa,yBAA0B,SAAQ,8BAU7C;CAAG;AAEL,eAAO,MAAM,6BAA6B,GACxC,QAAQ,kBAAkB,KACzB,yBAUC,CAAA;AAEJ,MAAM,WAAW,sBAAsB;IACrC,QAAQ,CAAC,oBAAoB,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAA;IAC5D,QAAQ,CAAC,gBAAgB,EAAE,CACzB,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,SAAS,EACjB,OAAO,EAAE,yBAAyB,KAC/B,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAA;IACpC,QAAQ,CAAC,sBAAsB,EAAE,CAC/B,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,SAAS,EACjB,OAAO,EAAE,yBAAyB,KAC/B,MAAM,CAAC,MAAM,CAAC,oBAAoB,EAAE,QAAQ,CAAC,CAAA;CACnD;;AAED,qBAAa,iBAAkB,SAAQ,sBAEtC;CAAG;AAEJ,eAAO,MAAM,qBAAqB,GAAI,SAAS,aAAa,KAAG,MAAM,GAAG,SAevE,CAAA;AAED,eAAO,MAAM,eAAe,GAC1B,QAAQ,SAAS,EACjB,SAAS,yBAAyB,EAClC,cAAc,MAAM,KACnB,aAAa,CAAC,MAAM,CActB,CAAA;AAED,eAAO,MAAM,2BAA2B,GAAI,OAAO,MAAM,KAAG,MAKxD,CAAA;AAEJ,eAAO,MAAM,gBAAgB,GAC3B,SAAS,yBAAyB,EAClC,wBAAwB,MAAM,KAC7B,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAWjC,CAAA;AAED,eAAO,MAAM,qBAAqB,GAAI,QAAQ,aAAa,CAAC,MAAM,CAAC,KAAG,MAAM,GAAG,SAanE,CAAA;AAEZ,eAAO,MAAM,0BAA0B,GACrC,OAAO,oBAAoB,EAC3B,QAAQ,aAAa,CAAC,MAAM,CAAC,KAC5B,oBAiCF,CAAA;AAED,eAAO,MAAM,sBAAsB,GACjC,UAAU,MAAM,EAChB,QAAQ,MAAM,EACd,YAAY,MAAM,KACjB,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,CAe9B,CAAA;AAED,eAAO,MAAM,wBAAwB,GACnC,UAAU,MAAM,EAChB,QAAQ,aAAa,CAAC,MAAM,CAAC,EAC7B,kBAAkB,OAAO,EACzB,YAAY,MAAM,KACjB,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,CAsB9B,CAAA;AAeD,MAAM,MAAM,wBAAwB,GAChC;IACE,QAAQ,CAAC,EAAE,EAAE,IAAI,CAAA;IACjB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAA;CACvB,GACD;IACE,QAAQ,CAAC,EAAE,EAAE,KAAK,CAAA;IAClB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAA;CACvB,CAAA;AAUL,eAAO,MAAM,oBAAoB,GAAI,MAAM,MAAM,KAAG,OACI,CAAA;AAExD,eAAO,MAAM,wBAAwB,GAAI,MAAM,MAAM,KAAG,OAiBvD,CAAA;AAmCD,eAAO,MAAM,wBAAwB,GAAI,QAAQ,MAAM,KAAG,wBAmCzD,CAAA;AAED,eAAO,MAAM,yBAAyB,GAAI,MAAM,MAAM,KAAG,oBAsGxD,CAAA;AA4CD,eAAO,MAAM,qBAAqB,GAChC,QAAQ,SAAS,EACjB,UAAU,sBAAsB,EAChC,mBAAkB,yBAA8D,KAC/E,iBA+MF,CAAA"}