@blaxel/core 0.2.49-dev.214 → 0.2.49-dev1

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,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 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() {
+        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/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,17 @@
+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>;
+    private sleep;
+    processWithRetry(retry?: number): Promise<void>;
+    process(): Promise<void>;
+    get authorization(): string;
+    get token(): string;
+}

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