@blaxel/core 0.2.0-dev1

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Beamlit, Inc
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/dist/agents/index.d.ts

@@ -0,0 +1,15 @@
+import { Agent } from "../client/index.js";
+declare class BlAgent {
+    agentName: string;
+    constructor(agentName: string);
+    get fallbackUrl(): import("url").URL | null;
+    get externalUrl(): import("url").URL;
+    get internalUrl(): import("url").URL;
+    get forcedUrl(): import("url").URL | null;
+    get url(): import("url").URL;
+    call(url: URL, input: Record<string, unknown> | string | undefined): Promise<Response>;
+    run(input: Record<string, unknown> | string | undefined): Promise<string>;
+}
+export declare const blAgent: (agentName: string) => BlAgent;
+export declare const getAgentMetadata: (agent: string) => Promise<Agent | null>;
+export {};

package/dist/agents/index.js

@@ -0,0 +1,114 @@
+"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 env_js_1 = require("../common/env.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 hash = (0, internal_js_1.getGlobalUniqueHash)(settings_js_1.settings.workspace, "agent", this.agentName);
+        return new URL(`${settings_js_1.settings.runInternalProtocol}://bl-${settings_js_1.settings.env}-${hash}.${settings_js_1.settings.runInternalHostname}`);
+    }
+    get forcedUrl() {
+        const envVar = this.agentName.replace(/-/g, "_").toUpperCase();
+        if (env_js_1.env[`BL_AGENT_${envVar}_URL`]) {
+            return new URL(env_js_1.env[`BL_AGENT_${envVar}_URL`]);
+        }
+        return null;
+    }
+    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/authentication/apikey.d.ts

@@ -0,0 +1,10 @@
+import { Credentials } from "./credentials.js";
+import { CredentialsType } from "./types.js";
+export declare class ApiKey extends Credentials {
+    private apiKey;
+    private credentials;
+    constructor(credentials: CredentialsType);
+    get workspace(): string;
+    get authorization(): string;
+    get token(): string;
+}

package/dist/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/authentication/clientcredentials.d.ts

@@ -0,0 +1,15 @@
+import { Credentials } from "./credentials.js";
+import { CredentialsType } from "./types.js";
+export declare class ClientCredentials extends Credentials {
+    private clientCredentials;
+    private accessToken;
+    private credentials;
+    private currentPromise;
+    constructor(credentials: CredentialsType);
+    get workspace(): string;
+    needRefresh(): boolean;
+    authenticate(): Promise<void>;
+    process(): Promise<void>;
+    get authorization(): string;
+    get token(): string;
+}

package/dist/authentication/clientcredentials.js

@@ -0,0 +1,70 @@
+"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.process();
+        return this.currentPromise;
+    }
+    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/authentication/credentials.d.ts

@@ -0,0 +1,6 @@
+export declare class Credentials {
+    authenticate(): Promise<void>;
+    get workspace(): string;
+    get authorization(): string;
+    get token(): string;
+}

package/dist/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/authentication/deviceMode.d.ts

@@ -0,0 +1,15 @@
+import { CredentialsType } from "./types.js";
+export declare class DeviceMode {
+    private refreshToken;
+    private deviceCode;
+    private accessToken;
+    private credentials;
+    private currentPromise;
+    constructor(credentials: CredentialsType);
+    get workspace(): string;
+    needRefresh(): boolean;
+    authenticate(): Promise<void>;
+    process(): Promise<void>;
+    get authorization(): string;
+    get token(): string;
+}

package/dist/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/authentication/index.d.ts

@@ -0,0 +1,3 @@
+import { Credentials } from "./credentials.js";
+import { DeviceMode } from "./deviceMode.js";
+export declare function authentication(): Credentials | DeviceMode;

package/dist/authentication/index.js

@@ -0,0 +1,63 @@
+"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 path_1 = require("path");
+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) {
+        return null;
+    }
+    try {
+        const homeDir = node_js_1.os.homedir();
+        const config = node_js_1.fs.readFileSync((0, path_1.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/authentication/types.d.ts

@@ -0,0 +1,9 @@
+export type CredentialsType = {
+    apiKey?: string;
+    clientCredentials?: string;
+    refresh_token?: string;
+    access_token?: string;
+    device_code?: string;
+    expires_in?: number;
+    workspace?: string;
+};

package/dist/authentication/types.js

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

package/dist/cache/index.d.ts

@@ -0,0 +1 @@
+export declare function findFromCache(resource: string, name: string): Promise<unknown | null>;

package/dist/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/client/authentication.d.ts

@@ -0,0 +1,25 @@
+import { OptionsLegacyParser } from "@hey-api/client-fetch";
+type OauthTokenData = {
+    body: {
+        grant_type: 'client_credentials' | 'refresh_token' | 'device_code';
+        client_id?: string;
+        client_secret?: string;
+        device_code?: string;
+        refresh_token?: string;
+    };
+    authenticated?: boolean;
+};
+type OauthTokenResponse = {
+    access_token: string;
+    refresh_token: string;
+    expires_in: number;
+    token_type: string;
+};
+type OauthTokenError = {
+    error: string;
+};
+/**
+ * Get a new oauth token
+ */
+export declare const oauthToken: <ThrowOnError extends boolean = false>(options: OptionsLegacyParser<OauthTokenData, ThrowOnError>) => import("@hey-api/client-fetch").RequestResult<OauthTokenResponse, OauthTokenError, ThrowOnError>;
+export {};

package/dist/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/client/client.d.ts

@@ -0,0 +1,2 @@
+export * from './index.js';
+export { _Function as Function } from './types.gen.js';

package/dist/client/client.gen.d.ts

@@ -0,0 +1,12 @@
+import { type Config, type ClientOptions as DefaultClientOptions } from '@hey-api/client-fetch';
+import type { ClientOptions } from './types.gen.js';
+/**
+ * The `createClientConfig()` function will be called on client initialization
+ * and the returned object will become the client's initial configuration.
+ *
+ * You may want to initialize your client this way instead of calling
+ * `setConfig()`. This is useful for example if you're using Next.js
+ * to ensure your client always has the correct values.
+ */
+export type CreateClientConfig<T extends DefaultClientOptions = ClientOptions> = (override?: Config<DefaultClientOptions & T>) => Config<Required<DefaultClientOptions> & T>;
+export declare const client: import("@hey-api/client-fetch").Client;

package/dist/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/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/client/index.d.ts

@@ -0,0 +1,2 @@
+export * from './types.gen.js';
+export * from './sdk.gen.js';

package/dist/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/client/interceptors.d.ts

@@ -0,0 +1,3 @@
+type Interceptor = (request: Request, options: Record<string, unknown>) => Promise<Request | Response>;
+export declare const interceptors: Interceptor[];
+export {};

package/dist/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;
+    },
+];