feat: tools (#1180)
* feat: tools * feat: claude use tools * feat: openai use tools
This commit is contained in:
@@ -6,6 +6,8 @@
|
||||
---@field role "user" | "assistant"
|
||||
---@field content [AvanteBedrockClaudeTextMessage][]
|
||||
|
||||
local Claude = require("avante.providers.claude")
|
||||
|
||||
---@class AvanteBedrockModelHandler
|
||||
local M = {}
|
||||
|
||||
@@ -33,25 +35,7 @@ M.parse_messages = function(opts)
|
||||
return messages
|
||||
end
|
||||
|
||||
M.parse_response = function(ctx, data_stream, event_state, opts)
|
||||
if event_state == nil then
|
||||
if data_stream:match('"content_block_delta"') then
|
||||
event_state = "content_block_delta"
|
||||
elseif data_stream:match('"message_stop"') then
|
||||
event_state = "message_stop"
|
||||
end
|
||||
end
|
||||
if event_state == "content_block_delta" then
|
||||
local ok, json = pcall(vim.json.decode, data_stream)
|
||||
if not ok then return end
|
||||
opts.on_chunk(json.delta.text)
|
||||
elseif event_state == "message_stop" then
|
||||
opts.on_complete(nil)
|
||||
return
|
||||
elseif event_state == "error" then
|
||||
opts.on_complete(vim.json.decode(data_stream))
|
||||
end
|
||||
end
|
||||
M.parse_response = Claude.parse_response
|
||||
|
||||
---@param prompt_opts AvantePromptOptions
|
||||
---@param body_opts table
|
||||
@@ -60,7 +44,6 @@ M.build_bedrock_payload = function(prompt_opts, body_opts)
|
||||
local system_prompt = prompt_opts.system_prompt or ""
|
||||
local messages = M.parse_messages(prompt_opts)
|
||||
local max_tokens = body_opts.max_tokens or 2000
|
||||
local temperature = body_opts.temperature or 0.7
|
||||
local payload = {
|
||||
anthropic_version = "bedrock-2023-05-31",
|
||||
max_tokens = max_tokens,
|
||||
|
||||
Reference in New Issue
Block a user