@blocklet/aigne-hub 0.2.7

package/README.md

@@ -0,0 +1,29 @@
+# AIGNE Hub Lib
+
+AIGNE Hub common UX component
+
+## Usage
+
+```jsx
+import React from 'react';
+import { HelloWorld } from '@blocklet/aigne-hub';
+
+function App() {
+  return <HelloWorld />;
+}
+```
+
+## Develop
+
+```
+
+# develop with @blocklet/aigne-hub and storybook
+npm run dev
+
+# only develop with @blocklet/aigne-hub
+npm run watch
+
+# only develop with storybook
+npm run storybook
+
+```

package/lib/cjs/api/ai-kit.js

@@ -0,0 +1,60 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createImageGenerationApi = exports.createTextCompletionApi = exports.createStatusApi = void 0;
+const error_1 = require("./error");
+const createStatusApi = ({ axios, path }) => () => axios.get(path).then((res) => res.data);
+exports.createStatusApi = createStatusApi;
+const createTextCompletionApi = ({ fetch, path, timeout, }) => async (options) => {
+    const promise = options.stream
+        ? fetch(path, {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify(options),
+        }).then(async (res) => {
+            var _a, _b, _c;
+            if (res.status !== 200) {
+                const text = await res.text();
+                let json;
+                try {
+                    json = JSON.parse(text);
+                }
+                catch (_d) {
+                    // eslint-disable-next-line no-empty
+                }
+                if ((_a = json === null || json === void 0 ? void 0 : json.error) === null || _a === void 0 ? void 0 : _a.type) {
+                    throw new error_1.SubscriptionError((_b = json === null || json === void 0 ? void 0 : json.error) === null || _b === void 0 ? void 0 : _b.type);
+                }
+                throw new Error(((_c = json === null || json === void 0 ? void 0 : json.error) === null || _c === void 0 ? void 0 : _c.message) || (json === null || json === void 0 ? void 0 : json.message) || text || res.status);
+            }
+            return res.body;
+        })
+        : fetch(path, options)
+            .then((res) => res.json())
+            .then((data) => { var _a, _b, _c, _d, _e, _f, _g; return ({ text: (_d = (_a = data.text) !== null && _a !== void 0 ? _a : (_c = (_b = data.choices) === null || _b === void 0 ? void 0 : _b[0]) === null || _c === void 0 ? void 0 : _c.text) !== null && _d !== void 0 ? _d : (_g = (_f = (_e = data.choices) === null || _e === void 0 ? void 0 : _e[0]) === null || _f === void 0 ? void 0 : _f.message) === null || _g === void 0 ? void 0 : _g.content }); })
+            .catch(processResponseError);
+    if (!timeout) {
+        return promise;
+    }
+    return Promise.race([
+        promise,
+        new Promise((_, reject) => {
+            setTimeout(() => reject(new Error('Timeout')), timeout);
+        }),
+    ]);
+};
+exports.createTextCompletionApi = createTextCompletionApi;
+const createImageGenerationApi = ({ axios, path, }) => async (options) => {
+    return axios
+        .post(path, options)
+        .then((res) => res.data)
+        .catch(processResponseError);
+};
+exports.createImageGenerationApi = createImageGenerationApi;
+function processResponseError(error) {
+    var _a, _b, _c, _d, _e;
+    const msg = ((_c = (_b = (_a = error.response) === null || _a === void 0 ? void 0 : _a.data) === null || _b === void 0 ? void 0 : _b.error) === null || _c === void 0 ? void 0 : _c.message) || ((_e = (_d = error.response) === null || _d === void 0 ? void 0 : _d.data) === null || _e === void 0 ? void 0 : _e.message);
+    if (msg) {
+        throw new Error(msg);
+    }
+    throw error;
+}

package/lib/cjs/api/api.js

@@ -0,0 +1,17 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+var _a, _b;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.API_TIMEOUT = exports.PREFIX = void 0;
+const axios_1 = __importDefault(require("axios"));
+exports.PREFIX = (typeof window !== 'undefined' &&
+    ((_b = (_a = window.blocklet) === null || _a === void 0 ? void 0 : _a.componentMountPoints.find((i) => i.name === 'ai-kit')) === null || _b === void 0 ? void 0 : _b.mountPoint)) ||
+    '/';
+exports.API_TIMEOUT = 30 * 1000;
+const api = axios_1.default.create({
+    baseURL: exports.PREFIX,
+    timeout: exports.API_TIMEOUT,
+});
+exports.default = api;

