Is it possible to update two state within a node ?

[deva2zcodeops]

I am working on developing a multi-agent system and have encountered a situation where I need to update two different states simultaneously. For example, consider the following code snippet where I update the state for the convo agent. In addition to this, I need to update the state for another list, agent_type (not an individual agent/ node). In the subsequent node, I will use the values stored in agent_type to perform conditional operations and proceed further. Is there any way I can solve it ?

class AgentState(TypedDict):
    primary_convo_agent_response : Annotated[list, add_messages]
    agent_type : Annotated[list, add_messages]

async def PrimaryConvoAgent(state):
    messages = state['user_messages']
    user_message = messages[-1].content
    prompt = primary_convo_agent_prompt.format(
        user_message=user_message
    )
    primary_convo_res = openai_client.chat.completions.create(
        model="gpt-4o-mini",
        messages=[
            {"role": "system", "content": "You are a helpful assistant"},
            {"role": "user", "content": prompt}
        ]
    ).choices[0].message.content
    return {"primary_convo_agent_response": [AIMessage(content=primary_convo_res)]}

[vbarda]

@deva2zcodeops yes, you can send update for multiple state keys like this:

async def PrimaryConvoAgent(state):
    ...
    agent_type_update = ...
    return {"primary_convo_agent_response": [AIMessage(content=primary_convo_res)], "agent_type": [agent_type_update]}

[deva2zcodeops]

Thanks, Got it worked 👍