How to refresh session data.

[tsartrooper]

so i am using google oauth for authentication, so after signing, user should choose a role. Following page provides with that interface, and after the user chooses the role, the role is updated on the database, but the client side session updation happens only after the page is refreshed.

choose-role page handling function:

const handleChoose = async (role: string) => {
        if (session?.user.role != null) return;

        console.log("changed role: ", role);
        await execute(role);

        const result = await update({
            ...session,
            user: {
                ...session?.user,
                role: role,
            },
        });

        if (!isPending) {
            if (result) {
                toast.success(`Role's been set, refresh the page`);
                window.location.href = "/dashboard";
            } else {
                toast.error(`Failed to update the session ${result}`);
            }

            router.refresh();
            router.replace("/dashboard");
        }
    };

notice here i have to refresh the page to update the session.
server action :

export const chooseRole = authenticatedAction
  .createServerAction()
  .input(z.string())
  .handler(async ({ input, ctx }) => {
    if (!ctx.user?.id) {
      throw new Error("Unauthorized");
    }

    const user = await prisma.user.findUnique({
      where: {id: ctx.user.id}
    });

    if(user?.role != null) return user;

    const updatedUser = await prisma.user.update({
      where: { id: ctx.user.id },
      data: { role: input },
    });

    console.log("role set: ",updatedUser.role);

    return;
  });

these are callbacks:

callbacks: {

   async jwt({ token, user, trigger, session }) {

     console.log("jwt trigger:",trigger);
     if (trigger === "update") {
       
       token=  { ...token, ...session.user };
       console.log("updated token: ", session.user);
     }

     if (user) {
       token.id = user.id;
       token.name = user.name;
       token.email = user.email;
       token.image = user.image;
       token.role = user.role;
       
       const dbUser = await prisma.user.findUnique({ where: { id: token.id as string } });
       if (dbUser && dbUser.role) {
         token.role = dbUser.role;
       }
     } 

     // console.log("jwt callback: ", token);

     return token;
   },

   async session({ session, token, trigger}) {
     console.log("session: ",session);
     session.user.id = token.id as string;
     session.user.name = token.name as string;
     session.user.email = token.email as string;
     session.user.image = token.image as string;
     session.user.role = token.role as string;

     console.log("trigger:",trigger);

     console.log("session after db check on role: ", session);

     return session;
   },
 },

the session updation should happen imediately after the role is persisted on the database, and not through page refresh. Is that the expected behaviour?