# ThinkRetrieve

> ThinkRetrieve is an open-source Python library and research framework for
> retrieval-augmented test-time scaling. It retrieves worked examples while a
> reasoning model is thinking and injects them into the live reasoning trace.

## Primary links

- [Project page](https://itsvaibhav01.github.io/ThinkRetrieve/)
- [Research paper](https://arxiv.org/abs/2608.10928)
- [Paper PDF](https://arxiv.org/pdf/2608.10928)
- [Paper abstract and citation details](https://itsvaibhav01.github.io/ThinkRetrieve/paper/)
- [BibTeX citation](https://itsvaibhav01.github.io/ThinkRetrieve/paper/thinkretrieve.bib)
- [RIS citation](https://itsvaibhav01.github.io/ThinkRetrieve/paper/thinkretrieve.ris)
- [Source code](https://github.com/itsvaibhav01/ThinkRetrieve)
- [Python package](https://pypi.org/project/thinkretrieve/)
- [Tutorial](https://github.com/itsvaibhav01/ThinkRetrieve/blob/main/thinkretrieve/TUTORIAL.md)
- [Technical article](https://itsvaibhav01.github.io/ThinkRetrieve/article/)
- [Technical article in Markdown](https://itsvaibhav01.github.io/ThinkRetrieve/article/README.md)

## Scope and evidence

ThinkRetrieve retrieves complete worked examples using an intermediate reasoning
state and injects them inside the reasoning trace. The research paper evaluates
five reasoning models on GSM8K, MATH-500, AIME 2025, and SciQ. Consult the paper
for experimental settings, baselines, and limitations.

The technical article is a separate, exploratory six-case science-agent tool
dispatch demonstration using synthetic incidents and private routes, selected
after exploratory screening. Its 5/6 result for a local 9B model is not a general
agent accuracy estimate or a biological safety benchmark. The repository links
the runner and raw outputs. These results are distinct from the paper benchmarks.

## Citation

Singh, Vaibhav; Ghosal, Soumya Suvra; Gharat, Sarvesh; Pal, Soumyabrata;
Narayanam, Ramasuri; and Manocha, Dinesh. “ThinkRetrieve:
Retrieval-Augmented Reasoning Traces for Test-Time Scaling.” arXiv:2608.10928,
2026. https://doi.org/10.48550/arXiv.2608.10928

## License

The ThinkRetrieve Python library is available under the MIT License.