💄 style: add gemma model logo for ollama (lobehub#1369)

573dave · Feb 23, 2024 · e2fb3a3 · e2fb3a3
1 parent 4bbb773
commit e2fb3a3
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 1 deletion.
diff --git a/package.json b/package.json
@@ -82,7 +82,7 @@
     "@icons-pack/react-simple-icons": "^9",
     "@lobehub/chat-plugin-sdk": "latest",
     "@lobehub/chat-plugins-gateway": "latest",
-    "@lobehub/icons": "^1.11.1",
+    "@lobehub/icons": "^1.11.2",
     "@lobehub/tts": "latest",
     "@lobehub/ui": "^1.129.2",
     "@vercel/analytics": "^1",

diff --git a/src/components/ModelIcon/index.tsx b/src/components/ModelIcon/index.tsx
@@ -4,6 +4,7 @@ import {
   Baichuan,
   ChatGLM,
   Gemini,
+  Gemma,
   Meta,
   Minimax,
   Mistral,
@@ -29,6 +30,7 @@ const ModelIcon = memo<ModelProviderIconProps>(({ model, size = 12 }) => {
   if (model.includes('titan')) return <Aws.Avatar size={size} />;
   if (model.includes('llama')) return <Meta.Avatar size={size} />;
   if (model.includes('gemini')) return <Gemini.Avatar size={size} />;
+  if (model.includes('gemma')) return <Gemma.Avatar size={size} />;
   if (model.includes('qwen')) return <Tongyi.Avatar background={Tongyi.colorPrimary} size={size} />;
   if (model.includes('minmax')) return <Minimax.Avatar size={size} />;
   if (model.includes('moonshot')) return <Moonshot.Avatar size={size} />;

diff --git a/src/components/ModelTag/ModelIcon.tsx b/src/components/ModelTag/ModelIcon.tsx
@@ -4,6 +4,7 @@ import {
   Baichuan,
   ChatGLM,
   Gemini,
+  Gemma,
   Meta,
   Minimax,
   Mistral,
@@ -28,6 +29,7 @@ const ModelIcon = memo<ModelIconProps>(({ model, size = 12 }) => {
   if (model.includes('titan')) return <Aws size={size} />;
   if (model.includes('llama')) return <Meta size={size} />;
   if (model.includes('gemini')) return <Gemini size={size} />;
+  if (model.includes('gemma')) return <Gemma.Simple size={size} />;
   if (model.includes('moonshot')) return <Moonshot size={size} />;
   if (model.includes('qwen')) return <Tongyi size={size} />;
   if (model.includes('minmax')) return <Minimax size={size} />;

diff --git a/src/config/modelProviders/ollama.ts b/src/config/modelProviders/ollama.ts
@@ -2,6 +2,28 @@ import { ModelProviderCard } from '@/types/llm';
 
 const Ollama: ModelProviderCard = {
   chatModels: [
+    {
+      displayName: 'Gemma 7B',
+      functionCall: false,
+      id: 'gemma:7b',
+      tokens: 4000,
+      vision: false,
+    },
+    {
+      displayName: 'Gemma 2B',
+      functionCall: false,
+      id: 'gemma',
+      tokens: 4000,
+      vision: false,
+    },
+    {
+      displayName: 'Llama2 Chat 13B',
+      functionCall: false,
+      hidden: true,
+      id: 'llama2:13b',
+      tokens: 4000,
+      vision: false,
+    },
     {
       displayName: 'Llama2 Chat 7B',
       functionCall: false,