nx2/test-small-llms
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

more error feedback

Browse Source
This commit is contained in:
Lennart J. Kurzweg (Nx2)
2024-08-30 23:05:07 +02:00
parent e3df5d3e78
commit 065fff21e6
1 changed files with 19 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libs/runnables.py
View File

@@ -87,7 +87,9 @@ def one_tool_call_answer(model: Model, seed: int, test: Test, technique: Techniq
tool_msg = selected_tool.invoke(call) tool_msg = selected_tool.invoke(call)
except KeyError: except KeyError:
tool_msg = SystemMessage(f"Tool '{call['name'].lower()}' does not exist. Available are {tools_dict.keys()}") 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) messages.append(tool_msg)
ai_msg = llm.invoke(messages) ai_msg = llm.invoke(messages)
i = 0 i = 0