Chroma Context-1: Training a Self-Editing Search Agent·|·Chroma
Retrieval pipelines typically operate in a single pass, which poses a problem when the information required to answer a question is spread across multiple documents or requires intermediate reasoning to locate. In practice, many real-world queries require multi-hop retrieval, in which the output of one search informs the next. Recent work has shown that frontier LLMs perform this multi-hop search effectively through a process known as agentic search, simply defined as a loop of LLM calls with search tools. This mode of search often comes with significant cost and latency due to their use of frontier-scale LLMs. We introduce Chroma Context-1, a 20B parameter agentic search model derived from gpt-oss-20B that achieves retrieval performance comparable to frontier-scale LLMs at a fraction of the cost and up to 10x faster inference speed. Context-1 is designed to be used as a subagent in conjunction with a frontier reasoning model. Given a query, it produces a ranked list of documents that are relevant to satisfying the query. The model is trained to decompose queries into subqueries, iteratively search a corpus, and selectively edit its own context to free capacity for further exploration.
Introduction # Using search systems in conjunction with a large language model (LLM) is a common paradigm for enabling language models to access data beyond their training corpus. This approach, broadly known as retrieval-augmented-generation (RAG) , has traditionally relied on single-stage retrieval pipelines composed of vector search, lexical search, or regular expression matching, optionally followed by a learned reranker. While effective for straightforward lookup queries, these pipelines are fundamentally limited: they assume that the information needed to answer a question can be retriev
saved by
related reading
- Introducing SWE-grep and SWE-grep-mini: RL for Multi-Turn, Fast Context Retrieval | Cognitioncognition.ai
- Recursive Language Models | Alex L. Zhangalexzhang13.github.io
- frontier model training methodologies | Alex Wa's Blogdjdumpling.github.io
- Effective context engineering for AI agents \ Anthropicanthropic.com
- [2005.11401] Retrieval-Augmented Generation for Knowledge-Intensive NLP Tasksarxiv.org
- How we built our multi-agent research system \ Anthropicanthropic.com
- Stream of Search (SoS): Learning to Search in Languagearxiv.org
- Remember, Refine, Retrieve: A Context Engine for Enterprise Agents | Applied Computeappliedcompute.com
- Subquadratic — How SSA Makes Long Context Practicalsubq.ai
- Auto-Retrieval from a Vector Databasegpt-index.readthedocs.io
- Training a Specialist Code Search Agent with turbopufferappliedcompute.com
- [2506.06266] Cartridges: Lightweight and general-purpose long context representations via self-studyarxiv.org