@avalant/piece-promptx-gemini 0.0.1

package/README.md

@@ -0,0 +1,7 @@
+# pieces-promptx-gemini
+
+This library was generated with [Nx](https://nx.dev).
+
+## Building
+
+Run `nx build pieces-promptx-gemini` to build the library.

package/package.json

@@ -0,0 +1,39 @@
+{
+  "name": "@avalant/piece-promptx-gemini",
+  "version": "0.0.1",
+  "type": "commonjs",
+  "main": "./src/index.js",
+  "types": "./src/index.d.ts",
+  "dependencies": {
+    "@ai-sdk/anthropic": "1.2.12",
+    "@ai-sdk/google": "1.2.19",
+    "@ai-sdk/openai": "1.3.22",
+    "@ai-sdk/replicate": "0.2.8",
+    "@google/generative-ai": "0.21.0",
+    "@sinclair/typebox": "0.34.11",
+    "ai": "4.3.16",
+    "axios": "1.8.3",
+    "axios-retry": "4.4.1",
+    "deepmerge-ts": "7.1.0",
+    "fast-glob": "3.3.3",
+    "mime-types": "2.1.35",
+    "nanoid": "3.3.8",
+    "semver": "7.6.0",
+    "zod": "3.25.76",
+    "@activepieces/pieces-common": "0.6.0",
+    "@activepieces/pieces-framework": "0.15.0",
+    "@activepieces/shared": "0.17.2",
+    "tslib": "2.8.1"
+  },
+  "overrides": {
+    "@tryfabric/martian": {
+      "@notionhq/client": "$@notionhq/client"
+    },
+    "vite": {
+      "rollup": "npm:@rollup/wasm-node"
+    }
+  },
+  "resolutions": {
+    "rollup": "npm:@rollup/wasm-node"
+  }
+}

package/src/i18n/translation.json

@@ -0,0 +1,34 @@
+{
+  "Google Gemini": "Google Gemini",
+  "Talk to Google Gemini using your available PromptX credits. Use the many tools Gemini has to offer using your PromptX credits per request": "Talk to Google Gemini using your available PromptX credits. Use the many tools Gemini has to offer using your PromptX credits per request",
+  "\nFollow these instructions to get your API Key:\n1. Visit the following website: https://makersuite.google.com/app/apikey\n2. Once on the website, locate and click on the option to obtain your API Key.\nPlease note this piece uses a API in the beta phase that may change at any time.\n": "\nFollow these instructions to get your API Key:\n1. Visit the following website: https://makersuite.google.com/app/apikey\n2. Once on the website, locate and click on the option to obtain your API Key.\nPlease note this piece uses a API in the beta phase that may change at any time.\n",
+  "Generate Content": "Generate Content",
+  "Generate Content from Image": "Generate Content from Image",
+  "Chat Gemini": "Chat Gemini",
+  "Custom API Call": "Custom API Call",
+  "Generate content using Google Gemini using the \"gemini-pro\" model": "Generate content using Google Gemini using the \"gemini-pro\" model",
+  "Generate content using Google Gemini using the \"gemini-pro-vision\" model": "Generate content using Google Gemini using the \"gemini-pro-vision\" model",
+  "Chat with Google Gemini": "Chat with Google Gemini",
+  "Make a custom API call to a specific endpoint": "Make a custom API call to a specific endpoint",
+  "Prompt": "Prompt",
+  "Model": "Model",
+  "Image": "Image",
+  "Memory Key": "Memory Key",
+  "Method": "Method",
+  "Headers": "Headers",
+  "Query Parameters": "Query Parameters",
+  "Body": "Body",
+  "No Error on Failure": "No Error on Failure",
+  "Timeout (in seconds)": "Timeout (in seconds)",
+  "The prompt to generate content from.": "The prompt to generate content from.",
+  "The model which will generate the completion": "The model which will generate the completion",
+  "The image to generate content from.": "The image to generate content from.",
+  "A memory key that will keep the chat history. Keep it empty to leave Gemini without memory of previous messages.": "A memory key that will keep the chat history. Keep it empty to leave Gemini without memory of previous messages.",
+  "Authorization headers are injected automatically from your connection.": "Authorization headers are injected automatically from your connection.",
+  "GET": "GET",
+  "POST": "POST",
+  "PATCH": "PATCH",
+  "PUT": "PUT",
+  "DELETE": "DELETE",
+  "HEAD": "HEAD"
+}

package/src/index.d.ts

@@ -0,0 +1,10 @@
+export declare const googleGeminiAuth: import("@activepieces/pieces-framework").CustomAuthProperty<{
+    server: import("@activepieces/pieces-framework").StaticDropdownProperty<string, true>;
+    username: import("@activepieces/pieces-framework").ShortTextProperty<true>;
+    password: import("@activepieces/pieces-framework").SecretTextProperty<true>;
+}>;
+export declare const googleGemini: import("@activepieces/pieces-framework").Piece<import("@activepieces/pieces-framework").CustomAuthProperty<{
+    server: import("@activepieces/pieces-framework").StaticDropdownProperty<string, true>;
+    username: import("@activepieces/pieces-framework").ShortTextProperty<true>;
+    password: import("@activepieces/pieces-framework").SecretTextProperty<true>;
+}>>;

package/src/index.js

@@ -0,0 +1,90 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.googleGemini = exports.googleGeminiAuth = void 0;
+const tslib_1 = require("tslib");
+const pieces_framework_1 = require("@activepieces/pieces-framework");
+const shared_1 = require("@activepieces/shared");
+const chat_gemini_action_1 = require("./lib/actions/chat-gemini.action");
+const generate_content_from_image_action_1 = require("./lib/actions/generate-content-from-image.action");
+const generate_content_action_1 = require("./lib/actions/generate-content.action");
+const pmtx_api_1 = require("./lib/common/pmtx-api");
+exports.googleGeminiAuth = pieces_framework_1.PieceAuth.CustomAuth({
+    required: true,
+    props: {
+        server: pieces_framework_1.Property.StaticDropdown({
+            displayName: 'Server',
+            options: {
+                options: [
+                    {
+                        label: 'Production',
+                        value: 'production',
+                    },
+                    {
+                        label: 'Test',
+                        value: 'staging',
+                    },
+                ],
+            },
+            required: true,
+            defaultValue: 'production',
+        }),
+        username: pieces_framework_1.Property.ShortText({
+            displayName: 'Username',
+            description: 'PromptX username',
+            required: true,
+        }),
+        password: pieces_framework_1.PieceAuth.SecretText({
+            displayName: 'Password',
+            description: 'PromptX password',
+            required: true,
+        }),
+    },
+    validate: (_a) => tslib_1.__awaiter(void 0, [_a], void 0, function* ({ auth }) {
+        const { username, password } = auth;
+        if (!username || !password) {
+            return {
+                valid: false,
+                error: 'Empty Username or Password',
+            };
+        }
+        const loginUrl = pmtx_api_1.baseUrlMap[auth.server].loginUrl;
+        const isStaging = auth.server === 'staging';
+        const body = isStaging
+            ? new URLSearchParams({ username, password }).toString()
+            : JSON.stringify({ username, password });
+        const headers = {
+            'Content-Type': isStaging
+                ? 'application/x-www-form-urlencoded'
+                : 'application/json',
+        };
+        const response = yield fetch(loginUrl, {
+            method: 'POST',
+            body,
+            headers,
+        });
+        if (response.status === 200) {
+            return {
+                valid: true,
+            };
+        }
+        else {
+            const data = yield response.json();
+            return {
+                valid: false,
+                error: (data === null || data === void 0 ? void 0 : data.error) || (data === null || data === void 0 ? void 0 : data.message),
+            };
+        }
+    }),
+});
+exports.googleGemini = (0, pieces_framework_1.createPiece)({
+    displayName: 'PromptX Google Gemini',
+    auth: exports.googleGeminiAuth,
+    description: 'Talk to Google Gemini using your available PromptX credits. Use the many tools Gemini has to offer using your PromptX credits per request',
+    minimumSupportedRelease: '0.63.0',
+    logoUrl: 'https://cdn.activepieces.com/pieces/google-gemini.png',
+    categories: [shared_1.PieceCategory.ARTIFICIAL_INTELLIGENCE],
+    authors: ['tumrabert'],
+    actions: [generate_content_action_1.generateContentAction, generate_content_from_image_action_1.generateContentFromImageAction, chat_gemini_action_1.chatGemini],
+    triggers: [],
+});
+//# sourceMappingURL=index.js.map

package/src/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../../packages/pieces/community/promptx-gemini/src/index.ts"],"names":[],"mappings":";;;;AAAA,qEAIwC;AACxC,iDAAqD;AACrD,yEAA8D;AAC9D,yGAAkG;AAClG,mFAA8E;AAC9E,oDAAmD;AAEtC,QAAA,gBAAgB,GAAG,4BAAS,CAAC,UAAU,CAAC;IACnD,QAAQ,EAAE,IAAI;IACd,KAAK,EAAE;QACL,MAAM,EAAE,2BAAQ,CAAC,cAAc,CAAC;YAC9B,WAAW,EAAE,QAAQ;YACrB,OAAO,EAAE;gBACP,OAAO,EAAE;oBACP;wBACE,KAAK,EAAE,YAAY;wBACnB,KAAK,EAAE,YAAY;qBACpB;oBACD;wBACE,KAAK,EAAE,MAAM;wBACb,KAAK,EAAE,SAAS;qBACjB;iBACF;aACF;YACD,QAAQ,EAAE,IAAI;YACd,YAAY,EAAE,YAAY;SAC3B,CAAC;QACF,QAAQ,EAAE,2BAAQ,CAAC,SAAS,CAAC;YAC3B,WAAW,EAAE,UAAU;YACvB,WAAW,EAAE,kBAAkB;YAC/B,QAAQ,EAAE,IAAI;SACf,CAAC;QACF,QAAQ,EAAE,4BAAS,CAAC,UAAU,CAAC;YAC7B,WAAW,EAAE,UAAU;YACvB,WAAW,EAAE,kBAAkB;YAC/B,QAAQ,EAAE,IAAI;SACf,CAAC;KACH;IACD,QAAQ,EAAE,KAAiB,EAAE,oDAAZ,EAAE,IAAI,EAAE;QACvB,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC;QACpC,IAAI,CAAC,QAAQ,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC3B,OAAO;gBACL,KAAK,EAAE,KAAK;gBACZ,KAAK,EAAE,4BAA4B;aACpC,CAAC;QACJ,CAAC;QAED,MAAM,QAAQ,GAAG,qBAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC;QAClD,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC;QAC5C,MAAM,IAAI,GAAG,SAAS;YACpB,CAAC,CAAC,IAAI,eAAe,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE;YACxD,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC3C,MAAM,OAAO,GAAG;YACd,cAAc,EAAE,SAAS;gBACvB,CAAC,CAAC,mCAAmC;gBACrC,CAAC,CAAC,kBAAkB;SACvB,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE;YACrC,MAAM,EAAE,MAAM;YACd,IAAI;YACJ,OAAO;SACR,CAAC,CAAC;QAEH,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YAC5B,OAAO;gBACL,KAAK,EAAE,IAAI;aACZ,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,OAAO;gBACL,KAAK,EAAE,KAAK;gBACZ,KAAK,EAAE,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,KAAK,MAAI,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,OAAO,CAAA;aACpC,CAAC;QACJ,CAAC;IACH,CAAC,CAAA;CACF,CAAC,CAAC;AACU,QAAA,YAAY,GAAG,IAAA,8BAAW,EAAC;IACtC,WAAW,EAAE,uBAAuB;IACpC,IAAI,EAAE,wBAAgB;IACtB,WAAW,EAAE,2IAA2I;IACxJ,uBAAuB,EAAE,QAAQ;IACjC,OAAO,EAAE,uDAAuD;IAChE,UAAU,EAAE,CAAC,sBAAa,CAAC,uBAAuB,CAAC;IACnD,OAAO,EAAE,CAAC,WAAW,CAAC;IACtB,OAAO,EAAE,CAAC,+CAAqB,EAAE,mEAA8B,EAAE,+BAAU,CAAC;IAC5E,QAAQ,EAAE,EAAE;CACb,CAAC,CAAC"}

package/src/lib/actions/chat-gemini.action.d.ts

@@ -0,0 +1,9 @@
+export declare const chatGemini: import("@activepieces/pieces-framework").IAction<import("@activepieces/pieces-framework").CustomAuthProperty<{
+    server: import("@activepieces/pieces-framework").StaticDropdownProperty<string, true>;
+    username: import("@activepieces/pieces-framework").ShortTextProperty<true>;
+    password: import("@activepieces/pieces-framework").SecretTextProperty<true>;
+}>, {
+    model: import("@activepieces/pieces-framework").DropdownProperty<string, true>;
+    prompt: import("@activepieces/pieces-framework").LongTextProperty<true>;
+    memoryKey: import("@activepieces/pieces-framework").ShortTextProperty<false>;
+}>;

package/src/lib/actions/chat-gemini.action.js

@@ -0,0 +1,83 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.chatGemini = void 0;
+const tslib_1 = require("tslib");
+const generative_ai_1 = require("@google/generative-ai");
+const pieces_framework_1 = require("@activepieces/pieces-framework");
+const zod_1 = require("zod");
+const index_1 = require("../../index");
+const common_1 = require("../common/common");
+const pieces_common_1 = require("@activepieces/pieces-common");
+const pmtx_api_1 = require("../common/pmtx-api");
+exports.chatGemini = (0, pieces_framework_1.createAction)({
+    auth: index_1.googleGeminiAuth,
+    name: 'chat_gemini',
+    displayName: 'Chat Gemini',
+    description: 'Chat with Google Gemini',
+    props: {
+        model: pieces_framework_1.Property.Dropdown({
+            displayName: 'Model',
+            required: true,
+            description: 'The model which will generate the completion',
+            refreshers: [],
+            defaultValue: common_1.defaultLLM,
+            options: (_a) => tslib_1.__awaiter(void 0, [_a], void 0, function* ({ auth }) {
+                let geminiKey;
+                try {
+                    geminiKey = String(yield (0, pmtx_api_1.getApiKeyFormAuth)(auth));
+                }
+                catch (error) {
+                    console.error(error);
+                    return {
+                        disabled: true,
+                        placeholder: 'Unable to fetch OpenAI key. Check connection',
+                        options: [],
+                    };
+                }
+                return (0, common_1.getGeminiModelOptions)({ auth: geminiKey });
+            }),
+        }),
+        prompt: pieces_framework_1.Property.LongText({
+            displayName: 'Prompt',
+            required: true,
+            description: 'The prompt to generate content from.',
+        }),
+        memoryKey: pieces_framework_1.Property.ShortText({
+            displayName: 'Memory Key',
+            description: 'A memory key that will keep the chat history. Keep it empty to leave Gemini without memory of previous messages.',
+            required: false,
+        }),
+    },
+    run(_a) {
+        return tslib_1.__awaiter(this, arguments, void 0, function* ({ auth, propsValue, store }) {
+            yield pieces_common_1.propsValidation.validateZod(propsValue, {
+                memoryKey: zod_1.z.string().max(128).optional(),
+            });
+            const { model, prompt, memoryKey } = propsValue;
+            const geminiKey = yield (0, pmtx_api_1.getApiKeyFormAuth)(auth);
+            const genAI = new generative_ai_1.GoogleGenerativeAI(geminiKey);
+            const geminiModel = genAI.getGenerativeModel({ model });
+            let history = [];
+            if (memoryKey) {
+                const storedHistory = yield store.get(memoryKey, pieces_framework_1.StoreScope.PROJECT);
+                if (Array.isArray(storedHistory)) {
+                    history = storedHistory;
+                }
+            }
+            const chat = geminiModel.startChat({
+                history: history,
+            });
+            const result = yield chat.sendMessage(prompt);
+            const responseText = result.response.text();
+            if (memoryKey) {
+                const updatedHistory = yield chat.getHistory();
+                yield store.put(memoryKey, updatedHistory, pieces_framework_1.StoreScope.PROJECT);
+            }
+            return {
+                response: responseText,
+                history: history,
+            };
+        });
+    },
+});
+//# sourceMappingURL=chat-gemini.action.js.map

package/src/lib/actions/chat-gemini.action.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-gemini.action.js","sourceRoot":"","sources":["../../../../../../../../packages/pieces/community/promptx-gemini/src/lib/actions/chat-gemini.action.ts"],"names":[],"mappings":";;;;AAAA,yDAAoE;AACpE,qEAIwC;AACxC,6BAAwB;AACxB,uCAA+C;AAC/C,6CAAqE;AACrE,+DAA8D;AAC9D,iDAAwE;AAE3D,QAAA,UAAU,GAAG,IAAA,+BAAY,EAAC;IACrC,IAAI,EAAE,wBAAgB;IACtB,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE,aAAa;IAC1B,WAAW,EAAE,yBAAyB;IACtC,KAAK,EAAE;QACL,KAAK,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACvB,WAAW,EAAE,OAAO;YACpB,QAAQ,EAAE,IAAI;YACd,WAAW,EAAE,8CAA8C;YAC3D,UAAU,EAAE,EAAE;YACd,YAAY,EAAE,mBAAU;YACxB,OAAO,EAAE,KAAiB,EAAE,oDAAZ,EAAE,IAAI,EAAE;gBACtB,IAAI,SAAiB,CAAC;gBACtB,IAAI,CAAC;oBACH,SAAS,GAAG,MAAM,CAAC,MAAM,IAAA,4BAAiB,EAAC,IAAuB,CAAC,CAAC,CAAC;gBACvE,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBACrB,OAAO;wBACL,QAAQ,EAAE,IAAI;wBACd,WAAW,EAAE,8CAA8C;wBAC3D,OAAO,EAAE,EAAE;qBACZ,CAAC;gBACJ,CAAC;gBAED,OAAO,IAAA,8BAAqB,EAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;YACpD,CAAC,CAAA;SACF,CAAC;QACF,MAAM,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACxB,WAAW,EAAE,QAAQ;YACrB,QAAQ,EAAE,IAAI;YACd,WAAW,EAAE,sCAAsC;SACpD,CAAC;QACF,SAAS,EAAE,2BAAQ,CAAC,SAAS,CAAC;YAC5B,WAAW,EAAE,YAAY;YACzB,WAAW,EACT,kHAAkH;YACpH,QAAQ,EAAE,KAAK;SAChB,CAAC;KACH;IACK,GAAG;qEAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE;YACnC,MAAM,+BAAe,CAAC,WAAW,CAAC,UAAU,EAAE;gBAC5C,SAAS,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE;aAC1C,CAAC,CAAC;YAEH,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,UAAU,CAAC;YAChD,MAAM,SAAS,GAAW,MAAM,IAAA,4BAAiB,EAAC,IAAuB,CAAC,CAAC;YAC3E,MAAM,KAAK,GAAG,IAAI,kCAAkB,CAAC,SAAS,CAAC,CAAC;YAChD,MAAM,WAAW,GAAG,KAAK,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;YACxD,IAAI,OAAO,GAAc,EAAE,CAAC;YAE5B,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,aAAa,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,6BAAU,CAAC,OAAO,CAAC,CAAC;gBACrE,IAAI,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;oBACjC,OAAO,GAAG,aAAa,CAAC;gBAC1B,CAAC;YACH,CAAC;YAED,MAAM,IAAI,GAAG,WAAW,CAAC,SAAS,CAAC;gBACjC,OAAO,EAAE,OAAO;aACjB,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YAC9C,MAAM,YAAY,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YAE5C,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC/C,MAAM,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,cAAc,EAAE,6BAAU,CAAC,OAAO,CAAC,CAAC;YACjE,CAAC;YAED,OAAO;gBACL,QAAQ,EAAE,YAAY;gBACtB,OAAO,EAAE,OAAO;aACjB,CAAC;QACJ,CAAC;KAAA;CACF,CAAC,CAAC"}

package/src/lib/actions/generate-content-from-image.action.d.ts

@@ -0,0 +1,9 @@
+export declare const generateContentFromImageAction: import("@activepieces/pieces-framework").IAction<import("@activepieces/pieces-framework").CustomAuthProperty<{
+    server: import("@activepieces/pieces-framework").StaticDropdownProperty<string, true>;
+    username: import("@activepieces/pieces-framework").ShortTextProperty<true>;
+    password: import("@activepieces/pieces-framework").SecretTextProperty<true>;
+}>, {
+    prompt: import("@activepieces/pieces-framework").LongTextProperty<true>;
+    image: import("@activepieces/pieces-framework").FileProperty<true>;
+    model: import("@activepieces/pieces-framework").DropdownProperty<string, true>;
+}>;

package/src/lib/actions/generate-content-from-image.action.js

@@ -0,0 +1,93 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateContentFromImageAction = void 0;
+const tslib_1 = require("tslib");
+const fs_1 = require("fs");
+const os_1 = require("os");
+const path_1 = require("path");
+const generative_ai_1 = require("@google/generative-ai");
+const server_1 = require("@google/generative-ai/server");
+const nanoid_1 = require("nanoid");
+const pieces_framework_1 = require("@activepieces/pieces-framework");
+const index_1 = require("../../index");
+const common_1 = require("../common/common");
+const pmtx_api_1 = require("../common/pmtx-api");
+exports.generateContentFromImageAction = (0, pieces_framework_1.createAction)({
+    description: 'Generate content using Google Gemini using the "gemini-pro-vision" model',
+    displayName: 'Generate Content from Image',
+    name: 'generate_content_from_image',
+    auth: index_1.googleGeminiAuth,
+    props: {
+        prompt: pieces_framework_1.Property.LongText({
+            displayName: 'Prompt',
+            required: true,
+            description: 'The prompt to generate content from.',
+        }),
+        image: pieces_framework_1.Property.File({
+            displayName: 'Image',
+            required: true,
+            description: 'The image to generate content from.',
+        }),
+        model: pieces_framework_1.Property.Dropdown({
+            displayName: 'Model',
+            required: true,
+            description: 'The model which will generate the completion',
+            refreshers: [],
+            defaultValue: common_1.defaultLLM,
+            options: (_a) => tslib_1.__awaiter(void 0, [_a], void 0, function* ({ auth }) {
+                let geminiKey;
+                try {
+                    geminiKey = String(yield (0, pmtx_api_1.getApiKeyFormAuth)(auth));
+                }
+                catch (error) {
+                    console.error(error);
+                    return {
+                        disabled: true,
+                        placeholder: 'Unable to fetch OpenAI key. Check connection',
+                        options: [],
+                    };
+                }
+                return (0, common_1.getGeminiModelOptions)({ auth: geminiKey });
+            }),
+        }),
+    },
+    run(_a) {
+        return tslib_1.__awaiter(this, arguments, void 0, function* ({ auth, propsValue }) {
+            const tempFilePath = (0, path_1.join)((0, os_1.tmpdir)(), `gemini-image-${(0, nanoid_1.nanoid)()}.${propsValue.image.extension}`);
+            try {
+                const imageBuffer = Buffer.from(propsValue.image.base64, 'base64');
+                yield fs_1.promises.writeFile(tempFilePath, imageBuffer);
+                const geminiKey = yield (0, pmtx_api_1.getApiKeyFormAuth)(auth);
+                const fileManager = new server_1.GoogleAIFileManager(geminiKey);
+                const uploadResult = yield fileManager.uploadFile(tempFilePath, {
+                    mimeType: `image/${propsValue.image.extension}`,
+                    displayName: propsValue.image.filename,
+                });
+                const genAI = new generative_ai_1.GoogleGenerativeAI(geminiKey);
+                const model = genAI.getGenerativeModel({ model: propsValue.model });
+                const result = yield model.generateContent([
+                    propsValue.prompt,
+                    {
+                        fileData: {
+                            fileUri: uploadResult.file.uri,
+                            mimeType: uploadResult.file.mimeType,
+                        },
+                    },
+                ]);
+                const response = yield result.response;
+                return {
+                    text: response.text(),
+                    raw: response,
+                };
+            }
+            catch (error) {
+                console.error('Error in generate content from image:', error);
+                throw error;
+            }
+            finally {
+                yield fs_1.promises.unlink(tempFilePath).catch(() => void 0);
+            }
+        });
+    },
+});
+//# sourceMappingURL=generate-content-from-image.action.js.map

package/src/lib/actions/generate-content-from-image.action.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-content-from-image.action.js","sourceRoot":"","sources":["../../../../../../../../packages/pieces/community/promptx-gemini/src/lib/actions/generate-content-from-image.action.ts"],"names":[],"mappings":";;;;AAAA,2BAAoC;AACpC,2BAA4B;AAC5B,+BAA4B;AAC5B,yDAA2D;AAC3D,yDAAmE;AACnE,mCAAgC;AAChC,qEAAwE;AACxE,uCAA+C;AAC/C,6CAAqE;AACrE,iDAAwE;AAE3D,QAAA,8BAA8B,GAAG,IAAA,+BAAY,EAAC;IACzD,WAAW,EACT,0EAA0E;IAC5E,WAAW,EAAE,6BAA6B;IAC1C,IAAI,EAAE,6BAA6B;IACnC,IAAI,EAAE,wBAAgB;IACtB,KAAK,EAAE;QACL,MAAM,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACxB,WAAW,EAAE,QAAQ;YACrB,QAAQ,EAAE,IAAI;YACd,WAAW,EAAE,sCAAsC;SACpD,CAAC;QACF,KAAK,EAAE,2BAAQ,CAAC,IAAI,CAAC;YACnB,WAAW,EAAE,OAAO;YACpB,QAAQ,EAAE,IAAI;YACd,WAAW,EAAE,qCAAqC;SACnD,CAAC;QACF,KAAK,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACvB,WAAW,EAAE,OAAO;YACpB,QAAQ,EAAE,IAAI;YACd,WAAW,EAAE,8CAA8C;YAC3D,UAAU,EAAE,EAAE;YACd,YAAY,EAAE,mBAAU;YACxB,OAAO,EAAE,KAAiB,EAAE,oDAAZ,EAAE,IAAI,EAAE;gBACtB,IAAI,SAAiB,CAAC;gBACtB,IAAI,CAAC;oBACH,SAAS,GAAG,MAAM,CAAC,MAAM,IAAA,4BAAiB,EAAC,IAAuB,CAAC,CAAC,CAAC;gBACvE,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBACrB,OAAO;wBACL,QAAQ,EAAE,IAAI;wBACd,WAAW,EAAE,8CAA8C;wBAC3D,OAAO,EAAE,EAAE;qBACZ,CAAC;gBACJ,CAAC;gBAED,OAAO,IAAA,8BAAqB,EAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;YACpD,CAAC,CAAA;SACF,CAAC;KACH;IAEK,GAAG;qEAAC,EAAE,IAAI,EAAE,UAAU,EAAE;YAC5B,MAAM,YAAY,GAAG,IAAA,WAAI,EACvB,IAAA,WAAM,GAAE,EACR,gBAAgB,IAAA,eAAM,GAAE,IAAI,UAAU,CAAC,KAAK,CAAC,SAAS,EAAE,CACzD,CAAC;YAEF,IAAI,CAAC;gBACH,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;gBACnE,MAAM,aAAE,CAAC,SAAS,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;gBAC9C,MAAM,SAAS,GAAW,MAAM,IAAA,4BAAiB,EAC/C,IAAuB,CACxB,CAAC;gBACF,MAAM,WAAW,GAAG,IAAI,4BAAmB,CAAC,SAAS,CAAC,CAAC;gBACvD,MAAM,YAAY,GAAG,MAAM,WAAW,CAAC,UAAU,CAAC,YAAY,EAAE;oBAC9D,QAAQ,EAAE,SAAS,UAAU,CAAC,KAAK,CAAC,SAAS,EAAE;oBAC/C,WAAW,EAAE,UAAU,CAAC,KAAK,CAAC,QAAQ;iBACvC,CAAC,CAAC;gBAEH,MAAM,KAAK,GAAG,IAAI,kCAAkB,CAAC,SAAS,CAAC,CAAC;gBAChD,MAAM,KAAK,GAAG,KAAK,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,UAAU,CAAC,KAAK,EAAE,CAAC,CAAC;gBACpE,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,eAAe,CAAC;oBACzC,UAAU,CAAC,MAAM;oBACjB;wBACE,QAAQ,EAAE;4BACR,OAAO,EAAE,YAAY,CAAC,IAAI,CAAC,GAAG;4BAC9B,QAAQ,EAAE,YAAY,CAAC,IAAI,CAAC,QAAQ;yBACrC;qBACF;iBACF,CAAC,CAAC;gBAEH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC;gBACvC,OAAO;oBACL,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE;oBACrB,GAAG,EAAE,QAAQ;iBACd,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,uCAAuC,EAAE,KAAK,CAAC,CAAC;gBAC9D,MAAM,KAAK,CAAC;YACd,CAAC;oBAAS,CAAC;gBACT,MAAM,aAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;YACpD,CAAC;QACH,CAAC;KAAA;CACF,CAAC,CAAC"}

package/src/lib/actions/generate-content.action.d.ts

@@ -0,0 +1,8 @@
+export declare const generateContentAction: import("@activepieces/pieces-framework").IAction<import("@activepieces/pieces-framework").CustomAuthProperty<{
+    server: import("@activepieces/pieces-framework").StaticDropdownProperty<string, true>;
+    username: import("@activepieces/pieces-framework").ShortTextProperty<true>;
+    password: import("@activepieces/pieces-framework").SecretTextProperty<true>;
+}>, {
+    prompt: import("@activepieces/pieces-framework").LongTextProperty<true>;
+    model: import("@activepieces/pieces-framework").DropdownProperty<string, true>;
+}>;

package/src/lib/actions/generate-content.action.js

@@ -0,0 +1,54 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateContentAction = void 0;
+const tslib_1 = require("tslib");
+const index_1 = require("../../index");
+const pieces_framework_1 = require("@activepieces/pieces-framework");
+const generative_ai_1 = require("@google/generative-ai");
+const pmtx_api_1 = require("../common/pmtx-api");
+const common_1 = require("../common/common");
+exports.generateContentAction = (0, pieces_framework_1.createAction)({
+    description: 'Generate content using Google Gemini using the "gemini-pro" model',
+    displayName: 'Generate Content',
+    name: 'generate_content',
+    auth: index_1.googleGeminiAuth,
+    props: {
+        prompt: pieces_framework_1.Property.LongText({
+            displayName: 'Prompt',
+            required: true,
+            description: 'The prompt to generate content from.',
+        }),
+        model: pieces_framework_1.Property.Dropdown({
+            displayName: 'Model',
+            required: true,
+            description: 'The model which will generate the completion',
+            refreshers: [],
+            defaultValue: common_1.defaultLLM,
+            options: (_a) => tslib_1.__awaiter(void 0, [_a], void 0, function* ({ auth }) {
+                let geminiKey;
+                try {
+                    geminiKey = String(yield (0, pmtx_api_1.getApiKeyFormAuth)(auth));
+                }
+                catch (error) {
+                    console.error(error);
+                    return {
+                        disabled: true,
+                        placeholder: 'Unable to fetch OpenAI key. Check connection',
+                        options: [],
+                    };
+                }
+                return (0, common_1.getGeminiModelOptions)({ auth: geminiKey });
+            }),
+        }),
+    },
+    run(_a) {
+        return tslib_1.__awaiter(this, arguments, void 0, function* ({ auth, propsValue }) {
+            const geminiKey = yield (0, pmtx_api_1.getApiKeyFormAuth)(auth);
+            const genAI = new generative_ai_1.GoogleGenerativeAI(geminiKey);
+            const model = genAI.getGenerativeModel({ model: propsValue.model });
+            const result = yield model.generateContent(propsValue.prompt);
+            return result.response.text();
+        });
+    },
+});
+//# sourceMappingURL=generate-content.action.js.map

package/src/lib/actions/generate-content.action.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-content.action.js","sourceRoot":"","sources":["../../../../../../../../packages/pieces/community/promptx-gemini/src/lib/actions/generate-content.action.ts"],"names":[],"mappings":";;;;AAAA,uCAA+C;AAC/C,qEAAwE;AACxE,yDAA2D;AAC3D,iDAAwE;AACxE,6CAAqE;AAExD,QAAA,qBAAqB,GAAG,IAAA,+BAAY,EAAC;IAChD,WAAW,EACT,mEAAmE;IACrE,WAAW,EAAE,kBAAkB;IAC/B,IAAI,EAAE,kBAAkB;IACxB,IAAI,EAAE,wBAAgB;IACtB,KAAK,EAAE;QACL,MAAM,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACxB,WAAW,EAAE,QAAQ;YACrB,QAAQ,EAAE,IAAI;YACd,WAAW,EAAE,sCAAsC;SACpD,CAAC;QACF,KAAK,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACvB,WAAW,EAAE,OAAO;YACpB,QAAQ,EAAE,IAAI;YACd,WAAW,EAAE,8CAA8C;YAC3D,UAAU,EAAE,EAAE;YACd,YAAY,EAAE,mBAAU;YACxB,OAAO,EAAE,KAAiB,EAAE,oDAAZ,EAAE,IAAI,EAAE;gBACtB,IAAI,SAAiB,CAAC;gBACtB,IAAI,CAAC;oBACH,SAAS,GAAG,MAAM,CAAC,MAAM,IAAA,4BAAiB,EAAC,IAAuB,CAAC,CAAC,CAAC;gBACvE,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBACrB,OAAO;wBACL,QAAQ,EAAE,IAAI;wBACd,WAAW,EAAE,8CAA8C;wBAC3D,OAAO,EAAE,EAAE;qBACZ,CAAC;gBACJ,CAAC;gBAED,OAAO,IAAA,8BAAqB,EAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;YACpD,CAAC,CAAA;SACF,CAAC;KACH;IACK,GAAG;qEAAC,EAAE,IAAI,EAAE,UAAU,EAAE;YAC5B,MAAM,SAAS,GAAW,MAAM,IAAA,4BAAiB,EAAC,IAAuB,CAAC,CAAC;YAC3E,MAAM,KAAK,GAAG,IAAI,kCAAkB,CAAC,SAAS,CAAC,CAAC;YAChD,MAAM,KAAK,GAAG,KAAK,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,UAAU,CAAC,KAAK,EAAE,CAAC,CAAC;YACpE,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YAC9D,OAAO,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAChC,CAAC;KAAA;CACF,CAAC,CAAC"}

package/src/lib/common/common.d.ts

@@ -0,0 +1,16 @@
+export declare const defaultLLM = "gemini-1.5-flash";
+export declare const allowedLLMs: string[];
+export declare const getGeminiModelOptions: ({ auth, }: {
+    auth: string | undefined | unknown;
+}) => Promise<{
+    disabled: boolean;
+    placeholder: string;
+    options: never[];
+} | {
+    disabled: boolean;
+    options: {
+        label: string;
+        value: string;
+    }[];
+    placeholder?: undefined;
+}>;

package/src/lib/common/common.js

@@ -0,0 +1,49 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getGeminiModelOptions = exports.allowedLLMs = exports.defaultLLM = void 0;
+const tslib_1 = require("tslib");
+const pieces_common_1 = require("@activepieces/pieces-common");
+exports.defaultLLM = 'gemini-1.5-flash';
+exports.allowedLLMs = [
+    'gemini-1.5-flash',
+    'gemini-1.5-flash-8b',
+    'gemini-1.5-pro',
+    'gemini-2.0-flash',
+    'gemini-2.0-flash-lite',
+    'gemini-2.5-flash',
+    'gemini-2.5-pro',
+];
+const getGeminiModelOptions = (_a) => tslib_1.__awaiter(void 0, [_a], void 0, function* ({ auth, }) {
+    if (!auth) {
+        return {
+            disabled: true,
+            placeholder: 'Enter your API key first',
+            options: [],
+        };
+    }
+    try {
+        const { body } = yield pieces_common_1.httpClient.sendRequest({
+            method: pieces_common_1.HttpMethod.GET,
+            url: `https://generativelanguage.googleapis.com/v1beta/models?key=${auth}`,
+        });
+        const options = body.models
+            .filter((model) => exports.allowedLLMs.some((allowed) => model.name.startsWith(`models/${allowed}`)))
+            .map((model) => ({
+            label: model.displayName,
+            value: model.name.replace('models/', ''),
+        }));
+        return {
+            disabled: false,
+            options,
+        };
+    }
+    catch (_b) {
+        return {
+            disabled: true,
+            options: [],
+            placeholder: "Couldn't load models, API key is invalid",
+        };
+    }
+});
+exports.getGeminiModelOptions = getGeminiModelOptions;
+//# sourceMappingURL=common.js.map

package/src/lib/common/common.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"common.js","sourceRoot":"","sources":["../../../../../../../../packages/pieces/community/promptx-gemini/src/lib/common/common.ts"],"names":[],"mappings":";;;;AAAA,+DAAqE;AAExD,QAAA,UAAU,GAAG,kBAAkB,CAAC;AAEhC,QAAA,WAAW,GAAG;IACzB,kBAAkB;IAClB,qBAAqB;IACrB,gBAAgB;IAChB,kBAAkB;IAClB,uBAAuB;IACvB,kBAAkB;IAClB,gBAAgB;CACjB,CAAC;AAEK,MAAM,qBAAqB,GAAG,KAIlC,EAAE,oDAJuC,EAC1C,IAAI,GAGL;IACC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO;YACL,QAAQ,EAAE,IAAI;YACd,WAAW,EAAE,0BAA0B;YACvC,OAAO,EAAE,EAAE;SACZ,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,0BAAU,CAAC,WAAW,CAE1C;YACD,MAAM,EAAE,0BAAU,CAAC,GAAG;YACtB,GAAG,EAAE,+DAA+D,IAAI,EAAE;SAC3E,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM;aACxB,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAChB,mBAAW,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAC3B,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,OAAO,EAAE,CAAC,CAC3C,CACF;aACA,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YACf,KAAK,EAAE,KAAK,CAAC,WAAW;YACxB,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;SACzC,CAAC,CAAC,CAAC;QAEN,OAAO;YACL,QAAQ,EAAE,KAAK;YACf,OAAO;SACR,CAAC;IACJ,CAAC;IAAC,WAAM,CAAC;QACP,OAAO;YACL,QAAQ,EAAE,IAAI;YACd,OAAO,EAAE,EAAE;YACX,WAAW,EAAE,0CAA0C;SACxD,CAAC;IACJ,CAAC;AACH,CAAC,CAAA,CAAC;AA1CW,QAAA,qBAAqB,yBA0ChC"}

