Fix: fix oauth for knowledge source (#1762)

Signed-off-by: Daishan Peng <[email protected]>

Author: StrongMonkey

pkg/controller/handlers/toolinfo/toolinfo.go

@@ -12,7 +12,9 @@ import (
 	"github.com/obot-platform/obot/pkg/render"
 	v1 "github.com/obot-platform/obot/pkg/storage/apis/obot.obot.ai/v1"
 	apierror "k8s.io/apimachinery/pkg/api/errors"
+	"k8s.io/apimachinery/pkg/fields"
 	"k8s.io/apimachinery/pkg/util/sets"
+	kclient "sigs.k8s.io/controller-runtime/pkg/client"
 )
 
 type Handler struct {
@@ -105,6 +107,47 @@ func (h *Handler) RemoveUnneededCredentials(req router.Request, _ router.Respons
 		}
 	}
 
+	var knowledgeSets v1.KnowledgeSetList
+	switch req.Object.(type) {
+	case *v1.Workflow:
+		if err := req.List(&knowledgeSets, &kclient.ListOptions{
+			Namespace: req.Namespace,
+			FieldSelector: fields.SelectorFromSet(map[string]string{
+				"spec.workflowName": req.Object.GetName(),
+			}),
+		}); err != nil {
+			return err
+		}
+
+	case *v1.Agent:
+		if err := req.List(&knowledgeSets, &kclient.ListOptions{
+			Namespace: req.Namespace,
+			FieldSelector: fields.SelectorFromSet(map[string]string{
+				"spec.agentName": req.Object.GetName(),
+			}),
+		}); err != nil {
+			return err
+		}
+	}
+
+	for _, knowledgeSet := range knowledgeSets.Items {
+		var knowledgeSources v1.KnowledgeSourceList
+		if err := req.List(&knowledgeSources, &kclient.ListOptions{
+			Namespace: req.Namespace,
+			FieldSelector: fields.SelectorFromSet(map[string]string{
+				"spec.knowledgeSetName": knowledgeSet.Name,
+			}),
+		}); err != nil {
+			return err
+		}
+
+		for _, knowledgeSource := range knowledgeSources.Items {
+			if sourceType := string(knowledgeSource.Spec.Manifest.GetType()); sourceType != "" {
+				credentialNames[sourceType+".sync-file"] = struct{}{}
+			}
+		}
+	}
+
 	for _, cred := range creds {
 		if _, ok := credentialNames[cred.ToolName]; !ok {
 			if err := h.gptscript.DeleteCredential(req.Ctx, req.Object.GetName(), cred.ToolName); err != nil && !errors.As(err, &gptscript.ErrNotFound{}) {