Adding to state in sglang function

[eaubin]

Is there a way to add to state that's returned from an sgl function? I'd like to capture plain python values created during processing and also access variables bound with gen() in forks.

I get ProgramState' object does not support item assignment error naively trying:

@sgl.function
def test(s):
    s += "Hello"
    s['foo'] = 1
test()

[eaubin]

FWIW, the method I'm using right now is to store what I need in s.stream_executor.variables.

[hnyls2002]

@eaubin Yeah, a new PR has been submitted here #198