@codeworksh/aikit 0.8.0 → 0.9.0

package/index.mjs

@@ -1,8 +1,9 @@
 import { t as __exportAll } from "./rolldown-runtime-D7D4PA-g.mjs";
-import { S as NamedError, _ as registry, a as KnownProviderEnumSchema, c as calculateCost, d as getModels, f as getProviders, g as optionsKey, i as KnownProviderEnum, l as clampThinkingLevel, m as modelsAreEqual, n as APIMethodEnumSchema, p as model_exports, r as ActiveThinkingLevel, s as ThinkingBudgets$1, u as getModel, v as thinkingTokenBudgetField } from "./model-CBnredXj.mjs";
-import { C as ToolCallSchema, D as transformMessages, E as resolveJsonSchemaConstraint, S as AssistantMessageSchema, T as message_exports, g as shortHash, n as createOpenAICodex, r as openaiCodex, v as openAICodexTools, w as createAssistantMessage$1, x as resolveOpenAICodexToolConstraint } from "./openai-codex-t4B7g-gv.mjs";
+import { C as NamedError, _ as optionsKey, a as KnownProviderEnumSchema, c as calculateCost, d as getModels, f as getProviders, h as modelsAreEqual, i as KnownProviderEnum, l as clampThinkingLevel, m as model_exports, n as APIMethodEnumSchema, r as ActiveThinkingLevel, s as ThinkingBudgets$1, t as APIMethodEnum, u as getModel, v as registry, y as thinkingTokenBudgetField } from "./model-gCDbmGqT.mjs";
+import { C as ToolCallSchema, D as transformMessages, E as resolveJsonSchemaConstraint, S as AssistantMessageSchema, T as message_exports, g as shortHash, n as createOpenAICodex, r as openaiCodex, v as openAICodexTools, w as createAssistantMessage$1, x as resolveOpenAICodexToolConstraint } from "./openai-codex-Bqzti6Mc.mjs";
 import { r as normalize, t as failure_exports } from "./failure-SAPyG-w2.mjs";
-import { i as packageForModel, o as resolveLanguageModel, r as loadProviderFactory } from "./registry-DYdYTwFt.mjs";
+import { i as packageForModel, o as resolveLanguageModel, r as loadProviderFactory } from "./registry-ZxwIW0gX.mjs";
+import { i as githubCopilot, r as createGitHubCopilot } from "./github-copilot-SIUCJz1G.mjs";
 import Type$2, { Type, Type as Type$1 } from "typebox";
 import { jsonSchema, streamText } from "ai";
 import { parse } from "partial-json";
@@ -638,6 +639,11 @@ async function resolveAISDKLanguageModel(model, options, thinkingTokenBudget) {
 			...typeof factoryOptions.compat === "object" && factoryOptions.compat !== null ? factoryOptions.compat : {}
 		};
 	}
+	if (npm === "@codeworksh/ai-sdk-github-copilot") {
+		if (options.sessionId) factoryOptions.sessionId ??= options.sessionId;
+		const budgetField = thinkingTokenBudgetField(model);
+		if (budgetField && thinkingTokenBudget !== void 0 && thinkingTokenBudget > 0) factoryOptions.transformRequestBody = thinkingTokenBudgetTransform(budgetField, thinkingTokenBudget, factoryOptions.transformRequestBody);
+	}
 	const provider = createProvider(factoryOptions);
 	return resolveLanguageModel(provider, model, options.method);
 }
@@ -651,9 +657,14 @@ const ANTHROPIC_MIN_THINKING_BUDGET = 1024;
 * Adaptive models are exempt: they are sent an effort level and no budget, so
 * there is no floor to fall under.
 */
