Issue while saving session state

[MahorShekhar]

My agent executes multiple tool call simulataneously. One of tool is get_customer_data which saves customer_id in session state. But in database I don't find it persisted.

def get_customer_data(customer_id: str, tool_context: ToolContext) -> dict:
    """
        Retrieve customer data based on customer ID
    Args:
        customer_id (str): customer id.
        tool_context: ToolContext object.
    Returns:
        dict: dictionary containing customer data with fields - id and name, loan_amount, account_history, interest loan, pre-approved loan.
    """
    # this is not persisting, will figure out
    tool_context.state['customer_id'] = customer_id
    customer_data = make_get_call("get_customer_data", customer_id)

    if not customer_data:
        return {
            "name": "Unknown",
        }

    return customer_data

[boyangsvl]

What do you mean by "in database"? Do you mean the database that you use as a session service storage?

[MahorShekhar]

yes @boyangsvl @hangfei I don't find my customer_id persisted in state field of sessions table.

By the way, could this be the reason that 3 more tool calls get executed along with get_customer_data. But always get_customer_data gets executed first. But other tool calls don't update session state. So there are no chances of dirty read or write, right ?

[Jacksunwei]

You can do dirty read. Once tool_context.state['customer_id'] = customer_id, a later tool call in the same invocation will be able to read from tool_context.state['customer_id'].

I put together a sample to illustrate it: https://github.com/google/adk-python/tree/main/contributing/samples/session_state_agent

Feel free to re-open if you have more questions.