更新模型

This commit is contained in:
IndieKKY
2024-07-23 10:00:51 +08:00
parent 9320928b34
commit 02b7a09f42
2 changed files with 18 additions and 13 deletions

View File

@@ -11,6 +11,7 @@ import {
LANGUAGES, LANGUAGES,
MODEL_DEFAULT, MODEL_DEFAULT,
MODEL_MAP, MODEL_MAP,
MODEL_TIP,
MODELS, MODELS,
PAGE_MAIN, PAGE_MAIN,
PROMPT_DEFAULTS, PROMPT_DEFAULTS,
@@ -257,13 +258,18 @@ const Settings = () => {
{MODELS.map(model => <option key={model.code} value={model.code}>{model.name}</option>)} {MODELS.map(model => <option key={model.code} value={model.code}>{model.name}</option>)}
</select> </select>
</FormItem> </FormItem>
<div className='desc text-xs'>
{MODEL_TIP}
</div>
{modelValue === 'custom' && <FormItem title='模型名' htmlFor='customModel'> {modelValue === 'custom' && <FormItem title='模型名' htmlFor='customModel'>
<input id='customModel' type='text' className='input input-sm input-bordered w-full' placeholder='llama2' <input id='customModel' type='text' className='input input-sm input-bordered w-full' placeholder='llama2'
value={customModelValue} onChange={onChangeCustomModelValue}/> value={customModelValue} onChange={onChangeCustomModelValue}/>
</FormItem>} </FormItem>}
{modelValue === 'custom' && <FormItem title='Token上限' htmlFor='customModelTokens'> {modelValue === 'custom' && <FormItem title='Token上限' htmlFor='customModelTokens'>
<input id='customModelTokens' type='number' className='input input-sm input-bordered w-full' placeholder={''+CUSTOM_MODEL_TOKENS} <input id='customModelTokens' type='number' className='input input-sm input-bordered w-full'
value={customModelTokensValue} onChange={e => setCustomModelTokensValue(e.target.value?parseInt(e.target.value):undefined)}/> placeholder={'' + CUSTOM_MODEL_TOKENS}
value={customModelTokensValue}
onChange={e => setCustomModelTokensValue(e.target.value ? parseInt(e.target.value) : undefined)}/>
</FormItem>} </FormItem>}
</Section>} </Section>}
@@ -274,10 +280,12 @@ const Settings = () => {
</FormItem> </FormItem>
<div> <div>
<div className='desc text-xs'> <div className='desc text-xs'>
<div><a className='link link-primary' href='https://makersuite.google.com/app/apikey' target='_blank' <div><a className='link link-primary' href='https://makersuite.google.com/app/apikey'
target='_blank'
rel="noreferrer">Google AI Studio</a> () rel="noreferrer">Google AI Studio</a> ()
</div> </div>
<div className='text-xs text-error flex items-center'><IoWarning className='text-sm text-warning'/>!</div> <div className='text-xs text-error flex items-center'><IoWarning className='text-sm text-warning'/>!
</div>
</div> </div>
</div> </div>
</Section>} </Section>}

View File

@@ -222,24 +222,21 @@ export const ASK_ENABLED_DEFAULT = true
export const DEFAULT_SERVER_URL_OPENAI = 'https://api.openai.com' export const DEFAULT_SERVER_URL_OPENAI = 'https://api.openai.com'
export const CUSTOM_MODEL_TOKENS = 16385 export const CUSTOM_MODEL_TOKENS = 16385
export const MODEL_TIP = '推荐gpt-4o-mini能力强价格低token上限大'
export const MODELS = [{ export const MODELS = [{
code: 'gpt-3.5-turbo', code: 'gpt-4o-mini',
name: 'gpt-3.5-turbo', name: 'gpt-4o-mini',
tokens: 4096, tokens: 128000,
}, { }, {
code: 'gpt-3.5-turbo-0125', code: 'gpt-3.5-turbo-0125',
name: 'gpt-3.5-turbo-0125', name: 'gpt-3.5-turbo-0125',
tokens: 16385, tokens: 16385,
}, {
code: 'gpt-3.5-turbo-1106',
name: 'gpt-3.5-turbo-1106',
tokens: 16385,
}, { }, {
code: 'custom', code: 'custom',
name: '自定义', name: '自定义',
}] }]
export const GEMINI_TOKENS = 32768 export const GEMINI_TOKENS = 32768
export const MODEL_DEFAULT = MODELS[1].code export const MODEL_DEFAULT = MODELS[0].code
export const MODEL_MAP: {[key: string]: typeof MODELS[number]} = {} export const MODEL_MAP: {[key: string]: typeof MODELS[number]} = {}
for (const model of MODELS) { for (const model of MODELS) {
MODEL_MAP[model.code] = model MODEL_MAP[model.code] = model