Quickly access variable within ProgramState

[SimoneRaponi]

Hey everyone, is there any way to quickly access variables within the ProgramState?
Let me show you an example.

I am asking an LLM to generate a small JSON output, constrained by a regex.
The generated variable is called "value", as shown below.

s += f"Your response: " + sgl.gen("value", max_tokens=256, regex=regex, temperature=0.5)

I noticed that the model requires some time to elaborate the output, so I measured the time of each step.

start = time.time()
query = build_query(args)
print(f"build query: {time.time() - start}")

start = time.time()
result = sgl_assistant.run(query, num_values)
print(f"run query: {time.time() - start}")

start = time.time()
values = result["value"]
print(f"extract value: {time.time() - start}")

build query: 1.430511474609375e-06
run query: 0.00025844573974609375
extract value: 2.8307931423187256

It results that the whole time is dedicated to the extraction of the generated variable from the ProgramState object.

I was wondering if there was a quicker way to extract the variable from the object.
Thanks a lot for your kind support.