@jupyterlite/ai 0.17.0 → 0.19.0

package/src/icons.ts

@@ -1,11 +0,0 @@
-import { LabIcon } from '@jupyterlab/ui-components';
-
-import jupyternautSvg from '../style/icons/jupyternaut-lite.svg';
-
-export const jupyternautIcon = new LabIcon({
-  name: '@jupyterlite/ai:jupyternaut',
-  svgstr: jupyternautSvg
-});
-
-const AI_AVATAR_BASE64 = btoa(jupyternautIcon.svgstr);
-export const AI_AVATAR = `data:image/svg+xml;base64,${AI_AVATAR_BASE64}`;

package/src/providers/built-in-providers.ts

@@ -1,241 +0,0 @@
-import { createAnthropic } from '@ai-sdk/anthropic';
-import { createGoogleGenerativeAI } from '@ai-sdk/google';
-import { createMistral } from '@ai-sdk/mistral';
-import { createOpenAI } from '@ai-sdk/openai';
-import { createOpenAICompatible } from '@ai-sdk/openai-compatible';
-
-import { BUILT_IN_PROVIDER_MODEL_INFO } from './generated-context-windows';
-import type { IProviderInfo } from '../tokens';
-import type { IModelOptions } from './models';
-
-/**
- * Anthropic provider
- */
-export const anthropicProvider: IProviderInfo = {
-  id: 'anthropic',
-  name: 'Anthropic Claude',
-  apiKeyRequirement: 'required',
-  defaultModels: [
-    'claude-opus-4-6',
-    'claude-sonnet-4-6',
-    'claude-opus-4-5',
-    'claude-opus-4-5-20251101',
-    'claude-sonnet-4-5',
-    'claude-sonnet-4-5-20250929',
-    'claude-haiku-4-5',
-    'claude-haiku-4-5-20251001',
-    'claude-opus-4-1',
-    'claude-opus-4-1-20250805',
-    'claude-opus-4-0',
-    'claude-opus-4-20250514',
-    'claude-sonnet-4-0',
-    'claude-sonnet-4-20250514'
-  ],
-  modelInfo: BUILT_IN_PROVIDER_MODEL_INFO.anthropic,
-  supportsBaseURL: true,
-  supportsHeaders: true,
-  providerToolCapabilities: {
-    webSearch: { implementation: 'anthropic' },
-    webFetch: { implementation: 'anthropic' }
-  },
-  factory: (options: IModelOptions) => {
-    if (!options.apiKey) {
-      throw new Error('API key required for Anthropic');
-    }
-    const anthropic = createAnthropic({
-      apiKey: options.apiKey,
-      headers: {
-        'anthropic-dangerous-direct-browser-access': 'true',
-        ...options.headers
-      },
-      ...(options.baseURL && { baseURL: options.baseURL })
-    });
-    const modelName = options.model ?? '';
-    return anthropic(modelName);
-  }
-};
-
-/**
- * Google Generative AI provider
- */
-export const googleProvider: IProviderInfo = {
-  id: 'google',
-  name: 'Google Generative AI',
-  apiKeyRequirement: 'required',
-  defaultModels: [
-    'gemini-3.1-pro-preview',
-    'gemini-3.1-pro-preview-customtools',
-    'gemini-3.1-flash-image-preview',
-    'gemini-3.1-flash-lite-preview',
-    'gemini-3-pro-image-preview',
-    'gemini-3-flash-preview',
-    'gemini-2.5-pro',
-    'gemini-2.5-flash',
-    'gemini-2.5-flash-image',
-    'gemini-2.5-flash-lite',
-    'gemini-2.5-computer-use-preview-10-2025',
-    'deep-research-pro-preview-12-2025',
-    'gemini-pro-latest',
-    'gemini-flash-latest',
-    'gemini-flash-lite-latest'
-  ],
-  modelInfo: BUILT_IN_PROVIDER_MODEL_INFO.google,
-  supportsBaseURL: true,
-  factory: (options: IModelOptions) => {
-    if (!options.apiKey) {
-      throw new Error('API key required for Google Generative AI');
-    }
-    const google = createGoogleGenerativeAI({
-      apiKey: options.apiKey,
-      ...(options.baseURL && { baseURL: options.baseURL })
-    });
-    const modelName = options.model || 'gemini-2.5-flash';
-    return google(modelName);
-  }
-};
-
-/**
- * Mistral provider
- */
-export const mistralProvider: IProviderInfo = {
-  id: 'mistral',
-  name: 'Mistral AI',
-  apiKeyRequirement: 'required',
-  defaultModels: [
-    'mistral-large-latest',
-    'mistral-medium-latest',
-    'mistral-medium-2508',
-    'mistral-small-latest',
-    'mistral-small-2506',
-    'ministral-3b-latest',
-    'ministral-8b-latest',
-    'ministral-14b-latest',
-    'magistral-small-latest',
-    'magistral-medium-latest',
-    'pixtral-large-latest',
-    'codestral-latest',
-    'devstral-latest'
-  ],
-  modelInfo: BUILT_IN_PROVIDER_MODEL_INFO.mistral,
-  supportsBaseURL: true,
-  factory: (options: IModelOptions) => {
-    if (!options.apiKey) {
-      throw new Error('API key required for Mistral');
-    }
-    const mistral = createMistral({
-      apiKey: options.apiKey,
-      ...(options.baseURL && { baseURL: options.baseURL })
-    });
-    const modelName = options.model || 'mistral-large-latest';
-    return mistral(modelName);
-  }
-};
-
-/**
- * OpenAI provider
- */
-export const openaiProvider: IProviderInfo = {
-  id: 'openai',
-  name: 'OpenAI',
-  apiKeyRequirement: 'required',
-  defaultModels: [
-    'gpt-5.4',
-    'gpt-5.4-mini',
-    'gpt-5.4-nano',
-    'gpt-5.2',
-    'gpt-5.2-2025-12-11',
-    'gpt-5.2-chat-latest',
-    'gpt-5.2-pro',
-    'gpt-5.2-pro-2025-12-11',
-    'gpt-5.2-codex',
-    'gpt-5.1',
-    'gpt-5.1-2025-11-13',
-    'gpt-5.1-chat-latest',
-    'gpt-5',
-    'gpt-5-2025-08-07',
-    'gpt-5-chat-latest',
-    'gpt-5-mini',
-    'gpt-5-mini-2025-08-07',
-    'gpt-5-nano',
-    'gpt-5-nano-2025-08-07',
-    'o4-mini',
-    'o4-mini-2025-04-16',
-    'o3-pro',
-    'o3',
-    'o3-2025-04-16',
-    'o3-mini',
-    'o3-mini-2025-01-31',
-    'o1',
-    'o1-2024-12-17',
-    'gpt-4.1',
-    'gpt-4.1-2025-04-14',
-    'gpt-4.1-mini',
-    'gpt-4.1-mini-2025-04-14',
-    'gpt-4.1-nano',
-    'gpt-4.1-nano-2025-04-14',
-    'gpt-4o',
-    'gpt-4o-2024-05-13',
-    'gpt-4o-2024-08-06',
-    'gpt-4o-2024-11-20',
-    'gpt-4o-mini',
-    'gpt-4o-mini-2024-07-18',
-    'gpt-4o-search-preview',
-    'gpt-4o-search-preview-2025-03-11',
-    'gpt-4o-mini-search-preview',
-    'gpt-4o-mini-search-preview-2025-03-11',
-    'gpt-3.5-turbo',
-    'gpt-3.5-turbo-0125'
-  ],
-  modelInfo: BUILT_IN_PROVIDER_MODEL_INFO.openai,
-  supportsBaseURL: true,
-  supportsHeaders: true,
-  providerToolCapabilities: {
-    webSearch: { implementation: 'openai' }
-  },
-  factory: (options: IModelOptions) => {
-    if (!options.apiKey) {
-      throw new Error('API key required for OpenAI');
-    }
-    const openai = createOpenAI({
-      apiKey: options.apiKey,
-      ...(options.baseURL && { baseURL: options.baseURL }),
-      ...(options.headers && { headers: options.headers })
-    });
-    const modelName = options.model || 'gpt-4o';
-    return openai(modelName);
-  }
-};
-
-/**
- * Generic OpenAI-compatible provider
- */
-export const genericProvider: IProviderInfo = {
-  id: 'generic',
-  name: 'Generic (OpenAI-compatible)',
-  apiKeyRequirement: 'optional',
-  defaultModels: [],
-  supportsBaseURL: true,
-  supportsHeaders: true,
-  supportsToolCalling: true,
-  description: 'Uses /chat/completions endpoint',
-  baseUrls: [
-    {
-      url: 'http://localhost:4000',
-      description: 'Default for local LiteLLM server'
-    },
-    {
-      url: 'http://localhost:11434/v1',
-      description: 'Default for local Ollama server'
-    }
-  ],
-  factory: (options: IModelOptions) => {
-    const openaiCompatible = createOpenAICompatible({
-      name: options.provider,
-      apiKey: options.apiKey || 'dummy',
-      baseURL: options.baseURL ?? '',
-      ...(options.headers && { headers: options.headers })
-    });
-    const modelName = options.model || 'gpt-4o';
-    return openaiCompatible(modelName);
-  }
-};

