From 48e85c752b802167242aedab643d023c2b895f11 Mon Sep 17 00:00:00 2001 From: kaifronsdal Date: Wed, 31 Jul 2024 17:21:52 -0700 Subject: [PATCH] Fix select w/o streaming bug --- python/sglang/lang/interpreter.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/sglang/lang/interpreter.py b/python/sglang/lang/interpreter.py index 0133b97c749e..2096e49e850a 100644 --- a/python/sglang/lang/interpreter.py +++ b/python/sglang/lang/interpreter.py @@ -553,7 +553,8 @@ def _execute_select(self, expr: SglSelect): "output_token_logprobs": output_token_logprobs, } self.variable_event[name].set() - self.stream_var_event[name].set() + if self.stream_var_event: + self.stream_var_event[name].set() self.text_ += decision def _execute_variable(self, expr: SglVariable):