Merge pull request #113 from rgbkrk/append-messages-before-others

append messages right after request made

Author: rgbkrk

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(