package/src/providers/generated-context-windows.ts

@@ -1,102 +0,0 @@
-/**
- * This file is generated by `jlpm sync:model-context-windows`.
- * Source: https://models.dev/api.json
- * Backed by: https://github.com/anomalyco/models.dev
- * Generated: 2026-04-08T16:23:34.080Z
- */
-
-import type { IProviderModelInfo } from '../tokens';
-
-export const BUILT_IN_PROVIDER_MODEL_INFO: Record<
-  string,
-  Record<string, IProviderModelInfo>
-> = {
-  anthropic: {
-    'claude-opus-4-6': { contextWindow: 1000000 },
-    'claude-sonnet-4-6': { contextWindow: 1000000 },
-    'claude-opus-4-5': { contextWindow: 200000 },
-    'claude-opus-4-5-20251101': { contextWindow: 200000 },
-    'claude-sonnet-4-5': { contextWindow: 200000 },
-    'claude-sonnet-4-5-20250929': { contextWindow: 200000 },
-    'claude-haiku-4-5': { contextWindow: 200000 },
-    'claude-haiku-4-5-20251001': { contextWindow: 200000 },
-    'claude-opus-4-1': { contextWindow: 200000 },
-    'claude-opus-4-1-20250805': { contextWindow: 200000 },
-    'claude-opus-4-0': { contextWindow: 200000 },
-    'claude-opus-4-20250514': { contextWindow: 200000 },
-    'claude-sonnet-4-0': { contextWindow: 200000 },
-    'claude-sonnet-4-20250514': { contextWindow: 200000 }
-  },
-  google: {
-    'gemini-3.1-pro-preview': { contextWindow: 1048576 },
-    'gemini-3.1-pro-preview-customtools': { contextWindow: 1048576 },
-    'gemini-3.1-flash-image-preview': { contextWindow: 131072 },
-    'gemini-3.1-flash-lite-preview': { contextWindow: 1048576 },
-    'gemini-3-flash-preview': { contextWindow: 1048576 },
-    'gemini-2.5-pro': { contextWindow: 1048576 },
-    'gemini-2.5-flash': { contextWindow: 1048576 },
-    'gemini-2.5-flash-image': { contextWindow: 32768 },
-    'gemini-2.5-flash-lite': { contextWindow: 1048576 },
-    'gemini-flash-latest': { contextWindow: 1048576 },
-    'gemini-flash-lite-latest': { contextWindow: 1048576 }
-  },
-  mistral: {
-    'mistral-large-latest': { contextWindow: 262144 },
-    'mistral-medium-latest': { contextWindow: 128000 },
-    'mistral-medium-2508': { contextWindow: 262144 },
-    'mistral-small-latest': { contextWindow: 256000 },
-    'mistral-small-2506': { contextWindow: 128000 },
-    'ministral-3b-latest': { contextWindow: 128000 },
-    'ministral-8b-latest': { contextWindow: 128000 },
-    'magistral-small-latest': { contextWindow: 128000 },
-    'magistral-medium-latest': { contextWindow: 128000 },
-    'pixtral-large-latest': { contextWindow: 128000 },
-    'codestral-latest': { contextWindow: 256000 },
-    'devstral-latest': { contextWindow: 262144 },
-    'devstral-2512': { contextWindow: 262144 }
-  },
-  openai: {
-    'gpt-5.4': { contextWindow: 1050000 },
-    'gpt-5.4-mini': { contextWindow: 400000 },
-    'gpt-5.4-nano': { contextWindow: 400000 },
-    'gpt-5.2': { contextWindow: 400000 },
-    'gpt-5.2-2025-12-11': { contextWindow: 400000 },
-    'gpt-5.2-chat-latest': { contextWindow: 128000 },
-    'gpt-5.2-pro': { contextWindow: 400000 },
-    'gpt-5.2-pro-2025-12-11': { contextWindow: 400000 },
-    'gpt-5.2-codex': { contextWindow: 400000 },
-    'gpt-5.1': { contextWindow: 400000 },
-    'gpt-5.1-2025-11-13': { contextWindow: 400000 },
-    'gpt-5.1-chat-latest': { contextWindow: 128000 },
-    'gpt-5': { contextWindow: 400000 },
-    'gpt-5-2025-08-07': { contextWindow: 400000 },
-    'gpt-5-chat-latest': { contextWindow: 400000 },
-    'gpt-5-mini': { contextWindow: 400000 },
-    'gpt-5-mini-2025-08-07': { contextWindow: 400000 },
-    'gpt-5-nano': { contextWindow: 400000 },
-    'gpt-5-nano-2025-08-07': { contextWindow: 400000 },
-    'o4-mini': { contextWindow: 200000 },
-    'o4-mini-2025-04-16': { contextWindow: 200000 },
-    'o3-pro': { contextWindow: 200000 },
-    o3: { contextWindow: 200000 },
-    'o3-2025-04-16': { contextWindow: 200000 },
-    'o3-mini': { contextWindow: 200000 },
-    'o3-mini-2025-01-31': { contextWindow: 200000 },
-    o1: { contextWindow: 200000 },
-    'o1-2024-12-17': { contextWindow: 200000 },
-    'gpt-4.1': { contextWindow: 1047576 },
-    'gpt-4.1-2025-04-14': { contextWindow: 1047576 },
-    'gpt-4.1-mini': { contextWindow: 1047576 },
-    'gpt-4.1-mini-2025-04-14': { contextWindow: 1047576 },
-    'gpt-4.1-nano': { contextWindow: 1047576 },
-    'gpt-4.1-nano-2025-04-14': { contextWindow: 1047576 },
-    'gpt-4o': { contextWindow: 128000 },
-    'gpt-4o-2024-05-13': { contextWindow: 128000 },
-    'gpt-4o-2024-08-06': { contextWindow: 128000 },
-    'gpt-4o-2024-11-20': { contextWindow: 128000 },
-    'gpt-4o-mini': { contextWindow: 128000 },
-    'gpt-4o-mini-2024-07-18': { contextWindow: 128000 },
-    'gpt-3.5-turbo': { contextWindow: 16385 },
-    'gpt-3.5-turbo-0125': { contextWindow: 16385 }
-  }
-};

