Merge pull request #5483 from whybeyoung/main
fix: 修复function call数据集如果 function_call 值的为不合法json,异常提示且中断训练。
This commit is contained in:
commit
af8dfb58fd
|
@ -113,7 +113,7 @@ class FunctionFormatter(Formatter):
|
|||
functions.append((tool_call["name"], json.dumps(tool_call["arguments"], ensure_ascii=False)))
|
||||
|
||||
except json.JSONDecodeError:
|
||||
functions = []
|
||||
raise RuntimeError("Invalid JSON format in function message: {}".format(str([content]))) # flat string
|
||||
|
||||
elements = []
|
||||
for name, arguments in functions:
|
||||
|
@ -141,7 +141,7 @@ class ToolFormatter(Formatter):
|
|||
tools = json.loads(content)
|
||||
return [self.tool_utils.tool_formatter(tools) if len(tools) != 0 else ""]
|
||||
except json.JSONDecodeError:
|
||||
return [""]
|
||||
raise RuntimeError("Invalid JSON format in tool description: {}".format(str([content]))) # flat string
|
||||
|
||||
@override
|
||||
def extract(self, content: str) -> Union[str, List["FunctionCall"]]:
|
||||
|
|
Loading…
Reference in New Issue