1use std::pin::Pin;
14use std::sync::Arc;
15
16use a2a_protocol_types::proto as apb;
17use a2a_protocol_types::proto::convert::ConvertError;
18use tokio::sync::mpsc;
19use tokio_stream::wrappers::ReceiverStream;
20use tonic::{Request, Response, Status};
21
22use super::helpers::{server_error_to_status, validated_metadata};
23use super::pb::a2a_service_server::A2aService;
24use super::GrpcConfig;
25use crate::handler::{RequestHandler, SendMessageResult};
26
27type NativeStream =
29 Pin<Box<dyn tokio_stream::Stream<Item = Result<apb::StreamResponse, Status>> + Send + 'static>>;
30
31#[allow(clippy::needless_pass_by_value)]
33fn bad_request(err: ConvertError) -> Status {
34 Status::invalid_argument(err.to_string())
35}
36
37#[allow(clippy::needless_pass_by_value)]
40fn bad_response(err: ConvertError) -> Status {
41 Status::internal(format!("response not representable in protobuf: {err}"))
42}
43
44fn send_result_to_stream(
46 resp: a2a_protocol_types::responses::SendMessageResponse,
47) -> Result<apb::StreamResponse, ConvertError> {
48 let payload = match resp {
49 a2a_protocol_types::responses::SendMessageResponse::Task(t) => {
50 apb::stream_response::Payload::Task(t.try_into()?)
51 }
52 a2a_protocol_types::responses::SendMessageResponse::Message(m) => {
53 apb::stream_response::Payload::Message(m.try_into()?)
54 }
55 other => {
56 return Err(ConvertError {
57 field: "sendMessageResponse.payload",
58 reason: format!("unsupported response variant: {other:?}"),
59 })
60 }
61 };
62 Ok(apb::StreamResponse {
63 payload: Some(payload),
64 })
65}
66
67fn reader_to_native_stream(
69 mut reader: crate::streaming::InMemoryQueueReader,
70 capacity: usize,
71) -> NativeStream {
72 use crate::streaming::EventQueueReader;
73 let (tx, rx) = mpsc::channel(capacity);
74 tokio::spawn(async move {
75 loop {
76 match reader.read().await {
77 Some(Ok(event)) => {
78 let item = apb::StreamResponse::try_from(event).map_err(bad_response);
79 let is_err = item.is_err();
80 if tx.send(item).await.is_err() || is_err {
81 break;
82 }
83 }
84 Some(Err(_)) => {
85 let _ = tx.send(Err(Status::internal("event queue error"))).await;
86 break;
87 }
88 None => break,
89 }
90 }
91 });
92 Box::pin(ReceiverStream::new(rx))
93}
94
95pub struct A2aServiceImpl {
100 pub(super) handler: Arc<RequestHandler>,
101 pub(super) config: GrpcConfig,
102}
103
104#[tonic::async_trait]
105impl A2aService for A2aServiceImpl {
106 async fn send_message(
109 &self,
110 request: Request<apb::SendMessageRequest>,
111 ) -> Result<Response<apb::SendMessageResponse>, Status> {
112 let headers = validated_metadata(request.metadata())?;
113 let params: a2a_protocol_types::params::MessageSendParams =
114 request.into_inner().try_into().map_err(bad_request)?;
115 match self
116 .handler
117 .on_send_message(params, false, Some(&headers))
118 .await
119 {
120 Ok(SendMessageResult::Response(resp)) => {
121 Ok(Response::new(resp.try_into().map_err(bad_response)?))
122 }
123 Ok(SendMessageResult::Stream(_)) => Err(Status::internal(
124 "unexpected stream response for unary call",
125 )),
126 Err(e) => Err(server_error_to_status(&e)),
127 }
128 }
129
130 type SendStreamingMessageStream = NativeStream;
131
132 async fn send_streaming_message(
133 &self,
134 request: Request<apb::SendMessageRequest>,
135 ) -> Result<Response<Self::SendStreamingMessageStream>, Status> {
136 let headers = validated_metadata(request.metadata())?;
137 let params: a2a_protocol_types::params::MessageSendParams =
138 request.into_inner().try_into().map_err(bad_request)?;
139 match self
140 .handler
141 .on_send_message(params, true, Some(&headers))
142 .await
143 {
144 Ok(SendMessageResult::Stream(reader)) => Ok(Response::new(reader_to_native_stream(
145 reader,
146 self.config.stream_channel_capacity,
147 ))),
148 Ok(SendMessageResult::Response(resp)) => {
149 let payload = send_result_to_stream(resp).map_err(bad_response)?;
151 let stream = Box::pin(tokio_stream::once(Ok(payload)));
152 Ok(Response::new(stream as NativeStream))
153 }
154 Err(e) => Err(server_error_to_status(&e)),
155 }
156 }
157
158 async fn get_task(
161 &self,
162 request: Request<apb::GetTaskRequest>,
163 ) -> Result<Response<apb::Task>, Status> {
164 let headers = validated_metadata(request.metadata())?;
165 let params: a2a_protocol_types::params::TaskQueryParams =
166 request.into_inner().try_into().map_err(bad_request)?;
167 match self.handler.on_get_task(params, Some(&headers)).await {
168 Ok(task) => Ok(Response::new(task.try_into().map_err(bad_response)?)),
169 Err(e) => Err(server_error_to_status(&e)),
170 }
171 }
172
173 async fn list_tasks(
174 &self,
175 request: Request<apb::ListTasksRequest>,
176 ) -> Result<Response<apb::ListTasksResponse>, Status> {
177 let headers = validated_metadata(request.metadata())?;
178 let params: a2a_protocol_types::params::ListTasksParams =
179 request.into_inner().try_into().map_err(bad_request)?;
180 match self.handler.on_list_tasks(params, Some(&headers)).await {
181 Ok(resp) => Ok(Response::new(resp.try_into().map_err(bad_response)?)),
182 Err(e) => Err(server_error_to_status(&e)),
183 }
184 }
185
186 async fn cancel_task(
187 &self,
188 request: Request<apb::CancelTaskRequest>,
189 ) -> Result<Response<apb::Task>, Status> {
190 let headers = validated_metadata(request.metadata())?;
191 let params: a2a_protocol_types::params::CancelTaskParams =
192 request.into_inner().try_into().map_err(bad_request)?;
193 match self.handler.on_cancel_task(params, Some(&headers)).await {
194 Ok(task) => Ok(Response::new(task.try_into().map_err(bad_response)?)),
195 Err(e) => Err(server_error_to_status(&e)),
196 }
197 }
198
199 type SubscribeToTaskStream = NativeStream;
200
201 async fn subscribe_to_task(
202 &self,
203 request: Request<apb::SubscribeToTaskRequest>,
204 ) -> Result<Response<Self::SubscribeToTaskStream>, Status> {
205 let headers = validated_metadata(request.metadata())?;
206 let params: a2a_protocol_types::params::TaskIdParams = request.into_inner().into();
207 match self.handler.on_resubscribe(params, Some(&headers)).await {
208 Ok(reader) => Ok(Response::new(reader_to_native_stream(
209 reader,
210 self.config.stream_channel_capacity,
211 ))),
212 Err(e) => Err(server_error_to_status(&e)),
213 }
214 }
215
216 async fn create_task_push_notification_config(
219 &self,
220 request: Request<apb::TaskPushNotificationConfig>,
221 ) -> Result<Response<apb::TaskPushNotificationConfig>, Status> {
222 let headers = validated_metadata(request.metadata())?;
223 let config: a2a_protocol_types::push::TaskPushNotificationConfig =
224 request.into_inner().into();
225 match self
226 .handler
227 .on_set_push_config(config, Some(&headers))
228 .await
229 {
230 Ok(cfg) => Ok(Response::new(cfg.into())),
231 Err(e) => Err(server_error_to_status(&e)),
232 }
233 }
234
235 async fn get_task_push_notification_config(
236 &self,
237 request: Request<apb::GetTaskPushNotificationConfigRequest>,
238 ) -> Result<Response<apb::TaskPushNotificationConfig>, Status> {
239 let headers = validated_metadata(request.metadata())?;
240 let params: a2a_protocol_types::params::GetPushConfigParams = request.into_inner().into();
241 match self
242 .handler
243 .on_get_push_config(params, Some(&headers))
244 .await
245 {
246 Ok(cfg) => Ok(Response::new(cfg.into())),
247 Err(e) => Err(server_error_to_status(&e)),
248 }
249 }
250
251 async fn list_task_push_notification_configs(
252 &self,
253 request: Request<apb::ListTaskPushNotificationConfigsRequest>,
254 ) -> Result<Response<apb::ListTaskPushNotificationConfigsResponse>, Status> {
255 let headers = validated_metadata(request.metadata())?;
256 let params: a2a_protocol_types::params::ListPushConfigsParams =
257 request.into_inner().try_into().map_err(bad_request)?;
258 match self
259 .handler
260 .on_list_push_configs(¶ms.task_id, params.tenant.as_deref(), Some(&headers))
261 .await
262 {
263 Ok(configs) => Ok(Response::new(
264 apb::ListTaskPushNotificationConfigsResponse {
265 configs: configs.into_iter().map(Into::into).collect(),
266 next_page_token: String::new(),
267 },
268 )),
269 Err(e) => Err(server_error_to_status(&e)),
270 }
271 }
272
273 async fn delete_task_push_notification_config(
274 &self,
275 request: Request<apb::DeleteTaskPushNotificationConfigRequest>,
276 ) -> Result<Response<()>, Status> {
277 let headers = validated_metadata(request.metadata())?;
278 let params: a2a_protocol_types::params::DeletePushConfigParams =
279 request.into_inner().into();
280 match self
281 .handler
282 .on_delete_push_config(params, Some(&headers))
283 .await
284 {
285 Ok(()) => Ok(Response::new(())),
286 Err(e) => Err(server_error_to_status(&e)),
287 }
288 }
289
290 async fn get_extended_agent_card(
293 &self,
294 request: Request<apb::GetExtendedAgentCardRequest>,
295 ) -> Result<Response<apb::AgentCard>, Status> {
296 let headers = validated_metadata(request.metadata())?;
297 match self
298 .handler
299 .on_get_extended_agent_card(Some(&headers))
300 .await
301 {
302 Ok(card) => Ok(Response::new(card.try_into().map_err(bad_response)?)),
303 Err(e) => Err(server_error_to_status(&e)),
304 }
305 }
306}
307
308#[cfg(test)]
309mod tests {
310 use super::*;
311 use a2a_protocol_types::message::{Message, MessageId, MessageRole, Part, PartContent};
312 use a2a_protocol_types::responses::SendMessageResponse;
313 use a2a_protocol_types::task::{ContextId, TaskId};
314
315 #[test]
316 fn bad_request_maps_to_invalid_argument() {
317 let status = bad_request(ConvertError {
318 field: "message.role",
319 reason: "unknown Role number 9".into(),
320 });
321 assert_eq!(status.code(), tonic::Code::InvalidArgument);
322 assert!(status.message().contains("message.role"));
323 }
324
325 #[test]
326 fn bad_response_maps_to_internal() {
327 let status = bad_response(ConvertError {
328 field: "task.metadata",
329 reason: "boom".into(),
330 });
331 assert_eq!(status.code(), tonic::Code::Internal);
332 }
333
334 #[test]
335 fn send_result_message_wraps_into_stream_payload() {
336 let resp = SendMessageResponse::Message(Message {
337 id: MessageId("m".into()),
338 role: MessageRole::Agent,
339 parts: vec![Part {
340 content: PartContent::Text("hi".into()),
341 metadata: None,
342 filename: None,
343 media_type: None,
344 }],
345 task_id: Some(TaskId("t".into())),
346 context_id: Some(ContextId("c".into())),
347 reference_task_ids: None,
348 extensions: None,
349 metadata: None,
350 });
351 let stream = send_result_to_stream(resp).unwrap();
352 assert!(matches!(
353 stream.payload,
354 Some(apb::stream_response::Payload::Message(_))
355 ));
356 }
357
358 use crate::agent_executor;
371 use crate::builder::RequestHandlerBuilder;
372 use a2a_protocol_types::agent_card::{
373 AgentCapabilities, AgentCard, AgentInterface, AgentSkill,
374 };
375
376 struct NoopExecutor;
377 agent_executor!(NoopExecutor, |_ctx, _queue| async { Ok(()) });
378
379 struct NoopSender;
383
384 impl crate::push::PushSender for NoopSender {
385 fn send<'a>(
386 &'a self,
387 _url: &'a str,
388 _event: &'a a2a_protocol_types::events::StreamResponse,
389 _config: &'a a2a_protocol_types::push::TaskPushNotificationConfig,
390 ) -> Pin<
391 Box<
392 dyn std::future::Future<Output = a2a_protocol_types::error::A2aResult<()>>
393 + Send
394 + 'a,
395 >,
396 > {
397 Box::pin(async { Ok(()) })
398 }
399
400 fn allows_private_urls(&self) -> bool {
401 true
402 }
403 }
404
405 fn test_card() -> AgentCard {
406 AgentCard {
407 url: None,
408 name: "native-grpc-test-agent".into(),
409 description: "Fixture for the canonical gRPC binding".into(),
410 version: "1.0.0".into(),
411 supported_interfaces: vec![AgentInterface {
412 url: "grpc://localhost:50051".into(),
413 protocol_binding: "gRPC".into(),
414 protocol_version: "1.0.0".into(),
415 tenant: None,
416 }],
417 default_input_modes: vec!["text/plain".into()],
418 default_output_modes: vec!["text/plain".into()],
419 skills: vec![AgentSkill {
420 id: "noop".into(),
421 name: "Noop".into(),
422 description: "Does nothing".into(),
423 tags: vec!["test".into()],
424 examples: None,
425 input_modes: None,
426 output_modes: None,
427 security_requirements: None,
428 }],
429 capabilities: AgentCapabilities::none()
430 .with_extended_agent_card(true)
431 .with_push_notifications(true),
434 provider: None,
435 icon_url: None,
436 documentation_url: None,
437 security_schemes: None,
438 security_requirements: None,
439 signatures: None,
440 }
441 }
442
443 fn service() -> A2aServiceImpl {
444 A2aServiceImpl {
445 handler: Arc::new(
446 RequestHandlerBuilder::new(NoopExecutor)
447 .with_agent_card(test_card())
448 .with_push_sender(NoopSender)
449 .allow_unauthenticated_extended_card()
452 .build()
453 .expect("default build should succeed"),
454 ),
455 config: GrpcConfig::default(),
456 }
457 }
458
459 fn send_request(text: &str) -> apb::SendMessageRequest {
460 apb::SendMessageRequest {
461 tenant: String::new(),
462 message: Some(apb::Message {
463 message_id: format!("msg-{text}"),
464 context_id: String::new(),
465 task_id: String::new(),
466 role: apb::Role::User as i32,
467 parts: vec![apb::Part {
468 metadata: None,
469 filename: String::new(),
470 media_type: String::new(),
471 content: Some(apb::part::Content::Text(text.into())),
472 }],
473 metadata: None,
474 extensions: Vec::new(),
475 reference_task_ids: Vec::new(),
476 }),
477 configuration: None,
478 metadata: None,
479 }
480 }
481
482 async fn seed_task(svc: &A2aServiceImpl) -> String {
484 let resp = svc
485 .send_message(Request::new(send_request("seed")))
486 .await
487 .expect("send_message should succeed")
488 .into_inner();
489 match resp.payload {
490 Some(apb::send_message_response::Payload::Task(t)) => t.id,
491 other => panic!("expected a Task payload, got {other:?}"),
492 }
493 }
494
495 #[tokio::test]
496 async fn send_message_returns_a_populated_payload() {
497 let svc = service();
498 let resp = svc
499 .send_message(Request::new(send_request("hello")))
500 .await
501 .expect("send_message should succeed")
502 .into_inner();
503
504 let payload = resp.payload.expect("a default response carries no payload");
505 match payload {
506 apb::send_message_response::Payload::Task(t) => {
507 assert!(!t.id.is_empty(), "the created task must carry an id");
508 }
509 apb::send_message_response::Payload::Message(m) => {
510 assert!(!m.message_id.is_empty(), "the reply must carry an id");
511 }
512 }
513 }
514
515 #[tokio::test]
516 async fn get_task_returns_the_task_that_was_created() {
517 let svc = service();
518 let id = seed_task(&svc).await;
519
520 let task = svc
521 .get_task(Request::new(apb::GetTaskRequest {
522 tenant: String::new(),
523 id: id.clone(),
524 history_length: None,
525 }))
526 .await
527 .expect("get_task should find the seeded task")
528 .into_inner();
529
530 assert_eq!(task.id, id, "the id round-trips through the binding");
531 }
532
533 #[tokio::test]
534 async fn get_task_maps_a_missing_task_to_not_found() {
535 let svc = service();
536 let status = svc
537 .get_task(Request::new(apb::GetTaskRequest {
538 tenant: String::new(),
539 id: "no-such-task".into(),
540 history_length: None,
541 }))
542 .await
543 .expect_err("an unknown id must not resolve");
544
545 assert_eq!(status.code(), tonic::Code::NotFound);
546 }
547
548 #[tokio::test]
549 async fn list_tasks_returns_the_seeded_task() {
550 let svc = service();
551 let id = seed_task(&svc).await;
552
553 let resp = svc
554 .list_tasks(Request::new(apb::ListTasksRequest::default()))
555 .await
556 .expect("list_tasks should succeed")
557 .into_inner();
558
559 assert!(
560 resp.tasks.iter().any(|t| t.id == id),
561 "the seeded task must appear in the listing, got {:?}",
562 resp.tasks.iter().map(|t| &t.id).collect::<Vec<_>>()
563 );
564 }
565
566 #[tokio::test]
567 async fn cancel_task_moves_the_task_out_of_a_running_state() {
568 let svc = service();
569 let id = seed_task(&svc).await;
570
571 let task = svc
572 .cancel_task(Request::new(apb::CancelTaskRequest {
573 tenant: String::new(),
574 id: id.clone(),
575 metadata: None,
576 }))
577 .await
578 .expect("cancel_task should succeed")
579 .into_inner();
580
581 assert_eq!(task.id, id);
582 let status = task.status.expect("a cancelled task carries a status");
583 assert_eq!(
584 status.state,
585 apb::TaskState::Canceled as i32,
586 "cancel must actually transition the task"
587 );
588 }
589
590 #[tokio::test]
591 async fn cancel_task_maps_a_missing_task_to_not_found() {
592 let svc = service();
593 let status = svc
594 .cancel_task(Request::new(apb::CancelTaskRequest {
595 tenant: String::new(),
596 id: "no-such-task".into(),
597 metadata: None,
598 }))
599 .await
600 .expect_err("an unknown id must not resolve");
601
602 assert_eq!(status.code(), tonic::Code::NotFound);
603 }
604
605 async fn create_push(
610 svc: &A2aServiceImpl,
611 task_id: &str,
612 url: &str,
613 ) -> apb::TaskPushNotificationConfig {
614 svc.create_task_push_notification_config(Request::new(apb::TaskPushNotificationConfig {
615 tenant: String::new(),
616 id: String::new(),
617 task_id: task_id.to_owned(),
618 url: url.to_owned(),
619 token: String::new(),
620 authentication: None,
621 }))
622 .await
623 .expect("create_task_push_notification_config should succeed")
624 .into_inner()
625 }
626
627 async fn list_push(
628 svc: &A2aServiceImpl,
629 task_id: &str,
630 ) -> Vec<apb::TaskPushNotificationConfig> {
631 svc.list_task_push_notification_configs(Request::new(
632 apb::ListTaskPushNotificationConfigsRequest {
633 tenant: String::new(),
634 task_id: task_id.to_owned(),
635 page_size: 0,
636 page_token: String::new(),
637 },
638 ))
639 .await
640 .expect("list_task_push_notification_configs should succeed")
641 .into_inner()
642 .configs
643 }
644
645 #[tokio::test]
646 async fn create_push_config_echoes_the_registered_url() {
647 let svc = service();
648 let task_id = seed_task(&svc).await;
649
650 let created = create_push(&svc, &task_id, "https://example.test/hook").await;
651
652 assert_eq!(created.url, "https://example.test/hook");
653 assert_eq!(created.task_id, task_id);
654 }
655
656 #[tokio::test]
657 async fn get_push_config_returns_what_create_stored() {
658 let svc = service();
659 let task_id = seed_task(&svc).await;
660 let created = create_push(&svc, &task_id, "https://example.test/get").await;
661
662 let fetched = svc
663 .get_task_push_notification_config(Request::new(
664 apb::GetTaskPushNotificationConfigRequest {
665 tenant: String::new(),
666 task_id: task_id.clone(),
667 id: created.id.clone(),
668 },
669 ))
670 .await
671 .expect("the config was just created")
672 .into_inner();
673
674 assert_eq!(fetched.id, created.id);
675 assert_eq!(fetched.url, "https://example.test/get");
676 }
677
678 #[tokio::test]
679 async fn list_push_configs_includes_the_created_one() {
680 let svc = service();
681 let task_id = seed_task(&svc).await;
682 let created = create_push(&svc, &task_id, "https://example.test/list").await;
683
684 let configs = list_push(&svc, &task_id).await;
685
686 assert!(
687 configs.iter().any(|c| c.id == created.id),
688 "the created config must appear in the listing"
689 );
690 }
691
692 #[tokio::test]
699 async fn delete_push_config_removes_it_from_the_listing() {
700 let svc = service();
701 let task_id = seed_task(&svc).await;
702 let created = create_push(&svc, &task_id, "https://example.test/delete").await;
703 assert_eq!(list_push(&svc, &task_id).await.len(), 1, "precondition");
704
705 svc.delete_task_push_notification_config(Request::new(
706 apb::DeleteTaskPushNotificationConfigRequest {
707 tenant: String::new(),
708 task_id: task_id.clone(),
709 id: created.id.clone(),
710 },
711 ))
712 .await
713 .expect("delete_task_push_notification_config should succeed");
714
715 assert!(
716 list_push(&svc, &task_id).await.is_empty(),
717 "delete must actually remove the config"
718 );
719 }
720
721 #[tokio::test]
731 async fn reader_to_native_stream_stops_after_a_conversion_error() {
732 use a2a_protocol_types::events::StreamResponse;
733 use tokio_stream::StreamExt;
734
735 use crate::streaming::event_queue::new_in_memory_queue;
736 use crate::streaming::EventQueueWriter;
737
738 fn message(metadata: serde_json::Value) -> Message {
739 Message {
740 id: MessageId("m".into()),
741 role: MessageRole::Agent,
742 parts: vec![Part {
743 content: PartContent::Text("hi".into()),
744 metadata: None,
745 filename: None,
746 media_type: None,
747 }],
748 task_id: Some(TaskId("t".into())),
749 context_id: Some(ContextId("c".into())),
750 reference_task_ids: None,
751 extensions: None,
752 metadata: Some(metadata),
753 }
754 }
755
756 let (writer, reader) = new_in_memory_queue();
757 writer
760 .write(StreamResponse::Message(message(serde_json::json!(
761 "not-an-object"
762 ))))
763 .await
764 .expect("write of the unconvertible event");
765 writer
768 .write(StreamResponse::Message(message(
769 serde_json::json!({"ok": true}),
770 )))
771 .await
772 .expect("write of the convertible event");
773 drop(writer);
774
775 let mut stream = reader_to_native_stream(reader, 8);
776
777 let first = stream.next().await.expect("the error item is delivered");
778 let status = first.expect_err("an unconvertible event surfaces as an error");
779 assert_eq!(status.code(), tonic::Code::Internal);
780
781 assert!(
782 stream.next().await.is_none(),
783 "the stream must end at the conversion error, not carry on"
784 );
785 }
786
787 #[tokio::test]
788 async fn get_extended_agent_card_returns_the_configured_card() {
789 let svc = service();
790 let card = svc
791 .get_extended_agent_card(Request::new(apb::GetExtendedAgentCardRequest {
792 tenant: String::new(),
793 }))
794 .await
795 .expect("the card is configured and unauthenticated access is opted in")
796 .into_inner();
797
798 assert_eq!(
799 card.name, "native-grpc-test-agent",
800 "a default card would carry an empty name"
801 );
802 }
803}