fix: query still needs parsing

BCG-X-Official · Apr 8, 2024 · 56e1f9e · 56e1f9e
1 parent 3b0b278
commit 56e1f9e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backend/app/app/services/chat_agent/tools/library/basellm_tool/basellm_tool.py b/backend/app/app/services/chat_agent/tools/library/basellm_tool/basellm_tool.py
@@ -67,12 +67,12 @@ async def _arun(
         **kwargs: Any,
     ) -> str:
         """Use the tool asynchronously."""
-        tool_input = kwargs.get(
+        tool_input_str = kwargs.get(
             "query",
             args[0],
         )
 
-        query = standard_query_format(tool_input)
+        query = standard_query_format(ToolInputSchema.parse_raw(tool_input_str))
 
         try:
             messages = [