From 38949a41e537c6315f510084a6e27dd3dede3b7d Mon Sep 17 00:00:00 2001
From: Dillon Laird <dillonalaird@gmail.com>
Date: Sun, 11 Aug 2024 16:50:09 -0700
Subject: [PATCH] grab correct key if call fails (#195)

* grab correct key if call fails

* fixed format
---
 vision_agent/tools/tool_utils.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/vision_agent/tools/tool_utils.py b/vision_agent/tools/tool_utils.py
index e1fa69c3..0ff56177 100644
--- a/vision_agent/tools/tool_utils.py
+++ b/vision_agent/tools/tool_utils.py
@@ -62,7 +62,9 @@ def send_inference_request(
                 traceback_raw=[],
             )
             _LOGGER.error(f"Request failed: {res.status_code} {res.text}")
-            raise RemoteToolCallFailed(payload["tool"], res.status_code, res.text)
+            raise RemoteToolCallFailed(
+                payload["function_name"], res.status_code, res.text
+            )
 
         resp = res.json()
         tool_call_trace.response = resp