package/lib/cjs/api/app.js

@@ -0,0 +1,57 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.appStatus = appStatus;
+exports.appConfig = appConfig;
+exports.setAppConfig = setAppConfig;
+exports.appServiceRegister = appServiceRegister;
+exports.appUsedCredits = appUsedCredits;
+exports.unsubscribe = unsubscribe;
+exports.getCreditPaymentLink = getCreditPaymentLink;
+exports.getCreditBalance = getCreditBalance;
+exports.getCreditGrants = getCreditGrants;
+exports.getCreditTransactions = getCreditTransactions;
+const api_1 = __importDefault(require("./api"));
+async function appStatus() {
+    return Promise.all([api_1.default.get('/api/app/service/status'), appConfig()]).then(([res, config]) => ({
+        ...res.data,
+        config,
+    }));
+}
+async function appConfig() {
+    return api_1.default.get('/api/app/config').then((res) => res.data);
+}
+async function setAppConfig(payload) {
+    return api_1.default.patch('/api/app/config', payload).then((res) => res.data);
+}
+async function appServiceRegister() {
+    return api_1.default.post('/api/app/service/register').then((res) => res.data);
+}
+async function appUsedCredits(query, options) {
+    return api_1.default
+        .get((options === null || options === void 0 ? void 0 : options.useAIKitService) ? '/api/app/service/usage' : '/api/app/usage', { params: query })
+        .then((res) => res.data);
+}
+async function unsubscribe() {
+    return api_1.default.post('/api/app/service/unsubscribe');
+}
+async function getCreditPaymentLink(options) {
+    const endpoint = (options === null || options === void 0 ? void 0 : options.useAIKitService) ? '/api/user/service/credit/payment-link' : '/api/user/credit/payment-link';
+    return api_1.default.get(endpoint).then((res) => res.data);
+}
+// Get credit balance
+async function getCreditBalance(options) {
+    const endpoint = (options === null || options === void 0 ? void 0 : options.useAIKitService) ? '/api/user/service/credit/balance' : '/api/user/credit/balance';
+    return api_1.default.get(endpoint).then((res) => res.data);
+}
+// Get credit grants
+async function getCreditGrants(params, options) {
+    const endpoint = (options === null || options === void 0 ? void 0 : options.useAIKitService) ? '/api/user/service/credit/grants' : '/api/user/credit/grants';
+    return api_1.default.get(endpoint, { params }).then((res) => res.data);
+}
+async function getCreditTransactions(params, options) {
+    const endpoint = (options === null || options === void 0 ? void 0 : options.useAIKitService) ? '/api/user/service/credit/transactions' : '/api/user/credit/transactions';
+    return api_1.default.get(endpoint, { params }).then((res) => res.data);
+}

package/lib/cjs/api/call/api.js

@@ -0,0 +1,51 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.aiKitApi = void 0;
+exports.catchAndRethrowUpstreamError = catchAndRethrowUpstreamError;
+const axios_1 = __importStar(require("axios"));
+const constants_1 = require("../constants");
+const event_stream_1 = require("../utils/event-stream");
+exports.aiKitApi = axios_1.default.create({
+    baseURL: constants_1.AI_KIT_BASE_URL,
+});
+async function catchAndRethrowUpstreamError(response) {
+    return response.catch(async (error) => {
+        var _a, _b, _c;
+        if ((0, axios_1.isAxiosError)(error) && ((_a = error.response) === null || _a === void 0 ? void 0 : _a.data)) {
+            const { data } = error.response;
+            const json = typeof data[Symbol.iterator] === 'function'
+                ? await (0, event_stream_1.tryParseJsonFromResponseStream)(data)
+                : data;
+            const message = (_b = json === null || json === void 0 ? void 0 : json.error) === null || _b === void 0 ? void 0 : _b.message;
+            if (typeof message === 'string') {
+                const error = new Error(message);
+                error.type = (_c = json === null || json === void 0 ? void 0 : json.error) === null || _c === void 0 ? void 0 : _c.type;
+                throw error;
+            }
+        }
+        throw error;
+    });
+}

package/lib/cjs/api/call/app.js

