@jeffreycao/copilot-api 1.6.2 → 1.6.4

package/dist/main.js

@@ -23,7 +23,7 @@ if (typeof args["enterprise-url"] === "string") process.env.COPILOT_API_ENTERPRI
 const { auth } = await import("./auth-I-jV5RwF.js");
 const { checkUsage } = await import("./check-usage-DYLFqYhN.js");
 const { debug } = await import("./debug-DcC7ZPH0.js");
-const { start } = await import("./start-CcPjJUMF.js");
+const { start } = await import("./start-6DUmwwg0.js");
 const main = defineCommand({
 	meta: {
 		name: "copilot-api",

package/dist/{server-ABtVTO7w.js → server-DS5Brc8C.js}

@@ -756,6 +756,9 @@ function mapContent(content) {
 				image_url: { url: `data:${block.source.media_type};base64,${block.source.data}` }
 			});
 			break;
+		case "document":
+			contentParts.push(createDocumentTextPart());
+			break;
 		case "tool_reference":
 			contentParts.push({
 				type: "text",
@@ -765,6 +768,12 @@ function mapContent(content) {
 	}
 	return contentParts;
 }
+function createDocumentTextPart() {
+	return {
+		type: "text",
+		text: "A PDF document was attached, but this api cannot send PDF inputs directly. Analyze using other tools."
+	};
+}
 function translateAnthropicToolsToOpenAI(anthropicTools) {
 	if (!anthropicTools) return;
 	return anthropicTools.map((tool) => ({
@@ -1026,7 +1035,7 @@ const translateUserMessage = (message) => {
 			continue;
 		}
 		const converted = translateUserContentBlock(block);
-		if (converted) pendingContent.push(converted);
+		if (converted.length > 0) pendingContent.push(...converted);
 	}
 	flushPendingContent(pendingContent, items, { role: "user" });
 	return items;
@@ -1076,9 +1085,10 @@ const translateAssistantMessage = (message, model, applyPhase) => {
 };
 const translateUserContentBlock = (block) => {
 	switch (block.type) {
-		case "text": return createTextContent(block.text);
-		case "image": return createImageContent(block);
-		default: return;
+		case "text": return [createTextContent(block.text)];
+		case "image": return [createImageContent(block)];
+		case "document": return [createFileContent(block)];
+		default: return [];
 	}
 };
 const translateAssistantContentBlock = (block) => {
@@ -1122,6 +1132,11 @@ const createImageContent = (block) => ({
 	image_url: `data:${block.source.media_type};base64,${block.source.data}`,
 	detail: "auto"
 });
+const createFileContent = (block) => ({
+	type: "input_file",
+	file_data: `data:${block.source.media_type};base64,${block.source.data}`,
+	filename: block.title ?? "document.pdf"
+});
 const createReasoningContent = (block) => {
 	const { encryptedContent, id } = parseReasoningSignature(block.signature);
 	const thinking = block.thinking === THINKING_TEXT$1 ? "" : block.thinking;
@@ -1373,6 +1388,9 @@ const convertToolResultContent = (content) => {
 			case "image":
 				result.push(createImageContent(block));
 				break;
+			case "document":
+				result.push(createFileContent(block));
+				break;
 			case "tool_reference":
 				result.push(createTextContent(`Tool ${block.tool_name} loaded`));
 				break;
@@ -1940,6 +1958,7 @@ const TOOL_REFERENCE_TURN_BOUNDARY = "Tool loaded.";
 const IDE_EXECUTE_CODE_TOOL = "mcp__ide__executeCode";
 const IDE_GET_DIAGNOSTICS_TOOL = "mcp__ide__getDiagnostics";
 const IDE_GET_DIAGNOSTICS_DESCRIPTION = "Get language diagnostics from VS Code. Returns errors, warnings, information, and hints for files in the workspace.";
+const PDF_FILE_READ_PREFIX = "PDF file read:";
 const getCompactCandidateText = (message) => {
 	if (message.role !== "user") return "";
 	if (typeof message.content === "string") return message.content;
@@ -1989,6 +2008,126 @@ const mergeContentWithTexts = (tr, textBlocks) => {
 		content: [...tr.content, ...textBlocks]
 	};
 };
+const mergeContentWithAttachments = (tr, attachments) => {
+	if (typeof tr.content === "string") return {
+		...tr,
+		content: [{
+			type: "text",
+			text: tr.content
+		}, ...attachments]
+	};
+	return {
+		...tr,
+		content: [...tr.content, ...attachments]
+	};
+};
+const isAttachmentBlock = (block) => {
+	return block.type === "image" || block.type === "document";
+};
+const getMergeableToolResultIndices = (toolResults) => {
+	return toolResults.flatMap((block, index) => hasToolRef(block) ? [] : [index]);
+};
+const mergeAttachmentsIntoToolResults = (toolResults, attachmentsByToolResultIndex) => {
+	if (attachmentsByToolResultIndex.size === 0) return toolResults;
+	return toolResults.map((block, index) => {
+		const matchedAttachments = attachmentsByToolResultIndex.get(index);
+		if (!matchedAttachments) return block;
+		const orderedAttachments = [...matchedAttachments].sort((left, right) => left.order - right.order).map(({ attachment }) => attachment);
+		return mergeContentWithAttachments(block, orderedAttachments);
+	});
+};
+const assignAttachmentsToToolResults = (target, attachments, options) => {
+	const { toolResultIndices } = options;
+	const fallbackToolResultIndices = options.fallbackToolResultIndices ?? toolResultIndices;
+	if (attachments.length === 0) return;
+	if (toolResultIndices.length > 0 && toolResultIndices.length === attachments.length) {
+		for (const [index, toolResultIndex] of toolResultIndices.entries()) {
+			const currentAttachments$1 = target.get(toolResultIndex);
+			if (currentAttachments$1) {
+				currentAttachments$1.push(attachments[index]);
+				continue;
+			}
+			target.set(toolResultIndex, [attachments[index]]);
+		}
+		return;
+	}
+	const lastToolResultIndex = fallbackToolResultIndices.at(-1);
+	if (lastToolResultIndex === void 0) return;
+	const currentAttachments = target.get(lastToolResultIndex);
+	if (currentAttachments) {
+		currentAttachments.push(...attachments);
+		return;
+	}
+	target.set(lastToolResultIndex, [...attachments]);
+};
+const startsWithPdfFileRead = (toolResult) => {
+	if (typeof toolResult.content === "string") return toolResult.content.startsWith(PDF_FILE_READ_PREFIX);
+	if (toolResult.content.some((block) => block.type === "document")) return false;
+	if (toolResult.content.length === 0) return false;
+	const firstBlock = toolResult.content[0];
+	if (firstBlock.type !== "text") return false;
+	return firstBlock.text.startsWith(PDF_FILE_READ_PREFIX);
+};
+const collectMergeableUserContent = (content) => {
+	const toolResults = [];
+	const textBlocks = [];
+	const attachments = [];
+	for (const [order, block] of content.entries()) {
+		if (block.type === "tool_result") {
+			toolResults.push(block);
+			continue;
+		}
+		if (block.type === "text") {
+			textBlocks.push(block);
+			continue;
+		}
+		if (isAttachmentBlock(block)) {
+			attachments.push({
+				attachment: block,
+				order
+			});
+			continue;
+		}
+		return null;
+	}
+	return {
+		toolResults,
+		textBlocks,
+		attachments
+	};
+};
+const mergeAttachmentsForToolResults = (toolResults, attachments) => {
+	if (attachments.length === 0) return toolResults;
+	const documentBlocks = attachments.filter(({ attachment }) => attachment.type === "document");
+	const mergeableToolResultIndices = getMergeableToolResultIndices(toolResults);
+	const pdfReadToolResultIndices = mergeableToolResultIndices.filter((index) => startsWithPdfFileRead(toolResults[index]));
+	const attachmentsByToolResultIndex = /* @__PURE__ */ new Map();
+	let remainingAttachments = attachments;
+	let countMatchToolResultIndices = mergeableToolResultIndices;
+	if (documentBlocks.length > 0 && pdfReadToolResultIndices.length > 0) {
+		const matchedDocumentCount = Math.min(pdfReadToolResultIndices.length, documentBlocks.length);
+		const matchedDocuments = documentBlocks.slice(0, matchedDocumentCount);
+		const matchedDocumentOrders = new Set(matchedDocuments.map(({ order }) => order));
+		const matchedPdfToolResultIndices = pdfReadToolResultIndices.slice(0, matchedDocumentCount);
+		const matchedPdfToolResultIndexSet = new Set(matchedPdfToolResultIndices);
+		assignAttachmentsToToolResults(attachmentsByToolResultIndex, matchedDocuments, { toolResultIndices: matchedPdfToolResultIndices });
+		countMatchToolResultIndices = mergeableToolResultIndices.filter((index) => !matchedPdfToolResultIndexSet.has(index));
+		remainingAttachments = attachments.filter(({ attachment, order }) => attachment.type !== "document" || !matchedDocumentOrders.has(order));
+	}
+	assignAttachmentsToToolResults(attachmentsByToolResultIndex, remainingAttachments, {
+		toolResultIndices: countMatchToolResultIndices,
+		fallbackToolResultIndices: mergeableToolResultIndices
+	});
+	return mergeAttachmentsIntoToolResults(toolResults, attachmentsByToolResultIndex);
+};
+const mergeUserMessageContent = (content) => {
+	const mergeableContent = collectMergeableUserContent(content);
+	if (!mergeableContent) return null;
+	const { toolResults, textBlocks, attachments } = mergeableContent;
+	if (toolResults.length === 0 || textBlocks.length === 0 && attachments.length === 0) return null;
+	const mergedToolResults = textBlocks.length === 0 ? toolResults : mergeToolResult(toolResults, textBlocks);
+	return mergeAttachmentsForToolResults(mergedToolResults, attachments);
+};
 const mergeToolResult = (toolResults, textBlocks) => {
 	if (toolResults.length === textBlocks.length) return toolResults.map((tr, i) => mergeContentWithText(tr, textBlocks[i]));
 	const lastIndex = toolResults.length - 1;
@@ -2006,23 +2145,14 @@ const mergeToolResultForClaude = (anthropicPayload, options) => {
 	for (const [index, msg] of anthropicPayload.messages.entries()) {
 		if (options?.skipLastMessage && index === lastMessageIndex) continue;
 		if (msg.role !== "user" || !Array.isArray(msg.content)) continue;
-		const toolResults = [];
-		const textBlocks = [];
-		let valid = true;
-		for (const block of msg.content) if (block.type === "tool_result") toolResults.push(block);
-		else if (block.type === "text") textBlocks.push(block);
-		else {
-			valid = false;
-			break;
-		}
-		if (!valid || toolResults.length === 0 || textBlocks.length === 0) continue;
-		msg.content = mergeToolResult(toolResults, textBlocks);
+		const mergedContent = mergeUserMessageContent(msg.content);
+		if (mergedContent) msg.content = mergedContent;
 	}
 };
 const sanitizeIdeTools = (payload) => {
 	if (!payload.tools || payload.tools.length === 0) return;
 	payload.tools = payload.tools.flatMap((tool) => {
-		if (tool.name === IDE_EXECUTE_CODE_TOOL) return [];
+		if (tool.name === IDE_EXECUTE_CODE_TOOL && !tool.defer_loading) return [];
 		if (tool.name === IDE_GET_DIAGNOSTICS_TOOL) return [{
 			...tool,
 			description: IDE_GET_DIAGNOSTICS_DESCRIPTION
@@ -3035,4 +3165,4 @@ server.route("/:provider/v1/models", providerModelRoutes);
 
 //#endregion
 export { server };
-//# sourceMappingURL=server-ABtVTO7w.js.map
+//# sourceMappingURL=server-DS5Brc8C.js.map