From 3bcd470ec629f713ceb7529961ee32b83b8f3bfe Mon Sep 17 00:00:00 2001
From: JzoNg <jzongcode@gmail.com>
Date: Thu, 28 Nov 2024 17:22:36 +0800
Subject: [PATCH] fix: authorization in debugging plugin

---
 .../plugins/plugin-detail-panel/action-list.tsx   | 15 +++++++++------
 web/service/use-tools.ts                          |  2 +-
 2 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/web/app/components/plugins/plugin-detail-panel/action-list.tsx b/web/app/components/plugins/plugin-detail-panel/action-list.tsx
index 334587ce31..02e59a005c 100644
--- a/web/app/components/plugins/plugin-detail-panel/action-list.tsx
+++ b/web/app/components/plugins/plugin-detail-panel/action-list.tsx
@@ -1,4 +1,4 @@
-import React, { useState } from 'react'
+import React, { useMemo, useState } from 'react'
 import { useTranslation } from 'react-i18next'
 import { useAppContext } from '@/context/app-context'
 import Button from '@/app/components/base/button'
@@ -7,9 +7,9 @@ import Indicator from '@/app/components/header/indicator'
 import ToolItem from '@/app/components/tools/provider/tool-item'
 import ConfigCredential from '@/app/components/tools/setting/build-in/config-credentials'
 import {
-  useBuiltinProviderInfo,
+  useAllToolProviders,
   useBuiltinTools,
-  useInvalidateBuiltinProviderInfo,
+  useInvalidateAllToolProviders,
   useRemoveProviderCredentials,
   useUpdateProviderCredentials,
 } from '@/service/use-tools'
@@ -26,14 +26,17 @@ const ActionList = ({
   const { isCurrentWorkspaceManager } = useAppContext()
   const providerBriefInfo = detail.declaration.tool.identity
   const providerKey = `${detail.plugin_id}/${providerBriefInfo.name}`
-  const { data: provider } = useBuiltinProviderInfo(providerKey)
-  const invalidateProviderInfo = useInvalidateBuiltinProviderInfo()
+  const { data: collectionList = [] } = useAllToolProviders()
+  const invalidateAllToolProviders = useInvalidateAllToolProviders()
+  const provider = useMemo(() => {
+    return collectionList.find(collection => collection.name === providerKey)
+  }, [collectionList, providerKey])
   const { data } = useBuiltinTools(providerKey)
 
   const [showSettingAuth, setShowSettingAuth] = useState(false)
 
   const handleCredentialSettingUpdate = () => {
-    invalidateProviderInfo(providerKey)
+    invalidateAllToolProviders()
     Toast.notify({
       type: 'success',
       message: t('common.api.actionSuccess'),
diff --git a/web/service/use-tools.ts b/web/service/use-tools.ts
index bf9678332f..ceaa4b14b3 100644
--- a/web/service/use-tools.ts
+++ b/web/service/use-tools.ts
@@ -15,7 +15,7 @@ const NAME_SPACE = 'tools'
 
 const useAllToolProvidersKey = [NAME_SPACE, 'allToolProviders']
 export const useAllToolProviders = () => {
-  return useQuery({
+  return useQuery<Collection[]>({
     queryKey: useAllToolProvidersKey,
     queryFn: () => get<Collection[]>('/workspaces/current/tool-providers'),
   })