package/src/providers/model-info.ts

@@ -1,88 +0,0 @@
-import type {
-  IProviderConfig,
-  IProviderInfo,
-  IProviderModelInfo,
-  IProviderRegistry
-} from '../tokens';
-
-const DATE_SUFFIX = /^(.*)-\d{4}-\d{2}-\d{2}$/;
-const SHORT_VERSION_SUFFIX = /^(.*)-\d{4}$/;
-
-// Treat rolling aliases and dated releases as the same model family so they
-// can share provider metadata such as context windows.
-function normalizeModelId(modelId: string): string {
-  if (modelId.endsWith('-latest')) {
-    return modelId.slice(0, -7);
-  }
-
-  const dateSuffixMatch = modelId.match(DATE_SUFFIX);
-  if (dateSuffixMatch) {
-    return dateSuffixMatch[1];
-  }
-
-  const shortVersionSuffixMatch = modelId.match(SHORT_VERSION_SUFFIX);
-  if (shortVersionSuffixMatch) {
-    return shortVersionSuffixMatch[1];
-  }
-
-  return modelId;
-}
-
-function getCandidateModelIds(modelId: string): string[] {
-  const candidates = [modelId];
-  const normalizedModelId = normalizeModelId(modelId);
-
-  candidates.push(normalizedModelId);
-
-  if (normalizedModelId !== modelId) {
-    candidates.push(`${normalizedModelId}-latest`);
-  }
-
-  return [...new Set(candidates)];
-}
-
-export function getProviderModelInfo(
-  providerInfo: IProviderInfo | null | undefined,
-  model: string | undefined
-): IProviderModelInfo | undefined {
-  if (!providerInfo || !model) {
-    return undefined;
-  }
-
-  const modelInfo = providerInfo.modelInfo;
-  if (!modelInfo) {
-    return undefined;
-  }
-
-  for (const candidateId of getCandidateModelIds(model)) {
-    if (modelInfo[candidateId]) {
-      return modelInfo[candidateId];
-    }
-  }
-
-  const normalizedModelId = normalizeModelId(model);
-  // As a last resort, match any known model entry that normalizes to the same
-  // base ID, even if the exact alias/version string differs.
-  return Object.entries(modelInfo).find(([candidateId]) => {
-    return normalizeModelId(candidateId) === normalizedModelId;
-  })?.[1];
-}
-
-export function getEffectiveContextWindow(
-  providerConfig: IProviderConfig | undefined,
-  providerRegistry?: IProviderRegistry
-): number | undefined {
-  if (!providerConfig) {
-    return undefined;
-  }
-
-  if (providerConfig.parameters?.contextWindow !== undefined) {
-    return providerConfig.parameters.contextWindow;
-  }
-
-  const providerInfo = providerRegistry?.getProviderInfo(
-    providerConfig.provider
-  );
-  return getProviderModelInfo(providerInfo, providerConfig.model)
-    ?.contextWindow;
-}

