Skip to content

GitLab

Unverified Commit 8d0a7fae authored by Zhang Wenbin's avatar Zhang Wenbin Committed by GitHub
Browse files

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

parent c4e9ebe3
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
python/sglang/lang/interpreter.py
View file @ 8d0a7fae
...@@ -683,7 +683,7 @@ class ProgramState: ...@@ -683,7 +683,7 @@ class ProgramState:
if var_name is None: if var_name is None:
yield self.text() yield self.text()
else: else:
yield self.get_var(name) yield self.get_var(var_name)
async def text_async_iter( async def text_async_iter(
self, var_name: Optional[str] = None, return_meta_data: bool = False self, var_name: Optional[str] = None, return_meta_data: bool = False
...@@ -721,7 +721,7 @@ class ProgramState: ...@@ -721,7 +721,7 @@ class ProgramState:
if var_name is None: if var_name is None:
yield self.text() yield self.text()
else: else:
yield self.get_var(name) yield self.get_var(var_name)
def get_var(self, name): def get_var(self, name):
return self.stream_executor.get_var(name) return self.stream_executor.get_var(name)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment