(Feat) Implement remote Agent type

Makefile

@@ -111,8 +111,12 @@ build-all: buildx-create
 
 .PHONY: push-test-agent
 push-test-agent: build-kagent-adk
+	@echo "Pushing BYO test Agent"
 	$(DOCKER_BUILDER) build --push --build-arg DOCKER_REGISTRY=$(DOCKER_REGISTRY) --build-arg VERSION=$(VERSION) -t $(DOCKER_REGISTRY)/kebab:latest -f go/test/e2e/agents/kebab/Dockerfile ./go/test/e2e/agents/kebab
 	kubectl apply --namespace kagent --context kind-$(KIND_CLUSTER_NAME) -f go/test/e2e/agents/kebab/agent.yaml
+	@echo "Pushing Remote test Agent"
+	$(DOCKER_BUILDER) build --push --build-arg DOCKER_REGISTRY=$(DOCKER_REGISTRY) --build-arg VERSION=$(VERSION) -t $(DOCKER_REGISTRY)/remote-kebab:latest -f go/test/e2e/agents/remote-kebab/Dockerfile ./go/test/e2e/agents/remote-kebab
+	kubectl apply --namespace kagent --context kind-$(KIND_CLUSTER_NAME) -f go/test/e2e/agents/remote-kebab/agent.yaml
 
 .PHONY: create-kind-cluster
 create-kind-cluster:

go/api/v1alpha2/agent_types.go

@@ -29,27 +29,30 @@ import (
 )
 
 // AgentType represents the agent type
-// +kubebuilder:validation:Enum=Declarative;BYO
+// +kubebuilder:validation:Enum=Declarative;BYO;Remote
 type AgentType string
 
 const (
 	AgentType_Declarative AgentType = "Declarative"
 	AgentType_BYO         AgentType = "BYO"
+	AgentType_Remote      AgentType = "Remote"
 )
 
 // AgentSpec defines the desired state of Agent.
 // +kubebuilder:validation:XValidation:message="type must be specified",rule="has(self.type)"
-// +kubebuilder:validation:XValidation:message="type must be either Declarative or BYO",rule="self.type == 'Declarative' || self.type == 'BYO'"
-// +kubebuilder:validation:XValidation:message="declarative must be specified if type is Declarative, or byo must be specified if type is BYO",rule="(self.type == 'Declarative' && has(self.declarative)) || (self.type == 'BYO' && has(self.byo))"
+// +kubebuilder:validation:XValidation:message="type must be either Declarative, BYO, or Remote",rule="self.type == 'Declarative' || self.type == 'BYO' || self.type == 'Remote'"
+// +kubebuilder:validation:XValidation:message="declarative must be specified if type is Declarative, or byo must be specified if type is BYO, or remote must be specified if type is Remote",rule="(self.type == 'Declarative' && has(self.declarative)) || (self.type == 'BYO' && has(self.byo)) || (self.type == 'Remote' && has(self.remote))"
 type AgentSpec struct {
-	// +kubebuilder:validation:Enum=Declarative;BYO
+	// +kubebuilder:validation:Enum=Declarative;BYO;Remote
 	// +kubebuilder:default=Declarative
 	Type AgentType `json:"type"`
 
 	// +optional
 	BYO *BYOAgentSpec `json:"byo,omitempty"`
 	// +optional
 	Declarative *DeclarativeAgentSpec `json:"declarative,omitempty"`
+	// +optional
+	Remote *RemoteAgentSpec `json:"remote,omitempty"`
 
 	// +optional
 	Description string `json:"description,omitempty"`
@@ -111,6 +114,13 @@ type ByoDeploymentSpec struct {
 	SharedDeploymentSpec `json:",inline"`
 }
 
+type RemoteAgentSpec struct {
+	// DiscoveryURL is the URL of the agent server.
+	// The Agent Card is infered from the well known path.
+	// +kubebuilder:validation:MinLength=1
+	DiscoveryURL string `json:"discoveryUrl,omitempty"`
+}
+
 type SharedDeploymentSpec struct {
 	// If not specified, the default value is 1.
 	// +optional

go/cli/internal/cli/install.go

@@ -340,4 +340,3 @@ func checkHelmAvailable() error {
 	}
 	return nil
 }
-

go/config/crd/bases/kagent.dev_agents.yaml

@@ -6779,7 +6779,8 @@ spec:
                       If not specified, the default value is true.
                     type: boolean
                   systemMessage:
-                    minLength: 1
+                    description: SystemMessage is a string specifying the system message
+                      for the agent
                     type: string
                   systemMessageFrom:
                     description: SystemMessageFrom is a reference to a ConfigMap or
@@ -6905,14 +6906,25 @@ spec:
                   rule: '!has(self.systemMessage) || !has(self.systemMessageFrom)'
               description:
                 type: string
+              remote:
+                properties:
+                  discoveryUrl:
+                    description: |-
+                      DiscoveryURL is the URL of the agent server.
+                      The Agent Card is infered from the well known path.
+                    minLength: 1
+                    type: string
+                type: object
               type:
                 allOf:
                 - enum:
                   - Declarative
                   - BYO
+                  - Remote
                 - enum:
                   - Declarative
                   - BYO
+                  - Remote
                 default: Declarative
                 description: AgentType represents the agent type
                 type: string
@@ -6922,12 +6934,14 @@ spec:
             x-kubernetes-validations:
             - message: type must be specified
               rule: has(self.type)
-            - message: type must be either Declarative or BYO
-              rule: self.type == 'Declarative' || self.type == 'BYO'
+            - message: type must be either Declarative, BYO, or Remote
+              rule: self.type == 'Declarative' || self.type == 'BYO' || self.type
+                == 'Remote'
             - message: declarative must be specified if type is Declarative, or byo
-                must be specified if type is BYO
+                must be specified if type is BYO, or remote must be specified if type
+                is Remote
               rule: (self.type == 'Declarative' && has(self.declarative)) || (self.type
-                == 'BYO' && has(self.byo))
+                == 'BYO' && has(self.byo)) || (self.type == 'Remote' && has(self.remote))
           status:
             description: AgentStatus defines the observed state of Agent.
             properties:

