| 26 | |
| 27 | class EventHandler(AsyncAssistantEventHandler): |
| 28 | def __init__(self, assistant_name: str) -> None: |
| 29 | super().__init__() |
| 30 | self.current_message: cl.Message = None |
| 31 | self.current_step: cl.Step = None |
| 32 | self.current_tool_call = None |
| 33 | self.assistant_name = assistant_name |
| 34 | previous_steps = local_steps.get() or [] |
| 35 | parent_step = previous_steps[-1] if previous_steps else None |
| 36 | if parent_step: |
| 37 | self.parent_id = parent_step.id |
| 38 | |
| 39 | async def on_run_step_created(self, run_step: RunStep) -> None: |
| 40 | cl.user_session.set("run_step", run_step) |