feat: modify eino agent config, from Transport to A2AClient (#463)

Author: meguminnnnnnnnn

a2a/extension/eino/client.go

@@ -29,12 +29,11 @@ import (
 
 	"github.com/cloudwego/eino-ext/a2a/client"
 	"github.com/cloudwego/eino-ext/a2a/models"
-	"github.com/cloudwego/eino-ext/a2a/transport"
 	"github.com/cloudwego/eino-ext/a2a/utils"
 )
 
 type AgentConfig struct {
-	Transport transport.ClientTransport
+	Client *client.A2AClient
 
 	// optional, from AgentCard by default
 	Name        *string
@@ -50,16 +49,13 @@ type AgentConfig struct {
 }
 
 func NewAgent(ctx context.Context, cfg AgentConfig) (adk.Agent, error) {
-	cli, err := client.NewA2AClient(ctx, &client.Config{
-		Transport: cfg.Transport,
-	})
-	if err != nil {
-		return nil, fmt.Errorf("failed to create a2a client: %w", err)
+	if cfg.Client == nil {
+		return nil, errors.New("Client is required")
 	}
 	var name, desc string
 	var streaming bool
 	if cfg.Name == nil || cfg.Description == nil || cfg.Streaming == nil {
-		card, err := cli.AgentCard(ctx)
+		card, err := cfg.Client.AgentCard(ctx)
 		if err != nil {
 			return nil, err
 		}
@@ -83,7 +79,7 @@ func NewAgent(ctx context.Context, cfg AgentConfig) (adk.Agent, error) {
 		streaming:             streaming,
 		inputMessageConvertor: cfg.InputMessageConvertor,
 		outputConvertor:       cfg.OutputConvertor,
-		cli:                   cli,
+		cli:                   cfg.Client,
 	}
 	if a.inputMessageConvertor == nil {
 		a.inputMessageConvertor = func(ctx context.Context, messages []*schema.Message) (models.Message, error) {

a2a/extension/eino/examples/client/client.go

@@ -28,6 +28,7 @@ import (
 	"github.com/cloudwego/eino/adk"
 	"github.com/cloudwego/eino/schema"
 
+	"github.com/cloudwego/eino-ext/a2a/client"
 	"github.com/cloudwego/eino-ext/a2a/extension/eino"
 	"github.com/cloudwego/eino-ext/a2a/transport/jsonrpc"
 )
@@ -51,8 +52,15 @@ func nonStreamChat(ctx context.Context) {
 		BaseURL:     "http://127.0.0.1:8888",
 		HandlerPath: "/test",
 	})
+	if err != nil {
+		log.Fatal(err)
+	}
+	cli, err := client.NewA2AClient(ctx, &client.Config{Transport: t})
+	if err != nil {
+		log.Fatal(err)
+	}
 	a, err := eino.NewAgent(ctx, eino.AgentConfig{
-		Transport:   t,
+		Client:      cli,
 		Name:        nil,
 		Description: nil,
 		Streaming:   &streaming,
@@ -80,8 +88,15 @@ func streamChat(ctx context.Context) {
 		BaseURL:     "http://127.0.0.1:8888",
 		HandlerPath: "/test",
 	})
+	if err != nil {
+		log.Fatal(err)
+	}
+	cli, err := client.NewA2AClient(ctx, &client.Config{Transport: t})
+	if err != nil {
+		log.Fatal(err)
+	}
 	a, err := eino.NewAgent(ctx, eino.AgentConfig{
-		Transport: t,
+		Client:    cli,
 		Streaming: &streaming,
 	})
 	if err != nil {
@@ -107,8 +122,15 @@ func humanInTheLoop(ctx context.Context) {
 		BaseURL:     "http://127.0.0.1:8888",
 		HandlerPath: "/test",
 	})
+	if err != nil {
+		log.Fatal(err)
+	}
+	cli, err := client.NewA2AClient(ctx, &client.Config{Transport: t})
+	if err != nil {
+		log.Fatal(err)
+	}
 	a, err := eino.NewAgent(ctx, eino.AgentConfig{
-		Transport: t,
+		Client: cli,
 	})
 	if err != nil {
 		log.Fatal(err)
@@ -123,8 +145,15 @@ func streamHumanInTheLoop(ctx context.Context) {
 		BaseURL:     "http://127.0.0.1:8888",
 		HandlerPath: "/test",
 	})
+	if err != nil {
+		log.Fatal(err)
+	}
+	cli, err := client.NewA2AClient(ctx, &client.Config{Transport: t})
+	if err != nil {
+		log.Fatal(err)
+	}
 	a, err := eino.NewAgent(ctx, eino.AgentConfig{
-		Transport: t,
+		Client:    cli,
 		Streaming: &streaming,
 	})
 	if err != nil {