Hi. I'm porting my Prefect 1 tasks to 2. Some of my original tasks used multiple values from the

prefect.context

, notably the

flow_name

,

flow_run_name

, and

flow_run_id

. Is there a way to pass the entire flow run context to a task? Or, is there a way to access the flow run context from within a task? If I uncomment the middle line in my flow function (see in the thread), the flow run seems to become unresponsive (I guess it's because it's waiting for "this flow run"'s state, but the flow is still running...)

Hi @Stéphan Taljaard I usually just pass FlowRunContext into tasks as arguments as needed (which you could do with single arg if you defined a dataclass / pydantic basemodel that represents the subset of FlowRunContext you care about in a task context)

But I don't necessarily want to pass many arguments this way.

out of curiosity, what qualms do you have with doing it this way?

Thanks Nate! I'll check tomorrow if it works if I pass it while using a dataclass/basemodel

out of curiosity, what qualms do you have with doing it this way?

Personal preference, I guess. I don't like "extracting" may variables, passing them as args, causing long lines of codes in two places

Here's a minimal functional example of what that could look like

from prefect import flow, task
from prefect.context import get_run_context
from pydantic import BaseModel
from uuid import UUID

class ImportantFlowRunContext(BaseModel):
    name: str
    id: UUID
    flow_id: UUID

@task
def task_needing_context(flow_context: ImportantFlowRunContext):
    print(
        f"This task is running in a flow run called {flow_context.name}, "
        f"which has ID: {flow_context.id} "
        f"and where the flow has ID: {flow_context.flow_id}"
    )
    
@flow
def some_flow():
    
    important_context = ImportantFlowRunContext(
        **dict(get_run_context().flow_run)
    )
    
    task_needing_context(flow_context=important_context)

if __name__ == "__main__":
    some_flow()

This works perfectly, thanks Nate I'll play around to see what in my version made it hang, and give feedback here, for other people coming across this thread