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

otca unkown tool fix

Browse Source
This commit is contained in:
Lennart J. Kurzweg (Nx2)
2024-08-30 18:09:16 +02:00
parent faf37c8bdb
commit dc4a29a97c
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libs/runnables.py
View File

@@ -82,8 +82,12 @@ def one_tool_call_answer(model: Model, seed: int, test: Test, technique: Techniq
assert isinstance(ai_msg, AIMessage) assert isinstance(ai_msg, AIMessage)
calls = ai_msg.tool_calls calls = ai_msg.tool_calls
for call in calls: for call in calls:
try:
selected_tool = tools_dict[call["name"].lower()] selected_tool = tools_dict[call["name"].lower()]
tool_msg = selected_tool.invoke(call) tool_msg = selected_tool.invoke(call)
except KeyError:
tool_msg = SystemMessage(f"Tool '{call['name'].lower()}' does not exist. Available are {tools_dict.keys()}")
messages.append(tool_msg) messages.append(tool_msg)
ai_msg = llm.invoke(messages) ai_msg = llm.invoke(messages)
i = 0 i = 0