package/src/providers/models.ts

@@ -1,76 +0,0 @@
-import type { LanguageModel } from 'ai';
-import type { IProviderRegistry } from '../tokens';
-
-/**
- * Configuration options for creating language models.
- */
-export interface IModelOptions {
-  /**
-   * The provider name (e.g., 'openai', 'anthropic', 'huggingface')
-   */
-  provider: string;
-
-  /**
-   * The specific model name. If not provided, uses provider's default model
-   */
-  model?: string;
-
-  /**
-   * API key for authentication with the provider
-   */
-  apiKey?: string;
-
-  /**
-   * Additional HTTP headers to send with requests
-   */
-  headers?: Record<string, string>;
-
-  /**
-   * Custom base URL for the provider's API endpoint
-   */
-  baseURL?: string;
-}
-
-/**
- * Create a completion model using the provider registry.
- * Built-in providers are automatically registered during extension initialization.
- */
-export function createCompletionModel(
-  options: IModelOptions,
-  registry?: IProviderRegistry
-): LanguageModel {
-  if (!registry) {
-    throw new Error('Provider registry not available');
-  }
-
-  const model = registry.createCompletionModel(options.provider, options);
-  if (!model) {
-    throw new Error(
-      `Provider ${options.provider} not found or failed to create model`
-    );
-  }
-
-  return model;
-}
-
-/**
- * Create a chat model using the provider registry.
- * Built-in providers are automatically registered during extension initialization.
- */
-export function createModel(
-  options: IModelOptions,
-  registry?: IProviderRegistry
-): LanguageModel {
-  if (!registry) {
-    throw new Error('Provider registry not available');
-  }
-
-  const model = registry.createChatModel(options.provider, options);
-  if (!model) {
-    throw new Error(
-      `Provider ${options.provider} not found or failed to create model`
-    );
-  }
-
-  return model;
-}

