☰

Claude Fork Conversation

Adds forking functionality to claude.ai

  1. // ==UserScript==
  2. // @name         Claude Fork Conversation
  3. // @namespace    https://lugia19.com
  4. // @version      0.5.3
  5. // @description  Adds forking functionality to claude.ai
  6. // @match        https://claude.ai/*
  7. // @grant        none
  8. // @license      GPLv3
  9. // ==/UserScript==
  10.  
  11. (function () {
  12. 	'use strict';
  13. 	let pendingForkModel = null;
  14. 	let includeAttachments = true;
  15. 	let isProcessing = false;
  16. 	let pendingUseSummary = false;
  17.  
  18. 	//#region UI elements creation
  19. 	function createBranchButton() {
  20. 		const button = document.createElement('button');
  21. 		button.className = 'branch-button flex flex-row items-center gap-1 rounded-md p-1 py-0.5 text-xs transition-opacity delay-100 hover:bg-bg-200 group/button';
  22.  
  23. 		button.innerHTML = `
  24. 			<svg xmlns="http://www.w3.org/2000/svg" width="1.35em" height="1.35em" fill="currentColor" viewBox="0 0 22 22">
  25. 				<path d="M7 5C7 3.89543 7.89543 3 9 3C10.1046 3 11 3.89543 11 5C11 5.74028 10.5978 6.38663 10 6.73244V14.0396H11.7915C12.8961 14.0396 13.7915 13.1441 13.7915 12.0396V10.7838C13.1823 10.4411 12.7708 9.78837 12.7708 9.03955C12.7708 7.93498 13.6662 7.03955 14.7708 7.03955C15.8753 7.03955 16.7708 7.93498 16.7708 9.03955C16.7708 9.77123 16.3778 10.4111 15.7915 10.7598V12.0396C15.7915 14.2487 14.0006 16.0396 11.7915 16.0396H10V17.2676C10.5978 17.6134 11 18.2597 11 19C11 20.1046 10.1046 21 9 21C7.89543 21 7 20.1046 7 19C7 18.2597 7.4022 17.6134 8 17.2676V6.73244C7.4022 6.38663 7 5.74028 7 5Z"/>
  26. 			</svg>
  27. 			<span>Fork</span>
  28. 		`;
  29.  
  30. 		button.onclick = async (e) => {
  31. 			e.preventDefault();
  32. 			e.stopPropagation();
  33.  
  34. 			const modal = await createModal();
  35. 			document.body.appendChild(modal);
  36.  
  37. 			// Add event listeners
  38. 			modal.querySelector('#cancelFork').onclick = () => {
  39. 				modal.remove();
  40. 			};
  41.  
  42. 			// And in our modal click handler:
  43. 			modal.querySelector('#confirmFork').onclick = async () => {
  44. 				const model = modal.querySelector('select').value;
  45. 				const useSummary = modal.querySelector('#summaryMode').checked;
  46.  
  47. 				// Disable the button to prevent multiple clicks
  48. 				const confirmBtn = modal.querySelector('#confirmFork');
  49. 				confirmBtn.disabled = true;
  50. 				confirmBtn.textContent = 'Processing...';
  51.  
  52. 				await forkConversationClicked(model, button, modal, useSummary);
  53. 				modal.remove();
  54. 			};
  55.  
  56. 			// Click outside to cancel
  57. 			modal.onclick = (e) => {
  58. 				if (e.target === modal) {
  59. 					modal.remove();
  60. 				}
  61. 			};
  62. 		};
  63.  
  64. 		return button;
  65. 	}
  66.  
  67. 	async function createModal() {
  68. 		const modal = document.createElement('div');
  69. 		modal.className = 'fixed inset-0 bg-black bg-opacity-50 flex items-center justify-center z-50';
  70.  
  71. 		modal.innerHTML = `
  72. 		  <div class="bg-bg-100 rounded-lg p-6 shadow-xl max-w-sm w-full mx-4 border border-border-300">
  73. 			<h3 class="text-lg font-semibold mb-4 text-text-100">Choose Model for Fork</h3>
  74. 			<select class="w-full p-2 rounded mb-4 bg-bg-200 text-text-100 border border-border-300">
  75. 			  <option value="claude-sonnet-4-20250514">Sonnet 4</option>
  76. 			  <option value="claude-opus-4-20250514">Opus 4</option>
  77. 			  <option value="claude-3-7-sonnet-20250219">Sonnet 3.7</option>
  78. 			  <option value="claude-3-opus-20240229">Opus 3</option>
  79. 			  <option value="claude-3-5-haiku-20241022">Haiku 3.5</option>
  80. 			</select>
  81. 			
  82. 			<div class="mb-4 space-y-2">
  83. 			  <div class="flex items-center justify-between mb-3 p-2 bg-bg-200 rounded">
  84. 				<span class="text-text-100 font-medium">Fork Type:</span>
  85. 				<div class="flex items-center gap-4">
  86. 				  <label class="flex items-center space-x-2">
  87. 					<input type="radio" id="fullChatlog" name="forkType" value="full" checked class="accent-accent-main-100">
  88. 					<span class="text-text-100">Full Chatlog</span>
  89. 				  </label>
  90. 				  <label class="flex items-center space-x-2">
  91. 					<input type="radio" id="summaryMode" name="forkType" value="summary" class="accent-accent-main-100">
  92. 					<span class="text-text-100">Summary</span>
  93. 				  </label>
  94. 				</div>
  95. 			  </div>
  96. 			
  97. 			  <label class="flex items-center space-x-2">
  98. 				<input type="checkbox" id="includeFiles" class="rounded border-border-300" checked>
  99. 				<span class="text-text-100">Include files</span>
  100. 			  </label>
  101. 			</div>
  102. 			
  103. 			<p class="text-sm text-text-400 sm:text-[0.75rem]">Note: Should you choose a slow model such as Opus, you may need to wait and refresh the page for the response to appear.</p>
  104. 			<div class="mt-4 flex flex-col gap-2 sm:flex-row-reverse">
  105. 			  <button class="inline-flex items-center justify-center relative shrink-0 ring-offset-2 ring-offset-bg-300 ring-accent-main-100 focus-visible:outline-none focus-visible:ring-1 disabled:pointer-events-none disabled:opacity-50 disabled:shadow-none disabled:drop-shadow-none bg-accent-main-100 bg-gradient-to-r from-accent-main-100 via-accent-main-200/50 to-accent-main-200 bg-[length:200%_100%] hover:bg-right active:bg-accent-main-000 border-0.5 border-border-300 text-oncolor-100 font-medium font-styrene drop-shadow-sm transition-all shadow-[inset_0_0.5px_0px_rgba(255,255,0,0.15)] [text-shadow:_0_1px_2px_rgb(0_0_0_/_10%)] active:shadow-[inset_0_1px_6px_rgba(0,0,0,0.2)] hover:from-accent-main-200 hover:to-accent-main-200 h-9 px-4 py-2 rounded-lg min-w-[5rem] active:scale-[0.985] whitespace-nowrap" id="confirmFork">
  106. 				Fork Chat
  107. 			  </button>
  108. 			  <button class="inline-flex items-center justify-center relative shrink-0 ring-offset-2 ring-offset-bg-300 ring-accent-main-100 focus-visible:outline-none focus-visible:ring-1 disabled:pointer-events-none disabled:opacity-50 disabled:shadow-none disabled:drop-shadow-none bg-[radial-gradient(ellipse,_var(--tw-gradient-stops))] from-bg-500/10 from-50% to-bg-500/30 border-0.5 border-border-400 font-medium font-styrene text-text-100/90 transition-colors active:bg-bg-500/50 hover:text-text-000 hover:bg-bg-500/60 h-9 px-4 py-2 rounded-lg min-w-[5rem] active:scale-[0.985] whitespace-nowrap" id="cancelFork">
  109. 				Cancel
  110. 			  </button>
  111. 			</div>
  112. 		  </div>
  113. 		`;
  114.  
  115. 		try {
  116. 			const accountData = await fetchAccountSettings();
  117. 			originalSettings = accountData.settings;
  118. 		} catch (error) {
  119. 			console.error('Failed to fetch account settings:', error);
  120. 		}
  121.  
  122. 		return modal;
  123. 	}
  124.  
  125.  
  126. 	function findMessageControls(messageElement) {
  127. 		if (messageElement.classList.contains('font-user-message')) {
  128. 			const group = messageElement.closest('.group');
  129. 			const buttons = group?.querySelectorAll('button');
  130. 			if (!buttons) return;
  131. 			const editButton = Array.from(buttons).find(button =>
  132. 				button.textContent.includes('Edit')
  133. 			);
  134. 			return editButton?.closest('.justify-between');
  135. 		}
  136.  
  137. 		if (messageElement.classList.contains('font-claude-message')) {
  138. 			const group = messageElement.closest('.group');
  139. 			const buttons = group?.querySelectorAll('button');
  140. 			const retryButton = Array.from(buttons).find(button =>
  141. 				button.textContent.includes('Retry')
  142. 			);
  143. 			return retryButton?.closest('.justify-between');
  144. 		}
  145.  
  146. 		return null;
  147. 	}
  148.  
  149. 	function addBranchButtons() {
  150. 		if (isProcessing) return;
  151. 		try {
  152. 			isProcessing = true;
  153. 			const messages = document.querySelectorAll('.font-claude-message');
  154. 			messages.forEach((message) => {
  155. 				const controls = findMessageControls(message);
  156. 				if (controls && !controls.querySelector('.branch-button')) {
  157. 					const container = document.createElement('div');
  158. 					container.className = 'flex items-center gap-0.5';
  159. 					const divider = document.createElement('div');
  160. 					divider.className = 'w-px h-4/5 self-center bg-border-300 mr-0.5';
  161. 					const branchBtn = createBranchButton();
  162. 					container.appendChild(branchBtn);
  163. 					container.appendChild(divider);
  164. 					controls.insertBefore(container, controls.firstChild);
  165. 				}
  166. 			});
  167. 		} catch (error) {
  168. 			console.error('Error adding branch buttons:', error);
  169. 		} finally {
  170. 			isProcessing = false;
  171. 		}
  172. 	}
  173.  
  174. 	//#endregion
  175.  
  176. 	async function forkConversationClicked(model, forkButton, modal, useSummary = false) {
  177. 		// Get conversation ID from URL
  178. 		const conversationId = window.location.pathname.split('/').pop();
  179. 		console.log('Forking conversation', conversationId, 'with model', model);
  180.  
  181. 		if (originalSettings) {
  182. 			const newSettings = { ...originalSettings };
  183. 			newSettings.paprika_mode = null; // Ensure it's off when we create the conversation (will be overridden to on if needed)
  184. 			console.log('Updating settings:', newSettings);
  185. 			await updateAccountSettings(newSettings);
  186. 		}
  187.  
  188. 		// Set up our global to catch the next retry request
  189. 		pendingForkModel = model;
  190. 		includeAttachments = modal.querySelector('#includeFiles')?.checked ?? true;
  191. 		pendingUseSummary = useSummary;
  192.  
  193. 		// Find and click the retry button in the same control group as our fork button
  194. 		const buttonGroup = forkButton.closest('.justify-between');
  195. 		const retryButton = Array.from(buttonGroup.querySelectorAll('button'))
  196. 			.find(button => button.textContent.includes('Retry'));
  197.  
  198. 		if (retryButton) {
  199. 			// Dispatch pointerdown event which Radix UI components use
  200. 			retryButton.dispatchEvent(new PointerEvent('pointerdown', {
  201. 				bubbles: true,
  202. 				cancelable: true,
  203. 				view: window,
  204. 				pointerType: 'mouse'
  205. 			}));
  206.  
  207. 			// Wait for the dropdown to appear
  208. 			await new Promise(resolve => setTimeout(resolve, 300));
  209.  
  210. 			// Look for the dropdown menu with "With no changes" option
  211. 			const withNoChangesOption = Array.from(document.querySelectorAll('[role="menuitem"]'))
  212. 				.find(element => element.textContent.includes('With no changes'));
  213.  
  214. 			if (withNoChangesOption) {
  215. 				console.log('Detected retry dropdown, clicking "With no changes"');
  216. 				// For the menu item, a regular click should work
  217. 				withNoChangesOption.click();
  218. 			} else {
  219. 				console.log('No dropdown detected, assuming direct retry');
  220. 				retryButton.click();
  221. 				// If no dropdown appeared, the retry might have been triggered directly
  222. 			}
  223. 		} else {
  224. 			console.error('Could not find retry button');
  225. 		}
  226. 	}
  227.  
  228. 	//#region Convo extraction & Other API
  229.  
  230. 	let originalSettings = null;
  231.  
  232. 	async function fetchAccountSettings() {
  233. 		const response = await fetch('/api/account');
  234. 		console.log('Account settings response:', response);
  235. 		const data = await response.json();
  236. 		return data;
  237. 	}
  238.  
  239. 	async function updateAccountSettings(settings) {
  240. 		await fetch('/api/account', {
  241. 			method: 'PUT',
  242. 			headers: {
  243. 				'Content-Type': 'application/json'
  244. 			},
  245. 			body: JSON.stringify({ settings })
  246. 		});
  247. 	}
  248.  
  249. 	async function getConversationContext(orgId, conversationId, targetParentUuid) {
  250. 		const response = await fetch(`/api/organizations/${orgId}/chat_conversations/${conversationId}?tree=False&rendering_mode=messages&render_all_tools=true`);
  251. 		const conversationData = await response.json();
  252.  
  253. 		let messages = [];
  254. 		let projectUuid = conversationData?.project?.uuid || null;
  255. 		const chatName = conversationData.name;
  256. 		const files = []
  257. 		const syncsources = []
  258. 		const attachments = []
  259.  
  260. 		for (const message of conversationData.chat_messages) {
  261. 			let messageContent = [];
  262.  
  263. 			// Process content array
  264. 			for (const content of message.content) {
  265. 				if (content.text) {
  266. 					messageContent.push(content.text);
  267. 				}
  268. 				if (content.input?.code) {
  269. 					messageContent.push(content.input.code);
  270. 				}
  271. 				if (content.content?.text) {
  272. 					messageContent.push(content.content.text);
  273. 				}
  274. 			}
  275.  
  276. 			// Process files with download URLs
  277. 			if (message.files_v2) {
  278. 				for (const file of message.files_v2) {
  279. 					let fileUrl;
  280. 					if (file.file_kind === "image") {
  281. 						fileUrl = file.preview_asset.url;
  282. 					} else if (file.file_kind === "document") {
  283. 						fileUrl = file.document_asset.url;
  284. 					}
  285.  
  286. 					if (fileUrl) {
  287. 						files.push({
  288. 							uuid: file.file_uuid,
  289. 							url: fileUrl,
  290. 							kind: file.file_kind,
  291. 							name: file.file_name
  292. 						});
  293. 					}
  294. 				}
  295. 			}
  296.  
  297. 			// Add attachment objects
  298. 			if (message.attachments) {
  299. 				for (const attachment of message.attachments) {
  300. 					attachments.push(attachment);
  301. 				}
  302. 			}
  303.  
  304. 			// Process sync sources
  305. 			for (const sync of message.sync_sources) {
  306. 				syncsources.push(sync);
  307. 			}
  308.  
  309. 			messages.push(messageContent.join(' '));
  310.  
  311. 			// Process until we find a message that has our target UUID as parent
  312. 			if (message.parent_message_uuid === targetParentUuid) {
  313. 				break;
  314. 			}
  315. 		}
  316.  
  317. 		if (!includeAttachments) {
  318. 			return {
  319. 				chatName,
  320. 				messages,
  321. 				syncsources: [],
  322. 				attachments: [],
  323. 				files: [],
  324. 				projectUuid
  325. 			};
  326. 		}
  327.  
  328. 		return {
  329. 			chatName,
  330. 			messages,
  331. 			syncsources,
  332. 			attachments,
  333. 			files,
  334. 			projectUuid
  335. 		};
  336. 	}
  337.  
  338. 	//#region File handlers (download, upload, sync)
  339. 	async function downloadFiles(files) {
  340. 		const downloadedFiles = [];
  341.  
  342. 		for (const file of files) {
  343. 			try {
  344. 				const response = await fetch(file.url);
  345. 				const blob = await response.blob();
  346.  
  347. 				downloadedFiles.push({
  348. 					data: blob,
  349. 					name: file.name,
  350. 					kind: file.kind,
  351. 					originalUuid: file.uuid
  352. 				});
  353. 			} catch (error) {
  354. 				console.error(`Failed to download file ${file.name}:`, error);
  355. 			}
  356. 		}
  357.  
  358. 		return downloadedFiles;
  359. 	}
  360.  
  361. 	async function uploadFile(orgId, file) {
  362. 		const formData = new FormData();
  363. 		formData.append('file', file.data, file.name);
  364.  
  365. 		const response = await fetch(`/api/${orgId}/upload`, {
  366. 			method: 'POST',
  367. 			body: formData
  368. 		});
  369.  
  370. 		const uploadResult = await response.json();
  371. 		return uploadResult.file_uuid;
  372. 	}
  373.  
  374. 	async function processSyncSource(orgId, syncsource) {
  375. 		const response = await fetch(`/api/organizations/${orgId}/sync/chat`, {
  376. 			method: 'POST',
  377. 			headers: {
  378. 				'Content-Type': 'application/json'
  379. 			},
  380. 			body: JSON.stringify({
  381. 				sync_source_config: syncsource?.config,
  382. 				sync_source_type: syncsource?.type
  383. 			})
  384. 		})
  385.  
  386. 		if (!response.ok) {
  387. 			console.error(`Failed to process sync source: ${response.statusText}`);
  388. 			return null;
  389. 		}
  390. 		const result = await response.json();
  391. 		return result.uuid;
  392. 	}
  393. 	//#endregion
  394.  
  395. 	//#region Convo forking
  396. 	function generateUuid() {
  397. 		return 'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx'.replace(/[xy]/g, function (c) {
  398. 			const r = Math.random() * 16 | 0;
  399. 			const v = c === 'x' ? r : (r & 0x3 | 0x8);
  400. 			return v.toString(16);
  401. 		});
  402. 	}
  403.  
  404.  
  405. 	// 1. Create a standalone conversation creation function
  406. 	async function createConversation(orgId, name, model = null, projectUuid = null, thinking = false) {
  407. 		const newUuid = generateUuid();
  408. 		const bodyJSON = {
  409. 			uuid: newUuid,
  410. 			name: name,
  411. 			include_conversation_preferences: true,
  412. 			project_uuid: projectUuid,
  413. 		}
  414.  
  415. 		if (model) bodyJSON.model = model;
  416.  
  417. 		if (thinking) {
  418. 			let isFree = true;
  419. 			const statSigResponse = await fetch(`/api/bootstrap/${orgId}/statsig`, {
  420. 				method: 'GET',
  421. 				headers: {
  422. 					'Content-Type': 'application/json'
  423. 				},
  424. 			});
  425. 			if (statSigResponse.ok) {
  426. 				const statSigData = await statSigResponse.json();
  427. 				if (statSigData?.user?.custom?.orgType !== 'claude_free') {
  428. 					isFree = false;
  429. 				}
  430. 			}
  431.  
  432. 			if (!isFree) {
  433. 				bodyJSON.paprika_mode = "extended";
  434. 			}
  435.  
  436. 		}
  437. 		const createResponse = await fetch(`/api/organizations/${orgId}/chat_conversations`, {
  438. 			method: 'POST',
  439. 			headers: {
  440. 				'Content-Type': 'application/json'
  441. 			},
  442. 			body: JSON.stringify(bodyJSON)
  443. 		});
  444.  
  445. 		if (!createResponse.ok) {
  446. 			throw new Error('Failed to create conversation');
  447. 		}
  448.  
  449. 		return newUuid;
  450. 	}
  451.  
  452. 	async function createForkedConversation(orgId, context, model, styleData) {
  453. 		const newName = `Fork of ${context.chatName}`;
  454.  
  455. 		// Create a new chat conversation
  456. 		const newUuid = await createConversation(orgId, newName, model, context.projectUuid);
  457.  
  458. 		// Create the chatlog
  459. 		if (context.messages) {
  460. 			const chatlog = context.messages.map((msg, index) => {
  461. 				const role = index % 2 === 0 ? 'User' : 'Assistant';
  462. 				return `${role}\n${msg}`;
  463. 			}).join('\n\n');
  464.  
  465. 			context.attachments.push({
  466. 				"extracted_content": chatlog,
  467. 				"file_name": "chatlog.txt",
  468. 				"file_size": 0,
  469. 				"file_type": "text/plain"
  470. 			});
  471. 		}
  472.  
  473. 		const message = context.messages
  474. 			? "This conversation is forked from the attached chatlog.txt\nYou are Assistant. Simply say 'Acknowledged' and wait for user input."
  475. 			: "This conversation is forked based on the summary in conversation_summary.txt\nYou are Assistant. Simply say 'Acknowledged' and wait for user input.";
  476.  
  477. 		// Send initial message to set up conversation history
  478. 		const completionResponse = await fetch(`/api/organizations/${orgId}/chat_conversations/${newUuid}/completion`, {
  479. 			method: 'POST',
  480. 			headers: {
  481. 				'Content-Type': 'application/json'
  482. 			},
  483. 			body: JSON.stringify({
  484. 				prompt: message,
  485. 				model: model,
  486. 				parent_message_uuid: '00000000-0000-4000-8000-000000000000',
  487. 				attachments: context.attachments,
  488. 				files: context.files,
  489. 				sync_sources: context.syncsources,
  490. 				personalized_styles: styleData
  491. 			})
  492. 		});
  493.  
  494. 		if (!completionResponse.ok) {
  495. 			throw new Error('Failed to initialize conversation');
  496. 		}
  497.  
  498. 		// Sleep for 2 seconds to allow the response to be fully created
  499. 		await new Promise(r => setTimeout(r, 2000));
  500. 		return newUuid;
  501. 	}
  502.  
  503. 	async function generateSummary(orgId, context) {
  504. 		// Create a temporary conversation for summarization
  505. 		const summaryConvoName = `Temp_Summary_${Date.now()}`;
  506. 		const summaryConvoId = await createConversation(orgId, summaryConvoName, null, context.projectUuid, true);
  507.  
  508. 		try {
  509. 			// Create the chatlog
  510. 			const chatlog = context.messages.map((msg, index) => {
  511. 				const role = index % 2 === 0 ? 'User' : 'Assistant';
  512. 				return `${role}\n${msg}`;
  513. 			}).join('\n\n');
  514.  
  515. 			const summaryAttachments = [...context.attachments, {
  516. 				"extracted_content": chatlog,
  517. 				"file_name": "chatlog.txt",
  518. 				"file_size": 0,
  519. 				"file_type": "text/plain"
  520. 			}];
  521.  
  522. 			// Ask the model to create a summary
  523. 			const summaryPrompt = "I've attached a chatlog from a previous conversation. Please create a complete, detailed summary of the conversation that covers all important points, questions, and responses. This summary will be used to continue the conversation in a new chat, so make sure it provides enough context to understand the full discussion. Be through, and think things through. Don't include any information already present in the other attachments, as those will be forwarded to the new chat as well.";
  524.  
  525. 			const summaryResponse = await fetch(`/api/organizations/${orgId}/chat_conversations/${summaryConvoId}/completion`, {
  526. 				method: 'POST',
  527. 				headers: {
  528. 					'Content-Type': 'application/json'
  529. 				},
  530. 				body: JSON.stringify({
  531. 					prompt: summaryPrompt,
  532. 					parent_message_uuid: '00000000-0000-4000-8000-000000000000',
  533. 					attachments: summaryAttachments,
  534. 					files: [],
  535. 					sync_sources: []
  536. 				})
  537. 			});
  538.  
  539. 			if (!summaryResponse.ok) {
  540. 				console.error('Failed to generate summary');
  541. 				return null;
  542. 			}
  543.  
  544.  
  545. 			// Implement polling with timeout for assistant response
  546. 			const maxRetries = 6; // 6 retries * 5 seconds = 30 seconds max
  547. 			let assistantMessage = null;
  548.  
  549. 			for (let attempt = 0; attempt < maxRetries; attempt++) {
  550. 				// Wait 5 seconds between attempts
  551. 				await new Promise(r => setTimeout(r, 5000));
  552.  
  553. 				console.log(`Checking for summary response (attempt ${attempt + 1}/${maxRetries})...`);
  554.  
  555. 				// Fetch conversation to check for assistant response
  556. 				const convoResponse = await fetch(`/api/organizations/${orgId}/chat_conversations/${summaryConvoId}?tree=False&rendering_mode=messages&render_all_tools=true`);
  557. 				const convoData = await convoResponse.json();
  558.  
  559. 				// Find the assistant's response
  560. 				assistantMessage = convoData.chat_messages.find(msg => msg.sender === 'assistant');
  561.  
  562. 				if (assistantMessage) {
  563. 					console.log('Found assistant summary response');
  564. 					break;
  565. 				}
  566.  
  567. 				if (attempt === maxRetries - 1) {
  568. 					console.error('Could not find assistant summary response after maximum retries');
  569. 					throw new Error('Could not find assistant summary response after 30 seconds');
  570. 				}
  571. 			}
  572.  
  573. 			// Extract the text of the summary
  574. 			let summaryText = '';
  575. 			for (const content of assistantMessage.content) {
  576. 				if (content.text) {
  577. 					summaryText += content.text;
  578. 				}
  579. 				if (content.content?.text) {
  580. 					summaryText += content.text;
  581. 				}
  582. 			}
  583.  
  584. 			return summaryText;
  585. 		} finally {
  586. 			// Delete the temporary summarization conversation
  587. 			try {
  588. 				await fetch(`/api/organizations/${orgId}/chat_conversations/${summaryConvoId}`, {
  589. 					method: 'DELETE'
  590. 				});
  591. 			} catch (error) {
  592. 				console.error('Failed to delete temporary summary conversation:', error);
  593. 			}
  594. 		}
  595. 	}
  596. 	//#endregion
  597.  
  598. 	//#region Fetch patching
  599. 	const originalFetch = window.fetch;
  600. 	window.fetch = async (...args) => {
  601. 		const [input, config] = args;
  602.  
  603. 		// Get the URL string whether it's a string or Request object
  604.  
  605. 		let url = undefined
  606. 		if (input instanceof URL) {
  607. 			url = input.href
  608. 		} else if (typeof input === 'string') {
  609. 			url = input
  610. 		} else if (input instanceof Request) {
  611. 			url = input.url
  612. 		}
  613.  
  614. 		// In the fetch patching section
  615. 		if (url && url.includes('/retry_completion') && pendingForkModel) {
  616. 			console.log('Intercepted retry request:', config?.body);
  617. 			const bodyJSON = JSON.parse(config?.body);
  618. 			const messageID = bodyJSON?.parent_message_uuid;
  619. 			const urlParts = url.split('/');
  620. 			const orgId = urlParts[urlParts.indexOf('organizations') + 1];
  621. 			const conversationId = urlParts[urlParts.indexOf('chat_conversations') + 1];
  622.  
  623. 			let styleData = bodyJSON?.personalized_styles;
  624.  
  625. 			try {
  626. 				// Get conversation context
  627. 				console.log('Getting conversation context, includeAttachments:', includeAttachments);
  628. 				const context = await getConversationContext(orgId, conversationId, messageID);
  629.  
  630. 				// Process files and sync sources if needed
  631. 				if (includeAttachments) {
  632. 					const downloadedFiles = await downloadFiles(context.files);
  633.  
  634. 					// Parallel processing of files and syncs
  635. 					[context.files, context.syncsources] = await Promise.all([
  636. 						Promise.all(downloadedFiles.map(file => uploadFile(orgId, file))),
  637. 						Promise.all(context.syncsources.map(syncsource => processSyncSource(orgId, syncsource)))
  638. 					]);
  639. 				} else {
  640. 					context.files = [];
  641. 					context.syncsources = [];
  642. 				}
  643.  
  644. 				let newConversationId;
  645.  
  646. 				if (pendingUseSummary) {
  647. 					// Generate summary
  648. 					console.log("Generating summary for forking");
  649. 					const summary = await generateSummary(orgId, context, pendingForkModel);
  650. 					if (summary === null) {
  651. 						// Fall back to normal forking
  652. 						newConversationId = await createForkedConversation(orgId, context, pendingForkModel, styleData);
  653. 					} else {
  654. 						// Prepare context for summary-based fork
  655. 						const summaryContext = { ...context };
  656. 						summaryContext.messages = null;  // Don't generate chatlog
  657. 						summaryContext.attachments = [{
  658. 							"extracted_content": summary,
  659. 							"file_name": "conversation_summary.txt",
  660. 							"file_size": 0,
  661. 							"file_type": "text/plain"
  662. 						}];
  663. 						// Create forked conversation with summary
  664. 						newConversationId = await createForkedConversation(orgId, summaryContext, pendingForkModel, styleData);
  665. 					}
  666. 				} else {
  667. 					// Standard workflow with full chatlog
  668. 					newConversationId = await createForkedConversation(orgId, context, pendingForkModel, styleData);
  669. 				}
  670.  
  671. 				// Restore original settings
  672. 				if (originalSettings) {
  673. 					await updateAccountSettings(originalSettings);
  674. 				}
  675.  
  676. 				// Navigate to new conversation
  677. 				console.log('Forked conversation created:', newConversationId);
  678. 				window.location.href = `/chat/${newConversationId}`;
  679.  
  680. 			} catch (error) {
  681. 				console.error('Failed to fork conversation:', error);
  682. 				// Restore original settings even if forking fails
  683. 				if (originalSettings) {
  684. 					await updateAccountSettings(originalSettings);
  685. 				}
  686. 			}
  687.  
  688. 			originalSettings = null;
  689. 			pendingForkModel = null; // Clear the pending flag
  690. 			pendingUseSummary = false; // Clear the summary flag
  691. 			return new Response(JSON.stringify({ success: true }));
  692. 		}
  693.  
  694. 		return originalFetch(...args);
  695. 	};
  696. 	//#endregion
  697.  
  698. 	//Check for buttons every 3 seconds
  699. 	setInterval(addBranchButtons, 3000);
  700. })();

欢迎访问Greasy Fork镜像

土豆服务器，请按需使用

镜像地址随时可能被墙，建议加群获取最新地址

加入Q群

QingJ © 2025

镜像随时可能失效，请加Q群300939539或关注我们的公众号极客氢云获取最新地址