refactor: remove legacy session and task services (#2598)

Author: EItanya

go/adk/pkg/controllerclient/client.go

@@ -39,8 +39,6 @@ type Client struct {
 	maxMessageBytes int
 	agentName       string
 	tokenProvider   TokenProvider
-	sessionService  apiv1alpha1.SessionServiceClient
-	taskService     apiv1alpha1.TaskStoreServiceClient
 	memoryService   apiv1alpha1.MemoryServiceClient
 	closeOnce       sync.Once
 	closeErr        error
@@ -80,20 +78,10 @@ func New(config Config) (*Client, error) {
 		maxMessageBytes: config.MaxMessageBytes,
 		agentName:       config.AgentName,
 		tokenProvider:   config.TokenProvider,
-		sessionService:  apiv1alpha1.NewSessionServiceClient(connection),
-		taskService:     apiv1alpha1.NewTaskStoreServiceClient(connection),
 		memoryService:   apiv1alpha1.NewMemoryServiceClient(connection),
 	}, nil
 }
 
-func (client *Client) SessionService() apiv1alpha1.SessionServiceClient {
-	return client.sessionService
-}
-
-func (client *Client) TaskService() apiv1alpha1.TaskStoreServiceClient {
-	return client.taskService
-}
-
 func (client *Client) MemoryService() apiv1alpha1.MemoryServiceClient {
 	return client.memoryService
 }

go/adk/pkg/controllerclient/client_test.go

@@ -33,25 +33,25 @@ func (provider *mutableTokenProvider) set(token string) {
 	provider.token = token
 }
 
-type metadataSessionServer struct {
-	apiv1alpha1.UnimplementedSessionServiceServer
+type metadataMemoryServer struct {
+	apiv1alpha1.UnimplementedMemoryServiceServer
 	metadata  []metadata.MD
 	deadlines []bool
 }
 
-func (server *metadataSessionServer) ListSessions(ctx context.Context, _ *apiv1alpha1.ListSessionsRequest) (*apiv1alpha1.ListSessionsResponse, error) {
+func (server *metadataMemoryServer) List(ctx context.Context, _ *apiv1alpha1.MemoryServiceListRequest) (*apiv1alpha1.MemoryServiceListResponse, error) {
 	values, _ := metadata.FromIncomingContext(ctx)
 	_, hasDeadline := ctx.Deadline()
 	server.metadata = append(server.metadata, values)
 	server.deadlines = append(server.deadlines, hasDeadline)
-	return &apiv1alpha1.ListSessionsResponse{}, nil
+	return &apiv1alpha1.MemoryServiceListResponse{}, nil
 }
 
 func TestClientAddsDynamicMetadataAndDeadlines(t *testing.T) {
 	listener := bufconn.Listen(1024 * 1024)
-	service := &metadataSessionServer{}
+	service := &metadataMemoryServer{}
 	grpcServer := grpc.NewServer()
-	apiv1alpha1.RegisterSessionServiceServer(grpcServer, service)
+	apiv1alpha1.RegisterMemoryServiceServer(grpcServer, service)
 	go func() { _ = grpcServer.Serve(listener) }()
 	t.Cleanup(func() {
 		grpcServer.Stop()
@@ -71,13 +71,13 @@ func TestClientAddsDynamicMetadataAndDeadlines(t *testing.T) {
 	t.Cleanup(func() { require.NoError(t, client.Close()) })
 
 	ctx, cancel := client.CallContext(auth.WithUserID(t.Context(), "context-user"), "")
-	_, err = client.SessionService().ListSessions(ctx, &apiv1alpha1.ListSessionsRequest{})
+	_, err = client.MemoryService().List(ctx, &apiv1alpha1.MemoryServiceListRequest{})
 	cancel()
 	require.NoError(t, err)
 
 	tokens.set("second-token")
 	ctx, cancel = client.CallContext(t.Context(), "explicit-user")
-	_, err = client.SessionService().ListSessions(ctx, &apiv1alpha1.ListSessionsRequest{})
+	_, err = client.MemoryService().List(ctx, &apiv1alpha1.MemoryServiceListRequest{})
 	cancel()
 	require.NoError(t, err)
 

go/adk/pkg/runner/adapter.go

@@ -11,7 +11,6 @@ import (
 	"github.com/kagent-dev/kagent/go/adk/pkg/controllerclient"
 	kagentmemory "github.com/kagent-dev/kagent/go/adk/pkg/memory"
 	"github.com/kagent-dev/kagent/go/adk/pkg/sts"
-	"github.com/kagent-dev/kagent/go/adk/pkg/tools"
 	"github.com/kagent-dev/kagent/go/api/adk"
 	adkmemory "google.golang.org/adk/v2/memory"
 	adkplugin "google.golang.org/adk/v2/plugin"
@@ -48,23 +47,6 @@ func CreateRunnerConfig(
 		extraTools = append(extraTools, saveTool)
 	}
 
-	if agentConfig.ShareTools != nil && *agentConfig.ShareTools && controllerClient != nil {
-		createTool, err := tools.NewCreateShareLinkTool(controllerClient, appName)
-		if err != nil {
-			return runner.Config{}, fmt.Errorf("failed to create create_share_link tool: %w", err)
-		}
-		listTool, err := tools.NewListShareLinksTool(controllerClient, appName)
-		if err != nil {
-			return runner.Config{}, fmt.Errorf("failed to create list_share_links tool: %w", err)
-		}
-		deleteTool, err := tools.NewDeleteShareLinkTool(controllerClient, appName)
-		if err != nil {
-			return runner.Config{}, fmt.Errorf("failed to create delete_share_link tool: %w", err)
-		}
-		extraTools = append(extraTools, createTool, listTool, deleteTool)
-		log.Info("Share link tools enabled")
-	}
-
 	stsPlugin, err := buildTokenPropagationPlugin(ctx, log)
 	if err != nil {
 		return runner.Config{}, err