@allternit/gizzi-sdk 1.3.0

package/dist/providers/google/content-helpers.js

@@ -0,0 +1,218 @@
+/**
+ * @license
+ * Copyright 2024 Google LLC
+ * Modified for Allternit - Allternit Google AI Provider
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { AllternitGoogleAIError, AllternitGoogleAIRequestInputError } from "./errors";
+import { POSSIBLE_ROLES, } from "./types";
+export function formatSystemInstruction(input) {
+    if (input == null) {
+        return undefined;
+    }
+    else if (typeof input === "string") {
+        return { role: "system", parts: [{ text: input }] };
+    }
+    else if (input.text) {
+        return { role: "system", parts: [input] };
+    }
+    else if (input.parts) {
+        if (!input.role) {
+            return { role: "system", parts: input.parts };
+        }
+        else {
+            return input;
+        }
+    }
+    return undefined;
+}
+export function formatNewContent(request) {
+    const newParts = [];
+    if (typeof request === "string") {
+        newParts.push({ text: request });
+    }
+    else {
+        for (const partOrString of request) {
+            if (typeof partOrString === "string") {
+                newParts.push({ text: partOrString });
+            }
+            else {
+                newParts.push(partOrString);
+            }
+        }
+    }
+    return assignRoleToPartsAndValidateSendMessageRequest(newParts);
+}
+/**
+ * When multiple Part types are passed in a single Part array, we may need to assign
+ * different roles to each part.
+ */
+function assignRoleToPartsAndValidateSendMessageRequest(parts) {
+    const userContent = { role: "user", parts: [] };
+    const functionContent = { role: "function", parts: [] };
+    let hasUserContent = false;
+    let hasFunctionContent = false;
+    for (const part of parts) {
+        if ("functionResponse" in part) {
+            functionContent.parts.push(part);
+            hasFunctionContent = true;
+        }
+        else {
+            userContent.parts.push(part);
+            hasUserContent = true;
+        }
+    }
+    if (hasUserContent && hasFunctionContent) {
+        throw new AllternitGoogleAIError("Within a single message, FunctionResponse cannot be mixed with other type of part in the request for sending chat message.");
+    }
+    if (!hasUserContent && !hasFunctionContent) {
+        throw new AllternitGoogleAIError("No content is provided for sending chat message.");
+    }
+    return hasUserContent ? userContent : functionContent;
+}
+export function formatCountTokensInput(params, modelParams) {
+    let formattedGenerateContentRequest = {
+        model: modelParams?.model,
+        generationConfig: modelParams?.generationConfig,
+        safetySettings: modelParams?.safetySettings,
+        tools: modelParams?.tools,
+        toolConfig: modelParams?.toolConfig,
+        systemInstruction: modelParams?.systemInstruction,
+        cachedContent: modelParams?.cachedContent?.name,
+        contents: [],
+    };
+    const containsGenerateContentRequest = typeof params === "object" && "generateContentRequest" in params && params.generateContentRequest != null;
+    if (typeof params === "object" && "contents" in params && params.contents) {
+        if (containsGenerateContentRequest) {
+            throw new AllternitGoogleAIRequestInputError("CountTokensRequest must have one of contents or generateContentRequest, not both.");
+        }
+        formattedGenerateContentRequest.contents = params.contents;
+    }
+    else if (containsGenerateContentRequest) {
+        formattedGenerateContentRequest = {
+            ...formattedGenerateContentRequest,
+            ...params.generateContentRequest,
+        };
+    }
+    else {
+        // Array or string
+        const content = formatNewContent(params);
+        formattedGenerateContentRequest.contents = [content];
+    }
+    return { generateContentRequest: formattedGenerateContentRequest };
+}
+export function formatGenerateContentInput(params) {
+    let formattedRequest;
+    if (typeof params === "object" && "contents" in params && params.contents) {
+        formattedRequest = params;
+    }
+    else {
+        // Array or string
+        const content = formatNewContent(params);
+        formattedRequest = { contents: [content] };
+    }
+    // Handle systemInstruction if present
+    if (typeof params === "object" && "systemInstruction" in params && params.systemInstruction) {
+        formattedRequest.systemInstruction = formatSystemInstruction(params.systemInstruction);
+    }
+    return formattedRequest;
+}
+export function formatEmbedContentInput(params) {
+    if (typeof params === "string" || Array.isArray(params)) {
+        const content = formatNewContent(params);
+        return { content };
+    }
+    return params;
+}
+// https://ai.google.dev/api/rest/v1beta/Content#part
+const VALID_PART_FIELDS = [
+    "text",
+    "inlineData",
+    "functionCall",
+    "functionResponse",
+    "executableCode",
+    "codeExecutionResult",
+];
+const VALID_PARTS_PER_ROLE = {
+    user: ["text", "inlineData"],
+    function: ["functionResponse"],
+    model: ["text", "functionCall", "executableCode", "codeExecutionResult"],
+    system: ["text"],
+};
+export function validateChatHistory(history) {
+    let prevContent = false;
+    for (const currContent of history) {
+        const { role, parts } = currContent;
+        if (!prevContent && role !== "user") {
+            throw new AllternitGoogleAIError(`First content should be with role 'user', got ${role}`);
+        }
+        if (!POSSIBLE_ROLES.includes(role)) {
+            throw new AllternitGoogleAIError(`Each item should include role field. Got ${role} but valid roles are: ${JSON.stringify(POSSIBLE_ROLES)}`);
+        }
+        if (!Array.isArray(parts)) {
+            throw new AllternitGoogleAIError("Content should have 'parts' property with an array of Parts");
+        }
+        if (parts.length === 0) {
+            throw new AllternitGoogleAIError("Each Content should have at least one part");
+        }
+        const countFields = {
+            text: 0,
+            inlineData: 0,
+            functionCall: 0,
+            functionResponse: 0,
+            fileData: 0,
+            executableCode: 0,
+            codeExecutionResult: 0,
+        };
+        for (const part of parts) {
+            for (const key of VALID_PART_FIELDS) {
+                if (key in part) {
+                    countFields[key] += 1;
+                }
+            }
+        }
+        const validParts = VALID_PARTS_PER_ROLE[role];
+        for (const key of VALID_PART_FIELDS) {
+            if (!validParts.includes(key) && countFields[key] > 0) {
+                throw new AllternitGoogleAIError(`Content with role '${role}' can't contain '${key}' part`);
+            }
+        }
+        prevContent = true;
+    }
+}
+/**
+ * Returns true if the response is valid (could be appended to the history).
+ */
+export function isValidResponse(response) {
+    if (response.candidates === undefined || response.candidates.length === 0) {
+        return false;
+    }
+    const content = response.candidates[0]?.content;
+    if (content === undefined) {
+        return false;
+    }
+    if (content.parts === undefined || content.parts.length === 0) {
+        return false;
+    }
+    for (const part of content.parts) {
+        if (part == null || Object.keys(part).length === 0) {
+            return false;
+        }
+        if (typeof part === 'object' && 'text' in part && part.text === '') {
+            return false;
+        }
+    }
+    return true;
+}
+//# sourceMappingURL=content-helpers.js.map