@@ -0,0 +1,74 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.appStatus = appStatus;
+exports.appRegister = appRegister;
+exports.cancelSubscription = cancelSubscription;
+exports.recoverSubscription = recoverSubscription;
+const component_1 = require("@blocklet/sdk/lib/component");
+const auth_1 = require("../utils/auth");
+const api_1 = require("./api");
+async function appStatus({ description }, { useAIKitService, ...options } = {}) {
+    const response = await (0, api_1.catchAndRethrowUpstreamError)(useAIKitService
+        ? api_1.aiKitApi.get('/api/app/status', {
+            params: { description },
+            responseType: options.responseType,
+            headers: { ...(0, auth_1.getRemoteComponentCallHeaders)({}) },
+        })
+        : (0, component_1.call)({
+            name: 'ai-kit',
+            method: 'GET',
+            path: '/api/app/status',
+            params: { description },
+            data: {},
+            responseType: options === null || options === void 0 ? void 0 : options.responseType,
+        }));
+    if ((options === null || options === void 0 ? void 0 : options.responseType) === 'stream')
+        return response;
+    return response.data;
+}
+async function appRegister(payload, { useAIKitService, ...options } = {}) {
+    const response = await (0, api_1.catchAndRethrowUpstreamError)(useAIKitService
+        ? api_1.aiKitApi.post('/api/app/register', payload, { responseType: options.responseType })
+        : (0, component_1.call)({
+            name: 'ai-kit',
+            method: 'GET',
+            path: '/api/app/register',
+            data: payload,
+            responseType: options === null || options === void 0 ? void 0 : options.responseType,
+        }));
+    if ((options === null || options === void 0 ? void 0 : options.responseType) === 'stream')
+        return response;
+    return response.data;
+}
+async function cancelSubscription({ useAIKitService, ...options } = {}) {
+    const response = await (0, api_1.catchAndRethrowUpstreamError)(useAIKitService
+        ? api_1.aiKitApi.post('/api/app/subscription/cancel', {}, {
+            responseType: options.responseType,
+            headers: { ...(0, auth_1.getRemoteComponentCallHeaders)({}) },
+        })
+        : (0, component_1.call)({
+            name: 'ai-kit',
+            path: '/api/app/subscription/cancel',
+            data: {},
+            responseType: options === null || options === void 0 ? void 0 : options.responseType,
+        }));
+    if ((options === null || options === void 0 ? void 0 : options.responseType) === 'stream')
+        return response;
+    return response.data;
+}
+async function recoverSubscription({ useAIKitService, ...options } = {}) {
+    const response = await (0, api_1.catchAndRethrowUpstreamError)(useAIKitService
+        ? api_1.aiKitApi.post('/api/app/subscription/recover', {}, {
+            responseType: options.responseType,
+            headers: { ...(0, auth_1.getRemoteComponentCallHeaders)({}) },
+        })
+        : (0, component_1.call)({
+            name: 'ai-kit',
+            path: '/api/app/subscription/recover',
+            data: {},
+            responseType: options === null || options === void 0 ? void 0 : options.responseType,
+        }));
+    if ((options === null || options === void 0 ? void 0 : options.responseType) === 'stream')
+        return response;
+    return response.data;
+}

package/lib/cjs/api/call/index.js

@@ -0,0 +1,19 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./v1"), exports);
+__exportStar(require("./v2"), exports);
+__exportStar(require("./proxy"), exports);

package/lib/cjs/api/call/proxy.js

