Fix NVML import and sha2 dependency

chimera/agent/Cargo.toml

@@ -12,6 +12,7 @@ prost-types = "0.12"
 tracing = "0.1"
 tracing-subscriber = { version = "0.3", features = ["fmt", "env-filter"] }
 nvml-wrapper = "0.9"
+sha2 = "0.10"
 
 [build-dependencies]
 prost-build = "0.12"

chimera/agent/build.rs

@@ -6,8 +6,8 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
         .build_server(true)
         .build_client(true)
         // Specify the proto file
-        .compile(&["/app/proto/telemetry.proto"], &["/app/proto"])?;
+        .compile(&["../proto/telemetry.proto"], &["../proto"])?;
 
-    println!("cargo:rerun-if-changed=/app/proto/telemetry.proto");
+    println!("cargo:rerun-if-changed=../proto/telemetry.proto");
     Ok(())
 }

chimera/agent/src/grpc/mod.rs

@@ -1,5 +1,4 @@
 use tonic::{transport::Server, Request, Response, Status};
-use prost::Message;
 use crate::hash::trusted_hash;
 use crate::nvml::{collect_metrics, TelemetryData};
 use tokio::time::{sleep, Duration};

chimera/agent/src/main.rs

@@ -1,6 +1,6 @@
 #![forbid(unsafe_code)]
 
-use tracing::{info, Instrument};
+use tracing::Instrument;
 
 mod hash;
 mod nvml;

chimera/agent/src/nvml/mod.rs

@@ -1,4 +1,4 @@
-use nvml_wrapper::NVML;
+use nvml_wrapper::Nvml;
 use std::error::Error;
 
 #[derive(Default, Clone)]
@@ -10,7 +10,7 @@ pub struct TelemetryData {
 }
 
 pub fn collect_metrics() -> Result<TelemetryData, Box<dyn Error>> {
-    let nvml = NVML::init()?;
+    let nvml = Nvml::init()?;
     let device = nvml.device_by_index(0)?;
     let memory = device.memory_info()?;
     let temperature = device.temperature(0)?;