File size: 2,551 Bytes
3d700dd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 | import { useMutation, useQueryClient } from "@tanstack/react-query";
import AgentServerConversationService from "#/api/conversation-service/agent-server-conversation-service.api";
import { DirectConversationInfo } from "#/api/agent-server-adapter";
interface ForkConversationVariables {
/** The conversation being branched from. */
sourceConversationId: string;
/** The message the action was invoked on. */
eventId: string;
/** Set for an "edit message" branch: exclude the message and restore this text. */
editText?: string | null;
/** Optional title for the fork, so it reads distinctly from its source. */
title?: string;
}
interface ForkConversationResult {
info: DirectConversationInfo;
/** Whether the message was excluded; the caller only prefills when true. */
excluded: boolean;
}
/**
* Branches a conversation from a message. Edit-mode (`editText` set) resolves
* the message's parent (via getEventParentId) and branches there, excluding
* the message; otherwise it branches at the message (inclusive). Local
* agent-server only.
*/
export const useForkConversation = () => {
const queryClient = useQueryClient();
return useMutation({
mutationKey: ["fork-conversation"],
mutationFn: async ({
sourceConversationId,
eventId,
editText,
title,
}: ForkConversationVariables): Promise<ForkConversationResult> => {
let fromEventId = eventId;
let excluded = false;
if (editText != null) {
const parentId = await AgentServerConversationService.getEventParentId(
sourceConversationId,
eventId,
);
if (parentId) {
fromEventId = parentId;
excluded = true;
}
}
const info = await AgentServerConversationService.forkConversation(
sourceConversationId,
fromEventId,
title,
);
// Older agent-servers (< 1.31.0) ignore `from_event_id` and copy the
// whole conversation. When honored, the fork's HEAD is `fromEventId`; if
// not, the message wasn't excluded — don't prefill (would duplicate).
if (excluded) {
const leafEventId = (info as { leaf_event_id?: string | null })
.leaf_event_id;
if (leafEventId !== fromEventId) {
excluded = false;
}
}
return { info, excluded };
},
onSuccess: () => {
// Surface the fork in the sidebar without wiping loaded pages.
queryClient.invalidateQueries({ queryKey: ["user", "conversations"] });
},
});
};
|