@blaxel/core 0.2.43-dev.196 → 0.2.43-dev.199

package/dist/cjs-browser/agents/index.js

@@ -0,0 +1,109 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getAgentMetadata = exports.blAgent = void 0;
+const index_js_1 = require("../cache/index.js");
+const index_js_2 = require("../client/index.js");
+const internal_js_1 = require("../common/internal.js");
+const logger_js_1 = require("../common/logger.js");
+const settings_js_1 = require("../common/settings.js");
+const telemetry_js_1 = require("../telemetry/telemetry.js");
+class BlAgent {
+    agentName;
+    constructor(agentName) {
+        this.agentName = agentName;
+    }
+    get fallbackUrl() {
+        if (this.externalUrl != this.url) {
+            return this.externalUrl;
+        }
+        return null;
+    }
+    get externalUrl() {
+        return new URL(`${settings_js_1.settings.runUrl}/${settings_js_1.settings.workspace}/agents/${this.agentName}`);
+    }
+    get internalUrl() {
+        const url = (0, internal_js_1.generateInternalUrl)(settings_js_1.settings.workspace, "agent", this.agentName, settings_js_1.settings.env, settings_js_1.settings.runInternalProtocol, settings_js_1.settings.runInternalHostname, settings_js_1.settings.blCloud, settings_js_1.settings.workspaceId);
+        return new URL(url);
+    }
+    get forcedUrl() {
+        return (0, internal_js_1.getForcedUrl)('function', this.agentName);
+    }
+    get url() {
+        if (this.forcedUrl)
+            return this.forcedUrl;
+        if (settings_js_1.settings.runInternalHostname)
+            return this.internalUrl;
+        return this.externalUrl;
+    }
+    async call(url, input) {
+        let body = input;
+        if (typeof body != "string") {
+            body = JSON.stringify(body);
+        }
+        const response = await fetch(url, {
+            method: "POST",
+            headers: {
+                ...settings_js_1.settings.headers,
+                "Content-Type": "application/json",
+            },
+            body,
+        });
+        return response;
+    }
+    async run(input) {
+        logger_js_1.logger.debug(`Agent Calling: ${this.agentName}`);
+        const span = (0, telemetry_js_1.startSpan)(this.agentName, {
+            attributes: {
+                "agent.name": this.agentName,
+                "agent.args": JSON.stringify(input),
+                "span.type": "agent.run",
+            },
+            isRoot: false,
+        });
+        try {
+            const response = await this.call(this.url, input);
+            span.setAttribute("agent.run.result", await response.text());
+            return await response.text();
+        }
+        catch (err) {
+            if (err instanceof Error) {
+                if (!this.fallbackUrl) {
+                    span.setAttribute("agent.run.error", err.stack);
+                    throw err;
+                }
+                try {
+                    const response = await this.call(this.fallbackUrl, input);
+                    span.setAttribute("agent.run.result", await response.text());
+                    return await response.text();
+                }
+                catch (err) {
+                    if (err instanceof Error) {
+                        span.setAttribute("agent.run.error", err.stack);
+                    }
+                    throw err;
+                }
+            }
+            throw err;
+        }
+        finally {
+            span.end();
+        }
+    }
+}
+const blAgent = (agentName) => {
+    return new BlAgent(agentName);
+};
+exports.blAgent = blAgent;
+const getAgentMetadata = async (agent) => {
+    const cacheData = await (0, index_js_1.findFromCache)("Agent", agent);
+    if (cacheData) {
+        return cacheData;
+    }
+    const { data } = await (0, index_js_2.getAgent)({
+        path: {
+            agentName: agent,
+        },
+    });
+    return data || null;
+};
+exports.getAgentMetadata = getAgentMetadata;

package/dist/cjs-browser/authentication/apikey.js

@@ -0,0 +1,24 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ApiKey = void 0;
+const env_js_1 = require("../common/env.js");
+const credentials_js_1 = require("./credentials.js");
+class ApiKey extends credentials_js_1.Credentials {
+    apiKey;
+    credentials;
+    constructor(credentials) {
+        super();
+        this.apiKey = credentials.apiKey || "";
+        this.credentials = credentials;
+    }
+    get workspace() {
+        return this.credentials.workspace || env_js_1.env.BL_WORKSPACE || "";
+    }
+    get authorization() {
+        return `Bearer ${this.apiKey}`;
+    }
+    get token() {
+        return this.apiKey;
+    }
+}
+exports.ApiKey = ApiKey;

package/dist/cjs-browser/authentication/clientcredentials.js

