Find answers from the community

Updated 4 months ago

Hi This global context appears to be

Hi This global context appears to be broken. Any help workaround ?

Plain Text
from llama_index.core.workflow import (
    Context,
    Workflow,
    StartEvent,
    StopEvent,
    step,
)
import asyncio  # Import asyncio to run the async functions


class WaitExampleFlow(Workflow):
    @step
    async def setup(self, ctx: Context, ev: StartEvent) -> StopEvent:
        if hasattr(ev, "data"):
            ctx.data["some_database"] = ev.data

        return StopEvent(result=None)

    @step
    async def query(self, ctx: Context, ev: StartEvent) -> StopEvent | None:
        if hasattr(ev, "query"):
            # do we have any data?
            if "some_database" in ctx.data:
                data = ctx.data["some_database"]
                return StopEvent(result=f"Got the data {data}")
            else:
                # there's non data yet
                return None
        else:
            # this isn't a query
            return None


async def main():
    w = WaitExampleFlow(verbose=True)
    result = await w.run(query="Can I kick it?")
    if result is None:
        print("No you can't")
    print("---")
    result = await w.run(data="Yes you can")
    print("---")
    result = await w.run(query="Can I kick it?")
    print(result)


if __name__ == "__main__":
    asyncio.run(main())
a
L
7 comments
Running step query
Step query produced no event
Running step setup
Step setup produced event StopEvent
No you can't
---
Running step query
Step query produced no event
Running step setup
Step setup produced event StopEvent
---
Running step query
Step query produced no event
Running step setup
Step setup produced event StopEvent
None
step 1 --> step 2 -- > context is shared
step 1 --> stop event --> step 2 --> global context is missing data.

during the debuggin session we can see 2 workflow sessions and their respetive root context but ctx.data["some_database"] does not return any value the key is missing
so basically when we have multiple workflow sessions the context is lost
Context is not persistent between run() calls afaik (yet, anyways, working on a session UX)

I would rewrite this workflow

Plain Text
from llama_index.core.workflow import (
    Context,
    Workflow,
    StartEvent,
    StopEvent,
    step,
)
import asyncio  # Import asyncio to run the async functions

class QueryEvent(Event):
  query: str

class WaitExampleFlow(Workflow):
    @step
    async def setup(self, ctx: Context, ev: StartEvent) -> QueryEvent:
        if "data" not in ev.data:
          raise ValueError("Data is missing")
        ctx.data["some_database"] = ev.data
        
        if "query" not in ev.data:
          raise ValueError("Query is missing")         

        return QueryEvent(query=ev.query)

    @step
    async def query(self, ctx: Context, ev: QueryEvent) -> StopEvent:
        data = ctx.data["some_database"]
        return StopEvent(result=f"Got the data {data}")


async def main():
    w = WaitExampleFlow(verbose=True)
    result = await w.run(query=query, data=data)


if __name__ == "__main__":
    asyncio.run(main())
It might have been persitent between runs at one point, but made a small change to allow calling .run() on the same instance concurrently (i.e asyncio,gather(w.run(), w.run())
uh oh, will update, thanks for flagging
Add a reply
Sign up and join the conversation on Discord