package/src/providers/provider-registry.ts

@@ -1,88 +0,0 @@
-import { ISignal, Signal } from '@lumino/signaling';
-import type { LanguageModel } from 'ai';
-import type { IModelOptions } from './models';
-import { IProviderInfo, IProviderRegistry } from '../tokens';
-
-/**
- * Implementation of the provider registry
- */
-export class ProviderRegistry implements IProviderRegistry {
-  /**
-   * Get a copy of all registered providers
-   */
-  get providers(): Record<string, IProviderInfo> {
-    return { ...this._providers };
-  }
-
-  /**
-   * Signal emitted when providers are added or removed
-   */
-  get providersChanged(): ISignal<IProviderRegistry, void> {
-    return this._providersChanged;
-  }
-
-  /**
-   * Register a new provider
-   * @param info Provider information with factories for chat and completion
-   */
-  registerProvider(info: IProviderInfo): void {
-    if (info.id in this._providers) {
-      throw new Error(`Provider with id "${info.id}" is already registered`);
-    }
-    this._providers[info.id] = { ...info };
-    this._providersChanged.emit();
-  }
-
-  /**
-   * Get provider information by ID
-   * @param id Provider ID
-   * @returns Provider info or null if not found
-   */
-  getProviderInfo(id: string): IProviderInfo | null {
-    return this._providers[id] || null;
-  }
-
-  /**
-   * Create a chat model instance using the specified provider
-   * @param id Provider ID
-   * @param options Model configuration options
-   * @returns Chat model instance or null if creation fails
-   */
-  createChatModel(id: string, options: IModelOptions): LanguageModel | null {
-    const provider = this._providers[id];
-    if (!provider) {
-      return null;
-    }
-
-    return provider.factory(options);
-  }
-
-  /**
-   * Create a completion model instance using the specified provider
-   * @param id Provider ID
-   * @param options Model configuration options
-   * @returns Language model instance or null if creation fails
-   */
-  createCompletionModel(
-    id: string,
-    options: IModelOptions
-  ): LanguageModel | null {
-    const provider = this._providers[id];
-    if (!provider) {
-      return null;
-    }
-
-    return provider.factory(options);
-  }
-
-  /**
-   * Get list of all available provider IDs
-   * @returns Array of provider IDs
-   */
-  getAvailableProviders(): string[] {
-    return Object.keys(this._providers);
-  }
-
-  private _providers: Record<string, IProviderInfo> = {};
-  private _providersChanged = new Signal<IProviderRegistry, void>(this);
-}