fix(prompt/base.py): Proper parsing of dict to json string (#728)

Co-authored-by: GlockPL <[email protected]>

Author: GlockPL

packages/ragbits-core/CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ## Unreleased
 
+- Fix issue with improper convertion to json of tool call arguments (#737)
 - Added Google Drive support (#686)
 - Add LLM Usage to LLMResponseWithMetadata (#700)
 - Split usage per model type (#715)

packages/ragbits-core/src/ragbits/core/prompt/base.py

@@ -1,3 +1,4 @@
+import json
 from abc import ABCMeta, abstractmethod
 from typing import Any, Generic
 
@@ -99,7 +100,7 @@ def add_tool_use_message(
                             "type": "function",
                             "function": {
                                 "name": name,
-                                "arguments": str(arguments),
+                                "arguments": json.dumps(arguments),
                             },
                         }
                     ],

packages/ragbits-core/tests/unit/prompts/test_prompt.py

@@ -675,7 +675,7 @@ def test_base_prompt_with_parser_add_tool_use_message_no_history():
     assert prompt.chat[0]["tool_calls"][0]["id"] == "tool_123"
     assert prompt.chat[0]["tool_calls"][0]["type"] == "function"
     assert prompt.chat[0]["tool_calls"][0]["function"]["name"] == "test_function"
-    assert prompt.chat[0]["tool_calls"][0]["function"]["arguments"] == "{'param': 'value'}"
+    assert prompt.chat[0]["tool_calls"][0]["function"]["arguments"] == '{"param": "value"}'
 
     assert prompt.chat[1]["role"] == "tool"
     assert prompt.chat[1]["tool_call_id"] == "tool_123"