feat: support tools in bedrock (#1598)

2025-03-17 13:42:18 +08:00
parent 3504e64407
commit e204ca1cf3
7 changed files with 25 additions and 7 deletions
--- a/lua/avante/providers/bedrock.lua
+++ b/lua/avante/providers/bedrock.lua
@@ -34,6 +34,11 @@ function M:parse_response(ctx, data_stream, event_state, opts)
  return model_handler.parse_response(self, ctx, data_stream, event_state, opts)
 end

+function M:transform_tool(tool)
+  local model_handler = M.load_model_handler()
+  return model_handler.transform_tool(self, tool)
+end
+
 function M:build_bedrock_payload(prompt_opts, request_body)
  local model_handler = M.load_model_handler()
  return model_handler.build_bedrock_payload(self, prompt_opts, request_body)