From 11cd5f477f2b076cca6da90732273f97eca9bef3 Mon Sep 17 00:00:00 2001
From: arvinxx <arvinx@foxmail.com>
Date: Sun, 17 Mar 2024 12:09:37 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=92=84=20style:=20improve=20model=20tags?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 package.json                               | 2 +-
 src/components/ModelProviderIcon/index.tsx | 5 +++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/package.json b/package.json
index ea7dfa8c9d4e9..dbfc9c253719e 100644
--- a/package.json
+++ b/package.json
@@ -90,7 +90,7 @@
     "@lobehub/chat-plugins-gateway": "latest",
     "@lobehub/icons": "latest",
     "@lobehub/tts": "latest",
-    "@lobehub/ui": "latest",
+    "@lobehub/ui": "^1.133.3",
     "@vercel/analytics": "^1",
     "ahooks": "^3",
     "ai": "^3.0.0",
diff --git a/src/components/ModelProviderIcon/index.tsx b/src/components/ModelProviderIcon/index.tsx
index 741612b948797..6bd63c18cf79b 100644
--- a/src/components/ModelProviderIcon/index.tsx
+++ b/src/components/ModelProviderIcon/index.tsx
@@ -8,6 +8,7 @@ import {
   Moonshot,
   Ollama,
   OpenAI,
+  OpenRouter,
   Perplexity,
   Zhipu,
 } from '@lobehub/icons';
@@ -74,6 +75,10 @@ const ModelProviderIcon = memo<ModelProviderIconProps>(({ provider }) => {
       return <Groq size={20} />;
     }
 
+    case ModelProvider.OpenRouter: {
+      return <OpenRouter size={20} />;
+    }
+
     default: {
       return null;
     }