Merge pull request #1065 from parea-ai/fix-convert-tool-use-response

fix: convert output to tool use response

Author: joschkabraun

parea/schemas/log.py

@@ -87,8 +87,10 @@ def convert_to_jsonl_row_for_finetuning(self) -> dict:
                 jsonl_row["functions"] = tools
             else:
                 tool_calls = tool_calls if isinstance(tool_calls, List) else [tool_calls]
-                for tool_call in tool_calls:
-                    tool_call["arguments"] = json.dumps(tool_call["arguments"])
+                tool_calls = [
+                    {"id": tool["id"], "type": "function", "function": {"name": tool["function"]["name"], "arguments": json.dumps(tool["function"]["arguments"])}}
+                    for tool in tool_calls
+                ]
                 assistant_response = {
                     "role": "assistant",
                     "tool_calls": tool_calls,

pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "poetry.core.masonry.api"
 [tool.poetry]
 name = "parea-ai"
 packages = [{ include = "parea" }]
-version = "0.2.208"
+version = "0.2.209"
 description = "Parea python sdk"
 readme = "README.md"
 authors = ["joel-parea-ai <[email protected]>"]