Add counter telemetry(WIP)

Signed-off-by: Rohit Dandamudi <[email protected]>

Author: me-diru

Cargo.lock

@@ -23,6 +23,7 @@ spin-llm-remote-http = { path = "../llm-remote-http" }
 spin-locked-app = { path = "../locked-app" }
 spin-world = { path = "../world" }
 tracing = { workspace = true }
+spin-telemetry = { path = "../telemetry" }
 tokio = { version = "1", features = ["sync"] }
 toml = "0.8"
 url = { version = "2", features = ["serde"] }

crates/factor-llm/Cargo.toml

@@ -68,6 +68,7 @@ impl LlmEngine for RemoteHttpLlmEngine {
         prompt: String,
         params: v2::InferencingParams,
     ) -> Result<v2::InferencingResult, v2::Error> {
+        spin_telemetry::monotonic_counter!(spin.llm_infer = 1, model_name = model);
         self.infer(model, prompt, params).await
     }
 

crates/factor-llm/src/spin.rs

@@ -13,6 +13,7 @@ tokio = { version = "1", features = ["macros", "sync", "rt"] }
 spin-app = { path = "../app" }
 spin-core = { path = "../core" }
 spin-world = { path = "../world" }
+spin-telemetry = { path = "../telemetry" }
 table = { path = "../table" }
 tracing = { workspace = true }
 lru = "0.9.0"

crates/key-value/Cargo.toml

@@ -98,6 +98,8 @@ impl key_value::HostStore for KeyValueDispatch {
         store: Resource<key_value::Store>,
         key: String,
     ) -> Result<Result<Option<Vec<u8>>, Error>> {
+        spin_telemetry::monotonic_counter!(spin.key_value_get = 1, key = key);
+
         let store = self.get_store(store)?;
         Ok(store.get(&key).await)
     }
@@ -108,6 +110,8 @@ impl key_value::HostStore for KeyValueDispatch {
         key: String,
         value: Vec<u8>,
     ) -> Result<Result<(), Error>> {
+        spin_telemetry::monotonic_counter!(spin.key_value_set = 1, key = key);
+
         let store = self.get_store(store)?;
         Ok(store.set(&key, &value).await)
     }