From 85610651c0fa0a87cdd75e8b35e4e304b784cc07 Mon Sep 17 00:00:00 2001
From: Yi <yxiaoisme@gmail.com>
Date: Tue, 7 Jan 2025 16:57:01 +0800
Subject: [PATCH] node ui fix

---
 .../model-parameter-modal/agent-model-trigger.tsx             | 4 ----
 .../model-provider-page/model-selector/model-trigger.tsx      | 2 +-
 web/app/components/plugins/types.ts                           | 1 +
 web/app/components/workflow/nodes/agent/node.tsx              | 2 +-
 web/app/components/workflow/nodes/llm/node.tsx                | 1 +
 .../components/workflow/nodes/parameter-extractor/node.tsx    | 1 +
 .../components/workflow/nodes/question-classifier/node.tsx    | 1 +
 7 files changed, 6 insertions(+), 6 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 c5ef343fe6..6cc21699c4 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
@@ -24,8 +24,6 @@ import { InstallPluginButton } from '@/app/components/workflow/nodes/_base/compo
 import StatusIndicators from './status-indicators'
 import cn from '@/utils/classnames'
 import { useProviderContext } from '@/context/provider-context'
-import { useModalContextSelector } from '@/context/modal-context'
-import { useEventEmitterContextContext } from '@/context/event-emitter'
 import { RiEqualizer2Line } from '@remixicon/react'
 import { fetchPluginInfoFromMarketPlace } from '@/service/plugins'
 
@@ -51,10 +49,8 @@ const AgentModelTrigger: FC<AgentModelTriggerProps> = ({
 }) => {
   const { t } = useTranslation()
   const { modelProviders } = useProviderContext()
-  const setShowModelModal = useModalContextSelector(state => state.setShowModelModal)
   const updateModelProviders = useUpdateModelProviders()
   const updateModelList = useUpdateModelList()
-  const { eventEmitter } = useEventEmitterContextContext()
   const { modelProvider, needsConfiguration } = useMemo(() => {
     const modelProvider = modelProviders.find(item => item.provider === providerName)
     const needsConfiguration = modelProvider?.custom_configuration.status === CustomConfigurationStatusEnum.noConfigure && !(
diff --git a/web/app/components/header/account-setting/model-provider-page/model-selector/model-trigger.tsx b/web/app/components/header/account-setting/model-provider-page/model-selector/model-trigger.tsx
index 353e9890d9..009887acd7 100644
--- a/web/app/components/header/account-setting/model-provider-page/model-selector/model-trigger.tsx
+++ b/web/app/components/header/account-setting/model-provider-page/model-selector/model-trigger.tsx
@@ -34,7 +34,7 @@ const ModelTrigger: FC<ModelTriggerProps> = ({
   return (
     <div
       className={cn(
-        'group flex items-center p-1 h-6 gap-0.5 rounded-lg bg-components-input-bg-normal',
+        'group flex items-center p-1 h-8 gap-0.5 rounded-lg bg-components-input-bg-normal',
         !readonly && 'hover:bg-components-input-bg-hover cursor-pointer',
         open && 'bg-components-input-bg-hover',
         model.status !== ModelStatusEnum.active && 'bg-components-input-bg-disabled hover:bg-components-input-bg-disabled',
diff --git a/web/app/components/plugins/types.ts b/web/app/components/plugins/types.ts
index 92ac9dad28..b58c25f9e3 100644
--- a/web/app/components/plugins/types.ts
+++ b/web/app/components/plugins/types.ts
@@ -111,6 +111,7 @@ export type PluginDetail = {
 export type PluginInfoFromMarketPlace = {
   category: PluginType
   latest_package_identifier: string
+  latest_version: string
 }
 
 export type Plugin = {
diff --git a/web/app/components/workflow/nodes/agent/node.tsx b/web/app/components/workflow/nodes/agent/node.tsx
index 033827741d..f2c93cfc4d 100644
--- a/web/app/components/workflow/nodes/agent/node.tsx
+++ b/web/app/components/workflow/nodes/agent/node.tsx
@@ -111,7 +111,7 @@ const AgentNode: FC<NodeProps<AgentNodeType>> = (props) => {
         return <ModelSelector
           key={model.param}
           modelList={modelList}
-          triggerClassName='bg-workflow-block-parma-bg'
+          triggerClassName='bg-workflow-block-parma-bg !h-6 !rounded-md'
           defaultModel={
             'provider' in model
               ? {
diff --git a/web/app/components/workflow/nodes/llm/node.tsx b/web/app/components/workflow/nodes/llm/node.tsx
index 3d81c172be..ce676ba984 100644
--- a/web/app/components/workflow/nodes/llm/node.tsx
+++ b/web/app/components/workflow/nodes/llm/node.tsx
@@ -25,6 +25,7 @@ const Node: FC<NodeProps<LLMNodeType>> = ({
         <ModelSelector
           defaultModel={{ provider, model: modelId }}
           modelList={textGenerationModelList}
+          triggerClassName='!h-6 !rounded-md'
           readonly
         />
       )}
diff --git a/web/app/components/workflow/nodes/parameter-extractor/node.tsx b/web/app/components/workflow/nodes/parameter-extractor/node.tsx
index 8a1b08e924..d79ae717d9 100644
--- a/web/app/components/workflow/nodes/parameter-extractor/node.tsx
+++ b/web/app/components/workflow/nodes/parameter-extractor/node.tsx
@@ -21,6 +21,7 @@ const Node: FC<NodeProps<ParameterExtractorNodeType>> = ({
         <ModelSelector
           defaultModel={{ provider, model: modelId }}
           modelList={textGenerationModelList}
+          triggerClassName='!h-6 !rounded-md'
           readonly
         />
       )}
diff --git a/web/app/components/workflow/nodes/question-classifier/node.tsx b/web/app/components/workflow/nodes/question-classifier/node.tsx
index 8ca721fbef..8316c3b259 100644
--- a/web/app/components/workflow/nodes/question-classifier/node.tsx
+++ b/web/app/components/workflow/nodes/question-classifier/node.tsx
@@ -32,6 +32,7 @@ const Node: FC<NodeProps<QuestionClassifierNodeType>> = (props) => {
       {hasSetModel && (
         <ModelSelector
           defaultModel={{ provider, model: modelId }}
+          triggerClassName='!h-6 !rounded-md'
           modelList={textGenerationModelList}
           readonly
         />