@narumitw/pi-btw 0.49.3 → 0.49.5

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@narumitw/pi-btw",
-	"version": "0.49.3",
+	"version": "0.49.5",
 	"description": "Pi extension that adds a /btw side-question command.",
 	"type": "module",
 	"license": "MIT",

package/src/btw.ts

@@ -3,6 +3,7 @@ import {
 	clampThinkingLevel,
 	getSupportedThinkingLevels,
 	type Model,
+	type ProviderHeaders,
 } from "@earendil-works/pi-ai";
 import {
 	BorderedLoader,
@@ -76,15 +77,10 @@ interface LoadBtwThinkingLevelOptions {
 	warn?: (message: string) => void;
 }
 
-interface BtwModelRegistry {
-	find(provider: string, modelId: string): Model<Api> | undefined;
-	getApiKeyAndHeaders(
-		model: Model<Api>,
-	): Promise<
-		| { ok: true; apiKey?: string; headers?: Record<string, string>; env?: Record<string, string> }
-		| { ok: false; error: string }
-	>;
-}
+type BtwModelRegistry = Pick<
+	ExtensionCommandContext["modelRegistry"],
+	"find" | "getApiKeyAndHeaders"
+>;
 
 interface ResolveBtwModelOptions {
 	settings: BtwSettings;
@@ -154,11 +150,15 @@ export async function resolveBtwModel({
 function hasRequestAuth(auth: SideQuestionAuth): boolean {
 	return Boolean(
 		auth.apiKey ||
-			(auth.headers && Object.keys(auth.headers).length > 0) ||
+			providerHeadersHaveValue(auth.headers) ||
 			(auth.env && Object.keys(auth.env).length > 0),
 	);
 }
 
+function providerHeadersHaveValue(headers: ProviderHeaders | undefined): boolean {
+	return headers !== undefined && Object.values(headers).some((value) => value !== null);
+}
+
 export async function loadBtwThinkingLevel(
 	currentThinkingLevel: BtwThinkingLevel,
 	options: LoadBtwThinkingLevelOptions = {},

package/src/side-thread.ts

@@ -4,6 +4,7 @@ import type {
 	Context,
 	Message,
 	Model,
+	ProviderHeaders,
 	SimpleStreamOptions,
 	UserMessage,
 } from "@earendil-works/pi-ai";
@@ -22,7 +23,7 @@ export type BtwThinkingLevel = (typeof BTW_THINKING_LEVELS)[number];
 
 export interface SideQuestionAuth {
 	apiKey?: string;
-	headers?: Record<string, string>;
+	headers?: ProviderHeaders;
 	env?: Record<string, string>;
 }