feat: add appendAtom in chatAtoms

jotaijs · Sep 12, 2024 · 0792f8c · 0792f8c
1 parent c663f6f
commit 0792f8c
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/packages/jotai-ai/src/chatAtoms.ts b/packages/jotai-ai/src/chatAtoms.ts
@@ -369,6 +369,26 @@ export function chatAtoms(
         set(inputBaseAtom, event.target.value);
       },
     ),
+    appendAtom: atom(
+      (get) => get(isLoadingAtom),
+      async (
+        get,
+        set,
+        message: Message | CreateMessage,
+        options: ChatRequestOptions = {},
+        metadata?: Metadata,
+      ) => {
+        if (metadata) {
+          set(metadataAtom, (prevMetadata) => ({
+            ...prevMetadata,
+            ...metadata,
+          }));
+        }
+        return append(get, set, message, options).catch((err) =>
+          onError(get, set, err),
+        );
+      },
+    ),
     submitAtom: atom(
       (get) => get(isLoadingAtom),
       (