package/dist/providers/google/content-helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"content-helpers.js","sourceRoot":"","sources":["../../../src/providers/google/content-helpers.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAEH,OAAO,EAAE,sBAAsB,EAAE,kCAAkC,EAAE,MAAM,UAAU,CAAC;AACtF,OAAO,EACL,cAAc,GAQf,MAAM,SAAS,CAAC;AAEjB,MAAM,UAAU,uBAAuB,CACrC,KAAiD;IAEjD,IAAI,KAAK,IAAI,IAAI,EAAE,CAAC;QAClB,OAAO,SAAS,CAAC;IACnB,CAAC;SAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACrC,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;IACtD,CAAC;SAAM,IAAK,KAAc,CAAC,IAAI,EAAE,CAAC;QAChC,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,KAAa,CAAC,EAAE,CAAC;IACpD,CAAC;SAAM,IAAK,KAAiB,CAAC,KAAK,EAAE,CAAC;QACpC,IAAI,CAAE,KAAiB,CAAC,IAAI,EAAE,CAAC;YAC7B,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAG,KAAiB,CAAC,KAAK,EAAE,CAAC;QAC7D,CAAC;aAAM,CAAC;YACN,OAAO,KAAgB,CAAC;QAC1B,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,OAAsC;IAEtC,MAAM,QAAQ,GAAW,EAAE,CAAC;IAC5B,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;IACnC,CAAC;SAAM,CAAC;QACN,KAAK,MAAM,YAAY,IAAI,OAAO,EAAE,CAAC;YACnC,IAAI,OAAO,YAAY,KAAK,QAAQ,EAAE,CAAC;gBACrC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;YACxC,CAAC;iBAAM,CAAC;gBACN,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,8CAA8C,CAAC,QAAQ,CAAC,CAAC;AAClE,CAAC;AAED;;;GAGG;AACH,SAAS,8CAA8C,CAAC,KAAa;IACnE,MAAM,WAAW,GAAY,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACzD,MAAM,eAAe,GAAY,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACjE,IAAI,cAAc,GAAG,KAAK,CAAC;IAC3B,IAAI,kBAAkB,GAAG,KAAK,CAAC;IAE/B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,kBAAkB,IAAI,IAAI,EAAE,CAAC;YAC/B,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,kBAAkB,GAAG,IAAI,CAAC;QAC5B,CAAC;aAAM,CAAC;YACN,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7B,cAAc,GAAG,IAAI,CAAC;QACxB,CAAC;IACH,CAAC;IAED,IAAI,cAAc,IAAI,kBAAkB,EAAE,CAAC;QACzC,MAAM,IAAI,sBAAsB,CAC9B,4HAA4H,CAC7H,CAAC;IACJ,CAAC;IACD,IAAI,CAAC,cAAc,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC3C,MAAM,IAAI,sBAAsB,CAAC,kDAAkD,CAAC,CAAC;IACvF,CAAC;IAED,OAAO,cAAc,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,eAAe,CAAC;AACxD,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,MAA0D,EAC1D,WAAyB;IAEzB,IAAI,+BAA+B,GAAoC;QACrE,KAAK,EAAE,WAAW,EAAE,KAAK;QACzB,gBAAgB,EAAE,WAAW,EAAE,gBAAgB;QAC/C,cAAc,EAAE,WAAW,EAAE,cAAc;QAC3C,KAAK,EAAE,WAAW,EAAE,KAAK;QACzB,UAAU,EAAE,WAAW,EAAE,UAAU;QACnC,iBAAiB,EAAE,WAAW,EAAE,iBAAiB;QACjD,aAAa,EAAE,WAAW,EAAE,aAAa,EAAE,IAAI;QAC/C,QAAQ,EAAE,EAAE;KACb,CAAC;IAEF,MAAM,8BAA8B,GAClC,OAAO,MAAM,KAAK,QAAQ,IAAI,wBAAwB,IAAI,MAAM,IAAI,MAAM,CAAC,sBAAsB,IAAI,IAAI,CAAC;IAE5G,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,UAAU,IAAI,MAAM,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QAC1E,IAAI,8BAA8B,EAAE,CAAC;YACnC,MAAM,IAAI,kCAAkC,CAC1C,mFAAmF,CACpF,CAAC;QACJ,CAAC;QACD,+BAA+B,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;IAC7D,CAAC;SAAM,IAAI,8BAA8B,EAAE,CAAC;QAC1C,+BAA+B,GAAG;YAChC,GAAG,+BAA+B;YAClC,GAAG,MAAM,CAAC,sBAAsB;SACjC,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,kBAAkB;QAClB,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAuC,CAAC,CAAC;QAC1E,+BAA+B,CAAC,QAAQ,GAAG,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC;IAED,OAAO,EAAE,sBAAsB,EAAE,+BAA+B,EAAE,CAAC;AACrE,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,MAA8D;IAE9D,IAAI,gBAAwC,CAAC;IAC7C,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,UAAU,IAAI,MAAM,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QAC1E,gBAAgB,GAAG,MAAM,CAAC;IAC5B,CAAC;SAAM,CAAC;QACN,kBAAkB;QAClB,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAuC,CAAC,CAAC;QAC1E,gBAAgB,GAAG,EAAE,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;IAC7C,CAAC;IAED,sCAAsC;IACtC,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,mBAAmB,IAAI,MAAM,IAAI,MAAM,CAAC,iBAAiB,EAAE,CAAC;QAC5F,gBAAgB,CAAC,iBAAiB,GAAG,uBAAuB,CAC1D,MAAM,CAAC,iBAAiB,CACzB,CAAC;IACJ,CAAC;IAED,OAAO,gBAAgB,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,MAA2D;IAE3D,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QACxD,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACzC,OAAO,EAAE,OAAO,EAAE,CAAC;IACrB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,qDAAqD;AACrD,MAAM,iBAAiB,GAAG;IACxB,MAAM;IACN,YAAY;IACZ,cAAc;IACd,kBAAkB;IAClB,gBAAgB;IAChB,qBAAqB;CACtB,CAAC;AAEF,MAAM,oBAAoB,GAA6B;IACrD,IAAI,EAAE,CAAC,MAAM,EAAE,YAAY,CAAC;IAC5B,QAAQ,EAAE,CAAC,kBAAkB,CAAC;IAC9B,KAAK,EAAE,CAAC,MAAM,EAAE,cAAc,EAAE,gBAAgB,EAAE,qBAAqB,CAAC;IACxE,MAAM,EAAE,CAAC,MAAM,CAAC;CACjB,CAAC;AAEF,MAAM,UAAU,mBAAmB,CAAC,OAAkB;IACpD,IAAI,WAAW,GAAG,KAAK,CAAC;IACxB,KAAK,MAAM,WAAW,IAAI,OAAO,EAAE,CAAC;QAClC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,WAAW,CAAC;QAEpC,IAAI,CAAC,WAAW,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;YACpC,MAAM,IAAI,sBAAsB,CAC9B,iDAAiD,IAAI,EAAE,CACxD,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,IAAW,CAAC,EAAE,CAAC;YAC1C,MAAM,IAAI,sBAAsB,CAC9B,4CAA4C,IAAI,yBAAyB,IAAI,CAAC,SAAS,CACrF,cAAc,CACf,EAAE,CACJ,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,sBAAsB,CAC9B,6DAA6D,CAC9D,CAAC;QACJ,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,MAAM,IAAI,sBAAsB,CAAC,4CAA4C,CAAC,CAAC;QACjF,CAAC;QAED,MAAM,WAAW,GAA2B;YAC1C,IAAI,EAAE,CAAC;YACP,UAAU,EAAE,CAAC;YACb,YAAY,EAAE,CAAC;YACf,gBAAgB,EAAE,CAAC;YACnB,QAAQ,EAAE,CAAC;YACX,cAAc,EAAE,CAAC;YACjB,mBAAmB,EAAE,CAAC;SACvB,CAAC;QAEF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,KAAK,MAAM,GAAG,IAAI,iBAAiB,EAAE,CAAC;gBACpC,IAAI,GAAG,IAAI,IAAI,EAAE,CAAC;oBAChB,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBACxB,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,UAAU,GAAG,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAC9C,KAAK,MAAM,GAAG,IAAI,iBAAiB,EAAE,CAAC;YACpC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBACtD,MAAM,IAAI,sBAAsB,CAC9B,sBAAsB,IAAI,oBAAoB,GAAG,QAAQ,CAC1D,CAAC;YACJ,CAAC;QACH,CAAC;QAED,WAAW,GAAG,IAAI,CAAC;IACrB,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,eAAe,CAAC,QAE/B;IACC,IAAI,QAAQ,CAAC,UAAU,KAAK,SAAS,IAAI,QAAQ,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1E,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,OAAO,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC;IAChD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9D,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QACjC,IAAI,IAAI,IAAI,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnD,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,MAAM,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,EAAE,EAAE,CAAC;YACnE,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/providers/google/errors.d.ts

@@ -0,0 +1,59 @@
+/**
+ * @license
+ * Copyright 2024 Google LLC
+ * Modified for Allternit - Allternit Google AI Provider
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import type { ErrorDetails } from "./types";
+/**
+ * Basic error type for this SDK.
+ * @public
+ */
+export declare class AllternitGoogleAIError extends Error {
+    constructor(message: string);
+}
+/**
+ * Errors in the contents of a response from the model. This includes parsing
+ * errors, or responses including a safety block reason.
+ * @public
+ */
+export declare class AllternitGoogleAIResponseError<T> extends AllternitGoogleAIError {
+    response?: T;
+    constructor(message: string, response?: T);
+}
+/**
+ * Error class covering HTTP errors when calling the server. Includes HTTP
+ * status, statusText, and optional details, if provided in the server response.
+ * @public
+ */
+export declare class AllternitGoogleAIFetchError extends AllternitGoogleAIError {
+    status?: number;
+    statusText?: string;
+    errorDetails?: ErrorDetails[];
+    constructor(message: string, status?: number, statusText?: string, errorDetails?: ErrorDetails[]);
+}
+/**
+ * Errors in the contents of a request originating from user input.
+ * @public
+ */
+export declare class AllternitGoogleAIRequestInputError extends AllternitGoogleAIError {
+}
+/**
+ * Error thrown when a request is aborted, either due to a timeout or
+ * intentional cancellation by the user.
+ * @public
+ */
+export declare class AllternitGoogleAIAbortError extends AllternitGoogleAIError {
+}
+//# sourceMappingURL=errors.d.ts.map

package/dist/providers/google/errors.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.d.ts","sourceRoot":"","sources":["../../../src/providers/google/errors.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAE5C;;;GAGG;AACH,qBAAa,sBAAuB,SAAQ,KAAK;gBACnC,OAAO,EAAE,MAAM;CAG5B;AAED;;;;GAIG;AACH,qBAAa,8BAA8B,CAAC,CAAC,CAAE,SAAQ,sBAAsB;IAC3E,QAAQ,CAAC,EAAE,CAAC,CAAC;gBACD,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,CAAC;CAI1C;AAED;;;;GAIG;AACH,qBAAa,2BAA4B,SAAQ,sBAAsB;IACrE,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,YAAY,EAAE,CAAC;gBAE5B,OAAO,EAAE,MAAM,EACf,MAAM,CAAC,EAAE,MAAM,EACf,UAAU,CAAC,EAAE,MAAM,EACnB,YAAY,CAAC,EAAE,YAAY,EAAE;CAOhC;AAED;;;GAGG;AACH,qBAAa,kCAAmC,SAAQ,sBAAsB;CAAG;AAEjF;;;;GAIG;AACH,qBAAa,2BAA4B,SAAQ,sBAAsB;CAAG"}

package/dist/providers/google/errors.js

@@ -0,0 +1,68 @@
+/**
+ * @license
+ * Copyright 2024 Google LLC
+ * Modified for Allternit - Allternit Google AI Provider
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+/**
+ * Basic error type for this SDK.
+ * @public
+ */
+export class AllternitGoogleAIError extends Error {
+    constructor(message) {
+        super(`[AllternitGoogleAI Error]: ${message}`);
+    }
+}
+/**
+ * Errors in the contents of a response from the model. This includes parsing
+ * errors, or responses including a safety block reason.
+ * @public
+ */
+export class AllternitGoogleAIResponseError extends AllternitGoogleAIError {
+    response;
+    constructor(message, response) {
+        super(message);
+        this.response = response;
+    }
+}
+/**
+ * Error class covering HTTP errors when calling the server. Includes HTTP
+ * status, statusText, and optional details, if provided in the server response.
+ * @public
+ */
+export class AllternitGoogleAIFetchError extends AllternitGoogleAIError {
+    status;
+    statusText;
+    errorDetails;
+    constructor(message, status, statusText, errorDetails) {
+        super(message);
+        this.status = status;
+        this.statusText = statusText;
+        this.errorDetails = errorDetails;
+    }
+}
+/**
+ * Errors in the contents of a request originating from user input.
+ * @public
+ */
+export class AllternitGoogleAIRequestInputError extends AllternitGoogleAIError {
+}
+/**
+ * Error thrown when a request is aborted, either due to a timeout or
+ * intentional cancellation by the user.
+ * @public
+ */
+export class AllternitGoogleAIAbortError extends AllternitGoogleAIError {
+}
+//# sourceMappingURL=errors.js.map

package/dist/providers/google/errors.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.js","sourceRoot":"","sources":["../../../src/providers/google/errors.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAIH;;;GAGG;AACH,MAAM,OAAO,sBAAuB,SAAQ,KAAK;IAC/C,YAAY,OAAe;QACzB,KAAK,CAAC,8BAA8B,OAAO,EAAE,CAAC,CAAC;IACjD,CAAC;CACF;AAED;;;;GAIG;AACH,MAAM,OAAO,8BAAkC,SAAQ,sBAAsB;IAC3E,QAAQ,CAAK;IACb,YAAY,OAAe,EAAE,QAAY;QACvC,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;CACF;AAED;;;;GAIG;AACH,MAAM,OAAO,2BAA4B,SAAQ,sBAAsB;IACrE,MAAM,CAAU;IAChB,UAAU,CAAU;IACpB,YAAY,CAAkB;IAC9B,YACE,OAAe,EACf,MAAe,EACf,UAAmB,EACnB,YAA6B;QAE7B,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;IACnC,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,kCAAmC,SAAQ,sBAAsB;CAAG;AAEjF;;;;GAIG;AACH,MAAM,OAAO,2BAA4B,SAAQ,sBAAsB;CAAG"}

package/dist/providers/google/gen-ai.d.ts

@@ -0,0 +1,38 @@
+/**
+ * @license
+ * Copyright 2024 Google LLC
+ * Modified for Allternit - Allternit Google AI Provider
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { AllternitGenerativeModel } from "./models/generative-model";
+import type { CachedContent, ModelParams, RequestOptions } from "./types";
+export { ChatSession } from "./methods/chat-session";
+export { AllternitGenerativeModel };
+/**
+ * Top-level class for the Allternit Google AI SDK
+ * @public
+ */
+export declare class AllternitGoogleAI {
+    apiKey: string;
+    constructor(apiKey: string);
+    /**
+     * Gets an AllternitGenerativeModel instance for the provided model name.
+     */
+    getGenerativeModel(modelParams: ModelParams, requestOptions?: RequestOptions): AllternitGenerativeModel;
+    /**
+     * Creates an AllternitGenerativeModel instance from provided content cache.
+     */
+    getGenerativeModelFromCachedContent(cachedContent: CachedContent, modelParams?: Partial<ModelParams>, requestOptions?: RequestOptions): AllternitGenerativeModel;
+}
+//# sourceMappingURL=gen-ai.d.ts.map

package/dist/providers/google/gen-ai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gen-ai.d.ts","sourceRoot":"","sources":["../../../src/providers/google/gen-ai.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAEH,OAAO,EAAE,wBAAwB,EAAE,MAAM,2BAA2B,CAAC;AAErE,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AAE1E,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACrD,OAAO,EAAE,wBAAwB,EAAE,CAAC;AAEpC;;;GAGG;AACH,qBAAa,iBAAiB;IACT,MAAM,EAAE,MAAM;gBAAd,MAAM,EAAE,MAAM;IAEjC;;OAEG;IACH,kBAAkB,CAChB,WAAW,EAAE,WAAW,EACxB,cAAc,CAAC,EAAE,cAAc,GAC9B,wBAAwB;IAU3B;;OAEG;IACH,mCAAmC,CACjC,aAAa,EAAE,aAAa,EAC5B,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,EAClC,cAAc,CAAC,EAAE,cAAc,GAC9B,wBAAwB;CAyD5B"}

package/dist/providers/google/gen-ai.js

@@ -0,0 +1,83 @@
+/**
+ * @license
+ * Copyright 2024 Google LLC
+ * Modified for Allternit - Allternit Google AI Provider
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import { AllternitGenerativeModel } from "./models/generative-model";
+import { AllternitGoogleAIError, AllternitGoogleAIRequestInputError } from "./errors";
+export { ChatSession } from "./methods/chat-session";
+export { AllternitGenerativeModel };
+/**
+ * Top-level class for the Allternit Google AI SDK
+ * @public
+ */
+export class AllternitGoogleAI {
+    apiKey;
+    constructor(apiKey) {
+        this.apiKey = apiKey;
+    }
+    /**
+     * Gets an AllternitGenerativeModel instance for the provided model name.
+     */
+    getGenerativeModel(modelParams, requestOptions) {
+        if (!modelParams.model) {
+            throw new AllternitGoogleAIError(`Must provide a model name. ` +
+                `Example: genai.getGenerativeModel({ model: 'gemini-1.5-flash' })`);
+        }
+        return new AllternitGenerativeModel(this.apiKey, modelParams, requestOptions);
+    }
+    /**
+     * Creates an AllternitGenerativeModel instance from provided content cache.
+     */
+    getGenerativeModelFromCachedContent(cachedContent, modelParams, requestOptions) {
+        if (!cachedContent.name) {
+            throw new AllternitGoogleAIRequestInputError("Cached content must contain a `name` field.");
+        }
+        if (!cachedContent.model) {
+            throw new AllternitGoogleAIRequestInputError("Cached content must contain a `model` field.");
+        }
+        const disallowedDuplicates = ["model", "systemInstruction"];
+        for (const key of disallowedDuplicates) {
+            if (modelParams?.[key] &&
+                cachedContent[key] &&
+                modelParams[key] !==
+                    cachedContent[key]) {
+                if (key === "model") {
+                    const modelParamsComp = modelParams.model.startsWith("models/")
+                        ? modelParams.model.replace("models/", "")
+                        : modelParams.model;
+                    const cachedContentComp = cachedContent.model.startsWith("models/")
+                        ? cachedContent.model.replace("models/", "")
+                        : cachedContent.model;
+                    if (modelParamsComp === cachedContentComp) {
+                        continue;
+                    }
+                }
+                throw new AllternitGoogleAIRequestInputError(`Different value for "${key}" specified in modelParams` +
+                    ` (${modelParams[key]}) and cachedContent (${cachedContent[key]})`);
+            }
+        }
+        const modelParamsFromCache = {
+            ...modelParams,
+            model: cachedContent.model,
+            tools: cachedContent.tools,
+            toolConfig: cachedContent.toolConfig,
+            systemInstruction: cachedContent.systemInstruction,
+            cachedContent,
+        };
+        return new AllternitGenerativeModel(this.apiKey, modelParamsFromCache, requestOptions);
+    }
+}
+//# sourceMappingURL=gen-ai.js.map

package/dist/providers/google/gen-ai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gen-ai.js","sourceRoot":"","sources":["../../../src/providers/google/gen-ai.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAEH,OAAO,EAAE,wBAAwB,EAAE,MAAM,2BAA2B,CAAC;AACrE,OAAO,EAAE,sBAAsB,EAAE,kCAAkC,EAAE,MAAM,UAAU,CAAC;AAGtF,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACrD,OAAO,EAAE,wBAAwB,EAAE,CAAC;AAEpC;;;GAGG;AACH,MAAM,OAAO,iBAAiB;IACT;IAAnB,YAAmB,MAAc;QAAd,WAAM,GAAN,MAAM,CAAQ;IAAG,CAAC;IAErC;;OAEG;IACH,kBAAkB,CAChB,WAAwB,EACxB,cAA+B;QAE/B,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;YACvB,MAAM,IAAI,sBAAsB,CAC9B,6BAA6B;gBAC3B,kEAAkE,CACrE,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,wBAAwB,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;IAChF,CAAC;IAED;;OAEG;IACH,mCAAmC,CACjC,aAA4B,EAC5B,WAAkC,EAClC,cAA+B;QAE/B,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;YACxB,MAAM,IAAI,kCAAkC,CAC1C,6CAA6C,CAC9C,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;YACzB,MAAM,IAAI,kCAAkC,CAC1C,8CAA8C,CAC/C,CAAC;QACJ,CAAC;QAED,MAAM,oBAAoB,GAAG,CAAC,OAAO,EAAE,mBAAmB,CAAC,CAAC;QAC5D,KAAK,MAAM,GAAG,IAAI,oBAAoB,EAAE,CAAC;YACvC,IACE,WAAW,EAAE,CAAC,GAAwB,CAAC;gBACvC,aAAa,CAAC,GAA0B,CAAC;gBACzC,WAAW,CAAC,GAAwB,CAAC;oBACnC,aAAa,CAAC,GAA0B,CAAC,EAC3C,CAAC;gBACD,IAAI,GAAG,KAAK,OAAO,EAAE,CAAC;oBACpB,MAAM,eAAe,GAClB,WAAW,CAAC,KAAgB,CAAC,UAAU,CAAC,SAAS,CAAC;wBACjD,CAAC,CAAE,WAAW,CAAC,KAAgB,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;wBACtD,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC;oBACxB,MAAM,iBAAiB,GACpB,aAAa,CAAC,KAAgB,CAAC,UAAU,CAAC,SAAS,CAAC;wBACnD,CAAC,CAAE,aAAa,CAAC,KAAgB,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;wBACxD,CAAC,CAAC,aAAa,CAAC,KAAK,CAAC;oBAC1B,IAAI,eAAe,KAAK,iBAAiB,EAAE,CAAC;wBAC1C,SAAS;oBACX,CAAC;gBACH,CAAC;gBACD,MAAM,IAAI,kCAAkC,CAC1C,wBAAwB,GAAG,4BAA4B;oBACrD,KAAK,WAAW,CAAC,GAAwB,CAAC,wBACxC,aAAa,CAAC,GAA0B,CAC1C,GAAG,CACN,CAAC;YACJ,CAAC;QACH,CAAC;QAED,MAAM,oBAAoB,GAAgB;YACxC,GAAI,WAA2B;YAC/B,KAAK,EAAE,aAAa,CAAC,KAAK;YAC1B,KAAK,EAAE,aAAa,CAAC,KAAK;YAC1B,UAAU,EAAE,aAAa,CAAC,UAAU;YACpC,iBAAiB,EAAE,aAAa,CAAC,iBAAiB;YAClD,aAAa;SACd,CAAC;QAEF,OAAO,IAAI,wBAAwB,CACjC,IAAI,CAAC,MAAM,EACX,oBAAoB,EACpB,cAAc,CACf,CAAC;IACJ,CAAC;CACF"}

package/dist/providers/google/index.d.ts

@@ -0,0 +1,99 @@
+/**
+ * @license
+ * Copyright 2024 Google LLC
+ * Modified for Allternit - Allternit Google AI Provider
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+/**
+ * Allternit Google AI Provider
+ *
+ * A rebranded fork of the Google GenAI SDK (@google/generative-ai)
+ * for use within the Allternit platform.
+ *
+ * @packageDocumentation
+ */
+export * from "./types";
+export { AllternitGoogleAIError, AllternitGoogleAIResponseError, AllternitGoogleAIFetchError, AllternitGoogleAIRequestInputError, AllternitGoogleAIAbortError, } from "./errors";
+export { AllternitGoogleAI, AllternitGenerativeModel, ChatSession } from "./gen-ai";
+/**
+ * Gemini 1.5 Flash - Fast and efficient multimodal model
+ * @public
+ */
+export declare const GEMINI_1_5_FLASH = "gemini-1.5-flash";
+/**
+ * Gemini 1.5 Flash Latest - Latest version of Flash
+ * @public
+ */
+export declare const GEMINI_1_5_FLASH_LATEST = "gemini-1.5-flash-latest";
+/**
+ * Gemini 1.5 Pro - Most capable multimodal model
+ * @public
+ */
+export declare const GEMINI_1_5_PRO = "gemini-1.5-pro";
+/**
+ * Gemini 1.5 Pro Latest - Latest version of Pro
+ * @public
+ */
+export declare const GEMINI_1_5_PRO_LATEST = "gemini-1.5-pro-latest";
+/**
+ * Gemini 1.0 Pro - Original Pro model
+ * @public
+ */
+export declare const GEMINI_1_0_PRO = "gemini-1.0-pro";
+/**
+ * Gemini 1.0 Pro Latest - Latest version of 1.0 Pro
+ * @public
+ */
+export declare const GEMINI_1_0_PRO_LATEST = "gemini-1.0-pro-latest";
+/**
+ * Gemini 1.0 Pro Vision - Vision-capable model
+ * @public
+ */
+export declare const GEMINI_1_0_PRO_VISION = "gemini-1.0-pro-vision";
+/**
+ * Gemini 1.0 Pro Vision Latest - Latest version of Pro Vision
+ * @public
+ */
+export declare const GEMINI_1_0_PRO_VISION_LATEST = "gemini-1.0-pro-vision-latest";
+/**
+ * Text Embedding 004 - Latest text embedding model
+ * @public
+ */
+export declare const TEXT_EMBEDDING_004 = "text-embedding-004";
+/**
+ * Embedding 001 - Original embedding model
+ * @public
+ */
+export declare const EMBEDDING_001 = "embedding-001";
+/**
+ * AQA (Attributed Question Answering) model
+ * @public
+ */
+export declare const AQA_MODEL = "aqa";
+/**
+ * Type for valid Gemini model names
+ * @public
+ */
+export type GeminiModelName = "gemini-1.5-flash" | "gemini-1.5-flash-latest" | "gemini-1.5-pro" | "gemini-1.5-pro-latest" | "gemini-1.0-pro" | "gemini-1.0-pro-latest" | "gemini-1.0-pro-vision" | "gemini-1.0-pro-vision-latest" | "text-embedding-004" | "embedding-001" | "aqa";
+/**
+ * SDK Version
+ * @public
+ */
+export declare const VERSION = "0.1.0";
+/**
+ * Provider name
+ * @public
+ */
+export declare const PROVIDER_NAME = "allternit-google-ai";
+//# sourceMappingURL=index.d.ts.map

package/dist/providers/google/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/providers/google/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAEH;;;;;;;GAOG;AAGH,cAAc,SAAS,CAAC;AAGxB,OAAO,EACL,sBAAsB,EACtB,8BAA8B,EAC9B,2BAA2B,EAC3B,kCAAkC,EAClC,2BAA2B,GAC5B,MAAM,UAAU,CAAC;AAGlB,OAAO,EAAE,iBAAiB,EAAE,wBAAwB,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAMpF;;;GAGG;AACH,eAAO,MAAM,gBAAgB,qBAAqB,CAAC;AAEnD;;;GAGG;AACH,eAAO,MAAM,uBAAuB,4BAA4B,CAAC;AAEjE;;;GAGG;AACH,eAAO,MAAM,cAAc,mBAAmB,CAAC;AAE/C;;;GAGG;AACH,eAAO,MAAM,qBAAqB,0BAA0B,CAAC;AAE7D;;;GAGG;AACH,eAAO,MAAM,cAAc,mBAAmB,CAAC;AAE/C;;;GAGG;AACH,eAAO,MAAM,qBAAqB,0BAA0B,CAAC;AAE7D;;;GAGG;AACH,eAAO,MAAM,qBAAqB,0BAA0B,CAAC;AAE7D;;;GAGG;AACH,eAAO,MAAM,4BAA4B,iCAAiC,CAAC;AAE3E;;;GAGG;AACH,eAAO,MAAM,kBAAkB,uBAAuB,CAAC;AAEvD;;;GAGG;AACH,eAAO,MAAM,aAAa,kBAAkB,CAAC;AAE7C;;;GAGG;AACH,eAAO,MAAM,SAAS,QAAQ,CAAC;AAM/B;;;GAGG;AACH,MAAM,MAAM,eAAe,GACvB,kBAAkB,GAClB,yBAAyB,GACzB,gBAAgB,GAChB,uBAAuB,GACvB,gBAAgB,GAChB,uBAAuB,GACvB,uBAAuB,GACvB,8BAA8B,GAC9B,oBAAoB,GACpB,eAAe,GACf,KAAK,CAAC;AAMV;;;GAGG;AACH,eAAO,MAAM,OAAO,UAAU,CAAC;AAE/B;;;GAGG;AACH,eAAO,MAAM,aAAa,wBAAwB,CAAC"}