fix: pass context to provider for stream data parsing (#1475)

* fix: pass context to provider for stream data parsing

* fix: luatype

---------

Co-authored-by: yetone <yetoneful@gmail.com>
This commit is contained in:
brook hong
2025-03-04 00:20:27 +08:00
committed by GitHub
parent c6d5527ea8
commit 6bd966e8e2
4 changed files with 8 additions and 6 deletions

View File

@@ -154,6 +154,8 @@ end
function M._stream(opts)
local provider = opts.provider or Providers[Config.provider]
---@cast provider AvanteProviderFunctor
local prompt_opts = M.generate_prompts(opts)
---@type string
@@ -285,10 +287,10 @@ function M._stream(opts)
{ once = true }
)
end
provider.parse_stream_data(data, handler_opts)
provider.parse_stream_data(resp_ctx, data, handler_opts)
else
if provider.parse_stream_data ~= nil then
provider.parse_stream_data(data, handler_opts)
provider.parse_stream_data(resp_ctx, data, handler_opts)
else
parse_stream_data(data)
end