diff --git a/web-app/public/images/model-provider/ornith.webp b/web-app/public/images/model-provider/ornith.webp new file mode 100644 index 000000000..678646e83 Binary files /dev/null and b/web-app/public/images/model-provider/ornith.webp differ diff --git a/web-app/src/lib/__tests__/model-logo.test.ts b/web-app/src/lib/__tests__/model-logo.test.ts index 6ae95352a..8484225af 100644 --- a/web-app/src/lib/__tests__/model-logo.test.ts +++ b/web-app/src/lib/__tests__/model-logo.test.ts @@ -42,6 +42,7 @@ describe('iconKeyLogoSrc', () => { 'bytedance', 'inclusionai', 'nanbeige', + 'ornith', ] for (const key of keys) { expect(iconKeyLogoSrc(key), `missing logo for "${key}"`).toBeTruthy() @@ -99,6 +100,9 @@ describe('modelFamilyLogoSrc', () => { expect(modelFamilyLogoSrc('unsloth/MiniMax-M2.7-GGUF')).toBe( '/svg/minimax.svg' ) + expect(modelFamilyLogoSrc('AtomicChat/Ornith-1.5-35B-A3B-GGUF')).toBe( + '/images/model-provider/ornith.webp' + ) }) it('keeps Bonsai on its own mark rather than the Qwen base it was built from', () => { diff --git a/web-app/src/lib/model-logo.ts b/web-app/src/lib/model-logo.ts index fc7458802..2b2cad6a8 100644 --- a/web-app/src/lib/model-logo.ts +++ b/web-app/src/lib/model-logo.ts @@ -9,6 +9,7 @@ const FAMILY_LOGO_RULES: Array<[RegExp, string]> = [ [/gemma/i, '/svg/google-color.svg'], [/\bglm\b|chatglm/i, '/svg/zai.svg'], [/bonsai/i, '/images/model-provider/prism-ml.webp'], + [/ornith/i, '/images/model-provider/ornith.webp'], [/qwen|qwq/i, '/svg/qwen-color.svg'], [/(?> = { prism: '/images/model-provider/prism-ml.webp', inclusionai: '/images/model-provider/inclusionai.webp', nanbeige: '/images/model-provider/nanbeige.webp', + ornith: '/images/model-provider/ornith.webp', huggingface: '/images/model-provider/huggingface.svg', }