423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-14 23:58:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix function formatter

Browse Source 
Former-commit-id: 363a87376ad8fe4149b387f7ccd60f31f2a5fdf7
This commit is contained in:
hiyouga 2024-01-18 16:01:07 +08:00
parent 42a13fec46
commit 9b390c4bea
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]]]:
try:
function = json.loads(content)
name = json.dumps(function["name"], ensure_ascii=False)
name = function["name"]
arguments = json.dumps(function["arguments"], ensure_ascii=False)
except json.JSONDecodeError:
except Exception:
name, arguments = "", ""
elements = []
@ -98,5 +98,5 @@ class ToolFormatter:
if self.type == "default":
return [self._default(tools)]
except json.JSONDecodeError:
except Exception:
return [""]