feat: chat_completion的函数开关支持

This commit is contained in:
RockChinQ
2023-07-29 16:29:16 +08:00
parent 8c69b8a1d9
commit 3c894fe70e
+14 -6
View File
@@ -69,12 +69,20 @@ class ChatCompletionRequest(RequestBase):
if self.pending_func_call is None: # 没有待处理的函数调用请求 if self.pending_func_call is None: # 没有待处理的函数调用请求
resp = self._req( args = {
model=self.model, "model": self.model,
messages=self.messages, "messages": self.messages,
functions=get_func_schema_list(), }
**self.kwargs
) funcs = get_func_schema_list()
if len(funcs) > 0:
args['functions'] = funcs
# 拼接kwargs
args = {**args, **self.kwargs}
resp = self._req(**args)
choice0 = resp["choices"][0] choice0 = resp["choices"][0]