Fix interpreter.py `get_var(var_name)` in text iter when `stream` is not enabled (#198)

8d0a7fae · Zhang Wenbin · GitHub · c4e9ebe3 · 8d0a7fae
Unverified Commit 8d0a7fae authored Feb 24, 2024 by Zhang Wenbin Committed by GitHub Feb 24, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

python/sglang/lang/interpreter.py python/sglang/lang/interpreter.py +2 -2

No files found.
--- 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)