@@ -0,0 +1,64 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.proxyToAIKit = proxyToAIKit;
+const http_1 = __importDefault(require("http"));
+const https_1 = __importDefault(require("https"));
+const component_1 = require("@blocklet/sdk/lib/component");
+const verify_sign_1 = require("@blocklet/sdk/lib/util/verify-sign");
+const lodash_1 = require("lodash");
+const ufo_1 = require("ufo");
+const constants_1 = require("../constants");
+const auth_1 = require("../utils/auth");
+function proxyToAIKit(path, { useAIKitService, proxyReqHeaders = ['accept', 'content-type'], proxyResHeaders = ['content-type', 'cache-control'], } = {}) {
+    const parseReqBody = path !== '/api/v1/audio/transcriptions';
+    return (req, res, next) => {
+        var _a;
+        const url = (0, ufo_1.parseURL)((0, ufo_1.withQuery)((0, ufo_1.joinURL)(useAIKitService ? constants_1.AI_KIT_BASE_URL : (0, component_1.getComponentWebEndpoint)('ai-kit'), path), req.query));
+        const userDid = ((_a = req.user) === null || _a === void 0 ? void 0 : _a.did) || (req === null || req === void 0 ? void 0 : req.get('x-app-user-did'));
+        const proxyReq = (url.protocol === 'https:' ? https_1.default : http_1.default).request((0, ufo_1.stringifyParsedURL)(url), {
+            headers: {
+                ...(0, lodash_1.pick)(req.headers, ...proxyReqHeaders),
+                ...(useAIKitService
+                    ? (0, auth_1.getRemoteComponentCallHeaders)(req.body || {}, userDid)
+                    : (() => {
+                        const { iat, exp, sig, version } = (0, verify_sign_1.getSignData)({
+                            data: req.body,
+                            params: req.query,
+                            method: req.method,
+                            url: (0, ufo_1.stringifyParsedURL)(url),
+                        });
+                        return {
+                            'x-component-sig': sig,
+                            'x-component-sig-iat': iat,
+                            'x-component-sig-exp': exp,
+                            'x-component-sig-version': version,
+                        };
+                    })()),
+            },
+            method: req.method,
+        }, (proxyRes) => {
+            if (proxyRes.statusCode)
+                res.status(proxyRes.statusCode);
+            res.setHeader('X-Accel-Buffering', 'no');
+            for (const [k, v] of Object.entries((0, lodash_1.pick)(proxyRes.headers, ...proxyResHeaders))) {
+                if (!(0, lodash_1.isNil)(v))
+                    res.setHeader(k, v);
+            }
+            proxyRes.pipe(res);
+        });
+        proxyReq.on('error', (e) => next(e));
+        req.on('aborted', () => {
+            proxyReq.destroy();
+        });
+        if (parseReqBody) {
+            proxyReq.write(JSON.stringify(req.body));
+            proxyReq.end();
+        }
+        else {
+            req.pipe(proxyReq);
+        }
+    };
+}

package/lib/cjs/api/call/v1.js

