testnet points api

Author: erhant

compute/src/node/mod.rs

@@ -122,7 +122,7 @@ impl DriaComputeNode {
             };
 
         let model_names = config.executors.get_model_names();
-        let points_client = DriaPointsClient::new(&config.address)?;
+        let points_client = DriaPointsClient::new(&config.address, &config.network)?;
 
         let spec_collector = SpecCollector::new(model_names.clone(), config.version);
         Ok((

compute/src/utils/points.rs

@@ -1,7 +1,6 @@
+use dkn_utils::DriaNetwork;
 use eyre::Context;
 
-const POINTS_API_BASE_URL: &str = "https://mainnet.dkn.dria.co/points/v0/total/node/";
-
 pub struct DriaPointsClient {
     pub url: String,
     client: reqwest::Client,
@@ -18,13 +17,21 @@ pub struct DriaPoints {
 }
 
 impl DriaPointsClient {
+    /// The base URL for the points API, w.r.t network.
+    pub fn base_url(network: &DriaNetwork) -> &'static str {
+        match network {
+            DriaNetwork::Mainnet => "https://mainnet.dkn.dria.co/points/v0/total/node/",
+            DriaNetwork::Testnet => "https://testnet.dkn.dria.co/points/v0/total/node/",
+        }
+    }
+
     /// Creates a new `DriaPointsClient` for the given address.
-    pub fn new(address: &str) -> eyre::Result<Self> {
+    pub fn new(address: &str, network: &DriaNetwork) -> eyre::Result<Self> {
         const USER_AGENT: &str = concat!(env!("CARGO_PKG_NAME"), "/", env!("CARGO_PKG_VERSION"));
 
         let url = format!(
             "{}/0x{}",
-            POINTS_API_BASE_URL,
+            Self::base_url(network),
             address.trim_start_matches("0x")
         );
 
@@ -66,7 +73,11 @@ mod tests {
 
     #[tokio::test]
     async fn test_get_points() {
-        let client = DriaPointsClient::new("0xa43536a6032a3907ccf60e8109429ee1047b207c").unwrap();
+        let client = DriaPointsClient::new(
+            "0xa43536a6032a3907ccf60e8109429ee1047b207c",
+            &DriaNetwork::Mainnet,
+        )
+        .unwrap();
         let steps = client.get_points().await.unwrap();
         assert!(steps.score >= 0.0);
         assert!(steps.percentile <= 100);