go/config/rbac/role.yaml

@@ -4,56 +4,54 @@ kind: ClusterRole
 metadata:
   name: manager-role
 rules:
-  - apiGroups:
-      - ""
-    resources:
-      - configmaps
-      - secrets
-      - services
-    verbs:
-      - get
-      - list
-      - watch
-  - apiGroups:
-      - kagent.dev
-    resources:
-      - agents
-      - memories
-      - modelconfigs
-      - remotemcpservers
-    verbs:
-      - create
-      - delete
-      - get
-      - list
-      - patch
-      - update
-      - watch
-  - apiGroups:
-      - kagent.dev
-    resources:
-      - agents/finalizers
-      - memories/finalizers
-      - modelconfigs/finalizers
-      - remotemcpservers/finalizers
-    verbs:
-      - update
-  - apiGroups:
-      - kagent.dev
-    resources:
-      - agents/status
-      - memories/status
-      - modelconfigs/status
-      - remotemcpservers/status
-    verbs:
-      - get
-      - patch
-      - update
-  - apiGroups:
-      - kagent.dev
-    resources:
-      - mcpservers
-    verbs:
-      - get
-      - list
-      - watch
+- apiGroups:
+  - ""
+  resources:
+  - services
+  verbs:
+  - get
+  - list
+  - watch
+- apiGroups:
+  - kagent.dev
+  resources:
+  - agents
+  - memories
+  - modelconfigs
+  - remotemcpservers
+  verbs:
+  - create
+  - delete
+  - get
+  - list
+  - patch
+  - update
+  - watch
+- apiGroups:
+  - kagent.dev
+  resources:
+  - agents/finalizers
+  - memories/finalizers
+  - modelconfigs/finalizers
+  - remotemcpservers/finalizers
+  verbs:
+  - update
+- apiGroups:
+  - kagent.dev
+  resources:
+  - agents/status
+  - memories/status
+  - modelconfigs/status
+  - remotemcpservers/status
+  verbs:
+  - get
+  - patch
+  - update
+- apiGroups:
+  - kagent.dev
+  resources:
+  - mcpservers
+  verbs:
+  - get
+  - list
+  - watch

go/internal/a2a/a2a_handler_mux.go

@@ -6,11 +6,14 @@ import (
 	"sync"
 
 	"github.com/gorilla/mux"
+	"github.com/kagent-dev/kagent/go/api/v1alpha2"
+	"github.com/kagent-dev/kagent/go/internal/database"
 	authimpl "github.com/kagent-dev/kagent/go/internal/httpserver/auth"
 	common "github.com/kagent-dev/kagent/go/internal/utils"
 	"github.com/kagent-dev/kagent/go/pkg/auth"
 	"trpc.group/trpc-go/trpc-a2a-go/client"
 	"trpc.group/trpc-go/trpc-a2a-go/server"
+	"trpc.group/trpc-go/trpc-a2a-go/taskmanager"
 )
 
 // A2AHandlerMux is an interface that defines methods for adding, getting, and removing agentic task handlers.
@@ -19,6 +22,7 @@ type A2AHandlerMux interface {
 		agentRef string,
 		client *client.A2AClient,
 		card server.AgentCard,
+		agentType v1alpha2.AgentType,
 	) error
 	RemoveAgentHandler(
 		agentRef string,
@@ -31,24 +35,34 @@ type handlerMux struct {
 	lock           sync.RWMutex
 	basePathPrefix string
 	authenticator  auth.AuthProvider
+	dbClient       database.Client
 }
 
 var _ A2AHandlerMux = &handlerMux{}
 
-func NewA2AHttpMux(pathPrefix string, authenticator auth.AuthProvider) *handlerMux {
+func NewA2AHttpMux(pathPrefix string, authenticator auth.AuthProvider, dbClient database.Client) *handlerMux {
 	return &handlerMux{
 		handlers:       make(map[string]http.Handler),
 		basePathPrefix: pathPrefix,
 		authenticator:  authenticator,
+		dbClient:       dbClient,
 	}
 }
 
 func (a *handlerMux) SetAgentHandler(
 	agentRef string,
 	client *client.A2AClient,
 	card server.AgentCard,
+	agentType v1alpha2.AgentType,
 ) error {
-	srv, err := server.NewA2AServer(card, NewPassthroughManager(client), server.WithMiddleWare(authimpl.NewA2AAuthenticator(a.authenticator)))
+	var manager taskmanager.TaskManager
+	if agentType == v1alpha2.AgentType_Remote {
+		manager = NewRecordingManager(client, a.dbClient, agentRef)
+	} else {
+		manager = NewPassthroughManager(client)
+	}
+
+	srv, err := server.NewA2AServer(card, manager, server.WithMiddleWare(authimpl.NewA2AAuthenticator(a.authenticator)))
 	if err != nil {
 		return fmt.Errorf("failed to create A2A server: %w", err)
 	}

go/internal/a2a/a2a_utils.go

@@ -12,6 +12,8 @@ func ExtractText(message protocol.Message) string {
 	for _, part := range message.Parts {
 		if textPart, ok := part.(*protocol.TextPart); ok {
 			builder.WriteString(textPart.Text)
+		} else if textPart, ok := part.(protocol.TextPart); ok {
+			builder.WriteString(textPart.Text)
 		}
 	}
 	return builder.String()