@@ -0,0 +1,166 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.status = status;
+exports.chatCompletions = chatCompletions;
+exports.imageGenerations = imageGenerations;
+exports.embeddings = embeddings;
+exports.audioTranscriptions = audioTranscriptions;
+exports.audioSpeech = audioSpeech;
+const web_1 = require("stream/web");
+const component_1 = require("@blocklet/sdk/lib/component");
+const verify_sign_1 = require("@blocklet/sdk/lib/util/verify-sign");
+const axios_1 = __importDefault(require("axios"));
+const form_data_1 = __importDefault(require("form-data"));
+const json_stable_stringify_1 = __importDefault(require("json-stable-stringify"));
+const ufo_1 = require("ufo");
+const types_1 = require("../types");
+const auth_1 = require("../utils/auth");
+const event_stream_1 = require("../utils/event-stream");
+const api_1 = require("./api");
+async function status({ useAIKitService, ...options } = {}) {
+    const response = await (0, api_1.catchAndRethrowUpstreamError)(useAIKitService
+        ? api_1.aiKitApi
+            .get('/api/v1/status', {
+            responseType: options.responseType,
+            headers: { ...(0, auth_1.getRemoteComponentCallHeaders)({}) },
+        })
+            .then((res) => res.data)
+        : (0, component_1.call)({ name: 'ai-kit', method: 'GET', path: '/api/v1/status', responseType: options === null || options === void 0 ? void 0 : options.responseType }).then((res) => res.data));
+    if ((options === null || options === void 0 ? void 0 : options.responseType) === 'stream')
+        return response;
+    return response.data;
+}
+async function chatCompletions(input, { useAIKitService, ...options } = {}) {
+    const response = (0, api_1.catchAndRethrowUpstreamError)(useAIKitService
+        ? (0, api_1.aiKitApi)('/api/v1/chat/completions', {
+            responseType: 'stream',
+            method: 'POST',
+            data: (0, json_stable_stringify_1.default)(input),
+            headers: {
+                ...(0, auth_1.getRemoteComponentCallHeaders)(input),
+                Accept: 'text/event-stream',
+                'Content-Type': 'application/json',
+            },
+        })
+        : (0, component_1.call)({
+            name: 'ai-kit',
+            path: 'api/v1/chat/completions',
+            data: input,
+            responseType: 'stream',
+            headers: { Accept: 'text/event-stream' },
+        }));
+    if ((options === null || options === void 0 ? void 0 : options.responseType) === 'stream')
+        return response;
+    return new web_1.ReadableStream({
+        async start(controller) {
+            try {
+                const stream = (0, event_stream_1.readableToWeb)((await response).data)
+                    .pipeThrough(new web_1.TextDecoderStream())
+                    .pipeThrough(new event_stream_1.EventSourceParserStream());
+                for await (const chunk of stream) {
+                    if ((0, types_1.isChatCompletionError)(chunk)) {
+                        if (chunk.error.type) {
+                            const error = new Error(chunk.error.message);
+                            error.type = chunk.error.type;
+                            error.timestamp = chunk.error.timestamp;
+                            controller.error(error);
+                        }
+                        else {
+                            controller.error(new Error(chunk.error.message));
+                        }
+                        break;
+                    }
+                    controller.enqueue(chunk);
+                }
+            }
+            catch (error) {
+                controller.error(error);
+            }
+            finally {
+                controller.close();
+            }
+        },
+    });
+}
+async function imageGenerations(input, { useAIKitService, ...options } = {}) {
+    const response = await (0, api_1.catchAndRethrowUpstreamError)(useAIKitService
+        ? api_1.aiKitApi.post('/api/v1/image/generations', input, {
+            responseType: options.responseType,
+            headers: { ...(0, auth_1.getRemoteComponentCallHeaders)(input) },
+        })
+        : // @ts-ignore
+            (0, component_1.call)({
+                name: 'ai-kit',
+                path: '/api/v1/image/generations',
+                data: input,
+                responseType: options === null || options === void 0 ? void 0 : options.responseType,
+                timeout: options === null || options === void 0 ? void 0 : options.timeout,
+            }));
+    if ((options === null || options === void 0 ? void 0 : options.responseType) === 'stream')
+        return response;
+    return response.data;
+}
+async function embeddings(input, { useAIKitService, ...options } = {}) {
+    const response = await (0, api_1.catchAndRethrowUpstreamError)(useAIKitService
+        ? api_1.aiKitApi.post('/api/v1/embeddings', input, {
+            responseType: options.responseType,
+            headers: { ...(0, auth_1.getRemoteComponentCallHeaders)(input) },
+        })
+        : (0, component_1.call)({
+            name: 'ai-kit',
+            path: '/api/v1/embeddings',
+            data: input,
+            responseType: options === null || options === void 0 ? void 0 : options.responseType,
+        }));
+    if ((options === null || options === void 0 ? void 0 : options.responseType) === 'stream')
+        return response;
+    return response.data;
+}
+async function audioTranscriptions(input, { useAIKitService, ...options } = {}) {
+    const form = new form_data_1.default();
+    for (const [key, val] of Object.entries(input)) {
+        form.append(key, val);
+    }
+    const response = await (0, api_1.catchAndRethrowUpstreamError)(useAIKitService
+        ? api_1.aiKitApi.post('/api/v1/audio/transcriptions', form, {
+            responseType: options.responseType,
+            headers: { ...(0, auth_1.getRemoteComponentCallHeaders)({}) },
+        })
+        : (() => {
+            const { iat, exp, sig, version } = (0, verify_sign_1.getSignData)({
+                data: {},
+                params: {},
+                method: 'post',
+                url: '/api/v1/audio/transcriptions',
+            });
+            return axios_1.default.post((0, ufo_1.joinURL)((0, component_1.getComponentWebEndpoint)('ai-kit'), '/api/v1/audio/transcriptions'), form, {
+                headers: {
+                    'x-component-sig': sig,
+                    'x-component-sig-iat': iat,
+                    'x-component-sig-exp': exp,
+                    'x-component-sig-version': version,
+                },
+                responseType: options === null || options === void 0 ? void 0 : options.responseType,
+            });
+        })());
+    if ((options === null || options === void 0 ? void 0 : options.responseType) === 'stream')
+        return response;
+    return response.data;
+}
+async function audioSpeech(input, { useAIKitService } = {}) {
+    const response = await (0, api_1.catchAndRethrowUpstreamError)(useAIKitService
+        ? api_1.aiKitApi.post('/api/v1/audio/speech', input, {
+            responseType: 'stream',
+            headers: { ...(0, auth_1.getRemoteComponentCallHeaders)(input) },
+        })
+        : (0, component_1.call)({
+            name: 'ai-kit',
+            path: '/api/v1/audio/speech',
+            data: input,
+            responseType: 'stream',
+        }));
+    return response;
+}

