From 3a09f43f707e6f40446fac75367e91c4c77885ae Mon Sep 17 00:00:00 2001
From: Yi <yxiaoisme@gmail.com>
Date: Tue, 31 Dec 2024 11:46:36 +0800
Subject: [PATCH] feat: update the install logic

---
 .../agent-model-trigger.tsx                    | 18 +++++++++++++++---
 .../model-selector/index.tsx                   |  1 +
 2 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/web/app/components/header/account-setting/model-provider-page/model-parameter-modal/agent-model-trigger.tsx b/web/app/components/header/account-setting/model-provider-page/model-parameter-modal/agent-model-trigger.tsx
index e1a8873cde..adffecb82a 100644
--- a/web/app/components/header/account-setting/model-provider-page/model-parameter-modal/agent-model-trigger.tsx
+++ b/web/app/components/header/account-setting/model-provider-page/model-parameter-modal/agent-model-trigger.tsx
@@ -39,6 +39,7 @@ export type AgentModelTriggerProps = {
   providerName?: string
   modelId?: string
   hasDeprecated?: boolean
+  scope?: string
 }
 
 const AgentModelTrigger: FC<AgentModelTriggerProps> = ({
@@ -48,6 +49,7 @@ const AgentModelTrigger: FC<AgentModelTriggerProps> = ({
   providerName,
   modelId,
   hasDeprecated,
+  scope,
 }) => {
   const { t } = useTranslation()
   const { modelProviders } = useProviderContext()
@@ -136,9 +138,19 @@ const AgentModelTrigger: FC<AgentModelTriggerProps> = ({
     try {
       const { all_installed } = await installPackageFromMarketPlace(pluginInfo.latest_package_identifier)
       if (all_installed) {
-        setInstalled(true)
+        [
+          ModelTypeEnum.textGeneration,
+          ModelTypeEnum.textEmbedding,
+          ModelTypeEnum.rerank,
+          ModelTypeEnum.moderation,
+          ModelTypeEnum.speech2text,
+          ModelTypeEnum.tts,
+        ].forEach((type: ModelTypeEnum) => {
+          if (scope?.includes(type))
+            updateModelList(type)
+        })
         updateModelProviders()
-        updateModelList(ModelTypeEnum.textGeneration)
+        setInstalled(true)
       }
     }
     catch (error) {
@@ -190,7 +202,7 @@ const AgentModelTrigger: FC<AgentModelTriggerProps> = ({
               t={t}
             />
           )}
-          {modelProvider && !disabled && (
+          {modelProvider && !disabled && !needsConfiguration && (
             <div className="flex pr-1 items-center">
               <RiEqualizer2Line className="w-4 h-4 text-text-tertiary group-hover:text-text-secondary" />
             </div>
diff --git a/web/app/components/plugins/plugin-detail-panel/model-selector/index.tsx b/web/app/components/plugins/plugin-detail-panel/model-selector/index.tsx
index ce8b70776a..00b0cd74eb 100644
--- a/web/app/components/plugins/plugin-detail-panel/model-selector/index.tsx
+++ b/web/app/components/plugins/plugin-detail-panel/model-selector/index.tsx
@@ -190,6 +190,7 @@ const ModelParameterModal: FC<ModelParameterModalProps> = ({
                   currentModel={currentModel}
                   providerName={value?.provider}
                   modelId={value?.model}
+                  scope={scope}
                 />
                 : <Trigger
                   disabled={disabled}