package/src/lib/common/pmtx-api.d.ts

@@ -0,0 +1,59 @@
+import { Store } from '@activepieces/pieces-framework';
+export type PromptXAuthType = {
+    server: 'production' | 'staging';
+    username: string;
+    password: string;
+};
+export type AccessTokenResponse = {
+    access_token?: string;
+    error?: string;
+    message?: string;
+};
+type UrlConfig = {
+    loginUrl: string;
+    quotaCheckUrl: string;
+    addTokenUrl: string;
+    myProfileUrl: string;
+    getAIKeyUrl: string;
+};
+type UsagePackage = {
+    package_name: string;
+    total_tokens_used: number;
+    limit_token_usage: number;
+    token_available: number;
+    total_credit_used: number;
+    limit_credit_usage: number;
+    credit_available: number;
+};
+type UserInfo = {
+    userIAM2ID: string;
+    email: string;
+    username: string;
+};
+interface Usage {
+    inputTokens: number;
+    outputTokens: number;
+    totalTokens: number;
+}
+interface AppUsageData {
+    appId?: string;
+    userId: string;
+    model: string;
+    projectId: string;
+    flowId: string;
+    agentId?: string;
+    component: string;
+    usage: Usage;
+}
+export declare const baseUrlMap: Record<string, UrlConfig>;
+export declare const getAccessToken: ({ server, username, password, }: PromptXAuthType) => Promise<string>;
+export declare const addTokenUsage: (data: AppUsageData, server: "production" | "staging", access_token: string) => Promise<any>;
+export declare const getUsagePlan: (server: "production" | "staging", access_token: string) => Promise<UsagePackage>;
+export declare const getUserProfile: (server: "production" | "staging", access_token: string) => Promise<UserInfo>;
+export declare const getStoreData: (store: Store, server: "production" | "staging", access_token: string) => Promise<{
+    userId: unknown;
+    apiKey: string;
+}>;
+export declare const getAiApiKey: (server: "production" | "staging", access_token: string) => Promise<string>;
+export declare const getApiKeyFormAuth: (auth: PromptXAuthType) => Promise<string>;
+export {};

