diff --git a/langchain/llms/huggingface_text_gen_inference.py b/langchain/llms/huggingface_text_gen_inference.py index 1bd150de653..012dd29a5c7 100644 --- a/langchain/llms/huggingface_text_gen_inference.py +++ b/langchain/llms/huggingface_text_gen_inference.py @@ -255,4 +255,5 @@ class HuggingFaceTextGenInference(LLM): if not token.special: if text_callback: await text_callback(token.text) + text += token.text return text