feat: update text-to-image model regex to include Janus

This commit is contained in:
kangfenmao 2025-02-04 12:28:17 +08:00
parent 507efda688
commit 2208ab7277

View File

@ -144,8 +144,7 @@ const visionAllowedModels = [
'pixtral',
'gpt-4(?:-[\\w-]+)',
'gpt-4o(?:-[\\w-]+)?',
'chatgpt-4o(?:-[\\w-]+)?',
'janus'
'chatgpt-4o(?:-[\\w-]+)?'
]
const visionExcludedModels = ['gpt-4-\\d+-preview', 'gpt-4-turbo-preview', 'gpt-4-32k', 'gpt-4-\\d+']
@ -155,7 +154,7 @@ export const VISION_REGEX = new RegExp(
'i'
)
export const TEXT_TO_IMAGE_REGEX = /flux|diffusion|stabilityai|sd-|dall|cogview/i
export const TEXT_TO_IMAGE_REGEX = /flux|diffusion|stabilityai|sd-|dall|cogview|janus/i
export const EMBEDDING_REGEX = /(?:^text-|embed|rerank|davinci|babbage|bge-|e5-|LLM2Vec|retrieval|uae-|gte-|jina)/i
export const NOT_SUPPORTED_REGEX = /(?:^tts|rerank|whisper|speech)/i