@blaxel/vercel 0.2.49-dev.213 → 0.2.49-dev1

package/dist/model.js

@@ -0,0 +1,94 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.blModel = void 0;
+const anthropic_1 = require("@ai-sdk/anthropic");
+const cerebras_1 = require("@ai-sdk/cerebras");
+const cohere_1 = require("@ai-sdk/cohere");
+const google_1 = require("@ai-sdk/google");
+const groq_1 = require("@ai-sdk/groq");
+const openai_1 = require("@ai-sdk/openai");
+const core_1 = require("@blaxel/core");
+const blModel = async (model, options) => {
+    const url = `${core_1.settings.runUrl}/${core_1.settings.workspace}/models/${model}`;
+    const modelData = await (0, core_1.getModelMetadata)(model);
+    if (!modelData) {
+        throw new Error(`Model ${model} not found`);
+    }
+    await (0, core_1.authenticate)();
+    const type = modelData?.spec?.runtime?.type || "openai";
+    const modelId = modelData?.spec?.runtime?.model || "gpt-4o";
+    // Custom fetch function that refreshes authentication on each request
+    const authenticatedFetch = async (input, init) => {
+        await (0, core_1.authenticate)();
+        const headers = {
+            ...init?.headers,
+            ...core_1.settings.headers,
+        };
+        return fetch(input, {
+            ...init,
+            headers,
+        });
+    };
+    try {
+        if (type === "gemini") {
+            return (0, google_1.createGoogleGenerativeAI)({
+                apiKey: "replaced",
+                fetch: async (_, options) => {
+                    await (0, core_1.authenticate)();
+                    const headers = {
+                        ...options?.headers,
+                        ...core_1.settings.headers,
+                    };
+                    return fetch(`${url}/v1beta/models/${modelId}:generateContent`, {
+                        ...options,
+                        headers,
+                    });
+                },
+                ...options,
+            })(modelId);
+        }
+        else if (type === "anthropic") {
+            return (0, anthropic_1.createAnthropic)({
+                apiKey: "replaced",
+                baseURL: `${url}/v1`,
+                fetch: authenticatedFetch,
+                ...options,
+            })(modelId);
+        }
+        else if (type === "groq") {
+            return (0, groq_1.createGroq)({
+                apiKey: "replaced",
+                baseURL: `${url}`,
+                fetch: authenticatedFetch,
+                ...options,
+            })(modelId);
+        }
+        else if (type === "cerebras") {
+            return (0, cerebras_1.createCerebras)({
+                apiKey: "replaced",
+                baseURL: `${url}/v1`,
+                fetch: authenticatedFetch,
+                ...options,
+            })(modelId);
+        }
+        else if (type === "cohere") {
+            return (0, cohere_1.createCohere)({
+                apiKey: "replaced",
+                baseURL: `${url}/v2`,
+                fetch: authenticatedFetch,
+                ...options,
+            })(modelId);
+        }
+        return (0, openai_1.createOpenAI)({
+            apiKey: "replaced",
+            baseURL: `${url}/v1`,
+            fetch: authenticatedFetch,
+            ...options,
+        })(modelId);
+    }
+    catch (err) {
+        (0, core_1.handleDynamicImportError)(err);
+        throw err;
+    }
+};
+exports.blModel = blModel;

package/dist/tools.d.ts

@@ -0,0 +1,4 @@
+import { ToolOptions } from "@blaxel/core/tools/mcpTool";
+import type { Tool } from "ai";
+export declare const blTool: (name: string, options?: ToolOptions | number) => Promise<Record<string, Tool>>;
+export declare const blTools: (names: string[], options?: ToolOptions | number) => Promise<Record<string, Tool>>;

package/dist/tools.js

@@ -0,0 +1,37 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.blTools = exports.blTool = void 0;
+const core_1 = require("@blaxel/core");
+const ai_1 = require("ai");
+const blTool = async (name, options) => {
+    try {
+        const toolFormated = {};
+        const blaxelTool = await (0, core_1.getTool)(name, options);
+        for (const t of blaxelTool) {
+            // @ts-ignore - Type instantiation depth issue with ai package in some environments
+            const toolInstance = (0, ai_1.tool)({
+                description: t.description,
+                parameters: t.inputSchema,
+                execute: t.call.bind(t),
+            });
+            toolFormated[t.name] = toolInstance;
+        }
+        return toolFormated;
+    }
+    catch (err) {
+        (0, core_1.handleDynamicImportError)(err);
+        throw err;
+    }
+};
+exports.blTool = blTool;
+const blTools = async (names, options) => {
+    const toolArrays = await Promise.all(names.map((n) => (0, exports.blTool)(n, options)));
+    const toolFormated = {};
+    for (const toolServer of toolArrays) {
+        for (const toolName in toolServer) {
+            toolFormated[toolName] = toolServer[toolName];
+        }
+    }
+    return toolFormated;
+};
+exports.blTools = blTools;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blaxel/vercel",
-  "version": "0.2.49-dev.213",
+  "version": "0.2.49-dev1",
   "description": "Blaxel SDK for TypeScript",
   "license": "MIT",
   "author": "Blaxel, INC (https://blaxel.ai)",
@@ -48,7 +48,7 @@
     "@ai-sdk/openai": "^2.0.57",
     "@ai-sdk/provider": "^2.0.0",
     "ai": "^5.0.82",
-    "@blaxel/core": "0.2.49-dev.213"
+    "@blaxel/core": "0.2.49-dev1"
   },
   "devDependencies": {
     "@eslint/js": "^9.26.0",