Assistance Needed for Properly Nesting Traces in Arize Integration

Hi Everyone, I am currently integrating arize in my product and I am not able to do nested traces properly. Can you guys help me on how to do nesting properly. Here is my use case: I am using llama index workflows (root trace starts from here), now from here step functions are called and these step fns invoke some handler fns and also make llm api calls (gemini or openai). I want these to be nested properly in the hierarchy. Currently when i try to do the same all llama index workflows gets traced as a separately and handler fn calls separately under the same root trace, they are not nested properly.