inference_gateway_adk/
lib.rs

1pub mod a2a_types;
2pub mod client;
3pub mod config;
4pub mod server;
5
6pub use client::{A2AClient, HealthStatus};
7pub use config::{AgentConfig, ClientConfig, Config};
8pub use server::{A2AServer, A2AServerBuilder, Agent, AgentBuilder, AgentCardOverrides};
9
10#[cfg(test)]
11mod tests {
12    #[test]
13    fn test_a2a_types_module_exists() {
14        use crate::a2a_types::JsonrpcMessage;
15        let _type_exists = std::mem::size_of::<JsonrpcMessage>();
16    }
17
18    #[test]
19    fn test_a2a_types_serialization() {
20        use crate::a2a_types::*;
21
22        let message = JsonrpcMessage {
23            jsonrpc: "2.0".to_string(),
24            id: Some(JsonrpcMessageId::String("test-id".to_string())),
25        };
26
27        let serialized = serde_json::to_string(&message).expect("Should serialize");
28        assert!(serialized.contains("\"jsonrpc\":\"2.0\""));
29        assert!(serialized.contains("\"id\":\"test-id\""));
30
31        let _deserialized: JsonrpcMessage =
32            serde_json::from_str(&serialized).expect("Should deserialize");
33    }
34}