@botpress/zai 1.0.1-beta.1 → 1.0.1-beta.3

package/dist/esm/zai.js

@@ -1,161 +0,0 @@
-var __defProp = Object.defineProperty;
-var __defProps = Object.defineProperties;
-var __getOwnPropDescs = Object.getOwnPropertyDescriptors;
-var __getOwnPropSymbols = Object.getOwnPropertySymbols;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __propIsEnum = Object.prototype.propertyIsEnumerable;
-var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
-var __spreadValues = (a, b) => {
-  for (var prop in b || (b = {}))
-    if (__hasOwnProp.call(b, prop))
-      __defNormalProp(a, prop, b[prop]);
-  if (__getOwnPropSymbols)
-    for (var prop of __getOwnPropSymbols(b)) {
-      if (__propIsEnum.call(b, prop))
-        __defNormalProp(a, prop, b[prop]);
-    }
-  return a;
-};
-var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
-var __publicField = (obj, key, value) => __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
-import sdk from "@botpress/sdk";
-import { getWasmTokenizer } from "@botpress/wasm";
-import { TableAdapter } from "./adapters/botpress-table";
-import { MemoryAdapter } from "./adapters/memory";
-import { Models } from "./models";
-import { BotpressClient } from "./utils";
-const { z } = sdk;
-const ActiveLearning = z.object({
-  enable: z.boolean().describe("Whether to enable active learning").default(false),
-  tableName: z.string().regex(
-    /^[A-Za-z0-9_/-]{1,100}Table$/,
-    "Namespace must be alphanumeric and contain only letters, numbers, underscores, hyphens and slashes"
-  ).describe("The name of the table to store active learning tasks").default("ActiveLearningTable"),
-  taskId: z.string().regex(
-    /^[A-Za-z0-9_/-]{1,100}$/,
-    "Namespace must be alphanumeric and contain only letters, numbers, underscores, hyphens and slashes"
-  ).describe("The ID of the task").default("default")
-});
-const ZaiConfig = z.object({
-  client: BotpressClient,
-  userId: z.string().describe("The ID of the user consuming the API").optional(),
-  retry: z.object({ maxRetries: z.number().min(0).max(100) }).default({ maxRetries: 3 }),
-  modelId: z.custom(
-    (value) => {
-      if (typeof value !== "string" || !value.includes("__")) {
-        return false;
-      }
-      return true;
-    },
-    {
-      message: "Invalid model ID"
-    }
-  ).describe("The ID of the model you want to use").default("openai__gpt-4o-mini-2024-07-18"),
-  activeLearning: ActiveLearning.default({ enable: false }),
-  namespace: z.string().regex(
-    /^[A-Za-z0-9_/-]{1,100}$/,
-    "Namespace must be alphanumeric and contain only letters, numbers, underscores, hyphens and slashes"
-  ).default("zai")
-});
-const _Zai = class _Zai {
-  constructor(config) {
-    __publicField(this, "client");
-    __publicField(this, "originalConfig");
-    __publicField(this, "userId");
-    __publicField(this, "integration");
-    __publicField(this, "model");
-    __publicField(this, "retry");
-    __publicField(this, "Model");
-    __publicField(this, "namespace");
-    __publicField(this, "adapter");
-    __publicField(this, "activeLearning");
-    var _a;
-    this.originalConfig = config;
-    const parsed = ZaiConfig.parse(config);
-    this.client = parsed.client;
-    const [integration, modelId] = parsed.modelId.split("__");
-    if (!(integration == null ? void 0 : integration.length) || !(modelId == null ? void 0 : modelId.length)) {
-      throw new Error(`Invalid model ID: ${parsed.modelId}. Expected format: <integration>__<modelId>`);
-    }
-    this.integration = integration;
-    this.model = modelId;
-    this.namespace = parsed.namespace;
-    this.userId = parsed.userId;
-    this.retry = parsed.retry;
-    this.Model = Models.find((m) => m.id === parsed.modelId);
-    this.activeLearning = parsed.activeLearning;
-    this.adapter = ((_a = parsed.activeLearning) == null ? void 0 : _a.enable) ? new TableAdapter({ client: this.client, tableName: parsed.activeLearning.tableName }) : new MemoryAdapter([]);
-  }
-  /** @internal */
-  async callModel(props) {
-    let retries = this.retry.maxRetries;
-    while (retries-- >= 0) {
-      try {
-        return await this._callModel(props);
-      } catch (e) {
-        if (retries >= 0) {
-          await new Promise((resolve) => setTimeout(resolve, 1e3));
-        } else {
-          throw new Error("Failed to call model after multiple retries");
-        }
-      }
-    }
-    throw new Error("Failed to call model after multiple retries");
-  }
-  /** @internal */
-  async _callModel(props) {
-    let retries = this.retry.maxRetries;
-    do {
-      const start = Date.now();
-      const input = __spreadValues({
-        messages: [],
-        temperature: 0,
-        topP: 1,
-        model: { id: this.model },
-        userId: this.userId
-      }, props);
-      const { output } = await this.client.callAction({
-        type: `${this.integration}:generateContent`,
-        input
-      });
-      const latency = Date.now() - start;
-      return __spreadProps(__spreadValues({}, output), {
-        metadata: {
-          model: this.model,
-          latency,
-          cost: { input: output.usage.inputCost, output: output.usage.outputCost },
-          tokens: { input: output.usage.inputTokens, output: output.usage.outputTokens }
-        }
-      });
-    } while (--retries > 0);
-  }
-  async getTokenizer() {
-    var _a;
-    (_a = _Zai.tokenizer) != null ? _a : _Zai.tokenizer = await (async () => {
-      while (!getWasmTokenizer) {
-        await new Promise((resolve) => setTimeout(resolve, 25));
-      }
-      return getWasmTokenizer();
-    })();
-    return _Zai.tokenizer;
-  }
-  get taskId() {
-    if (!this.activeLearning.enable) {
-      return void 0;
-    }
-    return `${this.namespace}/${this.activeLearning.taskId}`.replace(/\/+/g, "/");
-  }
-  with(options) {
-    return new _Zai(__spreadValues(__spreadValues({}, this.originalConfig), options));
-  }
-  learn(taskId) {
-    return new _Zai(__spreadProps(__spreadValues({}, this.originalConfig), {
-      activeLearning: __spreadProps(__spreadValues({}, this.activeLearning), { taskId, enable: true })
-    }));
-  }
-};
-__publicField(_Zai, "tokenizer", null);
-let Zai = _Zai;
-export {
-  Zai
-};