@nick3/copilot-api 1.6.5 → 1.6.9

package/dist/{server-kG7N_-Aj.js → server-BhOhCgni.js}

@@ -1,9 +1,9 @@
-import { A as captureOutboundHeadersSnapshot, D as prepareMessageProxyHeaders, E as prepareInteractionHeaders, M as requestContext, N as resolveTraceId, O as accountFromState, T as prepareForCompact, _ as HTTPError, b as copilotHeaders, c as getUUID, d as parseUserIdMetadata, f as resolveAffinityKey, g as getCopilotUsage, h as getDeviceCode, k as state, l as isNullish, m as getGitHubUser, o as generateRequestIdFromPayload, p as sleep, s as getRootSessionId, t as pollAccessToken, u as normalizeStableSessionId, v as forwardError, w as normalizeDomain, y as copilotBaseUrl } from "./poll-access-token-tKPQmH_6.js";
-import { _ as DEFAULT_IDENTITY_ENTERPRISE_DOMAIN, a as getAccountClientIdentityByLoginAndApp, b as getCurrentIdentityEnvironment, d as loadRegistry, g as saveRegistry, h as saveAccountToken, l as listAccountsFromRegistry, m as removeAccountToken, p as removeAccountFromRegistry, r as addAccountToRegistry, t as isAccountType } from "./account-Db80RUHX.js";
+import { A as captureOutboundHeadersSnapshot, D as prepareMessageProxyHeaders, E as prepareInteractionHeaders, M as requestContext, N as resolveTraceId, O as accountFromState, T as prepareForCompact, _ as HTTPError, b as copilotHeaders, c as getUUID, d as parseUserIdMetadata, f as resolveAffinityKey, g as getCopilotUsage, h as getDeviceCode, k as state, l as isNullish, m as getGitHubUser, o as generateRequestIdFromPayload, p as sleep, s as getRootSessionId, t as pollAccessToken, u as normalizeStableSessionId, v as forwardError, w as normalizeDomain, y as copilotBaseUrl } from "./poll-access-token-c2IYJJHM.js";
+import { _ as DEFAULT_IDENTITY_ENTERPRISE_DOMAIN, a as getAccountClientIdentityByLoginAndApp, b as getCurrentIdentityEnvironment, d as loadRegistry, g as saveRegistry, h as saveAccountToken, l as listAccountsFromRegistry, m as removeAccountToken, p as removeAccountFromRegistry, r as addAccountToRegistry, t as isAccountType } from "./account-B4EBsn8H.js";
 import { r as ensurePaths, t as PATHS } from "./paths-DGlr310R.js";
-import "./get-copilot-token-Buo3CA1x.js";
-import { i as getRequestOutboundStore, r as getRedactedHeaderKeys } from "./request-outbound-C0g49Ab6.js";
-import { A as isMessageStartInputTokensFallbackEnabled, C as getModelAliasesInfo, D as getSmallModel, E as getReasoningEffortForModel, F as resolveModelAlias, I as shouldCompactUseSmallModel, M as isResponsesApiContextManagementModel, N as isResponsesApiWebSearchEnabled, O as isAccountAffinityEnabled, P as mergeConfigWithDefaults, S as getModelAliases, T as getProviderConfig, _ as getAnthropicApiKey, a as getClientIpInfo, b as getExtraPromptForModel, c as normalizeChatCompletionsUsage, d as toLocalDateString, f as copilotFetch, g as getAliasTargetSet, h as PROVIDER_TYPE_ANTHROPIC, i as extractResponsesUsageFromStreamEvent, j as isMessagesApiEnabled, k as isForceAgentEnabled, l as normalizeEmbeddingsUsage, m as isDevModeEnabled, n as applySharedSessionAffinityRetention, o as getRequestHistoryStore, p as flushPendingCapture, r as extractResponsesUsageFromResult, s as getStatsStore, t as accountsManager, u as normalizeMessagesUsage, v as getClaudeTokenMultiplier, w as getModelRefreshIntervalMs, x as getLogLevel, y as getConfig } from "./accounts-manager-B4eOuKSK.js";
+import "./get-copilot-token-COIPGosP.js";
+import { i as getRequestOutboundStore, r as getRedactedHeaderKeys } from "./request-outbound-qyTeXbzy.js";
+import { A as isMessageStartInputTokensFallbackEnabled, C as getModelAliasesInfo, D as getSmallModel, E as getReasoningEffortForModel, F as resolveModelAlias, I as shouldCompactUseSmallModel, M as isResponsesApiContextManagementModel, N as isResponsesApiWebSearchEnabled, O as isAccountAffinityEnabled, P as mergeConfigWithDefaults, S as getModelAliases, T as getProviderConfig, _ as getAnthropicApiKey, a as getClientIpInfo, b as getExtraPromptForModel, c as normalizeChatCompletionsUsage, d as toLocalDateString, f as copilotFetch, g as getAliasTargetSet, h as PROVIDER_TYPE_ANTHROPIC, i as extractResponsesUsageFromStreamEvent, j as isMessagesApiEnabled, k as isForceAgentEnabled, l as normalizeEmbeddingsUsage, m as isDevModeEnabled, n as applySharedSessionAffinityRetention, o as getRequestHistoryStore, p as flushPendingCapture, r as extractResponsesUsageFromResult, s as getStatsStore, t as accountsManager, u as normalizeMessagesUsage, v as getClaudeTokenMultiplier, w as getModelRefreshIntervalMs, x as getLogLevel, y as getConfig } from "./accounts-manager-BNVadA3o.js";
 import consola from "consola";
 import fs, { readFile } from "node:fs/promises";
 import { randomUUID, timingSafeEqual } from "node:crypto";
