Find answers from the community

Updated 2 months ago

Passing-context-to-set-account-balance-in-workflow

Hi, I’m trying to understand this code from the https://github.com/run-llama/multi-agent-concierge/blob/main/workflows.py#L274
I need to from utils import get_account_balance and use it within that workflow. However, I’m confused about how to pass the context (ctx) so that I can set ctx.data["user"]["account_balance"] = 1000.

Does anyone know how can I achieve this?

10 comments

LLogan M

You don't need to pass the context, its already available in the parent scope here
https://github.com/run-llama/multi-agent-concierge/blob/0288e8c06a63f12e477a8f27e33081a3ae080861/workflows.py#L264

NNull

thats correct but if i have

Plain Text

def get_account_balance(account_id: str) -> str:
                """Useful for looking up an account balance."""
                print(f"Looking up account balance for {account_id}")
                ctx.data["user"]["account_balance"] = 1000
                return f"Account {account_id} has a balance of ${ctx.data['user']['account_balance']}"

NNull

in other file, ctx is undefined

LLogan M

Right, thats why its defined as a nested function

If you want to define it in another file, you still need some wrapper defined as a nested function so that you can pass in the context from the parent scope

NNull

how would that look like?

LLogan M

Plain Text

def account_balance(self, ctx: Context, ev: AccountBalanceEvent) -> AuthenticateEvent | ConciergeEvent:
  ...
  def get_account_balance(account_id: str):
    return actual_get_account_balance(account_id, ctx)

LLogan M

Something like that

NNull

that worked!! thanks 👍

LLogan M

Awesome!!

Add a reply