@upstash/qstash 2.6.4-workflow-alpha.2 → 2.6.4

package/{chunk-Z3TALRVS.js → chunk-MZSHBRF6.js}

@@ -1,484 +1,4 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class; var _class2;// src/receiver.ts
-var _jose = require('jose'); var jose = _interopRequireWildcard(_jose);
-var _cryptojs = require('crypto-js'); var _cryptojs2 = _interopRequireDefault(_cryptojs);
-var SignatureError = class extends Error {
-  constructor(message) {
-    super(message);
-    this.name = "SignatureError";
-  }
-};
-var Receiver = class {
-  
-  
-  constructor(config) {
-    this.currentSigningKey = config.currentSigningKey;
-    this.nextSigningKey = config.nextSigningKey;
-  }
-  /**
-   * Verify the signature of a request.
-   *
-   * Tries to verify the signature with the current signing key.
-   * If that fails, maybe because you have rotated the keys recently, it will
-   * try to verify the signature with the next signing key.
-   *
-   * If that fails, the signature is invalid and a `SignatureError` is thrown.
-   */
-  async verify(request) {
-    const isValid = await this.verifyWithKey(this.currentSigningKey, request);
-    if (isValid) {
-      return true;
-    }
-    return this.verifyWithKey(this.nextSigningKey, request);
-  }
-  /**
-   * Verify signature with a specific signing key
-   */
-  async verifyWithKey(key, request) {
-    const jwt = await jose.jwtVerify(request.signature, new TextEncoder().encode(key), {
-      issuer: "Upstash",
-      clockTolerance: request.clockTolerance
-    }).catch((error) => {
-      throw new SignatureError(error.message);
-    });
-    const p = jwt.payload;
-    if (request.url !== void 0 && p.sub !== request.url) {
-      throw new SignatureError(`invalid subject: ${p.sub}, want: ${request.url}`);
-    }
-    const bodyHash = _cryptojs2.default.SHA256(request.body).toString(_cryptojs2.default.enc.Base64url);
-    const padding = new RegExp(/=+$/);
-    if (p.body.replace(padding, "") !== bodyHash.replace(padding, "")) {
-      throw new SignatureError(`body hash does not match, want: ${p.body}, got: ${bodyHash}`);
-    }
-    return true;
-  }
-};
-
-// src/client/error.ts
-var QstashError = class extends Error {
-  constructor(message) {
-    super(message);
-    this.name = "QstashError";
-  }
-};
-var QstashRatelimitError = class extends QstashError {
-  
-  
-  
-  constructor(args) {
-    super(`Exceeded burst rate limit. ${JSON.stringify(args)} `);
-    this.name = "QstashRatelimitError";
-    this.limit = args.limit;
-    this.remaining = args.remaining;
-    this.reset = args.reset;
-  }
-};
-var QstashChatRatelimitError = class extends QstashError {
-  
-  
-  
-  
-  
-  
-  constructor(args) {
-    super(`Exceeded chat rate limit. ${JSON.stringify(args)} `);
-    this.limitRequests = args["limit-requests"];
-    this.limitTokens = args["limit-tokens"];
-    this.remainingRequests = args["remaining-requests"];
-    this.remainingTokens = args["remaining-tokens"];
-    this.resetRequests = args["reset-requests"];
-    this.resetTokens = args["reset-tokens"];
-  }
-};
-var QstashDailyRatelimitError = class extends QstashError {
-  
-  
-  
-  constructor(args) {
-    super(`Exceeded daily rate limit. ${JSON.stringify(args)} `);
-    this.limit = args.limit;
-    this.remaining = args.remaining;
-    this.reset = args.reset;
-    this.name = "QstashChatRatelimitError";
-  }
-};
-var QstashWorkflowError = class extends QstashError {
-  constructor(message) {
-    super(message);
-    this.name = "QstashWorkflowError";
-  }
-};
-var QstashWorkflowAbort = class extends Error {
-  
-  
-  constructor(stepName, stepInfo) {
-    super(`Aborting workflow after executing step '${stepName}'.`);
-    this.name = "QstashWorkflowAbort";
-    this.stepName = stepName;
-    this.stepInfo = stepInfo;
-  }
-};
-
-// src/client/llm/chat.ts
-var Chat = (_class = class _Chat {
-  
-  
-  constructor(http, token) {;_class.prototype.__init.call(this);_class.prototype.__init2.call(this);_class.prototype.__init3.call(this);
-    this.http = http;
-    this.token = token;
-  }
-  static toChatRequest(request) {
-    const messages = [];
-    messages.push(
-      { role: "system", content: request.system },
-      { role: "user", content: request.user }
-    );
-    const chatRequest = { ...request, messages };
-    return chatRequest;
-  }
-  /**
-   * Calls the Upstash completions api given a ChatRequest.
-   *
-   * Returns a ChatCompletion or a stream of ChatCompletionChunks
-   * if stream is enabled.
-   *
-   * @param request ChatRequest with messages
-   * @returns Chat completion or stream
-   */
-  __init() {this.create = async (request) => {
-    if (request.provider.owner != "upstash")
-      return this.createThirdParty(request);
-    const body = JSON.stringify(request);
-    if ("stream" in request && request.stream) {
-      return this.http.requestStream({
-        path: ["llm", "v1", "chat", "completions"],
-        method: "POST",
-        headers: {
-          "Content-Type": "application/json",
-          Connection: "keep-alive",
-          Accept: "text/event-stream",
-          "Cache-Control": "no-cache",
-          Authorization: `Bearer ${this.token}`
-        },
-        body
-      });
-    }
-    return this.http.request({
-      path: ["llm", "v1", "chat", "completions"],
-      method: "POST",
-      headers: { "Content-Type": "application/json", Authorization: `Bearer ${this.token}` },
-      body
-    });
-  }}
-  /**
-   * Calls the Upstash completions api given a ChatRequest.
-   *
-   * Returns a ChatCompletion or a stream of ChatCompletionChunks
-   * if stream is enabled.
-   *
-   * @param request ChatRequest with messages
-   * @returns Chat completion or stream
-   */
-  __init2() {this.createThirdParty = async (request) => {
-    const { baseUrl, token, owner } = request.provider;
-    if (owner === "upstash")
-      throw new Error("Upstash is not 3rd party provider!");
-    delete request.provider;
-    delete request.system;
-    const body = JSON.stringify(request);
-    if ("stream" in request && request.stream) {
-      return this.http.requestStream({
-        path: ["v1", "chat", "completions"],
-        method: "POST",
-        headers: {
-          "Content-Type": "application/json",
-          Connection: "keep-alive",
-          Accept: "text/event-stream",
-          "Cache-Control": "no-cache",
-          Authorization: `Bearer ${token}`
-        },
-        body,
-        baseUrl
-      });
-    }
-    return this.http.request({
-      path: ["v1", "chat", "completions"],
-      method: "POST",
-      headers: {
-        "Content-Type": "application/json",
-        Authorization: `Bearer ${token}`
-      },
-      body,
-      baseUrl
-    });
-  }}
-  /**
-   * Calls the Upstash completions api given a PromptRequest.
-   *
-   * Returns a ChatCompletion or a stream of ChatCompletionChunks
-   * if stream is enabled.
-   *
-   * @param request PromptRequest with system and user messages.
-   *    Note that system parameter shouldn't be passed in the case of
-   *    mistralai/Mistral-7B-Instruct-v0.2 model.
-   * @returns Chat completion or stream
-   */
-  __init3() {this.prompt = async (request) => {
-    const chatRequest = _Chat.toChatRequest(request);
-    return this.create(chatRequest);
-  }}
-}, _class);
-
-// src/client/messages.ts
-var Messages = class {
-  
-  constructor(http) {
-    this.http = http;
-  }
-  /**
-   * Get a message
-   */
-  async get(messageId) {
-    const messagePayload = await this.http.request({
-      method: "GET",
-      path: ["v2", "messages", messageId]
-    });
-    const message = {
-      ...messagePayload,
-      urlGroup: messagePayload.topicName
-    };
-    return message;
-  }
-  /**
-   * Cancel a message
-   */
-  async delete(messageId) {
-    return await this.http.request({
-      method: "DELETE",
-      path: ["v2", "messages", messageId],
-      parseResponseAsJson: false
-    });
-  }
-  async deleteMany(messageIds) {
-    const result = await this.http.request({
-      method: "DELETE",
-      path: ["v2", "messages"],
-      headers: { "Content-Type": "application/json" },
-      body: JSON.stringify({ messageIds })
-    });
-    return result.cancelled;
-  }
-  async deleteAll() {
-    const result = await this.http.request({
-      method: "DELETE",
-      path: ["v2", "messages"]
-    });
-    return result.cancelled;
-  }
-};
-
-// src/client/utils.ts
-var isIgnoredHeader = (header) => {
-  const lowerCaseHeader = header.toLowerCase();
-  return lowerCaseHeader.startsWith("content-type") || lowerCaseHeader.startsWith("upstash-");
-};
-function prefixHeaders(headers) {
-  const keysToBePrefixed = [...headers.keys()].filter((key) => !isIgnoredHeader(key));
-  for (const key of keysToBePrefixed) {
-    const value = headers.get(key);
-    if (value !== null) {
-      headers.set(`Upstash-Forward-${key}`, value);
-    }
-    headers.delete(key);
-  }
-  return headers;
-}
-function processHeaders(request) {
-  const headers = prefixHeaders(new Headers(request.headers));
-  headers.set("Upstash-Method", _nullishCoalesce(request.method, () => ( "POST")));
-  if (request.delay !== void 0) {
-    headers.set("Upstash-Delay", `${request.delay.toFixed(0)}s`);
-  }
-  if (request.notBefore !== void 0) {
-    headers.set("Upstash-Not-Before", request.notBefore.toFixed(0));
-  }
-  if (request.deduplicationId !== void 0) {
-    headers.set("Upstash-Deduplication-Id", request.deduplicationId);
-  }
-  if (request.contentBasedDeduplication !== void 0) {
-    headers.set("Upstash-Content-Based-Deduplication", "true");
-  }
-  if (request.retries !== void 0) {
-    headers.set("Upstash-Retries", request.retries.toFixed(0));
-  }
-  if (request.callback !== void 0) {
-    headers.set("Upstash-Callback", request.callback);
-  }
-  if (request.failureCallback !== void 0) {
-    headers.set("Upstash-Failure-Callback", request.failureCallback);
-  }
-  if (request.timeout !== void 0) {
-    headers.set("Upstash-Timeout", `${request.timeout}s`);
-  }
-  return headers;
-}
-function getRequestPath(request) {
-  return _nullishCoalesce(_nullishCoalesce(_nullishCoalesce(request.url, () => ( request.urlGroup)), () => ( request.topic)), () => ( `api/${_optionalChain([request, 'access', _ => _.api, 'optionalAccess', _2 => _2.name])}`));
-}
-
-// src/client/schedules.ts
-var Schedules = class {
-  
-  constructor(http) {
-    this.http = http;
-  }
-  /**
-   * Create a schedule
-   */
-  async create(request) {
-    const headers = prefixHeaders(new Headers(request.headers));
-    if (!headers.has("Content-Type")) {
-      headers.set("Content-Type", "application/json");
-    }
-    headers.set("Upstash-Cron", request.cron);
-    if (request.method !== void 0) {
-      headers.set("Upstash-Method", request.method);
-    }
-    if (request.delay !== void 0) {
-      headers.set("Upstash-Delay", `${request.delay.toFixed(0)}s`);
-    }
-    if (request.retries !== void 0) {
-      headers.set("Upstash-Retries", request.retries.toFixed(0));
-    }
-    if (request.callback !== void 0) {
-      headers.set("Upstash-Callback", request.callback);
-    }
-    if (request.failureCallback !== void 0) {
-      headers.set("Upstash-Failure-Callback", request.failureCallback);
-    }
-    if (request.timeout !== void 0) {
-      headers.set("Upstash-Timeout", `${request.timeout}s`);
-    }
-    return await this.http.request({
-      method: "POST",
-      headers,
-      path: ["v2", "schedules", request.destination],
-      body: request.body
-    });
-  }
-  /**
-   * Get a schedule
-   */
-  async get(scheduleId) {
-    return await this.http.request({
-      method: "GET",
-      path: ["v2", "schedules", scheduleId]
-    });
-  }
-  /**
-   * List your schedules
-   */
-  async list() {
-    return await this.http.request({
-      method: "GET",
-      path: ["v2", "schedules"]
-    });
-  }
-  /**
-   * Delete a schedule
-   */
-  async delete(scheduleId) {
-    return await this.http.request({
-      method: "DELETE",
-      path: ["v2", "schedules", scheduleId],
-      parseResponseAsJson: false
-    });
-  }
-  /**
-   * Pauses the schedule.
-   *
-   * A paused schedule will not deliver messages until
-   * it is resumed.
-   */
-  async pause({ schedule }) {
-    await this.http.request({
-      method: "PATCH",
-      path: ["v2", "schedules", schedule, "pause"],
-      parseResponseAsJson: false
-    });
-  }
-  /**
-   * Resumes the schedule.
-   */
-  async resume({ schedule }) {
-    await this.http.request({
-      method: "PATCH",
-      path: ["v2", "schedules", schedule, "resume"],
-      parseResponseAsJson: false
-    });
-  }
-};
-
-// src/client/url-groups.ts
-var UrlGroups = class {
-  
-  constructor(http) {
-    this.http = http;
-  }
-  /**
-   * Create a new url group with the given name and endpoints
-   */
-  async addEndpoints(request) {
-    await this.http.request({
-      method: "POST",
-      path: ["v2", "topics", request.name, "endpoints"],
-      headers: { "Content-Type": "application/json" },
-      body: JSON.stringify({ endpoints: request.endpoints }),
-      parseResponseAsJson: false
-    });
-  }
-  /**
-   * Remove endpoints from a url group.
-   */
-  async removeEndpoints(request) {
-    await this.http.request({
-      method: "DELETE",
-      path: ["v2", "topics", request.name, "endpoints"],
-      headers: { "Content-Type": "application/json" },
-      body: JSON.stringify({ endpoints: request.endpoints }),
-      parseResponseAsJson: false
-    });
-  }
-  /**
-   * Get a list of all url groups.
-   */
-  async list() {
-    return await this.http.request({
-      method: "GET",
-      path: ["v2", "topics"]
-    });
-  }
-  /**
-   * Get a single url group
-   */
-  async get(name) {
-    return await this.http.request({
-      method: "GET",
-      path: ["v2", "topics", name]
-    });
-  }
-  /**
-   * Delete a url group
-   */
-  async delete(name) {
-    return await this.http.request({
-      method: "DELETE",
-      path: ["v2", "topics", name],
-      parseResponseAsJson: false
-    });
-  }
-};
-
-// src/client/dlq.ts
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class; var _class2;// src/client/dlq.ts
 var DLQ = class {
   
   constructor(http) {
@@ -489,15 +9,15 @@ var DLQ = class {
    */
   async listMessages(options) {
     const filterPayload = {
-      ..._optionalChain([options, 'optionalAccess', _3 => _3.filter]),
-      topicName: _optionalChain([options, 'optionalAccess', _4 => _4.filter, 'optionalAccess', _5 => _5.urlGroup])
+      ..._optionalChain([options, 'optionalAccess', _ => _.filter]),
+      topicName: _optionalChain([options, 'optionalAccess', _2 => _2.filter, 'optionalAccess', _3 => _3.urlGroup])
     };
     const messagesPayload = await this.http.request({
       method: "GET",
       path: ["v2", "dlq"],
       query: {
-        cursor: _optionalChain([options, 'optionalAccess', _6 => _6.cursor]),
-        count: _optionalChain([options, 'optionalAccess', _7 => _7.count]),
+        cursor: _optionalChain([options, 'optionalAccess', _4 => _4.cursor]),
+        count: _optionalChain([options, 'optionalAccess', _5 => _5.count]),
         ...filterPayload
       }
     });
@@ -535,13 +55,60 @@ var DLQ = class {
   }
 };
 
+// src/client/error.ts
+var QstashError = class extends Error {
+  constructor(message) {
+    super(message);
+    this.name = "QstashError";
+  }
+};
+var QstashRatelimitError = class extends QstashError {
+  
+  
+  
+  constructor(args) {
+    super(`Exceeded burst rate limit. ${JSON.stringify(args)} `);
+    this.limit = args.limit;
+    this.remaining = args.remaining;
+    this.reset = args.reset;
+  }
+};
+var QstashChatRatelimitError = class extends QstashError {
+  
+  
+  
+  
+  
+  
+  constructor(args) {
+    super(`Exceeded chat rate limit. ${JSON.stringify(args)} `);
+    this.limitRequests = args["limit-requests"];
+    this.limitTokens = args["limit-tokens"];
+    this.remainingRequests = args["remaining-requests"];
+    this.remainingTokens = args["remaining-tokens"];
+    this.resetRequests = args["reset-requests"];
+    this.resetTokens = args["reset-tokens"];
+  }
+};
+var QstashDailyRatelimitError = class extends QstashError {
+  
+  
+  
+  constructor(args) {
+    super(`Exceeded daily rate limit. ${JSON.stringify(args)} `);
+    this.limit = args.limit;
+    this.remaining = args.remaining;
+    this.reset = args.reset;
+  }
+};
+
 // src/client/http.ts
-var HttpClient = (_class2 = class {
+var HttpClient = (_class = class {
   
   
   
   
-  constructor(config) {;_class2.prototype.__init4.call(this);_class2.prototype.__init5.call(this);
+  constructor(config) {;_class.prototype.__init.call(this);_class.prototype.__init2.call(this);
     this.baseUrl = config.baseUrl.replace(/\/$/, "");
     this.authorization = config.authorization;
     this.retry = // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition
@@ -549,8 +116,8 @@ var HttpClient = (_class2 = class {
       attempts: 1,
       backoff: () => 0
     } : {
-      attempts: _optionalChain([config, 'access', _8 => _8.retry, 'optionalAccess', _9 => _9.retries]) ? config.retry.retries + 1 : 5,
-      backoff: _nullishCoalesce(_optionalChain([config, 'access', _10 => _10.retry, 'optionalAccess', _11 => _11.backoff]), () => ( ((retryCount) => Math.exp(retryCount) * 50)))
+      attempts: _optionalChain([config, 'access', _6 => _6.retry, 'optionalAccess', _7 => _7.retries]) ? config.retry.retries + 1 : 5,
+      backoff: _nullishCoalesce(_optionalChain([config, 'access', _8 => _8.retry, 'optionalAccess', _9 => _9.backoff]), () => ( ((retryCount) => Math.exp(retryCount) * 50)))
     };
   }
   async request(request) {
@@ -590,7 +157,7 @@ var HttpClient = (_class2 = class {
       await reader.cancel();
     }
   }
-  __init4() {this.requestWithBackoff = async (request) => {
+  __init() {this.requestWithBackoff = async (request) => {
     const [url, requestOptions] = this.processRequest(request);
     let response = void 0;
     let error = void 0;
@@ -612,7 +179,7 @@ var HttpClient = (_class2 = class {
       error
     };
   }}
-  __init5() {this.processRequest = (request) => {
+  __init2() {this.processRequest = (request) => {
     const headers = new Headers(request.headers);
     if (!headers.has("Authorization")) {
       headers.set("Authorization", this.authorization);
@@ -662,30 +229,351 @@ var HttpClient = (_class2 = class {
       throw new QstashError(body.length > 0 ? body : `Error: status=${response.status}`);
     }
   }
+}, _class);
+
+// src/client/llm/providers.ts
+var setupAnalytics = (analytics, providerApiKey, providerBaseUrl, provider) => {
+  if (!analytics)
+    return {};
+  switch (analytics.name) {
+    case "helicone": {
+      switch (provider) {
+        case "upstash": {
+          return {
+            baseURL: "https://qstash.helicone.ai/llm/v1/chat/completions",
+            defaultHeaders: {
+              "Helicone-Auth": `Bearer ${analytics.token}`,
+              Authorization: `Bearer ${providerApiKey}`
+            }
+          };
+        }
+        default: {
+          return {
+            baseURL: "https://gateway.helicone.ai/v1/chat/completions",
+            defaultHeaders: {
+              "Helicone-Auth": `Bearer ${analytics.token}`,
+              "Helicone-Target-Url": providerBaseUrl,
+              Authorization: `Bearer ${providerApiKey}`
+            }
+          };
+        }
+      }
+    }
+    default: {
+      throw new Error("Unknown analytics provider");
+    }
+  }
+};
+var upstash = () => {
+  return {
+    owner: "upstash",
+    baseUrl: "https://qstash.upstash.io/llm",
+    token: ""
+  };
+};
+var openai = ({
+  token
+}) => {
+  return { token, owner: "openai", baseUrl: "https://api.openai.com" };
+};
+var custom = ({
+  baseUrl,
+  token
+}) => {
+  const trimmedBaseUrl = baseUrl.replace(/\/(v1\/)?chat\/completions$/, "");
+  return {
+    token,
+    owner: "custom",
+    baseUrl: trimmedBaseUrl
+  };
+};
+
+// src/client/llm/chat.ts
+var Chat = (_class2 = class _Chat {
+  
+  
+  constructor(http, token) {;_class2.prototype.__init3.call(this);_class2.prototype.__init4.call(this);_class2.prototype.__init5.call(this);
+    this.http = http;
+    this.token = token;
+  }
+  static toChatRequest(request) {
+    const messages = [];
+    messages.push(
+      { role: "system", content: request.system },
+      { role: "user", content: request.user }
+    );
+    const chatRequest = { ...request, messages };
+    return chatRequest;
+  }
+  /**
+   * Calls the Upstash completions api given a ChatRequest.
+   *
+   * Returns a ChatCompletion or a stream of ChatCompletionChunks
+   * if stream is enabled.
+   *
+   * @param request ChatRequest with messages
+   * @returns Chat completion or stream
+   */
+  __init3() {this.create = async (request) => {
+    if (request.provider.owner != "upstash")
+      return this.createThirdParty(request);
+    const body = JSON.stringify(request);
+    let baseUrl = void 0;
+    let headers = {
+      "Content-Type": "application/json",
+      Authorization: `Bearer ${this.token}`,
+      ..."stream" in request && request.stream ? {
+        Connection: "keep-alive",
+        Accept: "text/event-stream",
+        "Cache-Control": "no-cache"
+      } : {}
+    };
+    if (request.analytics) {
+      const { baseURL, defaultHeaders } = setupAnalytics(
+        { name: "helicone", token: request.analytics.token },
+        this.getAuthorizationToken(),
+        request.provider.baseUrl,
+        "upstash"
+      );
+      headers = { ...headers, ...defaultHeaders };
+      baseUrl = baseURL;
+    }
+    const path = request.analytics ? [] : ["llm", "v1", "chat", "completions"];
+    return "stream" in request && request.stream ? this.http.requestStream({
+      path,
+      method: "POST",
+      headers,
+      baseUrl,
+      body
+    }) : this.http.request({
+      path,
+      method: "POST",
+      headers,
+      baseUrl,
+      body
+    });
+  }}
+  /**
+   * Calls the Upstash completions api given a ChatRequest.
+   *
+   * Returns a ChatCompletion or a stream of ChatCompletionChunks
+   * if stream is enabled.
+   *
+   * @param request ChatRequest with messages
+   * @returns Chat completion or stream
+   */
+  // eslint-disable-next-line @typescript-eslint/require-await
+  __init4() {this.createThirdParty = async (request) => {
+    const { baseUrl, token, owner } = request.provider;
+    if (owner === "upstash")
+      throw new Error("Upstash is not 3rd party provider!");
+    delete request.provider;
+    delete request.system;
+    const analytics = request.analytics;
+    delete request.analytics;
+    const body = JSON.stringify(request);
+    const isAnalyticsEnabled = _optionalChain([analytics, 'optionalAccess', _12 => _12.name]) && analytics.token;
+    const analyticsConfig = _optionalChain([analytics, 'optionalAccess', _13 => _13.name]) && analytics.token ? setupAnalytics({ name: analytics.name, token: analytics.token }, token, baseUrl, owner) : { defaultHeaders: void 0, baseURL: baseUrl };
+    const isStream = "stream" in request && request.stream;
+    const headers = {
+      "Content-Type": "application/json",
+      Authorization: `Bearer ${token}`,
+      ...isStream ? {
+        Connection: "keep-alive",
+        Accept: "text/event-stream",
+        "Cache-Control": "no-cache"
+      } : {},
+      ...analyticsConfig.defaultHeaders
+    };
+    const response = await this.http[isStream ? "requestStream" : "request"]({
+      path: isAnalyticsEnabled ? [] : ["v1", "chat", "completions"],
+      method: "POST",
+      headers,
+      body,
+      baseUrl: analyticsConfig.baseURL
+    });
+    return response;
+  }}
+  // Helper method to get the authorization token
+  getAuthorizationToken() {
+    const authHeader = String(this.http.authorization);
+    const match = authHeader.match(/Bearer (.+)/);
+    if (!match) {
+      throw new Error("Invalid authorization header format");
+    }
+    return match[1];
+  }
+  /**
+   * Calls the Upstash completions api given a PromptRequest.
+   *
+   * Returns a ChatCompletion or a stream of ChatCompletionChunks
+   * if stream is enabled.
+   *
+   * @param request PromptRequest with system and user messages.
+   *    Note that system parameter shouldn't be passed in the case of
+   *    mistralai/Mistral-7B-Instruct-v0.2 model.
+   * @returns Chat completion or stream
+   */
+  __init5() {this.prompt = async (request) => {
+    const chatRequest = _Chat.toChatRequest(request);
+    return this.create(chatRequest);
+  }}
 }, _class2);
 
 // src/client/llm/utils.ts
-function appendLLMOptionsIfNeeded(request, headers) {
-  if (_optionalChain([request, 'access', _12 => _12.api, 'optionalAccess', _13 => _13.provider, 'optionalAccess', _14 => _14.owner]) === "upstash") {
-    request.api = { name: "llm" };
+function appendLLMOptionsIfNeeded(request, headers, http) {
+  if (!request.api)
+    return;
+  const provider = request.api.provider;
+  const analytics = request.api.analytics;
+  if (_optionalChain([provider, 'optionalAccess', _14 => _14.owner]) === "upstash") {
+    handleUpstashProvider(request, headers, http, analytics);
+    return;
+  }
+  if (!("provider" in request.api))
     return;
+  const { baseUrl, token } = validateProviderConfig(provider);
+  const analyticsConfig = analytics ? setupAnalytics({ name: analytics.name, token: analytics.token }, token, baseUrl, "custom") : void 0;
+  if (analyticsConfig) {
+    setAnalyticsHeaders(headers, analyticsConfig);
+    request.url = analyticsConfig.baseURL;
+  } else {
+    request.url = `${baseUrl}/v1/chat/completions`;
+    headers.set("Authorization", `Bearer ${token}`);
   }
-  if (request.api && "provider" in request.api) {
-    const provider = request.api.provider;
-    if (!_optionalChain([provider, 'optionalAccess', _15 => _15.baseUrl]))
-      throw new Error("baseUrl cannot be empty or undefined!");
-    if (!provider.token)
-      throw new Error("token cannot be empty or undefined!");
-    request.url = `${provider.baseUrl}/v1/chat/completions`;
-    headers.set("Authorization", `Bearer ${provider.token}`);
+}
+function handleUpstashProvider(request, headers, http, analytics) {
+  if (analytics) {
+    const analyticsConfig = setupAnalytics(
+      { name: analytics.name, token: analytics.token },
+      //@ts-expect-error hacky way to get bearer token
+      String(http.authorization).split("Bearer ")[1],
+      _optionalChain([request, 'access', _15 => _15.api, 'optionalAccess', _16 => _16.provider, 'optionalAccess', _17 => _17.baseUrl]),
+      "upstash"
+    );
+    setAnalyticsHeaders(headers, analyticsConfig);
+    request.url = analyticsConfig.baseURL;
+  } else {
+    request.api = { name: "llm" };
+  }
+}
+function validateProviderConfig(provider) {
+  if (!_optionalChain([provider, 'optionalAccess', _18 => _18.baseUrl]))
+    throw new Error("baseUrl cannot be empty or undefined!");
+  if (!provider.token)
+    throw new Error("token cannot be empty or undefined!");
+  return { baseUrl: provider.baseUrl, token: provider.token };
+}
+function setAnalyticsHeaders(headers, analyticsConfig) {
+  headers.set("Helicone-Auth", _nullishCoalesce(_optionalChain([analyticsConfig, 'access', _19 => _19.defaultHeaders, 'optionalAccess', _20 => _20["Helicone-Auth"]]), () => ( "")));
+  headers.set("Authorization", _nullishCoalesce(_optionalChain([analyticsConfig, 'access', _21 => _21.defaultHeaders, 'optionalAccess', _22 => _22.Authorization]), () => ( "")));
+  if (_optionalChain([analyticsConfig, 'access', _23 => _23.defaultHeaders, 'optionalAccess', _24 => _24["Helicone-Target-Url"]])) {
+    headers.set("Helicone-Target-Url", analyticsConfig.defaultHeaders["Helicone-Target-Url"]);
   }
 }
 function ensureCallbackPresent(request) {
-  if (_optionalChain([request, 'access', _16 => _16.api, 'optionalAccess', _17 => _17.name]) === "llm" && !request.callback) {
+  if (_optionalChain([request, 'access', _25 => _25.api, 'optionalAccess', _26 => _26.name]) === "llm" && !request.callback) {
     throw new TypeError("Callback cannot be undefined when using LLM");
   }
 }
 
+// src/client/messages.ts
+var Messages = class {
+  
+  constructor(http) {
+    this.http = http;
+  }
+  /**
+   * Get a message
+   */
+  async get(messageId) {
+    const messagePayload = await this.http.request({
+      method: "GET",
+      path: ["v2", "messages", messageId]
+    });
+    const message = {
+      ...messagePayload,
+      urlGroup: messagePayload.topicName
+    };
+    return message;
+  }
+  /**
+   * Cancel a message
+   */
+  async delete(messageId) {
+    return await this.http.request({
+      method: "DELETE",
+      path: ["v2", "messages", messageId],
+      parseResponseAsJson: false
+    });
+  }
+  async deleteMany(messageIds) {
+    const result = await this.http.request({
+      method: "DELETE",
+      path: ["v2", "messages"],
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ messageIds })
+    });
+    return result.cancelled;
+  }
+  async deleteAll() {
+    const result = await this.http.request({
+      method: "DELETE",
+      path: ["v2", "messages"]
+    });
+    return result.cancelled;
+  }
+};
+
+// src/client/utils.ts
+var isIgnoredHeader = (header) => {
+  const lowerCaseHeader = header.toLowerCase();
+  return lowerCaseHeader.startsWith("content-type") || lowerCaseHeader.startsWith("upstash-");
+};
+function prefixHeaders(headers) {
+  const keysToBePrefixed = [...headers.keys()].filter((key) => !isIgnoredHeader(key));
+  for (const key of keysToBePrefixed) {
+    const value = headers.get(key);
+    if (value !== null) {
+      headers.set(`Upstash-Forward-${key}`, value);
+    }
+    headers.delete(key);
+  }
+  return headers;
+}
+function processHeaders(request) {
+  const headers = prefixHeaders(new Headers(request.headers));
+  headers.set("Upstash-Method", _nullishCoalesce(request.method, () => ( "POST")));
+  if (request.delay !== void 0) {
+    headers.set("Upstash-Delay", `${request.delay.toFixed(0)}s`);
+  }
+  if (request.notBefore !== void 0) {
+    headers.set("Upstash-Not-Before", request.notBefore.toFixed(0));
+  }
+  if (request.deduplicationId !== void 0) {
+    headers.set("Upstash-Deduplication-Id", request.deduplicationId);
+  }
+  if (request.contentBasedDeduplication !== void 0) {
+    headers.set("Upstash-Content-Based-Deduplication", "true");
+  }
+  if (request.retries !== void 0) {
+    headers.set("Upstash-Retries", request.retries.toFixed(0));
+  }
+  if (request.callback !== void 0) {
+    headers.set("Upstash-Callback", request.callback);
+  }
+  if (request.failureCallback !== void 0) {
+    headers.set("Upstash-Failure-Callback", request.failureCallback);
+  }
+  if (request.timeout !== void 0) {
+    headers.set("Upstash-Timeout", `${request.timeout}s`);
+  }
+  return headers;
+}
+function getRequestPath(request) {
+  return _nullishCoalesce(_nullishCoalesce(_nullishCoalesce(request.url, () => ( request.urlGroup)), () => ( request.topic)), () => ( `api/${_optionalChain([request, 'access', _27 => _27.api, 'optionalAccess', _28 => _28.name])}`));
+}
+
 // src/client/queue.ts
 var Queue = class {
   
@@ -774,7 +662,7 @@ var Queue = class {
     const headers = prefixHeaders(new Headers(request.headers));
     headers.set("Content-Type", "application/json");
     ensureCallbackPresent(request);
-    appendLLMOptionsIfNeeded(request, headers);
+    appendLLMOptionsIfNeeded(request, headers, this.http);
     const response = await this.enqueue({
       ...request,
       body: JSON.stringify(request.body),
@@ -813,6 +701,159 @@ var Queue = class {
   }
 };
 
+// src/client/schedules.ts
+var Schedules = class {
+  
+  constructor(http) {
+    this.http = http;
+  }
+  /**
+   * Create a schedule
+   */
+  async create(request) {
+    const headers = prefixHeaders(new Headers(request.headers));
+    if (!headers.has("Content-Type")) {
+      headers.set("Content-Type", "application/json");
+    }
+    headers.set("Upstash-Cron", request.cron);
+    if (request.method !== void 0) {
+      headers.set("Upstash-Method", request.method);
+    }
+    if (request.delay !== void 0) {
+      headers.set("Upstash-Delay", `${request.delay.toFixed(0)}s`);
+    }
+    if (request.retries !== void 0) {
+      headers.set("Upstash-Retries", request.retries.toFixed(0));
+    }
+    if (request.callback !== void 0) {
+      headers.set("Upstash-Callback", request.callback);
+    }
+    if (request.failureCallback !== void 0) {
+      headers.set("Upstash-Failure-Callback", request.failureCallback);
+    }
+    if (request.timeout !== void 0) {
+      headers.set("Upstash-Timeout", `${request.timeout}s`);
+    }
+    return await this.http.request({
+      method: "POST",
+      headers,
+      path: ["v2", "schedules", request.destination],
+      body: request.body
+    });
+  }
+  /**
+   * Get a schedule
+   */
+  async get(scheduleId) {
+    return await this.http.request({
+      method: "GET",
+      path: ["v2", "schedules", scheduleId]
+    });
+  }
+  /**
+   * List your schedules
+   */
+  async list() {
+    return await this.http.request({
+      method: "GET",
+      path: ["v2", "schedules"]
+    });
+  }
+  /**
+   * Delete a schedule
+   */
+  async delete(scheduleId) {
+    return await this.http.request({
+      method: "DELETE",
+      path: ["v2", "schedules", scheduleId],
+      parseResponseAsJson: false
+    });
+  }
+  /**
+   * Pauses the schedule.
+   *
+   * A paused schedule will not deliver messages until
+   * it is resumed.
+   */
+  async pause({ schedule }) {
+    await this.http.request({
+      method: "PATCH",
+      path: ["v2", "schedules", schedule, "pause"],
+      parseResponseAsJson: false
+    });
+  }
+  /**
+   * Resumes the schedule.
+   */
+  async resume({ schedule }) {
+    await this.http.request({
+      method: "PATCH",
+      path: ["v2", "schedules", schedule, "resume"],
+      parseResponseAsJson: false
+    });
+  }
+};
+
+// src/client/url-groups.ts
+var UrlGroups = class {
+  
+  constructor(http) {
+    this.http = http;
+  }
+  /**
+   * Create a new url group with the given name and endpoints
+   */
+  async addEndpoints(request) {
+    await this.http.request({
+      method: "POST",
+      path: ["v2", "topics", request.name, "endpoints"],
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ endpoints: request.endpoints }),
+      parseResponseAsJson: false
+    });
+  }
+  /**
+   * Remove endpoints from a url group.
+   */
+  async removeEndpoints(request) {
+    await this.http.request({
+      method: "DELETE",
+      path: ["v2", "topics", request.name, "endpoints"],
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ endpoints: request.endpoints }),
+      parseResponseAsJson: false
+    });
+  }
+  /**
+   * Get a list of all url groups.
+   */
+  async list() {
+    return await this.http.request({
+      method: "GET",
+      path: ["v2", "topics"]
+    });
+  }
+  /**
+   * Get a single url group
+   */
+  async get(name) {
+    return await this.http.request({
+      method: "GET",
+      path: ["v2", "topics", name]
+    });
+  }
+  /**
+   * Delete a url group
+   */
+  async delete(name) {
+    return await this.http.request({
+      method: "DELETE",
+      path: ["v2", "topics", name],
+      parseResponseAsJson: false
+    });
+  }
+};
+
 // src/client/client.ts
 var Client = class {
   
@@ -873,7 +914,7 @@ var Client = class {
    * Create, read, update or delete queues.
    */
   queue(request) {
-    return new Queue(this.http, _optionalChain([request, 'optionalAccess', _18 => _18.queueName]));
+    return new Queue(this.http, _optionalChain([request, 'optionalAccess', _29 => _29.queueName]));
   }
   /**
    * Access the Chat API
@@ -901,7 +942,7 @@ var Client = class {
     const headers = prefixHeaders(new Headers(request.headers));
     headers.set("Content-Type", "application/json");
     ensureCallbackPresent(request);
-    appendLLMOptionsIfNeeded(request, headers);
+    appendLLMOptionsIfNeeded(request, headers, this.http);
     const response = await this.publish({
       ...request,
       headers,
@@ -932,8 +973,7 @@ var Client = class {
       },
       method: "POST"
     });
-    const arrayResposne = Array.isArray(response) ? response : [response];
-    return arrayResposne;
+    return response;
   }
   /**
    * Batch publish messages to QStash, serializing each body to JSON.
@@ -945,7 +985,7 @@ var Client = class {
       }
       message.headers = new Headers(message.headers);
       ensureCallbackPresent(message);
-      appendLLMOptionsIfNeeded(message, message.headers);
+      appendLLMOptionsIfNeeded(message, message.headers, this.http);
       message.headers.set("Content-Type", "application/json");
     }
     const response = await this.batch(request);
@@ -972,10 +1012,10 @@ var Client = class {
    */
   async events(request) {
     const query = {};
-    if (_optionalChain([request, 'optionalAccess', _19 => _19.cursor]) && request.cursor > 0) {
+    if (_optionalChain([request, 'optionalAccess', _30 => _30.cursor]) && request.cursor > 0) {
       query.cursor = request.cursor.toString();
     }
-    for (const [key, value] of Object.entries(_nullishCoalesce(_optionalChain([request, 'optionalAccess', _20 => _20.filter]), () => ( {})))) {
+    for (const [key, value] of Object.entries(_nullishCoalesce(_optionalChain([request, 'optionalAccess', _31 => _31.filter]), () => ( {})))) {
       if (typeof value === "number" && value < 0) {
         continue;
       }
@@ -1016,4 +1056,4 @@ var Client = class {
 
 
 
-exports.SignatureError = SignatureError; exports.Receiver = Receiver; exports.QstashError = QstashError; exports.QstashRatelimitError = QstashRatelimitError; exports.QstashChatRatelimitError = QstashChatRatelimitError; exports.QstashDailyRatelimitError = QstashDailyRatelimitError; exports.QstashWorkflowError = QstashWorkflowError; exports.QstashWorkflowAbort = QstashWorkflowAbort; exports.Chat = Chat; exports.Messages = Messages; exports.Schedules = Schedules; exports.UrlGroups = UrlGroups; exports.Client = Client;
+exports.QstashError = QstashError; exports.QstashRatelimitError = QstashRatelimitError; exports.QstashChatRatelimitError = QstashChatRatelimitError; exports.QstashDailyRatelimitError = QstashDailyRatelimitError; exports.setupAnalytics = setupAnalytics; exports.upstash = upstash; exports.openai = openai; exports.custom = custom; exports.Chat = Chat; exports.Messages = Messages; exports.Schedules = Schedules; exports.UrlGroups = UrlGroups; exports.Client = Client;