more error feedback

2024-08-30 23:05:07 +02:00
parent e3df5d3e78
commit 065fff21e6
1 changed files with 19 additions and 17 deletions
--- a/libs/runnables.py
+++ b/libs/runnables.py
@@ -87,7 +87,9 @@ def one_tool_call_answer(model: Model, seed: int, test: Test, technique: Techniq
                tool_msg = selected_tool.invoke(call)
            except KeyError:
                tool_msg = SystemMessage(f"Tool '{call['name'].lower()}' does not exist. Available are {tools_dict.keys()}")
-                
+            except Exception as e:
+                tool_msg = SystemMessage(f"Tool '{call['name'].lower()}' returned a input validation error:" + "\n" + str(e))
+            finally:
                messages.append(tool_msg)
                ai_msg = llm.invoke(messages)
                i = 0