ai 2.1.23 → 2.1.24

package/solid/dist/index.mjs

@@ -0,0 +1,386 @@
+var __defProp = Object.defineProperty;
+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 __async = (__this, __arguments, generator) => {
+  return new Promise((resolve, reject) => {
+    var fulfilled = (value) => {
+      try {
+        step(generator.next(value));
+      } catch (e) {
+        reject(e);
+      }
+    };
+    var rejected = (value) => {
+      try {
+        step(generator.throw(value));
+      } catch (e) {
+        reject(e);
+      }
+    };
+    var step = (x) => x.done ? resolve(x.value) : Promise.resolve(x.value).then(fulfilled, rejected);
+    step((generator = generator.apply(__this, __arguments)).next());
+  });
+};
+
+// solid/use-chat.ts
+import { createSignal } from "solid-js";
+import { useSWRStore } from "solid-swr-store";
+import { createSWRStore } from "swr-store";
+
+// shared/utils.ts
+import { customAlphabet } from "nanoid/non-secure";
+var nanoid = customAlphabet(
+  "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
+  7
+);
+function createChunkDecoder() {
+  const decoder = new TextDecoder();
+  return function(chunk) {
+    if (!chunk)
+      return "";
+    return decoder.decode(chunk, { stream: true });
+  };
+}
+
+// solid/use-chat.ts
+var uniqueId = 0;
+var store = {};
+var chatApiStore = createSWRStore({
+  get: (key) => __async(void 0, null, function* () {
+    var _a;
+    return (_a = store[key]) != null ? _a : [];
+  })
+});
+function useChat({
+  api = "/api/chat",
+  id,
+  initialMessages = [],
+  initialInput = "",
+  sendExtraMessageFields,
+  onResponse,
+  onFinish,
+  onError,
+  credentials,
+  headers,
+  body
+} = {}) {
+  const chatId = id || `chat-${uniqueId++}`;
+  const key = `${api}|${chatId}`;
+  const data = useSWRStore(chatApiStore, () => [key], {
+    initialData: initialMessages
+  });
+  const mutate = (data2) => {
+    store[key] = data2;
+    return chatApiStore.mutate([key], {
+      status: "success",
+      data: data2
+    });
+  };
+  const messages = data;
+  const [error, setError] = createSignal(void 0);
+  const [isLoading, setIsLoading] = createSignal(false);
+  let abortController = null;
+  function triggerRequest(messagesSnapshot, options) {
+    return __async(this, null, function* () {
+      try {
+        setIsLoading(true);
+        abortController = new AbortController();
+        const previousMessages = chatApiStore.get([key], {
+          shouldRevalidate: false
+        });
+        mutate(messagesSnapshot);
+        const res = yield fetch(api, {
+          method: "POST",
+          body: JSON.stringify(__spreadValues(__spreadValues({
+            messages: sendExtraMessageFields ? messagesSnapshot : messagesSnapshot.map(({ role, content }) => ({
+              role,
+              content
+            }))
+          }, body), options == null ? void 0 : options.body)),
+          headers: __spreadValues(__spreadValues({}, headers), options == null ? void 0 : options.headers),
+          signal: abortController.signal,
+          credentials
+        }).catch((err) => {
+          if (previousMessages.status === "success") {
+            mutate(previousMessages.data);
+          }
+          throw err;
+        });
+        if (onResponse) {
+          try {
+            yield onResponse(res);
+          } catch (err) {
+            throw err;
+          }
+        }
+        if (!res.ok) {
+          if (previousMessages.status === "success") {
+            mutate(previousMessages.data);
+          }
+          throw new Error(
+            (yield res.text()) || "Failed to fetch the chat response."
+          );
+        }
+        if (!res.body) {
+          throw new Error("The response body is empty.");
+        }
+        let result = "";
+        const createdAt = /* @__PURE__ */ new Date();
+        const replyId = nanoid();
+        const reader = res.body.getReader();
+        const decoder = createChunkDecoder();
+        while (true) {
+          const { done, value } = yield reader.read();
+          if (done) {
+            break;
+          }
+          result += decoder(value);
+          mutate([
+            ...messagesSnapshot,
+            {
+              id: replyId,
+              createdAt,
+              content: result,
+              role: "assistant"
+            }
+          ]);
+          if (abortController === null) {
+            reader.cancel();
+            break;
+          }
+        }
+        if (onFinish) {
+          onFinish({
+            id: replyId,
+            createdAt,
+            content: result,
+            role: "assistant"
+          });
+        }
+        abortController = null;
+        return result;
+      } catch (err) {
+        if (err.name === "AbortError") {
+          abortController = null;
+          return null;
+        }
+        if (onError && err instanceof Error) {
+          onError(err);
+        }
+        setError(err);
+      } finally {
+        setIsLoading(false);
+      }
+    });
+  }
+  const append = (message, options) => __async(this, null, function* () {
+    var _a;
+    if (!message.id) {
+      message.id = nanoid();
+    }
+    return triggerRequest(
+      ((_a = messages()) != null ? _a : []).concat(message),
+      options
+    );
+  });
+  const reload = (options) => __async(this, null, function* () {
+    const messagesSnapshot = messages();
+    if (!messagesSnapshot || messagesSnapshot.length === 0)
+      return null;
+    const lastMessage = messagesSnapshot[messagesSnapshot.length - 1];
+    if (lastMessage.role === "assistant") {
+      return triggerRequest(messagesSnapshot.slice(0, -1), options);
+    }
+    return triggerRequest(messagesSnapshot, options);
+  });
+  const stop = () => {
+    if (abortController) {
+      abortController.abort();
+      abortController = null;
+    }
+  };
+  const setMessages = (messages2) => {
+    mutate(messages2);
+  };
+  const [input, setInput] = createSignal(initialInput);
+  const handleSubmit = (e) => {
+    e.preventDefault();
+    const inputValue = input();
+    if (!inputValue)
+      return;
+    append({
+      content: inputValue,
+      role: "user",
+      createdAt: /* @__PURE__ */ new Date()
+    });
+    setInput("");
+  };
+  return {
+    messages,
+    append,
+    error,
+    reload,
+    stop,
+    setMessages,
+    input,
+    setInput,
+    handleSubmit,
+    isLoading
+  };
+}
+
+// solid/use-completion.ts
+import { createSignal as createSignal2 } from "solid-js";
+import { createSWRStore as createSWRStore2 } from "swr-store";
+import { useSWRStore as useSWRStore2 } from "solid-swr-store";
+var uniqueId2 = 0;
+var store2 = {};
+var completionApiStore = createSWRStore2({
+  get: (key) => __async(void 0, null, function* () {
+    var _a;
+    return (_a = store2[key]) != null ? _a : [];
+  })
+});
+function useCompletion({
+  api = "/api/completion",
+  id,
+  initialCompletion = "",
+  initialInput = "",
+  credentials,
+  headers,
+  body,
+  onResponse,
+  onFinish,
+  onError
+} = {}) {
+  const completionId = id || `completion-${uniqueId2++}`;
+  const key = `${api}|${completionId}`;
+  const data = useSWRStore2(completionApiStore, () => [key], {
+    initialData: initialCompletion
+  });
+  const mutate = (data2) => {
+    store2[key] = data2;
+    return completionApiStore.mutate([key], {
+      data: data2,
+      status: "success"
+    });
+  };
+  const completion = data;
+  const [error, setError] = createSignal2(void 0);
+  const [isLoading, setIsLoading] = createSignal2(false);
+  let abortController = null;
+  function triggerRequest(prompt, options) {
+    return __async(this, null, function* () {
+      try {
+        setIsLoading(true);
+        abortController = new AbortController();
+        mutate("");
+        const res = yield fetch(api, {
+          method: "POST",
+          body: JSON.stringify(__spreadValues(__spreadValues({
+            prompt
+          }, body), options == null ? void 0 : options.body)),
+          headers: __spreadValues(__spreadValues({}, headers), options == null ? void 0 : options.headers),
+          signal: abortController.signal,
+          credentials
+        }).catch((err) => {
+          throw err;
+        });
+        if (onResponse) {
+          try {
+            yield onResponse(res);
+          } catch (err) {
+            throw err;
+          }
+        }
+        if (!res.ok) {
+          throw new Error(
+            (yield res.text()) || "Failed to fetch the chat response."
+          );
+        }
+        if (!res.body) {
+          throw new Error("The response body is empty.");
+        }
+        let result = "";
+        const reader = res.body.getReader();
+        const decoder = createChunkDecoder();
+        while (true) {
+          const { done, value } = yield reader.read();
+          if (done) {
+            break;
+          }
+          result += decoder(value);
+          mutate(result);
+          if (abortController === null) {
+            reader.cancel();
+            break;
+          }
+        }
+        if (onFinish) {
+          onFinish(prompt, result);
+        }
+        abortController = null;
+        return result;
+      } catch (err) {
+        if (err.name === "AbortError") {
+          abortController = null;
+          return null;
+        }
+        if (onError && error instanceof Error) {
+          onError(error);
+        }
+        setError(err);
+      } finally {
+        setIsLoading(false);
+      }
+    });
+  }
+  const complete = (prompt, options) => __async(this, null, function* () {
+    return triggerRequest(prompt, options);
+  });
+  const stop = () => {
+    if (abortController) {
+      abortController.abort();
+      abortController = null;
+    }
+  };
+  const setCompletion = (completion2) => {
+    mutate(completion2);
+  };
+  const [input, setInput] = createSignal2(initialInput);
+  const handleSubmit = (e) => {
+    e.preventDefault();
+    const inputValue = input();
+    if (!inputValue)
+      return;
+    return complete(inputValue);
+  };
+  return {
+    completion,
+    complete,
+    error,
+    stop,
+    setCompletion,
+    input,
+    setInput,
+    handleSubmit,
+    isLoading
+  };
+}
+export {
+  useChat,
+  useCompletion
+};