@jupyterlite/ai 0.3.0 → 0.5.0

package/src/llm-models/utils.ts

@@ -1,90 +0,0 @@
-import { ChatAnthropic } from '@langchain/anthropic';
-import { ChromeAI } from '@langchain/community/experimental/llms/chrome_ai';
-import { BaseChatModel } from '@langchain/core/language_models/chat_models';
-import { ChatMistralAI } from '@langchain/mistralai';
-import { ChatOpenAI } from '@langchain/openai';
-
-import { IBaseCompleter } from './base-completer';
-import { AnthropicCompleter } from './anthropic-completer';
-import { CodestralCompleter } from './codestral-completer';
-import { ReadonlyPartialJSONObject } from '@lumino/coreutils';
-import { ChromeCompleter } from './chrome-completer';
-import { OpenAICompleter } from './openai-completer';
-
-import chromeAI from '../_provider-settings/chromeAI.json';
-import mistralAI from '../_provider-settings/mistralAI.json';
-import anthropic from '../_provider-settings/anthropic.json';
-import openAI from '../_provider-settings/openAI.json';
-
-/**
- * Get an LLM completer from the name.
- */
-export function getCompleter(
-  name: string,
-  settings: ReadonlyPartialJSONObject
-): IBaseCompleter | null {
-  if (name === 'MistralAI') {
-    return new CodestralCompleter({ settings });
-  } else if (name === 'Anthropic') {
-    return new AnthropicCompleter({ settings });
-  } else if (name === 'ChromeAI') {
-    return new ChromeCompleter({ settings });
-  } else if (name === 'OpenAI') {
-    return new OpenAICompleter({ settings });
-  }
-  return null;
-}
-
-/**
- * Get an LLM chat model from the name.
- */
-export function getChatModel(
-  name: string,
-  settings: ReadonlyPartialJSONObject
-): BaseChatModel | null {
-  if (name === 'MistralAI') {
-    return new ChatMistralAI({ ...settings });
-  } else if (name === 'Anthropic') {
-    return new ChatAnthropic({ ...settings });
-  } else if (name === 'ChromeAI') {
-    // TODO: fix
-    // @ts-expect-error: missing properties
-    return new ChromeAI({ ...settings });
-  } else if (name === 'OpenAI') {
-    return new ChatOpenAI({ ...settings });
-  }
-  return null;
-}
-
-/**
- * Get the error message from provider.
- */
-export function getErrorMessage(name: string, error: any): string {
-  if (name === 'MistralAI') {
-    return error.message;
-  } else if (name === 'Anthropic') {
-    return error.error.error.message;
-  } else if (name === 'ChromeAI') {
-    return error.message;
-  } else if (name === 'OpenAI') {
-    return error.message;
-  }
-  return 'Unknown provider';
-}
-
-/*
- * Get an LLM completer from the name.
- */
-export function getSettings(name: string): any {
-  if (name === 'MistralAI') {
-    return mistralAI.properties;
-  } else if (name === 'Anthropic') {
-    return anthropic.properties;
-  } else if (name === 'ChromeAI') {
-    return chromeAI.properties;
-  } else if (name === 'OpenAI') {
-    return openAI.properties;
-  }
-
-  return null;
-}

package/src/slash-commands.tsx

@@ -1,55 +0,0 @@
-/**
- * TODO: reuse from Jupyter AI instead of copying?
- * https://github.com/jupyterlab/jupyter-ai/blob/main/packages/jupyter-ai/src/slash-autocompletion.tsx
- */
-
-import { Box, Typography } from '@mui/material';
-import { AutocompleteCommand } from '@jupyter/chat';
-
-import HideSource from '@mui/icons-material/HideSource';
-
-import React from 'react';
-
-const DEFAULT_SLASH_COMMAND_ICONS: Record<string, JSX.Element> = {
-  clear: <HideSource />
-};
-
-type SlashCommandOption = AutocompleteCommand & {
-  id: string;
-  description: string;
-};
-
-/**
- * Renders an option shown in the slash command autocomplete.
- */
-export function renderSlashCommandOption(
-  optionProps: React.HTMLAttributes<HTMLLIElement>,
-  option: SlashCommandOption
-): JSX.Element {
-  const icon =
-    option.id in DEFAULT_SLASH_COMMAND_ICONS
-      ? DEFAULT_SLASH_COMMAND_ICONS[option.id]
-      : DEFAULT_SLASH_COMMAND_ICONS.unknown;
-
-  return (
-    <li {...optionProps}>
-      <Box sx={{ lineHeight: 0, marginRight: 4, opacity: 0.618 }}>{icon}</Box>
-      <Box sx={{ flexGrow: 1 }}>
-        <Typography
-          component="span"
-          sx={{
-            fontSize: 'var(--jp-ui-font-size1)'
-          }}
-        >
-          {option.label}
-        </Typography>
-        <Typography
-          component="span"
-          sx={{ opacity: 0.618, fontSize: 'var(--jp-ui-font-size0)' }}
-        >
-          {' — ' + option.description}
-        </Typography>
-      </Box>
-    </li>
-  );
-}

package/src/token.ts

@@ -1,19 +0,0 @@
-import { BaseChatModel } from '@langchain/core/language_models/chat_models';
-import { Token } from '@lumino/coreutils';
-import { ISignal } from '@lumino/signaling';
-
-import { IBaseCompleter } from './llm-models';
-
-export interface IAIProvider {
-  name: string;
-  completer: IBaseCompleter | null;
-  chatModel: BaseChatModel | null;
-  modelChange: ISignal<IAIProvider, void>;
-  chatError: string;
-  completerError: string;
-}
-
-export const IAIProvider = new Token<IAIProvider>(
-  '@jupyterlite/ai:AIProvider',
-  'Provider for chat and completion LLM provider'
-);