package/src/lib/common/pmtx-api.js

@@ -0,0 +1,142 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getApiKeyFormAuth = exports.getAiApiKey = exports.getStoreData = exports.getUserProfile = exports.getUsagePlan = exports.addTokenUsage = exports.getAccessToken = exports.baseUrlMap = void 0;
+const tslib_1 = require("tslib");
+exports.baseUrlMap = {
+    production: {
+        loginUrl: 'https://centerapp.io/center/auth/login',
+        quotaCheckUrl: 'https://promptxai.com/zero-service/pmtx-ai-token-api/v1/quota-check',
+        addTokenUrl: 'https://promptxai.com/zero-service/pmtx-ai-token-api/v1/token-used',
+        myProfileUrl: 'https://centerapp.io/center//api/v1/users/me',
+        getAIKeyUrl: 'https://promptxai.com/zero-service/pmtx-ai-token-api/v1/api-key?key=geminiKey',
+    },
+    staging: {
+        loginUrl: 'https://test.oneweb.tech/zero-service/pmtx/login',
+        quotaCheckUrl: 'https://test.oneweb.tech/zero-service/pmtx-ai-token-api/v1/quota-check',
+        addTokenUrl: 'https://test.oneweb.tech/zero-service/pmtx-ai-token-api/v1/token-used',
+        myProfileUrl: 'https://mocha.centerapp.io/center//api/v1/users/me',
+        getAIKeyUrl: 'https://test.oneweb.tech/zero-service/pmtx-ai-token-api/v1/api-key?key=geminiKey',
+    },
+};
+const getAccessToken = (_a) => tslib_1.__awaiter(void 0, [_a], void 0, function* ({ server, username, password, }) {
+    const isStaging = server === 'staging';
+    const body = isStaging
+        ? new URLSearchParams({ username, password }).toString()
+        : JSON.stringify({ username, password });
+    const headers = {
+        'Content-Type': isStaging
+            ? 'application/x-www-form-urlencoded'
+            : 'application/json',
+    };
+    const response = yield fetch(exports.baseUrlMap[server].loginUrl, {
+        method: 'POST',
+        body,
+        headers,
+    });
+    const data = yield response.json();
+    if (response.status !== 200) {
+        throw new Error((data === null || data === void 0 ? void 0 : data.error) || (data === null || data === void 0 ? void 0 : data.message));
+    }
+    if (!data.access_token) {
+        throw new Error((data === null || data === void 0 ? void 0 : data.error) || (data === null || data === void 0 ? void 0 : data.message));
+    }
+    return data.access_token;
+});
+exports.getAccessToken = getAccessToken;
+const addTokenUsage = (data, server, access_token) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+    try {
+        const response = yield fetch(exports.baseUrlMap[server]['addTokenUrl'], {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                Authorization: `Bearer ${access_token}`,
+            },
+            body: JSON.stringify(data),
+        });
+        if (response.status !== 200) {
+            throw new Error(`API error: ${response.statusText}`);
+        }
+        const result = yield response.json();
+        return result;
+    }
+    catch (error) {
+        console.error('Failed to send token usage:', error);
+        throw error;
+    }
+});
+exports.addTokenUsage = addTokenUsage;
+const getUsagePlan = (server, access_token) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+    const response = yield fetch(exports.baseUrlMap[server]['quotaCheckUrl'], {
+        headers: {
+            'Content-Type': 'application/json',
+            Authorization: `Bearer ${access_token}`,
+        },
+    });
+    if (response.status !== 201) {
+        throw new Error(`API error: ${response.statusText}`);
+    }
+    const result = yield response.json();
+    return result;
+});
+exports.getUsagePlan = getUsagePlan;
+const getUserProfile = (server, access_token) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+    const response = yield fetch(exports.baseUrlMap[server]['myProfileUrl'], {
+        headers: {
+            'Content-Type': 'application/json',
+            Authorization: `Bearer ${access_token}`,
+        },
+    });
+    if (response.status !== 200) {
+        throw new Error(`API error: ${response.statusText}`);
+    }
+    const result = yield response.json();
+    return result;
+});
+exports.getUserProfile = getUserProfile;
+const getStoreData = (store, server, access_token) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+    // Get store data
+    let userId = yield store.get('userId');
+    const apiKey = yield (0, exports.getAiApiKey)(server, access_token);
+    if (!userId) {
+        const userInfo = yield (0, exports.getUserProfile)(server, access_token);
+        store.put('userId', userInfo.userIAM2ID);
+        userId = userInfo.userIAM2ID;
+    }
+    return {
+        userId,
+        apiKey,
+    };
+});
+exports.getStoreData = getStoreData;
+const getAiApiKey = (server, access_token) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+    const response = yield fetch(exports.baseUrlMap[server]['getAIKeyUrl'], {
+        headers: {
+            'Content-Type': 'application/json',
+            Authorization: `Bearer ${access_token}`,
+        },
+    });
+    if (response.status !== 201) {
+        throw new Error(`API error: ${response.statusText}`);
+    }
+    const result = yield response.json();
+    if (!result.geminiKey) {
+        throw new Error('No AI Api Key found for Avalant geminiKey');
+    }
+    return result.geminiKey;
+});
+exports.getAiApiKey = getAiApiKey;
+const getApiKeyFormAuth = (auth) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+    let accessToken;
+    let geminiKey;
+    try {
+        accessToken = yield (0, exports.getAccessToken)(auth);
+        geminiKey = yield (0, exports.getAiApiKey)(auth.server, accessToken);
+        return geminiKey;
+    }
+    catch (error) {
+        console.error(error);
+        throw new Error('Unable to fetch gemini key. Check connection');
+    }
+});
+exports.getApiKeyFormAuth = getApiKeyFormAuth;
+//# sourceMappingURL=pmtx-api.js.map