@@ -350,6 +350,52 @@ const _normalizeSdkModelId = (sdkModelId) => {
 	};
 };
 
+//#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/lib/request-initiator.ts
 function resolveEffectiveInitiator(baseInitiator, options) {
@@ -400,6 +446,7 @@ const createChatCompletions = async (payload, account, options) => {
 		requestId: options?.requestId,
 		callSite: "chat-completions"
 	});
+	logCopilotRateLimits(response.headers);
 	if (!response.ok) {
 		consola.error("Failed to create chat completions", response);
 		throw new HTTPError("Failed to create chat completions", response);
@@ -4354,7 +4401,7 @@ const createResponses = async (payload, { vision, initiator, upstreamRequestId,
 	};
 	prepareInteractionHeaders(sessionId, Boolean(subagentMarker), headers);
 	prepareForCompact(headers, compactType);
-	payload.service_tier = null;
+	payload.service_tier = void 0;
 	captureOutboundHeadersSnapshot(headers);
 	const response = await copilotFetch(`${copilotBaseUrl(ctx)}/responses`, {
 		method: "POST",
@@ -4364,6 +4411,7 @@ const createResponses = async (payload, { vision, initiator, upstreamRequestId,
 		requestId,
 		callSite: "responses"
 	});
+	logCopilotRateLimits(response.headers);
 	if (!response.ok) {
 		consola.error("Failed to create responses", response);
 		throw new HTTPError("Failed to create responses", response);
@@ -5419,6 +5467,7 @@ const createMessages = async (payload, account, options) => {
 		requestId: options?.requestId,
 		callSite: "messages"
 	});
+	logCopilotRateLimits(response.headers);
 	if (!response.ok) {
 		consola.error("Failed to create messages", response);
 		throw new HTTPError("Failed to create messages", response);
@@ -6985,6 +7034,7 @@ const handleResponses = async (c) => {
 		model: selectedModel.id
 	};
 	useFunctionApplyPatch(upstreamPayload);
+	removeUnsupportedTools(upstreamPayload);
 	applyResponsesApiContextManagement(upstreamPayload, selectedModel.capabilities.limits.max_prompt_tokens);
 	compactInputByLatestCompaction(upstreamPayload);
 	const premiumRemainingBefore = account.premiumRemaining;
@@ -7377,6 +7427,20 @@ async function handleNonStreamingResponses(params) {
 		});
 	}
 }
+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
@@ -7470,4 +7534,4 @@ server.route("/:provider/v1/models", providerModelRoutes);
 
 //#endregion
 export { server };
-//# sourceMappingURL=server-kG7N_-Aj.js.map
+//# sourceMappingURL=server-BhOhCgni.js.map