core/types: fix types to support async onResponse callback (#190)

vercel · Jun 21, 2023 · bde3898 · bde3898
1 parent 7815b11
commit bde3898
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/.changeset/gorgeous-kangaroos-sort.md b/.changeset/gorgeous-kangaroos-sort.md
@@ -0,0 +1,5 @@
+---
+'ai': patch
+---
+
+Allow an async onResponse callback in useChat/useCompletion
diff --git a/packages/core/shared/types.ts b/packages/core/shared/types.ts
@@ -42,7 +42,7 @@ export type UseChatOptions = {
   /**
    * Callback function to be called when the API response is received.
    */
-  onResponse?: (response: Response) => void
+  onResponse?: (response: Response) => void | Promise<void>
 
   /**
    * Callback function to be called when the chat is finished streaming.
@@ -107,7 +107,8 @@ export type UseCompletionOptions = {
   /**
    * Callback function to be called when the API response is received.
    */
-  onResponse?: (response: Response) => void
+  onResponse?: (response: Response) => void | Promise<void>
+
   /**
    * Callback function to be called when the completion is finished streaming.
    */