llamaIndex - 💡(How to fix) Fix [Bug]: AgentWorkflow leaks `initial_state` mutations across runs [1 pull requests]

import asyncio
from llama_index.core.agent.workflow import AgentWorkflow
from llama_index.core.workflow import Context
from llama_index.core.llms.mock import MockFunctionCallingLLM


async def bump_counter(ctx: Context) -> str:
      async with ctx.store.edit_state() as state:
          state["state"]["counter"] += 1
      return "bumped"


async def main() -> None:
      workflow = AgentWorkflow.from_tools_or_functions(
          [bump_counter],
          llm=MockFunctionCallingLLM(),
          system_prompt="Call bump_counter exactly once, then reply 'done'.",
          initial_state={"counter": 0},
      )
      await workflow.run(user_msg="bump it")
      handler = workflow.run(user_msg="bump it")
      await handler
      state = await handler.ctx.store.get("state")
      print(state)

asyncio.run(main())

Expected output : {'counter': 1} 
Actual output : {'counter': 2}