p04798526
/
LLaMA-Factory-310P3
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix function formatter

This commit is contained in:
hiyouga 2024-01-18 16:01:07 +08:00
parent ddd48ce8ab
commit 35aef8b287
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/llmtuner/data/formatter.py
View File

@ -44,9 +44,9 @@ class FunctionFormatter:
def __call__(self, content: str) -> List[Union[str, Dict[str, str]]]: def __call__(self, content: str) -> List[Union[str, Dict[str, str]]]:
try: try:
function = json.loads(content) function = json.loads(content)
name = json.dumps(function["name"], ensure_ascii=False) name = function["name"]
arguments = json.dumps(function["arguments"], ensure_ascii=False) arguments = json.dumps(function["arguments"], ensure_ascii=False)
except json.JSONDecodeError: except Exception:
name, arguments = "", "" name, arguments = "", ""
elements = [] elements = []
@ -98,5 +98,5 @@ class ToolFormatter:
if self.type == "default": if self.type == "default":
return [self._default(tools)] return [self._default(tools)]
except json.JSONDecodeError: except Exception:
return [""] return [""]