chore: update inferadb-common-storage-ledger config

Author: evansims

crates/api/tests/ledger_integration.rs

@@ -39,7 +39,9 @@ use axum::{
     middleware::{self, Next},
     response::Response,
 };
-use inferadb_common_storage_ledger::{LedgerBackend, LedgerBackendConfig};
+use inferadb_common_storage_ledger::{
+    ClientConfig, LedgerBackend, LedgerBackendConfig, ServerSource,
+};
 use inferadb_engine_api::AppState;
 use inferadb_engine_config::Config;
 use inferadb_engine_core::ipl::{RelationDef, RelationExpr, Schema, TypeDef};
@@ -110,13 +112,16 @@ fn create_test_config() -> Config {
 
 async fn create_ledger_backend() -> LedgerBackend {
     let vault_id = unique_vault_id();
-    let config = LedgerBackendConfig::builder()
-        .endpoints(vec![ledger_endpoint()])
+    let client_config = ClientConfig::builder()
+        .servers(ServerSource::from_static([ledger_endpoint()]))
         .client_id(format!("engine-test-{}", vault_id))
+        .build()
+        .expect("valid client config");
+    let config = LedgerBackendConfig::builder()
+        .client(client_config)
         .namespace_id(ledger_namespace_id())
         .vault_id(vault_id)
-        .build()
-        .expect("valid config");
+        .build();
 
     LedgerBackend::new(config).await.expect("backend creation should succeed")
 }

crates/auth/tests/signing_key_cache_ledger_integration.rs

@@ -20,7 +20,7 @@ use chrono::Utc;
 use ed25519_dalek::SigningKey;
 use inferadb_common_storage::auth::{PublicSigningKey, PublicSigningKeyStore};
 use inferadb_common_storage_ledger::{
-    LedgerBackend, LedgerBackendConfig, auth::LedgerSigningKeyStore,
+    ClientConfig, LedgerBackend, LedgerBackendConfig, ServerSource, auth::LedgerSigningKeyStore,
 };
 use inferadb_engine_auth::SigningKeyCache;
 use rand_core::OsRng;
@@ -74,12 +74,15 @@ fn create_test_key(kid: &str, client_id: i64) -> PublicSigningKey {
 
 /// Create a `LedgerSigningKeyStore` for testing.
 async fn create_key_store() -> Arc<dyn PublicSigningKeyStore> {
-    let config = LedgerBackendConfig::builder()
-        .endpoints(vec![ledger_endpoint()])
+    let client_config = ClientConfig::builder()
+        .servers(ServerSource::from_static([ledger_endpoint()]))
         .client_id(format!("test-cache-{}", unique_namespace_id()))
-        .namespace_id(1) // Namespace is specified per-operation, not here
         .build()
-        .expect("valid config");
+        .expect("valid client config");
+    let config = LedgerBackendConfig::builder()
+        .client(client_config)
+        .namespace_id(1) // Namespace is specified per-operation, not here
+        .build();
 
     let backend = LedgerBackend::new(config).await.expect("backend creation should succeed");
 

crates/engine/src/main.rs

@@ -7,7 +7,9 @@ use std::sync::Arc;
 use anyhow::Result;
 use clap::Parser;
 use inferadb_common_storage::MemoryBackend;
-use inferadb_common_storage_ledger::{LedgerBackend, LedgerBackendConfig};
+use inferadb_common_storage_ledger::{
+    ClientConfig, LedgerBackend, LedgerBackendConfig, ServerSource,
+};
 use inferadb_engine_auth::SigningKeyCache;
 use inferadb_engine_config::load_or_default;
 use inferadb_engine_core::ipl::{Schema, parse_schema};
@@ -158,13 +160,16 @@ async fn main() -> Result<()> {
                 .ledger
                 .namespace_id
                 .ok_or_else(|| anyhow::anyhow!("Ledger namespace_id is required"))?;
-            let ledger_config = LedgerBackendConfig::builder()
-                .endpoints(vec![endpoint])
+            let client_config = ClientConfig::builder()
+                .servers(ServerSource::from_static([endpoint]))
                 .client_id(client_id)
+                .build()
+                .map_err(|e| anyhow::anyhow!("Failed to build Ledger client config: {}", e))?;
+            let ledger_config = LedgerBackendConfig::builder()
+                .client(client_config)
                 .namespace_id(namespace_id)
                 .maybe_vault_id(config.ledger.vault_id)
-                .build()
-                .map_err(|e| anyhow::anyhow!("Failed to build Ledger config: {}", e))?;
+                .build();
             let ledger_backend = LedgerBackend::new(ledger_config)
                 .await
                 .map_err(|e| anyhow::anyhow!("Failed to connect to Ledger: {}", e))?;

crates/store/src/factory.rs

@@ -16,13 +16,18 @@
 //! let store: Arc<dyn InferaStore> = Arc::new(EngineStorage::builder().backend(MemoryBackend::new()).build());
 //!
 //! // Ledger backend
-//! use inferadb_common_storage_ledger::{LedgerBackend, LedgerBackendConfig};
+//! use inferadb_common_storage_ledger::{
+//!     ClientConfig, LedgerBackend, LedgerBackendConfig, ServerSource,
+//! };
 //!
-//! let config = LedgerBackendConfig::builder()
-//!     .endpoints(vec!["http://localhost:50051".to_string()])
+//! let client = ClientConfig::builder()
+//!     .servers(ServerSource::from_static(["http://localhost:50051"]))
 //!     .client_id("my-service")
-//!     .namespace_id(1)
 //!     .build()?;
+//! let config = LedgerBackendConfig::builder()
+//!     .client(client)
+//!     .namespace_id(1)
+//!     .build();
 //! let backend = LedgerBackend::new(config).await?;
 //! let store: Arc<dyn InferaStore> = Arc::new(EngineStorage::builder().backend(backend).build());
 //! ```