PRD-2078: Immediate stop generation behavior fix

[Tomass673]

Task: https://app.clickup.com/t/24336023/PRD-2078?comment=90150176699394

[Tomass673]

@veliseev93 please take a look

[veliseev93]

Could you check if we just need to tweak patchChatQueryData a bit?

It seems that in some cases, using patchChatQueryData doesn't trigger a rerender, causing isResponseGenerating in the component to return stale value.

Try this fix, for example (please be careful and test):

import { merge } from 'lodash-es';
import { queryClient } from '@open-webui-react-native/shared/data-access/query-client';
import { chatQueriesKeys } from '../chat-queries-keys';
import { ChatResponse } from '../models';

export const patchChatQueryData = (chatId: string, partialData: Partial<ChatResponse>): void => {
  queryClient.setQueryData<ChatResponse>(chatQueriesKeys.get(chatId).queryKey, (draft) => {
    if (!draft) {
      return undefined;
    }

    return new ChatResponse(merge(draft, partialData));
  });
};