@@ -0,0 +1,85 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ClientCredentials = void 0;
+const jwt_decode_1 = require("jwt-decode");
+const authentication_js_1 = require("../client/authentication.js");
+const env_js_1 = require("../common/env.js");
+const credentials_js_1 = require("./credentials.js");
+class ClientCredentials extends credentials_js_1.Credentials {
+    clientCredentials;
+    accessToken;
+    credentials;
+    currentPromise;
+    constructor(credentials) {
+        super();
+        this.clientCredentials = credentials.clientCredentials || "";
+        this.credentials = credentials;
+        this.accessToken = "";
+        this.currentPromise = null;
+    }
+    get workspace() {
+        return this.credentials.workspace || env_js_1.env.BL_WORKSPACE || "";
+    }
+    needRefresh() {
+        if (this.currentPromise)
+            return false;
+        if (this.accessToken) {
+            const decoded = (0, jwt_decode_1.jwtDecode)(this.accessToken);
+            const { exp, iat } = decoded;
+            if (!exp || !iat)
+                return true;
+            const expDate = new Date(exp * 1000);
+            const iatDate = new Date(iat * 1000);
+            const nowDate = new Date();
+            const diff = expDate.getTime() - nowDate.getTime();
+            const iatDiff = expDate.getTime() - iatDate.getTime();
+            const ratio = diff / iatDiff;
+            return ratio < 0.5;
+        }
+        return true;
+    }
+    async authenticate() {
+        if (!this.needRefresh()) {
+            return this.currentPromise || Promise.resolve();
+        }
+        this.currentPromise = this.processWithRetry();
+        return this.currentPromise;
+    }
+    sleep(ms) {
+        return new Promise((resolve) => setTimeout(resolve, ms));
+    }
+    async processWithRetry(retry = 3) {
+        try {
+            return await this.process();
+        }
+        catch (error) {
+            if (retry > 0) {
+                await this.sleep(1000);
+                return this.processWithRetry(retry - 1);
+            }
+            throw error;
+        }
+    }
+    async process() {
+        const response = await (0, authentication_js_1.oauthToken)({
+            headers: {
+                Authorization: `Basic ${this.clientCredentials}`,
+            },
+            body: {
+                grant_type: "client_credentials",
+            },
+        });
+        if (response.error) {
+            throw new Error(response.error.error);
+        }
+        this.accessToken = response.data?.access_token || "";
+        this.currentPromise = null;
+    }
+    get authorization() {
+        return `Bearer ${this.accessToken}`;
+    }
+    get token() {
+        return this.accessToken;
+    }
+}
+exports.ClientCredentials = ClientCredentials;

package/dist/cjs-browser/authentication/credentials.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Credentials = void 0;
+const env_js_1 = require("../common/env.js");
+class Credentials {
+    async authenticate() { }
+    get workspace() {
+        return env_js_1.env.BL_WORKSPACE || "";
+    }
+    get authorization() {
+        return "";
+    }
+    get token() {
+        return "";
+    }
+}
+exports.Credentials = Credentials;

package/dist/cjs-browser/authentication/deviceMode.js

@@ -0,0 +1,70 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DeviceMode = void 0;
+const jwt_decode_1 = require("jwt-decode");
+const authentication_js_1 = require("../client/authentication.js");
+const env_js_1 = require("../common/env.js");
+class DeviceMode {
+    refreshToken;
+    deviceCode;
+    accessToken;
+    credentials;
+    currentPromise;
+    // private expireIn: number;
+    constructor(credentials) {
+        this.refreshToken = credentials.refresh_token || "";
+        this.deviceCode = credentials.device_code || "";
+        this.accessToken = credentials.access_token || "";
+        this.credentials = credentials;
+        this.currentPromise = null;
+    }
+    get workspace() {
+        return this.credentials.workspace || env_js_1.env.BL_WORKSPACE || "";
+    }
+    needRefresh() {
+        if (this.currentPromise)
+            return false;
+        if (this.accessToken) {
+            const decoded = (0, jwt_decode_1.jwtDecode)(this.accessToken);
+            const { exp, iat } = decoded;
+            if (!exp || !iat)
+                return true;
+            const expDate = new Date(exp * 1000);
+            const iatDate = new Date(iat * 1000);
+            const nowDate = new Date();
+            const diff = expDate.getTime() - nowDate.getTime();
+            const iatDiff = expDate.getTime() - iatDate.getTime();
+            const ratio = diff / iatDiff;
+            return ratio < 0.5;
+        }
+        return true;
+    }
+    async authenticate() {
+        if (!this.needRefresh()) {
+            return this.currentPromise || Promise.resolve();
+        }
+        this.currentPromise = this.process();
+        return this.currentPromise;
+    }
+    async process() {
+        const response = await (0, authentication_js_1.oauthToken)({
+            body: {
+                grant_type: "refresh_token",
+                device_code: this.deviceCode,
+                refresh_token: this.refreshToken,
+            },
+        });
+        if (response.error) {
+            throw new Error(response.error.error);
+        }
+        this.accessToken = response.data?.access_token || "";
+        this.currentPromise = null;
+    }
+    get authorization() {
+        return `Bearer ${this.accessToken}`;
+    }
+    get token() {
+        return this.accessToken;
+    }
+}
+exports.DeviceMode = DeviceMode;

package/dist/cjs-browser/authentication/index.js

