agent: fix functionary tool_calls templating
This commit is contained in:
parent
eb9a5524eb
commit
3da30ed89e
1 changed files with 21 additions and 0 deletions
|
@ -171,6 +171,27 @@ class ChatTemplate(BaseModel):
|
||||||
|
|
||||||
messages = new_messages
|
messages = new_messages
|
||||||
|
|
||||||
|
# JSON!
|
||||||
|
messages = [m.model_dump() for m in messages]
|
||||||
|
|
||||||
|
if self.inferred_tool_style == ToolsPromptStyle.TYPESCRIPT_FUNCTIONARY_V2:
|
||||||
|
messages = [
|
||||||
|
{
|
||||||
|
**m,
|
||||||
|
"tool_calls": [
|
||||||
|
{
|
||||||
|
**tc,
|
||||||
|
"function": {
|
||||||
|
"name": tc["function"]["name"],
|
||||||
|
"arguments": json.dumps(tc["function"]["arguments"]),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for tc in m["tool_calls"]
|
||||||
|
] if m.get("tool_calls") else None
|
||||||
|
}
|
||||||
|
for m in messages
|
||||||
|
]
|
||||||
|
|
||||||
result = self._template.render(
|
result = self._template.render(
|
||||||
messages=messages,
|
messages=messages,
|
||||||
eos_token=self._eos_token,
|
eos_token=self._eos_token,
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue