mincomk/risuai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor getModelName function to import from "src/ts/model/names"

Browse Source
This commit is contained in:
kwaroran
2024-03-19 02:08:26 +09:00
parent ba43ea825b
commit e90d422586
2 changed files with 150 additions and 107 deletions
Download Patch File Download Diff File Expand all files Collapse all files

108
src/lib/UI/ModelList.svelte
View File

@@ -7,118 +7,12 @@
import CheckInput from "./GUI/CheckInput.svelte"; import CheckInput from "./GUI/CheckInput.svelte";
import { isTauri } from "src/ts/storage/globalApi"; import { isTauri } from "src/ts/storage/globalApi";
import {open} from '@tauri-apps/api/dialog' import {open} from '@tauri-apps/api/dialog'
import { getModelName } from "src/ts/model/names";
export let value = "" export let value = ""
export let onChange: (v:string) => void = (v) => {} export let onChange: (v:string) => void = (v) => {}
let openOptions = false let openOptions = false
function getModelName(name:string){
switch(name){
case "gpt35":
return "GPT-3.5 Turbo"
case "gpt35_0613":
return "GPT-3.5 Turbo 0613"
case "gpt35_0301":
return "GPT-3.5 Turbo 0301"
case "gpt35_16k":
return "GPT-3.5 Turbo 16k"
case "gpt35_16k_0613":
return "GPT-3.5 Turbo 16k 0613"
case 'instructgpt35':
return 'GPT-3.5 Turbo Instruct'
case "gpt4":
return "GPT-4"
case "gpt4_0301":
return "GPT-4 0301"
case "gpt4_32k":
return "GPT-4 32k"
case "gpt4_0613":
return "GPT-4 0613"
case "gpt4_32k_0613":
return "GPT-4 32k 0613"
case "gpt4_1106":
return "GPT-4 Turbo 1106"
case 'gpt45':
return 'GPT-4.5'
case "gpt35_1106":
return "GPT-3.5 Turbo 1106"
case 'local_gptq':
return 'Local Model GPTQ'
case "palm2":
return "PaLM2 Bison"
case "textgen_webui":
return "Oobabooga Legacy"
case 'ooba':
return 'Oobabooga'
case "mancer":
return "Mancer"
case "kobold":
return "Kobold"
case "custom":
return "Plugin"
case "novelai":
return "NovelAI Clio"
case "novelai_kayra":
return "NovelAI Kayra"
case "novellist":
return "NovelList SuperTrin"
case "novellist damsel":
return "NovelList Damsel"
case 'reverse_proxy':
return "Reverse Proxy"
case 'openrouter':
return "OpenRouter"
case 'gptvi4_1106':
return "GPT-4 Turbo 1106 Vision"
case 'palm2_unicorn':
return "PaLM2 Unicorn"
case 'mistral-tiny':
return "Mistral Tiny"
case 'mistral-small':
return "Mistral Small"
case 'mistral-medium':
return "Mistral Medium"
case 'gemini-pro':
return "Gemini Pro"
case 'horde:::auto':
return 'Horde Auto Model'
case 'gpt4_0125':
return 'GPT-4 Turbo 0125'
case 'gpt35_0125':
return 'GPT-3.5 Turbo 0125'
case 'gemini-ultra':
return 'Gemini Ultra'
case 'gemini-ultra-vision':
return 'Gemini Ultra Vision'
case 'claude-3-opus-20240229':
return 'Claude 3 Opus (20240229)'
case 'claude-3-sonnet-20240229':
return 'Claude 3 Sonnet (20240229)'
case 'mistral-large-latest':
return 'Mistral Large'
case 'mistral-small-latest':
return 'Mistral Small'
case 'mistral-medium-latest':
return 'Mistral Medium'
case 'claude-3-haiku-20240307':
return 'Claude 3 Haiku (20240307)'
default:
if(name.startsWith("horde:::")){
const split = name.split(":::")
return `Horde ${split[1]}`
}
if(name.startsWith('tf:::')){
const split = name.split(":::")
return `${split[1]}`
}
if(name.startsWith('local_')){
const realName = name.replace('local_', '').split(/(\\|\/)/g).at(-1)
return `GGUF ${realName}`
}
return name
}
}
function changeModel(name:string){ function changeModel(name:string){
value = name value = name
openOptions = false openOptions = false

149
src/ts/model/names.ts Normal file
View File

@@ -0,0 +1,149 @@
export function getModelName(name:string){
switch(name){
case "gpt35":
return "GPT-3.5 Turbo"
case "gpt35_0613":
return "GPT-3.5 Turbo 0613"
case "gpt35_0301":
return "GPT-3.5 Turbo 0301"
case "gpt35_16k":
return "GPT-3.5 Turbo 16k"
case "gpt35_16k_0613":
return "GPT-3.5 Turbo 16k 0613"
case 'instructgpt35':
return 'GPT-3.5 Turbo Instruct'
case "gpt4":
return "GPT-4"
case "gpt4_0301":
return "GPT-4 0301"
case "gpt4_32k":
return "GPT-4 32k"
case "gpt4_0613":
return "GPT-4 0613"
case "gpt4_32k_0613":
return "GPT-4 32k 0613"
case "gpt4_1106":
return "GPT-4 Turbo 1106"
case 'gpt45':
return 'GPT-4.5'
case "gpt35_1106":
return "GPT-3.5 Turbo 1106"
case 'local_gptq':
return 'Local Model GPTQ'
case "palm2":
return "PaLM2 Bison"
case "textgen_webui":
return "Oobabooga Legacy"
case 'ooba':
return 'Oobabooga'
case "mancer":
return "Mancer"
case "kobold":
return "Kobold"
case "custom":
return "Plugin"
case "novelai":
return "NovelAI Clio"
case "novelai_kayra":
return "NovelAI Kayra"
case "novellist":
return "NovelList SuperTrin"
case "novellist damsel":
return "NovelList Damsel"
case 'reverse_proxy':
return "Reverse Proxy"
case 'openrouter':
return "OpenRouter"
case 'gptvi4_1106':
return "GPT-4 Turbo 1106 Vision"
case 'palm2_unicorn':
return "PaLM2 Unicorn"
case 'mistral-tiny':
return "Mistral Tiny"
case 'mistral-small':
return "Mistral Small"
case 'mistral-medium':
return "Mistral Medium"
case 'gemini-pro':
return "Gemini Pro"
case 'horde:::auto':
return 'Horde Auto Model'
case 'gpt4_0125':
return 'GPT-4 Turbo 0125'
case 'gpt35_0125':
return 'GPT-3.5 Turbo 0125'
case 'gemini-ultra':
return 'Gemini Ultra'
case 'gemini-ultra-vision':
return 'Gemini Ultra Vision'
case 'claude-3-opus-20240229':
return 'Claude 3 Opus (20240229)'
case 'claude-3-sonnet-20240229':
return 'Claude 3 Sonnet (20240229)'
case 'mistral-large-latest':
return 'Mistral Large'
case 'mistral-small-latest':
return 'Mistral Small'
case 'mistral-medium-latest':
return 'Mistral Medium'
case 'claude-3-haiku-20240307':
return 'Claude 3 Haiku (20240307)'
default:
if(name.startsWith("horde:::")){
const split = name.split(":::")
return `Horde ${split[1]}`
}
if(name.startsWith('tf:::')){
const split = name.split(":::")
return `${split[1]}`
}
if(name.startsWith('local_')){
const realName = name.replace('local_', '').split(/(\\|\/)/g).at(-1)
return `GGUF ${realName}`
}
return name
}
}
export function getModelShortName(model:string){
if(model.startsWith("gpt35")){
return "GPT-3.5"
}
if(model.startsWith("gpt4")){
return "GPT-4"
}
if(model.startsWith("gptvi4")){
return "GPT-4V"
}
if(model.startsWith("mistral")){
return getModelName(model).split(" ").at(-1)
}
if(model.startsWith("mancer")){
return "Mancer"
}
if(model.startsWith('tf:::')){
const split = model.split(":::")
return split[1]
}
if(model.startsWith('local_')){
const realName = model.replace('local_', '').split(/(\\|\/)/g).at(-1)
return realName
}
if(model.startsWith('horde:::')){
const split = model.split(":::")
return split[1]
}
if(model.startsWith('claude-3')){
const split = model.split("-")
if(!isNaN(parseInt(split[split.length-1]))){
return split[split.length-2]
}
else{
return split[split.length-1]
}
}
return getModelName(model)
}