From 58d63c27aefdcffa4482c59a4a324edbac2b6669 Mon Sep 17 00:00:00 2001 From: Tanishq Dubey Date: Sun, 6 Oct 2024 23:54:25 -0400 Subject: [PATCH] sql gen --- main.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/main.py b/main.py index d3f1f4e..09a5729 100644 --- a/main.py +++ b/main.py @@ -578,9 +578,12 @@ def answer_question_tools_api( for tool_call in assistant_message["tool_calls"]: tool_name = tool_call["function"]["name"] tool_args = tool_call["function"]["arguments"] - tool_response = tool_manager.get_tool(tool_name).execute(tool_args) - conversation_history.append({"role": "tool", "content": tool_response}) - logger.info(f"API Tool response: {tool_response}") + if tool_name is not None and tool_args is not None: + tool_response = tool_manager.get_tool(tool_name).execute(tool_args) + conversation_history.append({"role": "tool", "content": tool_response}) + logger.info(f"API Tool response: {tool_response}") + else: + logger.warning(f"Skipping tool call due to missing tool name or arguments: {tool_call}") else: if "" in assistant_message["content"].lower(): reply_content = re.search(