a2aclient

type AuthCredential string

type AuthInterceptor struct {
	PassthroughInterceptor
	Service CredentialsService
}

type CallInterceptor interface {
	// Before allows to observe, modify or reject a Request.
	// A new context.Context can be returned to pass information to After.
	// If either the result (2nd return value) or the error (3rd return value) is non nil,
	// the network request will not be made and the value will be returned to the caller.
	Before(ctx context.Context, req *Request) (context.Context, any, error)

	// After allows to observe, modify or reject a Response.
	After(ctx context.Context, resp *Response) error
}

type Client struct {
	// contains filtered or unexported fields
}

type Config struct {
	// PushConfig specifies the default push notification configuration to apply for every Task.
	PushConfig *a2a.PushConfig
	// AcceptedOutputModes are MIME types passed with every Client message and might be used by an agent
	// to decide on the result format.
	// For example, an Agent might declare a skill with OutputModes: ["application/json", "image/png"]
	// and a Client that doesn't support images will pass AcceptedOutputModes: ["application/json"]
	// to get a result in the desired format.
	AcceptedOutputModes []string
	// PreferredTransports is used for selecting the most appropriate communication protocol.
	// The first transport from the list which is also supported by the server is going to be used
	// to establish a connection. If no preference is provided the server ordering will be used.
	// If there's no overlap in supported Transport Factory will return an error on Client
	// creation attempt.
	PreferredTransports []a2a.TransportProtocol
	// Whether client prefers to poll for task updates instead of blocking until a terminal state is reached.
	// If set to true, non-streaming send message result might be a Message or a Task in any (including non-terminal) state.
	// Callers are responsible for running the polling loop. This configuration does not apply to streaming requests.
	Polling bool
}

type CredentialsService interface {
	// Get retrieves the credential for the given session ID and security scheme name.
	Get(ctx context.Context, sid SessionID, scheme a2a.SecuritySchemeName) (AuthCredential, error)
}

type Factory struct {
	// contains filtered or unexported fields
}

type FactoryOption interface {
	// contains filtered or unexported methods
}

type InMemoryCredentialsStore struct {
	// contains filtered or unexported fields
}

type JSONRPCOption func(*jsonrpcTransport)

type PassthroughInterceptor struct{}

type RESTTransport struct {
	// contains filtered or unexported fields
}

type Request struct {
	// Method is the name of the method invoked on the A2A-server.
	Method string
	// BaseURL is the URL of the agent interface to which the Client is connected.
	BaseURL string
	// ServiceParams holds horizontally applicable context or parameters with case-insensitive keys.
	// The transmission mechanism for these service parameter key-value pairs is defined by the specific protocol binding
	ServiceParams ServiceParams
	// Card is the AgentCard of the agent the client is connected to. Might be nil if Client was
	// created directly from server URL and extended AgentCard was never fetched.
	Card *a2a.AgentCard
	// Payload is the request payload. It is nil if the method does not take any parameters. Otherwise, it is one of a2a package core types otherwise.
	Payload any
}

type Response struct {
	// Method is the name of the method invoked on the A2A-server.
	Method string
	// BaseURL is the URL of the agent interface to which the Client is connected.
	BaseURL string
	// Err is the error response. It is nil for successful invocations.
	Err error
	// ServiceParams holds horizontally applicable context or parameters with case-insensitive keys.
	// The transmission mechanism for these service parameter key-value pairs is defined by the specific protocol binding
	ServiceParams ServiceParams
	// Card is the AgentCard of the agent the client is connected to. Might be nil if Client was
	// created directly from server URL and extended AgentCard was never fetched.
	Card *a2a.AgentCard
	// Payload is the response. It is nil if method doesn't return anything or Err was returned. Otherwise, it is one of a2a package core types otherwise.
	Payload any
}

type ServiceParams map[string][]string

type SessionCredentials map[a2a.SecuritySchemeName]AuthCredential

type SessionID string

type Transport interface {
	// GetTask calls the 'GetTask' protocol method.
	GetTask(context.Context, ServiceParams, *a2a.GetTaskRequest) (*a2a.Task, error)

	// ListTasks calls the 'ListTasks' protocol method.
	ListTasks(context.Context, ServiceParams, *a2a.ListTasksRequest) (*a2a.ListTasksResponse, error)

	// CancelTask calls the 'CancelTask' protocol method.
	CancelTask(context.Context, ServiceParams, *a2a.CancelTaskRequest) (*a2a.Task, error)

	// SendMessage calls the 'SendMessage' protocol method (non-streaming).
	SendMessage(context.Context, ServiceParams, *a2a.SendMessageRequest) (a2a.SendMessageResult, error)

	// SubscribeToTask calls the `SubscribeToTask` protocol method.
	SubscribeToTask(context.Context, ServiceParams, *a2a.SubscribeToTaskRequest) iter.Seq2[a2a.Event, error]

	// SendStreamingMessage calls the 'SendStreamingMessage' protocol method (streaming).
	SendStreamingMessage(context.Context, ServiceParams, *a2a.SendMessageRequest) iter.Seq2[a2a.Event, error]

	// GetTaskPushConfig calls the `GetTaskPushNotificationConfig` protocol method.
	GetTaskPushConfig(context.Context, ServiceParams, *a2a.GetTaskPushConfigRequest) (*a2a.TaskPushConfig, error)

	// ListTaskPushConfigs calls the `ListTaskPushNotificationConfig` protocol method.
	ListTaskPushConfigs(context.Context, ServiceParams, *a2a.ListTaskPushConfigRequest) ([]*a2a.TaskPushConfig, error)

	// CreateTaskPushConfig calls the `CreateTaskPushNotificationConfig` protocol method.
	CreateTaskPushConfig(context.Context, ServiceParams, *a2a.CreateTaskPushConfigRequest) (*a2a.TaskPushConfig, error)

	// DeleteTaskPushConfig calls the `DeleteTaskPushNotificationConfig` protocol method.
	DeleteTaskPushConfig(context.Context, ServiceParams, *a2a.DeleteTaskPushConfigRequest) error

	// GetExtendedAgentCard calls the 'GetExtendedAgentCard' protocol method.
	GetExtendedAgentCard(context.Context, ServiceParams, *a2a.GetExtendedAgentCardRequest) (*a2a.AgentCard, error)

	// Destroy cleans up resources associated with the transport (eg. close a gRPC channel).
	Destroy() error
}

type TransportFactory interface {
	// Create creates an A2A protocol connection to the provided URL.
	Create(ctx context.Context, card *a2a.AgentCard, iface *a2a.AgentInterface) (Transport, error)
}

type TransportFactoryFn func(ctx context.Context, card *a2a.AgentCard, iface *a2a.AgentInterface) (Transport, error)