package/lib/cjs/api/call/v2.js

@@ -0,0 +1,101 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.chatCompletionsV2 = chatCompletionsV2;
+exports.imageGenerationsV2 = imageGenerationsV2;
+exports.embeddingsV2 = embeddingsV2;
+const web_1 = require("stream/web");
+const component_1 = require("@blocklet/sdk/lib/component");
+const json_stable_stringify_1 = __importDefault(require("json-stable-stringify"));
+const types_1 = require("../types");
+const auth_1 = require("../utils/auth");
+const event_stream_1 = require("../utils/event-stream");
+const api_1 = require("./api");
+async function chatCompletionsV2(input, { useAIKitService, ...options } = {}) {
+    const response = (0, api_1.catchAndRethrowUpstreamError)(useAIKitService
+        ? (0, api_1.aiKitApi)('/api/v2/chat/completions', {
+            responseType: 'stream',
+            method: 'POST',
+            data: (0, json_stable_stringify_1.default)(input),
+            headers: {
+                ...(0, auth_1.getRemoteComponentCallHeaders)(input, options.userDid),
+                Accept: 'text/event-stream',
+                'Content-Type': 'application/json',
+            },
+        })
+        : (0, component_1.call)({
+            name: 'ai-kit',
+            path: 'api/v2/chat/completions',
+            data: input,
+            responseType: 'stream',
+            headers: { Accept: 'text/event-stream' },
+        }));
+    if ((options === null || options === void 0 ? void 0 : options.responseType) === 'stream')
+        return response;
+    return new web_1.ReadableStream({
+        async start(controller) {
+            try {
+                const stream = (0, event_stream_1.readableToWeb)((await response).data)
+                    .pipeThrough(new web_1.TextDecoderStream())
+                    .pipeThrough(new event_stream_1.EventSourceParserStream());
+                for await (const chunk of stream) {
+                    if ((0, types_1.isChatCompletionError)(chunk)) {
+                        if (chunk.error.type) {
+                            const error = new Error(chunk.error.message);
+                            error.type = chunk.error.type;
+                            error.timestamp = chunk.error.timestamp;
+                            controller.error(error);
+                        }
+                        else {
+                            controller.error(new Error(chunk.error.message));
+                        }
+                        break;
+                    }
+                    controller.enqueue(chunk);
+                }
+            }
+            catch (error) {
+                controller.error(error);
+            }
+            finally {
+                controller.close();
+            }
+        },
+    });
+}
+async function imageGenerationsV2(input, { useAIKitService, ...options } = {}) {
+    const response = await (0, api_1.catchAndRethrowUpstreamError)(useAIKitService
+        ? api_1.aiKitApi.post('/api/v2/image/generations', input, {
+            responseType: options.responseType,
+            headers: { ...(0, auth_1.getRemoteComponentCallHeaders)(input, options.userDid) },
+        })
+        : // @ts-ignore
+            (0, component_1.call)({
+                name: 'ai-kit',
+                path: '/api/v2/image/generations',
+                data: input,
+                responseType: options === null || options === void 0 ? void 0 : options.responseType,
+                timeout: options === null || options === void 0 ? void 0 : options.timeout,
+            }));
+    if ((options === null || options === void 0 ? void 0 : options.responseType) === 'stream')
+        return response;
+    return response.data;
+}
+async function embeddingsV2(input, { useAIKitService, ...options } = {}) {
+    const response = await (0, api_1.catchAndRethrowUpstreamError)(useAIKitService
+        ? api_1.aiKitApi.post('/api/v2/embeddings', input, {
+            responseType: options.responseType,
+            headers: { ...(0, auth_1.getRemoteComponentCallHeaders)(input, options.userDid) },
+        })
+        : (0, component_1.call)({
+            name: 'ai-kit',
+            path: '/api/v2/embeddings',
+            data: input,
+            responseType: options === null || options === void 0 ? void 0 : options.responseType,
+        }));
+    if ((options === null || options === void 0 ? void 0 : options.responseType) === 'stream')
+        return response;
+    return response.data;
+}