JSONCallParser

class torchrl.envs.llm.transforms.JSONCallParser

Parser for JSON-style function-calling responses.

Expects responses in the format:

{

“message”: “…”, “tools”: [

{“tool”: “search”, “args”: {“query”: “…”}, “tag”: “A”}, {“tool”: “summarize”, “args”: {“text”: “…”}}

]

}

Examples

>>> parser = JSONCallParser()
>>> response = {
...     "message": "Let me search for that.",
...     "tools": [{"tool": "search", "args": {"query": "torchrl"}}]
... }
>>> result = parser(response)
>>> print(result["text"])
Let me search for that.
>>> print(result["calls"][0].tool)
search