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

This commit is contained in:
save196
2024-04-01 08:35:16 +03:00
committed by GitHub
parent bb63eba9c5
commit be138cbd77
2 changed files with 7 additions and 2 deletions
+1 -1
View File
@@ -47,7 +47,7 @@ class ChatSession:
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"]:
+6 -1
View File
@@ -121,7 +121,12 @@ class Handler:
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