feat: call fetch ai response with direct value

XD2Sketch · Oct 17, 2023 · bd68615 · bd68615
1 parent 0656501
commit bd68615
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/package.json b/package.json
@@ -1,7 +1,7 @@
 {
   "name": "@magicul/react-chat-stream",
   "description": "A React hook that lets you easily integrate your custom ChatGPT-like chat in React.",
-  "version": "0.1.7",
+  "version": "0.2.0",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "homepage": "https://github.com/XD2Sketch/react-chat-stream#readme",

diff --git a/src/hooks/useChatStream.ts b/src/hooks/useChatStream.ts
@@ -64,7 +64,7 @@ const useChatStream = (input: UseChatStreamInput): UseChatStreamResult => {
     });
   };
 
-  const fetchAndUpdateAIResponse = async () => {
+  const fetchAndUpdateAIResponse = async (message: string) => {
     const stream = await getStream(message, input.options, input.method);
     if (!stream) throw new Error();
 
@@ -82,7 +82,7 @@ const useChatStream = (input: UseChatStreamInput): UseChatStreamResult => {
     setMessage('');
 
     try {
-      await fetchAndUpdateAIResponse();
+      await fetchAndUpdateAIResponse(newMessage ?? message);
     } catch {
       addMessageToChat(BOT_ERROR_MESSAGE, 'bot');
     }