-function minimumThinkingBudget(model) {
+/** Whether the model talks Anthropic Messages — directly, or via Copilot's `/v1/messages` route. */
+function isAnthropicMessagesModel(model) {
 	const key = optionsKey(model);
-	if (key !== "anthropic" && key !== "google-vertex-anthropic") return 0;
+	if (key === "anthropic" || key === "google-vertex-anthropic") return true;
+	return key === "github-copilot" && model.api?.method === APIMethodEnum.messages;
+}
+function minimumThinkingBudget(model) {
+	if (!isAnthropicMessagesModel(model)) return 0;
 	return model.compat?.forceAdaptiveThinking ? 0 : ANTHROPIC_MIN_THINKING_BUDGET;
 }
 function resolvePlan(model, context, options) {
@@ -716,6 +727,10 @@ function disabledProviderOptions(model) {
 	if (!model.reasoning) return {};
 	const key = optionsKey(model);
 	if (key === "anthropic" || key === "google-vertex-anthropic") return { [key]: { thinking: { type: "disabled" } } };
+	if (key === "github-copilot") {
+		if (model.api?.method === APIMethodEnum.messages) return { [key]: { thinking: { type: "disabled" } } };
+		return {};
+	}
 	if (key === "google" || key === "google-vertex") {
 		if (model.thinkingLevelMap?.off === null) return {};
 		return { [key]: { thinkingConfig: googleDisabledThinkingConfig(model) } };
@@ -763,6 +778,24 @@ function reasoningProviderOptions(model, plan) {
 	const key = optionsKey(model);
 	const budget = plan.budget;
 	if (key === "openai" || key === "xai" || key === "openai-codex") return { [key]: { reasoningEffort: mapped } };
+	if (key === "github-copilot") {
+		const method = model.api?.method;
+		if (method === APIMethodEnum.responses) return { [key]: { reasoningEffort: mapped } };
+		if (method === APIMethodEnum.messages) {
+			if (model.compat?.forceAdaptiveThinking) return { [key]: {
+				thinking: {
+					type: "adaptive",
+					display: "summarized"
+				},
+				effort: anthropicEffort(model, level)
+			} };
+			return { [key]: { thinking: {
+				type: "enabled",
+				budgetTokens: budget
+			} } };
+		}
+		return {};
+	}
 	if (key === "openrouter") return { [key]: { reasoning: { effort: mapped === "off" ? "none" : mapped } } };
 	if (key === "google" || key === "google-vertex") {
 		const thinkingLevel = googleThinkingLevel(mapped);
@@ -1002,7 +1035,7 @@ function isRecord(value) {
 }
 function encodeOpenAIReasoningSignature(providerMetadata) {
 	if (!isRecord(providerMetadata)) return;
-	const metadata = providerMetadata.openai;
+	const metadata = providerMetadata.openai ?? providerMetadata["github-copilot"];
 	if (!isRecord(metadata) || typeof metadata.itemId !== "string") return;
 	const signature = { itemId: metadata.itemId };
 	if (typeof metadata.reasoningEncryptedContent === "string" || metadata.reasoningEncryptedContent === null) signature.reasoningEncryptedContent = metadata.reasoningEncryptedContent;
@@ -1048,6 +1081,25 @@ function assistantMessages(message, model) {
 		if (part.type === "thinking") {
 			const thinking = sanitizeSurrogates(part.thinking);
 			if (thinking.trim().length === 0 && !part.thinkingSignature) continue;
+			if (message.protocol === KnownProviderEnum.githubCopilot) {
+				const method = model.api?.method;
+				const openai = part.thinkingSignature ? decodeOpenAIReasoningSignature(part.thinkingSignature) : void 0;
+				if (method === APIMethodEnum.responses && openai) assistantContent.push({
+					type: "reasoning",
+					text: thinking,
+					providerOptions: { "github-copilot": openai }
+				});
+				else if (method === APIMethodEnum.messages && part.thinkingSignature && !openai) assistantContent.push({
+					type: "reasoning",
+					text: thinking,
+					providerOptions: { anthropic: { signature: part.thinkingSignature } }
+				});
+				else if (thinking.trim().length > 0) assistantContent.push({
+					type: "text",
+					text: thinking
+				});
+				continue;
+			}
 			const reasoning = {
 				type: "reasoning",
 				text: thinking
@@ -1217,7 +1269,7 @@ function cacheProviderOptions(model, options) {
 		if (retention === "long") openai.promptCacheRetention = "24h";
 		return Object.keys(openai).length > 0 ? { [key]: openai } : {};
 	}
-	if ((key === "anthropic" || key === "google-vertex-anthropic") && retention) return { [key]: { cacheControl: {
+	if (isAnthropicMessagesModel(model) && retention) return { [key]: { cacheControl: {
 		type: "ephemeral",
 		...retention === "long" ? { ttl: "1h" } : {}
 	} } };
@@ -1452,10 +1504,9 @@ function resolvePricing(model, pricing) {
 * unchanged -- subtracting there would quietly under-ask by the whole budget.
 */
 function resolveMaxOutputTokens(model, plan) {
-	const key = optionsKey(model);
-	if (key === "openai-codex") return void 0;
+	if (optionsKey(model) === "openai-codex") return void 0;
 	if (plan.maxTokens === void 0) return void 0;
-	return (key === "anthropic" || key === "google-vertex-anthropic") && !model.compat?.forceAdaptiveThinking ? plan.maxTokens - plan.budget : plan.maxTokens;
+	return isAnthropicMessagesModel(model) && !model.compat?.forceAdaptiveThinking ? plan.maxTokens - plan.budget : plan.maxTokens;
 }
 const stream$1 = (model, context, options) => {
 	const stream = new AssistantMessageEventStream();
@@ -1542,6 +1593,7 @@ function registerAISDKProtocols() {
 	registerProtocolProvider(createAISDKProtocol(KnownProviderEnum.openai));
 	registerProtocolProvider(createAISDKProtocol(KnownProviderEnum.openaiCompatible));
 	registerProtocolProvider(createAISDKProtocol(KnownProviderEnum.openaiCodex));
+	registerProtocolProvider(createAISDKProtocol(KnownProviderEnum.githubCopilot));
 	registerProtocolProvider(createAISDKProtocol(KnownProviderEnum.openrouter));
 	registerProtocolProvider(createAISDKProtocol(KnownProviderEnum.xai));
 }
@@ -1616,6 +1668,6 @@ function validateToolArguments(tool, toolExecution) {
 	return validateSchema(tool.parameters, toolExecution.rawArgs, `Tool "${toolExecution.name}"`);
 }
 //#endregion
-export { event_exports as Event, EventStream, failure_exports as Failure, message_exports as Message, model_exports as Model, protocol_exports as Protocol, ThinkingBudgets, Type, createAssistantMessageEventStream, createOpenAICodex, llm, openAICodexTools, openaiCodex, stream, validateSchema, validateToolArguments, validateToolCall };
+export { event_exports as Event, EventStream, failure_exports as Failure, message_exports as Message, model_exports as Model, protocol_exports as Protocol, ThinkingBudgets, Type, createAssistantMessageEventStream, createGitHubCopilot, createOpenAICodex, githubCopilot, llm, openAICodexTools, openaiCodex, stream, validateSchema, validateToolArguments, validateToolCall };
 
 //# sourceMappingURL=index.mjs.map