webui build

frontend/src/types/about.ts

@@ -0,0 +1 @@
+export type AboutContent = { [lang: string]: string }

frontend/src/types/chat.ts

@@ -0,0 +1,29 @@
+export const userName = 'M E';
+export const botName = 'A I';
+export const welcomeUuid = 'welcome';
+
+export enum MessageType {
+  Normal,
+  Error
+}
+
+export type Side = 'left' | 'right'
+export type Color = 'neutral' | 'brand' | 'colorful'
+export type MessageItem = {
+  sender: string,
+  type: MessageType,
+  color: Color,
+  avatarImg?: string,
+  time: string,
+  content: string,
+  side: Side,
+  done: boolean
+}
+export type Conversation = {
+  [uuid: string]: MessageItem
+}
+export type Role = 'assistant' | 'user' | 'system';
+export type ConversationMessage = {
+  role: Role;
+  content: string;
+}

frontend/src/types/completion.ts

@@ -0,0 +1,12 @@
+import { ApiParameters } from './configs';
+
+export type CompletionParams = Omit<ApiParameters, 'apiPort'> & {
+  stop: string,
+  injectStart: string,
+  injectEnd: string
+};
+export type CompletionPreset = {
+  name: string,
+  prompt: string,
+  params: CompletionParams
+}

frontend/src/types/composition.ts

@@ -0,0 +1,12 @@
+import { NoteSequence } from '@magenta/music/esm/protobuf';
+
+export type CompositionParams = {
+  prompt: string,
+  maxResponseToken: number,
+  temperature: number,
+  topP: number,
+  autoPlay: boolean,
+  useLocalSoundFont: boolean,
+  midi: ArrayBuffer | null,
+  ns: NoteSequence | null
+}

frontend/src/types/configs.ts

@@ -0,0 +1,28 @@
+export type ApiParameters = {
+  apiPort: number
+  maxResponseToken: number;
+  temperature: number;
+  topP: number;
+  presencePenalty: number;
+  frequencyPenalty: number;
+}
+export type Device = 'CPU' | 'CUDA' | 'CUDA-Beta' | 'WebGPU' | 'MPS' | 'Custom';
+export type Precision = 'fp16' | 'int8' | 'fp32';
+export type ModelParameters = {
+  // different models can not have the same name
+  modelName: string;
+  device: Device;
+  precision: Precision;
+  storedLayers: number;
+  maxStoredLayers: number;
+  useCustomCuda?: boolean;
+  customStrategy?: string;
+  useCustomTokenizer?: boolean;
+  customTokenizer?: string;
+}
+export type ModelConfig = {
+  // different configs can have the same name
+  name: string;
+  apiParameters: ApiParameters
+  modelParameters: ModelParameters
+}

frontend/src/types/downloads.ts

@@ -0,0 +1,11 @@
+export type DownloadStatus = {
+  name: string;
+  path: string;
+  url: string;
+  transferred: number;
+  size: number;
+  speed: number;
+  progress: number;
+  downloading: boolean;
+  done: boolean;
+}

frontend/src/types/home.ts

@@ -0,0 +1,11 @@
+import { ReactElement } from 'react';
+
+export type IntroductionContent = {
+  [lang: string]: string
+}
+export type NavCard = {
+  label: string;
+  desc: string;
+  path: string;
+  icon: ReactElement;
+};

frontend/src/types/models.ts

@@ -0,0 +1,14 @@
+export type ModelSourceItem = {
+  name: string;
+  size: number;
+  lastUpdated: string;
+  desc?: { [lang: string]: string | undefined; };
+  SHA256?: string;
+  url?: string;
+  downloadUrl?: string;
+  isComplete?: boolean;
+  isLocal?: boolean;
+  localSize?: number;
+  lastUpdatedMs?: number;
+  hide?: boolean;
+};

frontend/src/types/presets.ts

@@ -0,0 +1,30 @@
+import { ReactElement } from 'react';
+
+import { ConversationMessage } from './chat';
+
+export type PresetType = 'chat' | 'completion' | 'chatInCompletion'
+export type Preset = {
+  name: string,
+  tag: string,
+  // if name and sourceUrl are same, it will be overridden when importing
+  sourceUrl: string,
+  desc: string,
+  avatarImg: string,
+  type: PresetType,
+  // chat
+  welcomeMessage: string,
+  messages: ConversationMessage[],
+  displayPresetMessages: boolean,
+  // completion
+  prompt: string,
+  stop: string,
+  injectStart: string,
+  injectEnd: string,
+  presystem?: boolean,
+  userName?: string,
+  assistantName?: string
+}
+export type PresetsNavigationItem = {
+  icon: ReactElement;
+  element: ReactElement;
+};

frontend/src/types/settings.ts

@@ -0,0 +1,21 @@
+export const Languages = {
+  dev: 'English', // i18n default
+  zh: '简体中文',
+  ja: '日本語'
+};
+export type Language = keyof typeof Languages;
+export type SettingsType = {
+  language: Language
+  darkMode: boolean
+  autoUpdatesCheck: boolean
+  giteeUpdatesSource: boolean
+  cnMirror: boolean
+  host: string
+  dpiScaling: number
+  customModelsPath: string
+  customPythonPath: string
+  apiUrl: string
+  apiKey: string
+  apiChatModelName: string
+  apiCompletionModelName: string
+}

frontend/src/types/train.ts

@@ -0,0 +1,35 @@
+import { ReactElement } from 'react';
+
+export type DataProcessParameters = {
+  dataPath: string;
+  vocabPath: string;
+}
+export type LoraFinetunePrecision = 'bf16' | 'fp16' | 'tf32';
+export type LoraFinetuneParameters = {
+  baseModel: string;
+  ctxLen: number;
+  epochSteps: number;
+  epochCount: number;
+  epochBegin: number;
+  epochSave: number;
+  microBsz: number;
+  accumGradBatches: number;
+  preFfn: boolean;
+  headQk: boolean;
+  lrInit: string;
+  lrFinal: string;
+  warmupSteps: number;
+  beta1: number;
+  beta2: number;
+  adamEps: string;
+  devices: number;
+  precision: LoraFinetunePrecision;
+  gradCp: boolean;
+  loraR: number;
+  loraAlpha: number;
+  loraDropout: number;
+  loraLoad: string
+}
+export type TrainNavigationItem = {
+  element: ReactElement;
+};