fix tool name transform path

This commit is contained in:
dal 2025-02-11 08:23:58 -07:00
parent 8940fbf3b6
commit fcc15a0e6f
No known key found for this signature in database
GPG Key ID: 16F4B0E1E9F61122
1 changed files with 2 additions and 2 deletions

View File

@ -335,7 +335,7 @@ fn transform_tool_message(
progress: Option<MessageProgress>,
) -> Result<Vec<BusterThreadMessage>> {
match name.as_str() {
"data_catalog_search" => tool_data_catalog_search(id, content, progress),
"search_data_catalog" => tool_data_catalog_search(id, content, progress),
"stored_values_search" => tool_stored_values_search(id, content, progress),
"search_files" => tool_file_search(id, content, progress),
"create_files" => tool_create_file(id, content, progress),
@ -352,7 +352,7 @@ fn transform_assistant_tool_message(
) -> Result<Vec<BusterThreadMessage>> {
if let Some(tool_call) = tool_calls.first() {
match tool_call.function.name.as_str() {
"data_catalog_search" => assistant_data_catalog_search(id, progress),
"search_data_catalog" => assistant_data_catalog_search(id, progress),
"stored_values_search" => assistant_stored_values_search(id, progress),
"search_files" => assistant_file_search(id, progress),
"create_files" => assistant_create_file(id, tool_calls, progress),