@jeffreycao/copilot-api 1.6.7 → 1.6.9

package/dist/main.js

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

package/dist/{server-B4dnJhMM.js → server-DzviJdGg.js}

@@ -248,6 +248,52 @@ async function checkRateLimit(state$1) {
 	consola.info("Rate limit wait completed, proceeding with request");
 }
 
+//#endregion
+//#region src/lib/copilot-rate-limit.ts
+const copilotRateLimitTypes = ["session", "weekly"];
+const copilotRateLimitHeaders = {
+	session: "x-usage-ratelimit-session",
+	weekly: "x-usage-ratelimit-weekly"
+};
+const hasGetMethod = (headers) => {
+	return "get" in headers && typeof headers.get === "function";
+};
+const getHeaderValue = (headers, headerName) => {
+	if (hasGetMethod(headers)) return headers.get(headerName);
+	const normalizedHeaderName = headerName.toLowerCase();
+	return Object.entries(headers).find(([key]) => key.toLowerCase() === normalizedHeaderName)?.[1] ?? null;
+};
+const parseCopilotRateLimitHeader = (headerValue) => {
+	const params = new URLSearchParams(headerValue);
+	const remaining = params.get("rem");
+	const resetAt = params.get("rst");
+	if (!remaining || !resetAt) return null;
+	return {
+		remaining,
+		resetAt
+	};
+};
+const getCopilotRateLimitUsage = (headers, type) => {
+	const headerName = copilotRateLimitHeaders[type];
+	const headerValue = getHeaderValue(headers, headerName);
+	if (!headerValue) return null;
+	const parsed = parseCopilotRateLimitHeader(headerValue);
+	if (!parsed) return null;
+	return {
+		type,
+		...parsed
+	};
+};
+const logCopilotRateLimits = (headers) => {
+	for (const type of copilotRateLimitTypes) {
+		const usage = getCopilotRateLimitUsage(headers, type);
+		if (!usage) continue;
+		const d = new Date(usage.resetAt);
+		const dateStr = Number.isNaN(d.getTime()) ? usage.resetAt : d.toLocaleString();
+		consola.info(`Copilot ${usage.type} quota remaining: ${usage.remaining}, resets at: ${dateStr}`);
+	}
+};
+
 //#endregion
 //#region src/services/copilot/create-chat-completions.ts
 const createChatCompletions = async (payload, options) => {
@@ -270,6 +316,7 @@ const createChatCompletions = async (payload, options) => {
 		headers,
 		body: JSON.stringify(payload)
 	});
+	logCopilotRateLimits(response.headers);
 	if (!response.ok) {
 		consola.error("Failed to create chat completions", response);
 		throw new HTTPError("Failed to create chat completions", response);
@@ -962,6 +1009,7 @@ const createResponses = async (payload, { vision, initiator, subagentMarker, req
 		headers,
 		body: JSON.stringify(payload)
 	});
+	logCopilotRateLimits(response.headers);
 	if (!response.ok) {
 		consola.error("Failed to create responses", response);
 		throw new HTTPError("Failed to create responses", response);
@@ -1947,6 +1995,7 @@ const createMessages = async (payload, anthropicBetaHeader, options) => {
 		headers,
 		body: JSON.stringify(payload)
 	});
+	logCopilotRateLimits(response.headers);
 	if (!response.ok) {
 		consola.error("Failed to create messages", response);
 		throw new HTTPError("Failed to create messages", response);
@@ -2572,6 +2621,8 @@ const handleWithMessagesApi = async (c, anthropicPayload, options) => {
 			for await (const event of response) {
 				const eventName = event.event;
 				const data = event.data ?? "";
+				if (data === "[DONE]") break;
+				if (!data) continue;
 				debugLazy(logger$7, () => ["Messages raw stream event:", data]);
 				await stream.writeSSE({
 					event: eventName,
@@ -3024,6 +3075,7 @@ const handleResponses = async (c) => {
 	const sessionId = getUUID(requestId);
 	logger$1.debug("Extracted session ID:", sessionId);
 	useFunctionApplyPatch(payload);
+	removeUnsupportedTools(payload);
 	if (!isResponsesApiWebSearchEnabled()) removeWebSearchTool(payload);
 	compactInputByLatestCompaction(payload);
 	const selectedModel = state.models?.data.find((model) => model.id === payload.model);
@@ -3095,6 +3147,20 @@ const removeWebSearchTool = (payload) => {
 		return t.type !== "web_search";
 	});
 };
+const COPILOT_UNSUPPORTED_TOOL_TYPES = new Set(["image_generation"]);
+const removeUnsupportedTools = (payload) => {
+	if (!Array.isArray(payload.tools) || payload.tools.length === 0) return;
+	const dropped = [];
+	payload.tools = payload.tools.filter((t) => {
+		const type = t.type;
+		if (COPILOT_UNSUPPORTED_TOOL_TYPES.has(type)) {
+			dropped.push(type);
+			return false;
+		}
+		return true;
+	});
+	if (dropped.length > 0) logger$1.debug("Removed unsupported tools:", dropped);
+};
 
 //#endregion
 //#region src/routes/responses/route.ts
@@ -3168,4 +3234,4 @@ server.route("/:provider/v1/models", providerModelRoutes);
 
 //#endregion
 export { server };
-//# sourceMappingURL=server-B4dnJhMM.js.map
+//# sourceMappingURL=server-DzviJdGg.js.map