// Types for pi-custom-provider — imports from pi, only extension-specific types here
import type { Api, KnownApi, KnownProvider, Model } from "@earendil-works/pi-ai";

// ─── API-type predicates ─────────────────────────────────────────

/** APIs using Authorization: Bearer header (= OpenAI-compatible subset) */
export function usesAuthHeader(api: KnownApi): boolean {
  return api.startsWith("openai") || api === "mistral-conversations";
}

// ─── models.json config types ────────────────────────────────────

export interface ModelConfig
  extends Partial<
    Pick<Model<Api>, "baseUrl" | "reasoning" | "thinkingLevelMap" | "input" | "contextWindow" | "maxTokens" | "cost" | "headers">
  > {
  id: string;
  name?: string;
  api?: KnownApi;
  compat?: Record<string, unknown>;
}

export interface ProviderConfig {
  name?: string;
  baseUrl: string;
  api: KnownApi;
  apiKey?: string;
  /** Plugin-owned OAuth flow. Do not use models.json's built-in `oauth` field; Pi only accepts it for Radius. */
  oauthProvider?: KnownProvider;
  /** Optional OAuth credential JSON path. Empty means use the provider ID's stored Pi OAuth credential. */
  oauthJsonPath?: string;
  headers?: Record<string, string>;
  authHeader?: boolean;
  compat?: Record<string, unknown>;
  models: ModelConfig[];
}

export interface ModelsConfig {
  providers: Record<string, ProviderConfig>;
}

// ─── Extension-specific ──────────────────────────────────────────

/** A discovered model: same as ModelConfig but with name guaranteed */
export interface DiscoveredModel extends ModelConfig {
  name: string;
  /** How this candidate was found. Discovery-only; never written to models.json. */
  suggestedBy?: "api" | "model-id" | "base-url";
}