/**
 * Message within an agent thread with TextPart/ToolPart jsonb parts
 * @generated by @constructive-io/graphql-codegen
 * DO NOT EDIT - changes will be overwritten
 */
import { useMutation, useQueryClient } from '@tanstack/react-query';
import { getClient } from '../client';
import { buildSelectionArgs } from '../selection';
import { agentMessageKeys } from '../query-keys';
import { agentMessageMutationKeys } from '../mutation-keys';
export function useUpdateAgentMessageMutation(params) {
    const args = buildSelectionArgs(params.selection);
    const { selection: _selection, ...mutationOptions } = params ?? {};
    void _selection;
    const queryClient = useQueryClient();
    return useMutation({
        mutationKey: agentMessageMutationKeys.all,
        mutationFn: ({ id, agentMessagePatch }) => getClient()
            .agentMessage.update({
            where: {
                id,
            },
            data: agentMessagePatch,
            select: args.select,
        })
            .unwrap(),
        onSuccess: (_, variables) => {
            queryClient.invalidateQueries({
                queryKey: agentMessageKeys.detail(variables.id),
            });
            queryClient.invalidateQueries({
                queryKey: agentMessageKeys.lists(),
            });
        },
        ...mutationOptions,
    });
}
