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.
	Before(ctx context.Context, req *Request) (context.Context, error)

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

type CallMeta map[string][]string

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(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 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
	// CallMeta holds request metadata like auth headers and signatures.
	Meta CallMeta
	// 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
	// CallMeta holds request metadata like auth headers and signatures.
	Meta CallMeta
	// 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 SessionCredentials map[a2a.SecuritySchemeName]AuthCredential

type SessionID string

type Transport interface {
	// GetTask calls the 'tasks/get' protocol method.
	GetTask(ctx context.Context, query *a2a.TaskQueryParams) (*a2a.Task, error)

	// CancelTask calls the 'tasks/cancel' protocol method.
	CancelTask(ctx context.Context, id *a2a.TaskIDParams) (*a2a.Task, error)

	// SendMessage calls the 'message/send' protocol method (non-streaming).
	SendMessage(ctx context.Context, message *a2a.MessageSendParams) (a2a.SendMessageResult, error)

	// ResubscribeToTask calls the `tasks/resubscribe` protocol method.
	ResubscribeToTask(ctx context.Context, id *a2a.TaskIDParams) iter.Seq2[a2a.Event, error]

	// SendStreamingMessage calls the 'message/stream' protocol method (streaming).
	SendStreamingMessage(ctx context.Context, message *a2a.MessageSendParams) iter.Seq2[a2a.Event, error]

	// GetTaskPushNotificationConfig calls the `tasks/pushNotificationConfig/get` protocol method.
	GetTaskPushConfig(ctx context.Context, params *a2a.GetTaskPushConfigParams) (*a2a.TaskPushConfig, error)

	// ListTaskPushNotificationConfig calls the `tasks/pushNotificationConfig/list` protocol method.
	ListTaskPushConfig(ctx context.Context, params *a2a.ListTaskPushConfigParams) ([]*a2a.TaskPushConfig, error)

	// SetTaskPushConfig calls the `tasks/pushNotificationConfig/set` protocol method.
	SetTaskPushConfig(ctx context.Context, params *a2a.TaskPushConfig) (*a2a.TaskPushConfig, error)

	// DeleteTaskPushNotificationConfig calls the `tasks/pushNotificationConfig/delete` protocol method.
	DeleteTaskPushConfig(ctx context.Context, params *a2a.DeleteTaskPushConfigParams) error

	// GetAgentCard resolves the AgentCard.
	// If extended card is supported calls the 'agent/getAuthenticatedExtendedCard' protocol method.
	GetAgentCard(ctx context.Context) (*a2a.AgentCard, error)

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

type TransportFactory interface {
	Create(ctx context.Context, url string, card *a2a.AgentCard) (Transport, error)
}

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