LlamaIndex supports streaming the response as it’s being generated. This allows you to start printing or processing the beginning of the response before the full response is finished. This can drastically reduce the perceived latency of queries.
To enable streaming, you need to use an LLM that supports streaming.
Right now, streaming is supported by
HuggingFaceLLM, and most LangChain LLMs (via
Configure query engine to use streaming:
If you are using the high-level API, set
streaming=True when building a query engine.
query_engine = index.as_query_engine(streaming=True, similarity_top_k=1)
If you are using the low-level API to compose the query engine,
streaming=True when constructing the
from llama_index import get_response_synthesizer synth = get_response_synthesizer(streaming=True, ...) query_engine = RetrieverQueryEngine(response_synthesizer=synth, ...)
After properly configuring both the LLM and the query engine,
query now returns a
streaming_response = query_engine.query( "What did the author do growing up?", )
The response is returned immediately when the LLM call starts, without having to wait for the full completion.
Note: In the case where the query engine makes multiple LLM calls, only the last LLM call will be streamed and the response is returned when the last LLM call starts.
You can obtain a
Generator from the streaming response and iterate over the tokens as they arrive:
for text in streaming_response.response_gen: # do something with text as they arrive. pass
Alternatively, if you just want to print the text as they arrive:
See an end-to-end example