@@ -0,0 +1,62 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.authentication = authentication;
+const yaml_1 = __importDefault(require("yaml"));
+const env_js_1 = require("../common/env.js");
+const node_js_1 = require("../common/node.js");
+const apikey_js_1 = require("./apikey.js");
+const clientcredentials_js_1 = require("./clientcredentials.js");
+const credentials_js_1 = require("./credentials.js");
+const deviceMode_js_1 = require("./deviceMode.js");
+function getCredentials() {
+    if (env_js_1.env.BL_API_KEY) {
+        return {
+            apiKey: env_js_1.env.BL_API_KEY,
+            workspace: env_js_1.env.BL_WORKSPACE,
+        };
+    }
+    if (env_js_1.env.BL_CLIENT_CREDENTIALS) {
+        return {
+            clientCredentials: env_js_1.env.BL_CLIENT_CREDENTIALS,
+            workspace: env_js_1.env.BL_WORKSPACE,
+        };
+    }
+    if (node_js_1.os === null || node_js_1.fs === null || node_js_1.path === null) {
+        return null;
+    }
+    try {
+        const homeDir = node_js_1.os.homedir();
+        const config = node_js_1.fs.readFileSync(node_js_1.path.join(homeDir, ".blaxel/config.yaml"), "utf8");
+        const configJson = yaml_1.default.parse(config);
+        const workspaceName = env_js_1.env.BL_WORKSPACE || configJson.context.workspace;
+        const credentials = configJson.workspaces.find((wk) => wk.name === workspaceName)?.credentials;
+        if (!credentials) {
+            return null;
+        }
+        credentials.workspace = workspaceName;
+        return credentials;
+    }
+    catch {
+        // If any error (e.g., running in browser), just return null
+        return null;
+    }
+}
+function authentication() {
+    const credentials = getCredentials();
+    if (!credentials) {
+        return new credentials_js_1.Credentials();
+    }
+    if (credentials.apiKey) {
+        return new apikey_js_1.ApiKey(credentials);
+    }
+    if (credentials.clientCredentials) {
+        return new clientcredentials_js_1.ClientCredentials(credentials);
+    }
+    if (credentials.device_code) {
+        return new deviceMode_js_1.DeviceMode(credentials);
+    }
+    return new credentials_js_1.Credentials();
+}

package/dist/cjs-browser/authentication/types.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/cjs-browser/cache/index.js

@@ -0,0 +1,26 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.findFromCache = findFromCache;
+const yaml_1 = __importDefault(require("yaml"));
+const node_js_1 = require("../common/node.js");
+const cache = new Map();
+try {
+    if (node_js_1.fs) {
+        const cacheString = node_js_1.fs.readFileSync(".cache.yaml", "utf8");
+        const cacheData = yaml_1.default.parseAllDocuments(cacheString);
+        for (const doc of cacheData) {
+            const jsonDoc = doc.toJSON();
+            const cacheKey = `${jsonDoc.kind}/${jsonDoc.metadata.name}`;
+            cache.set(cacheKey, jsonDoc);
+        }
+    }
+    /* eslint-disable */
+}
+catch (error) { }
+async function findFromCache(resource, name) {
+    const cacheKey = `${resource}/${name}`;
+    return cache.get(cacheKey);
+}

package/dist/cjs-browser/client/authentication.js

@@ -0,0 +1,15 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.oauthToken = void 0;
+const client_gen_js_1 = require("./client.gen.js");
+/**
+ * Get a new oauth token
+ */
+const oauthToken = (options) => {
+    options.authenticated = false;
+    return client_gen_js_1.client.post({
+        ...options,
+        url: '/oauth/token',
+    });
+};
+exports.oauthToken = oauthToken;

package/dist/cjs-browser/client/client.gen.js

@@ -0,0 +1,8 @@
+"use strict";
+// This file is auto-generated by @hey-api/openapi-ts
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.client = void 0;
+const client_fetch_1 = require("@hey-api/client-fetch");
+exports.client = (0, client_fetch_1.createClient)((0, client_fetch_1.createConfig)({
+    baseUrl: 'https://api.blaxel.ai/v0'
+}));

package/dist/cjs-browser/client/client.js

@@ -0,0 +1,17 @@
+"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("./index.js"), exports);

package/dist/cjs-browser/client/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 });
+// This file is auto-generated by @hey-api/openapi-ts
+__exportStar(require("./types.gen.js"), exports);
+__exportStar(require("./sdk.gen.js"), exports);

package/dist/cjs-browser/client/interceptors.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.interceptors = void 0;
+const settings_js_1 = require("../common/settings.js");
+exports.interceptors = [
+    // Authentication interceptor
+    async (request, options) => {
+        if (options.authenticated === false) {
+            return request;
+        }
+        await settings_js_1.settings.authenticate();
+        for (const header in settings_js_1.settings.headers) {
+            request.headers.set(header, settings_js_1.settings.headers[header]);
+        }
+        return request;
+    },
+];