@jeffreycao/copilot-api 1.10.1 → 1.10.3

package/dist/{server-DpVPS3zt.js → server-UsHsMP_2.js}

@@ -1,6 +1,6 @@
 import { t as PATHS } from "./paths-DC-mqCY3.js";
-import { D as generateTraceId, E as prepareMessageProxyHeaders, I as state, M as compactMessageSections, O as requestContext, P as compactSystemPromptStarts, T as prepareInteractionHeaders, _ as copilotWebSocketHeaders, c as getUUID, d as sleep, f as getCopilotUsage, g as copilotHeaders, h as copilotBaseUrl, j as compactAutoContinuePromptStarts, k as resolveTraceId$1, l as isNullish, m as forwardError, n as cacheModels, o as generateRequestIdFromPayload, p as HTTPError, s as getRootSessionId, u as parseUserIdMetadata, w as prepareForCompact } from "./utils-C5ej0z8n.js";
-import { a as getConfig, c as getReasoningEffortForModel, d as isResponsesApiContextManagementModel, f as isResponsesApiWebSearchEnabled, i as getClaudeTokenMultiplier, l as getSmallModel, o as getExtraPromptForModel, p as isResponsesApiWebSocketEnabled, r as getAnthropicApiKey, s as getProviderConfig, t as getProxyEnvDispatcher, u as isMessagesApiEnabled } from "./proxy-De0Po8kG.js";
+import { D as generateTraceId, E as prepareMessageProxyHeaders, I as state, M as compactMessageSections, O as requestContext, P as compactSystemPromptStarts, T as prepareInteractionHeaders, _ as copilotWebSocketHeaders, c as getUUID, d as sleep, f as getCopilotUsage, g as copilotHeaders, h as copilotBaseUrl, j as compactAutoContinuePromptStarts, k as resolveTraceId$1, l as isNullish, m as forwardError, n as cacheModels, o as generateRequestIdFromPayload, p as HTTPError, s as getRootSessionId, u as parseUserIdMetadata, w as prepareForCompact } from "./utils-jHLgqAq2.js";
+import { _ as setModelMappings, a as getConfig, c as getProviderConfig, d as isMessagesApiEnabled, f as isResponsesApiContextManagementModel, g as resolveMappedModel, i as getClaudeTokenMultiplier, l as getReasoningEffortForModel, m as isResponsesApiWebSocketEnabled, o as getExtraPromptForModel, p as isResponsesApiWebSearchEnabled, r as getAnthropicApiKey, s as getModelMappings, t as getProxyEnvDispatcher, u as getSmallModel } from "./proxy-DQLzdeq3.js";
 import consola from "consola";
 import fs from "node:fs/promises";
 import path from "node:path";
@@ -12,6 +12,7 @@ import fs$1, { readFileSync } from "node:fs";
 import { streamSSE } from "hono/streaming";
 import util from "node:util";
 import { events } from "fetch-event-stream";
