append messages right after request made

rgbkrk · Nov 15, 2023 · 62af83d · 62af83d
1 parent 06db076
commit 62af83d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/chatlab/chat.py b/chatlab/chat.py
@@ -246,6 +246,8 @@ async def submit(self, *messages: Union[ChatCompletionMessageParam, str], stream
                     temperature=kwargs.get("temperature", 0),
                 )
 
+                self.append(*messages)
+
                 finish_reason, function_call_request = await self.__process_stream(streaming_response)
             else:
                 full_response = await client.chat.completions.create(
@@ -256,6 +258,8 @@ async def submit(self, *messages: Union[ChatCompletionMessageParam, str], stream
                     temperature=kwargs.get("temperature", 0),
                 )
 
+                self.append(*messages)
+
                 (
                     finish_reason,
                     function_call_request,
@@ -268,8 +272,6 @@ async def submit(self, *messages: Union[ChatCompletionMessageParam, str], stream
 
             return
 
-        self.append(*messages)
-
         if finish_reason == "function_call":
             if function_call_request is None:
                 raise ValueError(