VincentLambert 4f3711d37f fix: handle missing 'total' key causing KeyError in deep research retrieval (#13942) ...

## Summary

- When KB retrieval fails (e.g. ES `AssertionError` on empty
`index_names`), `kbinfos` falls back to a dict without a `total` key
- `_async_update_chunk_info` then iterates over `chunk_info.keys()`
(which includes `total`) and tries `kbinfos['total']`, raising a
`KeyError`
- This error surfaces when using Tavily web retrieval in a chat with no
knowledge base attached

## Changes

- Add `'total': 0` to all default `kbinfos` dicts in
`_retrieve_information`
- Add `setdefault('total', 0)` guard after successful KB retrieval to
handle cases where the retrieval result omits the key
- Accumulate `total` correctly in the merge branch of
`_async_update_chunk_info`

## Test plan

- [ ] Start a chat with Tavily configured and no knowledge base
- [ ] Verify no `KeyError: 'total'` is raised
- [ ] Verify Tavily results are returned correctly

---------

Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com>

2026-05-09 10:57:51 +08:00

..

__init__.py

Feat: support tree structured deep-research policy. (#12559)

2026-01-13 09:41:35 +08:00

tree_structured_query_decomposition_retrieval.py

fix: handle missing 'total' key causing KeyError in deep research retrieval (#13942)

2026-05-09 10:57:51 +08:00