EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Zhang Wenbin
2024-02-24 16:27:34 +08:00
committed by GitHub
parent c4e9ebe3a4
commit 8d0a7fae3b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/sglang/lang/interpreter.py
View File

@@ -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:
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)