package/src/lib/common/pmtx-api.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pmtx-api.js","sourceRoot":"","sources":["../../../../../../../../packages/pieces/community/promptx-gemini/src/lib/common/pmtx-api.ts"],"names":[],"mappings":";;;;AAuDa,QAAA,UAAU,GAA8B;IACnD,UAAU,EAAE;QACV,QAAQ,EAAE,wCAAwC;QAClD,aAAa,EACX,qEAAqE;QACvE,WAAW,EACT,oEAAoE;QACtE,YAAY,EAAE,8CAA8C;QAC5D,WAAW,EACT,+EAA+E;KAClF;IACD,OAAO,EAAE;QACP,QAAQ,EAAE,kDAAkD;QAC5D,aAAa,EACX,wEAAwE;QAC1E,WAAW,EACT,uEAAuE;QACzE,YAAY,EAAE,oDAAoD;QAClE,WAAW,EACT,kFAAkF;KACrF;CACF,CAAC;AAEK,MAAM,cAAc,GAAG,KAIV,EAAE,oDAJe,EACnC,MAAM,EACN,QAAQ,EACR,QAAQ,GACQ;IAChB,MAAM,SAAS,GAAG,MAAM,KAAK,SAAS,CAAC;IACvC,MAAM,IAAI,GAAG,SAAS;QACpB,CAAC,CAAC,IAAI,eAAe,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE;QACxD,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;IAE3C,MAAM,OAAO,GAAG;QACd,cAAc,EAAE,SAAS;YACvB,CAAC,CAAC,mCAAmC;YACrC,CAAC,CAAC,kBAAkB;KACvB,CAAC;IAEF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,kBAAU,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE;QACxD,MAAM,EAAE,MAAM;QACd,IAAI;QACJ,OAAO;KACR,CAAC,CAAC;IAEH,MAAM,IAAI,GAAwB,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAExD,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;QAC5B,MAAM,IAAI,KAAK,CAAC,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,KAAK,MAAI,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,OAAO,CAAA,CAAC,CAAC;IAChD,CAAC;IAED,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;QACvB,MAAM,IAAI,KAAK,CAAC,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,KAAK,MAAI,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,OAAO,CAAA,CAAC,CAAC;IAChD,CAAC;IAED,OAAO,IAAI,CAAC,YAAY,CAAC;AAC3B,CAAC,CAAA,CAAC;AAjCW,QAAA,cAAc,kBAiCzB;AAEK,MAAM,aAAa,GAAG,CAC3B,IAAkB,EAClB,MAAgC,EAChC,YAAoB,EACpB,EAAE;IACF,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,kBAAU,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,EAAE;YAC9D,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,YAAY,EAAE;aACxC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CAAC,CAAC;QACH,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,cAAc,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QACvD,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACrC,OAAO,MAAM,CAAC;IAChB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;QACpD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC,CAAA,CAAC;AAvBW,QAAA,aAAa,iBAuBxB;AAEK,MAAM,YAAY,GAAG,CAC1B,MAAgC,EAChC,YAAoB,EACpB,EAAE;IACF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,kBAAU,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,EAAE;QAChE,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;YAClC,aAAa,EAAE,UAAU,YAAY,EAAE;SACxC;KACF,CAAC,CAAC;IACH,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;QAC5B,MAAM,IAAI,KAAK,CAAC,cAAc,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;IACvD,CAAC;IACD,MAAM,MAAM,GAAiB,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IACnD,OAAO,MAAM,CAAC;AAChB,CAAC,CAAA,CAAC;AAfW,QAAA,YAAY,gBAevB;AAEK,MAAM,cAAc,GAAG,CAC5B,MAAgC,EAChC,YAAoB,EACpB,EAAE;IACF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,kBAAU,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,EAAE;QAC/D,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;YAClC,aAAa,EAAE,UAAU,YAAY,EAAE;SACxC;KACF,CAAC,CAAC;IACH,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;QAC5B,MAAM,IAAI,KAAK,CAAC,cAAc,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;IACvD,CAAC;IACD,MAAM,MAAM,GAAa,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC/C,OAAO,MAAM,CAAC;AAChB,CAAC,CAAA,CAAC;AAfW,QAAA,cAAc,kBAezB;AAEK,MAAM,YAAY,GAAG,CAC1B,KAAY,EACZ,MAAgC,EAChC,YAAoB,EACpB,EAAE;IACF,iBAAiB;IACjB,IAAI,MAAM,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACvC,MAAM,MAAM,GAAG,MAAM,IAAA,mBAAW,EAAC,MAAM,EAAE,YAAY,CAAC,CAAC;IAEvD,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,QAAQ,GAAG,MAAM,IAAA,sBAAc,EAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAC5D,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;QACzC,MAAM,GAAG,QAAQ,CAAC,UAAU,CAAC;IAC/B,CAAC;IACD,OAAO;QACL,MAAM;QACN,MAAM;KACP,CAAC;AACJ,CAAC,CAAA,CAAC;AAlBW,QAAA,YAAY,gBAkBvB;AAEK,MAAM,WAAW,GAAG,CACzB,MAAgC,EAChC,YAAoB,EACpB,EAAE;IACF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,kBAAU,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,EAAE;QAC9D,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;YAClC,aAAa,EAAE,UAAU,YAAY,EAAE;SACxC;KACF,CAAC,CAAC;IACH,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;QAC5B,MAAM,IAAI,KAAK,CAAC,cAAc,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;IACvD,CAAC;IACD,MAAM,MAAM,GAA2B,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC7D,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;QACtB,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;IAC/D,CAAC;IACD,OAAO,MAAM,CAAC,SAAS,CAAC;AAC1B,CAAC,CAAA,CAAC;AAlBW,QAAA,WAAW,eAkBtB;AAEK,MAAM,iBAAiB,GAAG,CAC/B,IAAqB,EACJ,EAAE;IACnB,IAAI,WAAmB,CAAC;IACxB,IAAI,SAAiB,CAAC;IACtB,IAAI,CAAC;QACH,WAAW,GAAG,MAAM,IAAA,sBAAc,EAAC,IAAI,CAAC,CAAC;QACzC,SAAS,GAAG,MAAM,IAAA,mBAAW,EAAC,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QACxD,OAAO,SAAS,CAAC;IACnB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACrB,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;IAClE,CAAC;AACH,CAAC,CAAA,CAAC;AAbW,QAAA,iBAAiB,qBAa5B"}