Add counter telemetry(WIP)

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

Author: me-diru

Cargo.lock

@@ -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,9 @@ impl key_value::HostStore for KeyValueDispatch {
         store: Resource<key_value::Store>,
         key: String,
     ) -> Result<Result<Option<Vec<u8>>, Error>> {
+        // Log key value host component get feature
+        spin_telemetry::counter!(spin.key_value_get = 1, key = key);
+
         let store = self.get_store(store)?;
         Ok(store.get(&key).await)
     }
@@ -108,6 +111,9 @@ impl key_value::HostStore for KeyValueDispatch {
         key: String,
         value: Vec<u8>,
     ) -> Result<Result<(), Error>> {
+        // Log key value host component set feature
+        spin_telemetry::counter!(spin.key_value_set = 1, key = key);
+
         let store = self.get_store(store)?;
         Ok(store.set(&key, &value).await)
     }

crates/key-value/src/lib.rs

@@ -70,6 +70,13 @@ impl RemoteHttpLlmEngine {
         );
         spin_telemetry::inject_trace_context(&mut headers);
 
+        // capturing llm host component metrics
+        spin_telemetry::counter!(
+            spin.llm_infer = 1,
+            model_name = model,
+            prompt_given = prompt
+        );
+
         let inference_options = InferRequestBodyParams {
             max_tokens: params.max_tokens,
             repeat_penalty: params.repeat_penalty,