@blaxel/vercel 0.2.10 → 0.2.11-dev.89

package/dist/model.js

@@ -3,6 +3,7 @@ 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");
@@ -17,10 +18,13 @@ const blModel = async (model, options) => {
     const type = modelData?.spec?.runtime?.type || "openai";
     const modelId = modelData?.spec?.runtime?.model || "gpt-4o";
     try {
-        if (type === "google") {
+        if (type === "gemini") {
             return (0, google_1.createGoogleGenerativeAI)({
                 apiKey: core_1.settings.token,
-                baseURL: `${url}/v1`,
+                headers: core_1.settings.headers,
+                fetch: (_, options) => {
+                    return fetch(`${url}/v1beta/models/${modelId}:generateContent`, options);
+                },
                 ...options,
             })(modelId);
         }
@@ -46,6 +50,13 @@ const blModel = async (model, options) => {
                 ...options,
             })(modelId);
         }
+        else if (type === "cohere") {
+            return (0, cohere_1.createCohere)({
+                apiKey: core_1.settings.token,
+                baseURL: `${url}/v2`,
+                ...options,
+            })(modelId);
+        }
         return (0, openai_1.createOpenAI)({
             apiKey: core_1.settings.token,
             baseURL: `${url}/v1`,

package/dist/tools.d.ts

@@ -1,3 +1,3 @@
 import type { Tool } from "ai";
-export declare const blTool: (name: string) => Promise<Record<string, Tool>>;
-export declare const blTools: (names: string[]) => Promise<Record<string, Tool>>;
+export declare const blTool: (name: string, ms?: number) => Promise<Record<string, Tool>>;
+export declare const blTools: (names: string[], ms?: number) => Promise<Record<string, Tool>>;

package/dist/tools.js

@@ -3,10 +3,10 @@ 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) => {
+const blTool = async (name, ms) => {
     try {
         const toolFormated = {};
-        const blaxelTool = await (0, core_1.getTool)(name);
+        const blaxelTool = await (0, core_1.getTool)(name, ms);
         for (const t of blaxelTool) {
             const toolInstance = (0, ai_1.tool)({
                 description: t.description,
@@ -23,8 +23,8 @@ const blTool = async (name) => {
     }
 };
 exports.blTool = blTool;
-const blTools = async (names) => {
-    const toolArrays = await Promise.all(names.map(exports.blTool));
+const blTools = async (names, ms) => {
+    const toolArrays = await Promise.all(names.map((n) => (0, exports.blTool)(n, ms)));
     const toolFormated = {};
     for (const toolServer of toolArrays) {
         for (const toolName in toolServer) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blaxel/vercel",
-  "version": "0.2.10",
+  "version": "0.2.11-dev.89",
   "description": "Blaxel SDK for TypeScript",
   "license": "MIT",
   "author": "Blaxel, INC (https://blaxel.ai)",
@@ -55,11 +55,12 @@
   "dependencies": {
     "@ai-sdk/anthropic": "^1.2.10",
     "@ai-sdk/cerebras": "^0.2.13",
+    "@ai-sdk/cohere": "^1.2.10",
     "@ai-sdk/google": "^1.2.14",
     "@ai-sdk/groq": "^1.2.8",
     "@ai-sdk/openai": "^1.3.21",
     "ai": "^4.3.13",
-    "@blaxel/core": "0.2.10"
+    "@blaxel/core": "0.2.11-dev.89"
   },
   "devDependencies": {
     "@eslint/js": "^9.26.0",