From 77ed7e17d5ea97f7b61f4bb4175c8a61c265ea5d Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Thu, 11 Jul 2024 18:02:50 +0800
Subject: [PATCH] feat: show modal info

---
 .../settings/components/ModelListPopup.tsx    | 33 ++++++++++++++++---
 src/renderer/src/types/index.ts               |  1 +
 src/renderer/src/utils/index.ts               |  5 +++
 3 files changed, 34 insertions(+), 5 deletions(-)

diff --git a/src/renderer/src/pages/settings/components/ModelListPopup.tsx b/src/renderer/src/pages/settings/components/ModelListPopup.tsx
index d2d2118c..b2823a3a 100644
--- a/src/renderer/src/pages/settings/components/ModelListPopup.tsx
+++ b/src/renderer/src/pages/settings/components/ModelListPopup.tsx
@@ -1,11 +1,11 @@
-import { LoadingOutlined, MinusOutlined, PlusOutlined } from '@ant-design/icons'
+import { LoadingOutlined, MinusOutlined, PlusOutlined, QuestionCircleOutlined } from '@ant-design/icons'
 import { SYSTEM_MODELS } from '@renderer/config/models'
 import { useProvider } from '@renderer/hooks/useProvider'
 import { fetchModels } from '@renderer/services/api'
 import { getModelLogo } from '@renderer/services/provider'
 import { Model, Provider } from '@renderer/types'
-import { getDefaultGroupName, runAsyncFunction } from '@renderer/utils'
-import { Avatar, Button, Empty, Flex, Modal } from 'antd'
+import { getDefaultGroupName, isFreeModel, runAsyncFunction } from '@renderer/utils'
+import { Avatar, Button, Empty, Flex, Modal, Tag } from 'antd'
 import Search from 'antd/es/input/Search'
 import { groupBy, isEmpty, uniqBy } from 'lodash'
 import { useEffect, useState } from 'react'
@@ -67,7 +67,9 @@ const PopupContainer: React.FC<Props> = ({ provider: _provider, resolve }) => {
             // @ts-ignore name
             name: model.name || model.id,
             provider: _provider.id,
-            group: getDefaultGroupName(model.id)
+            group: getDefaultGroupName(model.id),
+            // @ts-ignore name
+            description: model?.description
           }))
         )
         setLoading(false)
@@ -115,7 +117,15 @@ const PopupContainer: React.FC<Props> = ({ provider: _provider, resolve }) => {
                     <Avatar src={getModelLogo(model.id)} size={24}>
                       {model.name[0].toUpperCase()}
                     </Avatar>
-                    <ListItemName>{model.name}</ListItemName>
+                    <ListItemName>
+                      {model.name}
+                      {isFreeModel(model) && (
+                        <Tag style={{ marginLeft: 10 }} color="green">
+                          Free
+                        </Tag>
+                      )}
+                      {!isEmpty(model.description) && <Question onClick={() => onShowModelInfo(model)} />}
+                    </ListItemName>
                   </ListItemHeader>
                   {hasModel ? (
                     <Button type="default" onClick={() => onRemoveModel(model)} icon={<MinusOutlined />} />
@@ -133,6 +143,13 @@ const PopupContainer: React.FC<Props> = ({ provider: _provider, resolve }) => {
   )
 }
 
+const onShowModelInfo = (model: Model) => {
+  window.modal.info({
+    title: model.name,
+    content: model.description
+  })
+}
+
 const SearchContainer = styled.div`
   display: flex;
   flex-direction: row;
@@ -189,6 +206,12 @@ const ModelHeaderTitle = styled.div`
   margin-right: 10px;
 `
 
+const Question = styled(QuestionCircleOutlined)`
+  cursor: pointer;
+  margin: 0 10px;
+  color: #888;
+`
+
 export default class ModelListPopup {
   static topviewId = 0
   static hide() {
diff --git a/src/renderer/src/types/index.ts b/src/renderer/src/types/index.ts
index 569dc473..512bcdb2 100644
--- a/src/renderer/src/types/index.ts
+++ b/src/renderer/src/types/index.ts
@@ -46,6 +46,7 @@ export type Model = {
   provider: string
   name: string
   group: string
+  description?: string
 }
 
 export type SystemAssistant = {
diff --git a/src/renderer/src/utils/index.ts b/src/renderer/src/utils/index.ts
index c8be6590..53c8c7de 100644
--- a/src/renderer/src/utils/index.ts
+++ b/src/renderer/src/utils/index.ts
@@ -1,5 +1,6 @@
 import { v4 as uuidv4 } from 'uuid'
 import imageCompression from 'browser-image-compression'
+import { Model } from '@renderer/types'
 
 export const runAsyncFunction = async (fn: () => void) => {
   await fn()
@@ -93,3 +94,7 @@ export function droppableReorder<T>(list: T[], startIndex: number, endIndex: num
 export const firstLetter = (str?: string) => {
   return str ? str[0] : ''
 }
+
+export function isFreeModel(model: Model) {
+  return (model.id + model.name).toLocaleLowerCase().includes('free')
+}