@blaxel/core 0.2.36-dev.185 → 0.2.36-dev.190

package/dist/esm/agents/index.js

@@ -0,0 +1,104 @@
+import { findFromCache } from "../cache/index.js";
+import { getAgent } from "../client/index.js";
+import { getForcedUrl, getGlobalUniqueHash } from "../common/internal.js";
+import { logger } from "../common/logger.js";
+import { settings } from "../common/settings.js";
+import { startSpan } from '../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.runUrl}/${settings.workspace}/agents/${this.agentName}`);
+    }
+    get internalUrl() {
+        const hash = getGlobalUniqueHash(settings.workspace, "agent", this.agentName);
+        return new URL(`${settings.runInternalProtocol}://bl-${settings.env}-${hash}.${settings.runInternalHostname}`);
+    }
+    get forcedUrl() {
+        return getForcedUrl('function', this.agentName);
+    }
+    get url() {
+        if (this.forcedUrl)
+            return this.forcedUrl;
+        if (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.headers,
+                "Content-Type": "application/json",
+            },
+            body,
+        });
+        return response;
+    }
+    async run(input) {
+        logger.debug(`Agent Calling: ${this.agentName}`);
+        const span = 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();
+        }
+    }
+}
+export const blAgent = (agentName) => {
+    return new BlAgent(agentName);
+};
+export const getAgentMetadata = async (agent) => {
+    const cacheData = await findFromCache("Agent", agent);
+    if (cacheData) {
+        return cacheData;
+    }
+    const { data } = await getAgent({
+        path: {
+            agentName: agent,
+        },
+    });
+    return data || null;
+};

package/dist/esm/authentication/apikey.js

@@ -0,0 +1,20 @@
+import { env } from "../common/env.js";
+import { Credentials } from "./credentials.js";
+export class ApiKey extends Credentials {
+    apiKey;
+    credentials;
+    constructor(credentials) {
+        super();
+        this.apiKey = credentials.apiKey || "";
+        this.credentials = credentials;
+    }
+    get workspace() {
+        return this.credentials.workspace || env.BL_WORKSPACE || "";
+    }
+    get authorization() {
+        return `Bearer ${this.apiKey}`;
+    }
+    get token() {
+        return this.apiKey;
+    }
+}

package/dist/esm/authentication/clientcredentials.js

@@ -0,0 +1,81 @@
+import { jwtDecode } from "jwt-decode";
+import { oauthToken } from "../client/authentication.js";
+import { env } from "../common/env.js";
+import { Credentials } from "./credentials.js";
+export class ClientCredentials extends 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.BL_WORKSPACE || "";
+    }
+    needRefresh() {
+        if (this.currentPromise)
+            return false;
+        if (this.accessToken) {
+            const decoded = 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 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;
+    }
+}

package/dist/esm/authentication/credentials.js

@@ -0,0 +1,13 @@
+import { env } from "../common/env.js";
+export class Credentials {
+    async authenticate() { }
+    get workspace() {
+        return env.BL_WORKSPACE || "";
+    }
+    get authorization() {
+        return "";
+    }
+    get token() {
+        return "";
+    }
+}

package/dist/esm/authentication/deviceMode.js

@@ -0,0 +1,66 @@
+import { jwtDecode } from "jwt-decode";
+import { oauthToken } from "../client/authentication.js";
+import { env } from "../common/env.js";
+export 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.BL_WORKSPACE || "";
+    }
+    needRefresh() {
+        if (this.currentPromise)
+            return false;
+        if (this.accessToken) {
+            const decoded = 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 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;
+    }
+}

package/dist/esm/authentication/index.js

@@ -0,0 +1,56 @@
+import yaml from 'yaml';
+import { env } from "../common/env.js";
+import { fs, os, path } from "../common/node.js";
+import { ApiKey } from "./apikey.js";
+import { ClientCredentials } from "./clientcredentials.js";
+import { Credentials } from "./credentials.js";
+import { DeviceMode } from "./deviceMode.js";
+function getCredentials() {
+    if (env.BL_API_KEY) {
+        return {
+            apiKey: env.BL_API_KEY,
+            workspace: env.BL_WORKSPACE,
+        };
+    }
+    if (env.BL_CLIENT_CREDENTIALS) {
+        return {
+            clientCredentials: env.BL_CLIENT_CREDENTIALS,
+            workspace: env.BL_WORKSPACE,
+        };
+    }
+    if (os === null || fs === null || path === null) {
+        return null;
+    }
+    try {
+        const homeDir = os.homedir();
+        const config = fs.readFileSync(path.join(homeDir, ".blaxel/config.yaml"), "utf8");
+        const configJson = yaml.parse(config);
+        const workspaceName = 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;
+    }
+}
+export function authentication() {
+    const credentials = getCredentials();
+    if (!credentials) {
+        return new Credentials();
+    }
+    if (credentials.apiKey) {
+        return new ApiKey(credentials);
+    }
+    if (credentials.clientCredentials) {
+        return new ClientCredentials(credentials);
+    }
+    if (credentials.device_code) {
+        return new DeviceMode(credentials);
+    }
+    return new Credentials();
+}

package/dist/esm/authentication/types.js

@@ -0,0 +1 @@
+export {};

package/dist/esm/cache/index.js

@@ -0,0 +1,20 @@
+import yaml from "yaml";
+import { fs } from "../common/node.js";
+const cache = new Map();
+try {
+    if (fs) {
+        const cacheString = fs.readFileSync(".cache.yaml", "utf8");
+        const cacheData = yaml.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) { }
+export async function findFromCache(resource, name) {
+    const cacheKey = `${resource}/${name}`;
+    return cache.get(cacheKey);
+}

package/dist/esm/client/authentication.js

@@ -0,0 +1,11 @@
+import { client } from "./client.gen.js";
+/**
+ * Get a new oauth token
+ */
+export const oauthToken = (options) => {
+    options.authenticated = false;
+    return client.post({
+        ...options,
+        url: '/oauth/token',
+    });
+};

package/dist/esm/client/client.gen.js

@@ -0,0 +1,5 @@
+// This file is auto-generated by @hey-api/openapi-ts
+import { createClient, createConfig } from '@hey-api/client-fetch';
+export const client = createClient(createConfig({
+    baseUrl: 'https://api.blaxel.ai/v0'
+}));

package/dist/esm/client/client.js

@@ -0,0 +1 @@
+export * from './index.js';

package/dist/esm/client/index.js

@@ -0,0 +1,3 @@
+// This file is auto-generated by @hey-api/openapi-ts
+export * from './types.gen.js';
+export * from './sdk.gen.js';

package/dist/esm/client/interceptors.js

@@ -0,0 +1,14 @@
+import { settings } from "../common/settings.js";
+export const interceptors = [
+    // Authentication interceptor
+    async (request, options) => {
+        if (options.authenticated === false) {
+            return request;
+        }
+        await settings.authenticate();
+        for (const header in settings.headers) {
+            request.headers.set(header, settings.headers[header]);
+        }
+        return request;
+    },
+];