+import { z } from "zod";
 import { WebSocket } from "undici";
 //#region src/lib/request-auth.ts
 function normalizeApiKeys(apiKeys) {
@@ -26,6 +27,14 @@ function normalizeApiKeys(apiKeys) {
 function getConfiguredApiKeys() {
 	return normalizeApiKeys(getConfig().auth?.apiKeys);
 }
+function normalizeApiKey(apiKey) {
+	if (typeof apiKey !== "string") return null;
+	return apiKey.trim() || null;
+}
+function getConfiguredAdminApiKeys() {
+	const adminApiKey = normalizeApiKey(getConfig().auth?.adminApiKey);
+	return adminApiKey ? [adminApiKey] : [];
+}
 function extractRequestApiKey(c) {
 	const xApiKey = c.req.header("x-api-key")?.trim();
 	if (xApiKey) return xApiKey;
@@ -46,11 +55,14 @@ function createAuthMiddleware(options = {}) {
 	const getApiKeys = options.getApiKeys ?? getConfiguredApiKeys;
 	const allowUnauthenticatedPaths = options.allowUnauthenticatedPaths ?? ["/"];
 	const allowOptionsBypass = options.allowOptionsBypass ?? true;
+	const allowWhenNoApiKeys = options.allowWhenNoApiKeys ?? true;
+	const shouldSkipPath = options.shouldSkipPath ?? (() => false);
 	return async (c, next) => {
 		if (allowOptionsBypass && c.req.method === "OPTIONS") return next();
+		if (shouldSkipPath(c.req.path)) return next();
 		if (allowUnauthenticatedPaths.includes(c.req.path)) return next();
 		const apiKeys = getApiKeys();
-		if (apiKeys.length === 0) return next();
+		if (apiKeys.length === 0) return allowWhenNoApiKeys ? next() : createUnauthorizedResponse(c);
 		const requestApiKey = extractRequestApiKey(c);
 		if (!requestApiKey || !apiKeys.includes(requestApiKey)) return createUnauthorizedResponse(c);
 		return next();
@@ -1085,6 +1097,32 @@ completionRoutes.post("/", async (c) => {
 	}
 });
 //#endregion
+//#region src/routes/admin/config/route.ts
+const configRoutes = new Hono();
+const modelMappingsRequestSchema = z.object({ modelMappings: z.record(z.string(), z.string()) });
+configRoutes.get("/model-mappings", (c) => {
+	return c.json({
+		configPath: PATHS.CONFIG_PATH,
+		modelMappings: getModelMappings()
+	});
+});
+configRoutes.post("/model-mappings", async (c) => {
+	try {
+		const parseResult = modelMappingsRequestSchema.safeParse(await c.req.json());
+		if (!parseResult.success) return c.json({ error: {
+			message: parseResult.error.issues[0]?.message ?? "Invalid request body.",
+			type: "invalid_request_error"
+		} }, 400);
+		const updatedModelMappings = setModelMappings(parseResult.data.modelMappings);
+		return c.json({
+			configPath: PATHS.CONFIG_PATH,
+			modelMappings: updatedModelMappings
+		});
+	} catch (error) {
+		return await forwardError(c, error);
+	}
+});
+//#endregion
 //#region src/services/copilot/create-embeddings.ts
 const createEmbeddings = async (payload) => {
 	if (!state.copilotToken) throw new Error("Copilot token not found");
@@ -1819,6 +1857,7 @@ async function countTokensViaAnthropic(c, payload) {
 */
 async function handleCountTokens(c) {
 	const anthropicPayload = await c.req.json();
+	anthropicPayload.model = resolveMappedModel(anthropicPayload.model);
 	const providerModelAlias = parseProviderModelAlias(anthropicPayload.model);
 	if (providerModelAlias) {
 		anthropicPayload.model = providerModelAlias.model;
@@ -2777,6 +2816,12 @@ const removeResponsesWebSocketPoolEntry = (poolKey, entry) => {
 const unrefTimer = (timer) => {
 	if (typeof timer === "object" && "unref" in timer && typeof timer.unref === "function") timer.unref();
 };
+const createResponsesWebSocketError = (message, event) => {
+	const reason = event?.error ?? event?.message;
+	if (reason === void 0 || reason === "") return new Error(message);
+	const cause = toError(reason);
+	return new Error(`${message}: ${cause.message}`, { cause });
+};
 const openResponsesWebSocket = async ({ headers, url }) => await new Promise((resolve, reject) => {
 	const dispatcher = getProxyEnvDispatcher();
 	const websocket = new WebSocket(url, dispatcher ? {
@@ -2791,9 +2836,9 @@ const openResponsesWebSocket = async ({ headers, url }) => await new Promise((re
 		cleanup();
 		resolve(websocket);
 	};
-	const onError = () => {
+	const onError = (event) => {
 		cleanup();
-		reject(/* @__PURE__ */ new Error("Failed to create responses websocket"));
+		reject(createResponsesWebSocketError("Failed to create responses websocket", event));
 	};
 	websocket.addEventListener("open", onOpen);
 	websocket.addEventListener("error", onError);
@@ -2815,8 +2860,8 @@ const createWebSocketMessageStream = async function* (websocket) {
 		closed = true;
 		wake();
 	};
-	const onError = () => {
-		error = /* @__PURE__ */ new Error("Responses websocket stream error");
+	const onError = (event) => {
+		error = createResponsesWebSocketError("Responses websocket stream error", event);
 		wake();
 	};
 	websocket.addEventListener("message", onMessage);
@@ -2905,14 +2950,23 @@ const MESSAGE_TYPE = "message";
 const COMPACTION_SIGNATURE_PREFIX = "cm1#";
 const COMPACTION_SIGNATURE_SEPARATOR = "@";
 const THINKING_TEXT = "Thinking...";
-const translateAnthropicMessagesToResponsesPayload = (payload) => {
+const buildPromptCacheKey = (basePromptCacheKey, subagentAgentId) => {
+	if (!basePromptCacheKey) return null;
+	const normalizedSubagentAgentId = subagentAgentId?.trim() || null;
+	if (!normalizedSubagentAgentId) return basePromptCacheKey;
+	return `${basePromptCacheKey}:agent:${normalizedSubagentAgentId}`;
+};
+const translateAnthropicMessagesToResponsesPayload = (payload, subagentAgentId) => {
 	const input = [];
 	const applyPhase = shouldApplyPhase(payload.model);
 	for (const message of payload.messages) input.push(...translateMessage(message, payload.model, applyPhase));
+	const hasOriginalTools = Array.isArray(payload.tools) && payload.tools.length > 0;
 	const translatedTools = convertAnthropicTools(payload.tools);
 	const toolChoice = convertAnthropicToolChoice(payload.tool_choice);
-	const { sessionId: promptCacheKey } = parseUserIdMetadata(payload.metadata?.user_id);
-	return {
+	const { sessionId: metadataPromptCacheKey } = parseUserIdMetadata(payload.metadata?.user_id);
+	const sessionAffinity = requestContext.getStore()?.sessionAffinity?.trim() || null;
+	const promptCacheKey = buildPromptCacheKey(metadataPromptCacheKey ?? sessionAffinity, subagentAgentId);
+	const responsesPayload = {
 		model: payload.model,
 		input,
 		instructions: translateSystemPrompt(payload.system, payload.model),
@@ -2922,7 +2976,6 @@ const translateAnthropicMessagesToResponsesPayload = (payload) => {
 		tools: translatedTools,
 		tool_choice: toolChoice,
 		metadata: payload.metadata ? { ...payload.metadata } : null,
-		prompt_cache_key: promptCacheKey,
 		stream: payload.stream ?? null,
 		store: false,
 		parallel_tool_calls: true,
@@ -2932,6 +2985,8 @@ const translateAnthropicMessagesToResponsesPayload = (payload) => {
 		},
 		include: ["reasoning.encrypted_content"]
 	};
+	if (hasOriginalTools) responsesPayload.prompt_cache_key = promptCacheKey;
+	return responsesPayload;
 };
 const encodeCompactionCarrierSignature = (compaction) => {
 	return `${COMPACTION_SIGNATURE_PREFIX}${compaction.encrypted_content}${COMPACTION_SIGNATURE_SEPARATOR}${compaction.id}`;
@@ -3047,8 +3102,8 @@ const resolveAssistantPhase = (_model, content, applyPhase) => {
 	if (!content.some((block) => block.type === "text")) return;
 	return content.some((block) => block.type === "tool_use") ? "commentary" : "final_answer";
 };
-const shouldApplyPhase = (model) => {
-	return getExtraPromptForModel(model).includes("## Intermediary updates");
+const shouldApplyPhase = (_model) => {
+	return true;
 };
 const createTextContent = (text) => ({
 	type: "input_text",
@@ -4116,7 +4171,7 @@ const handleWithChatCompletions = async (c, anthropicPayload, options) => {
 };
 const handleWithResponsesApi = async (c, anthropicPayload, options) => {
 	const { logger, selectedModel, ...requestOptions } = options;
-	const responsesPayload = translateAnthropicMessagesToResponsesPayload(anthropicPayload);
+	const responsesPayload = translateAnthropicMessagesToResponsesPayload(anthropicPayload, requestOptions.subagentMarker?.agent_id);
 	const recordUsage = createCopilotUsageRecorder({
 		endpoint: "responses",
 		fallbackSessionId: requestOptions.sessionId,
@@ -4322,6 +4377,9 @@ const messagesFlowHandlers = {
 };
 async function handleCompletion(c) {
 	const anthropicPayload = await c.req.json();
+	const requestedModel = anthropicPayload.model;
+	anthropicPayload.model = resolveMappedModel(anthropicPayload.model);
+	if (anthropicPayload.model !== requestedModel) logger$3.debug(`Resolved model mapping: ${requestedModel} -> ${anthropicPayload.model}`);
 	const providerModelAlias = parseProviderModelAlias(anthropicPayload.model);
 	if (providerModelAlias) {
 		anthropicPayload.model = providerModelAlias.model;
@@ -4707,11 +4765,19 @@ const server = new Hono();
 server.use(traceIdMiddleware);
 server.use(logger());
 server.use(cors());
-server.use("*", createAuthMiddleware({ allowUnauthenticatedPaths: [
-	"/",
-	"/usage-viewer",
-	"/usage-viewer/"
-] }));
+server.use("*", createAuthMiddleware({
+	allowUnauthenticatedPaths: [
+		"/",
+		"/usage-viewer",
+		"/usage-viewer/"
+	],
+	shouldSkipPath: (path) => path.startsWith("/admin/")
+}));
+server.use("/admin/*", createAuthMiddleware({
+	getApiKeys: getConfiguredAdminApiKeys,
+	allowUnauthenticatedPaths: [],
+	allowWhenNoApiKeys: false
+}));
 server.get("/", (c) => c.text("Server running"));
 server.get("/usage-viewer", (c) => {
 	const usageViewerFileUrl = new URL("../pages/index.html", import.meta.url);
@@ -4719,6 +4785,7 @@ server.get("/usage-viewer", (c) => {
 });
 server.get("/usage-viewer/", (c) => c.redirect("/usage-viewer", 301));
 server.route("/chat/completions", completionRoutes);
+server.route("/admin/config", configRoutes);
 server.route("/models", modelRoutes);
 server.route("/embeddings", embeddingRoutes);
 server.route("/usage", usageRoute);
@@ -4735,4 +4802,4 @@ server.route("/:provider/v1/models", providerModelRoutes);
 //#endregion
 export { server };
 
-//# sourceMappingURL=server-DpVPS3zt.js.map
+//# sourceMappingURL=server-UsHsMP_2.js.map