From 8d0a7fae3b89eecae8cd2a755a673462c23ab31f Mon Sep 17 00:00:00 2001 From: Zhang Wenbin <22811973@qq.com> Date: Sat, 24 Feb 2024 16:27:34 +0800 Subject: [PATCH] Fix interpreter.py `get_var(var_name)` in text iter when `stream` is not enabled (#198) --- python/sglang/lang/interpreter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/sglang/lang/interpreter.py b/python/sglang/lang/interpreter.py index be525d8e2..e3de5b6d2 100644 --- a/python/sglang/lang/interpreter.py +++ b/python/sglang/lang/interpreter.py @@ -683,7 +683,7 @@ class ProgramState: if var_name is None: yield self.text() else: - yield self.get_var(name) + yield self.get_var(var_name) async def text_async_iter( self, var_name: Optional[str] = None, return_meta_data: bool = False @@ -721,7 +721,7 @@ class ProgramState: if var_name is None: yield self.text() else: - yield self.get_var(name) + yield self.get_var(var_name) def get_var(self, name): return self.stream_executor.get_var(name)