Chat and REPL modes output fix after function calls (#525)

TheR1D · Apr 1, 2024 · be138cb · be138cb
1 parent bb63eba
commit be138cb
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/sgpt/handlers/chat_handler.py b/sgpt/handlers/chat_handler.py
@@ -47,7 +47,7 @@ def wrapper(*args: Any, **kwargs: Any) -> Generator[str, None, None]:
             if not kwargs.get("messages"):
                 return
             if not chat_id:
-                yield from func(*args, **kwargs, caching=False)
+                yield from func(*args, **kwargs)
                 return
             previous_messages = self._read(chat_id)
             for message in kwargs["messages"]:

diff --git a/sgpt/handlers/handler.py b/sgpt/handlers/handler.py
@@ -121,7 +121,12 @@ def get_completion(
                 if chunk.choices[0].finish_reason == "function_call":
                     yield from self.handle_function_call(messages, name, arguments)
                     yield from self.get_completion(
-                        model, temperature, top_p, messages, functions, caching=False
+                        model=model,
+                        temperature=temperature,
+                        top_p=top_p,
+                        messages=messages,
+                        functions=functions,
+                        caching=False,
                     )
                     return