feat(llm): Update tracing of llm host components

Signed-off-by: Caleb Schoepp <[email protected]>

Author: calebschoepp

crates/llm-local/Cargo.toml

@@ -31,3 +31,6 @@ tracing = { workspace = true }
 default = []
 metal = ["llm/metal"]
 cublas = ["llm/cublas"]
+
+[lints]
+workspace = true

crates/llm-local/src/lib.rs

@@ -33,7 +33,7 @@ pub struct LocalLlmEngine {
 
 #[async_trait]
 impl LlmEngine for LocalLlmEngine {
-    #[instrument(name = "generate_inference_local_llm", skip(self, prompt), err(level = Level::INFO))]
+    #[instrument(name = "spin_llm_local.infer", skip(self, prompt), err(level = Level::INFO))]
     async fn infer(
         &mut self,
         model: wasi_llm::InferencingModel,
@@ -93,7 +93,7 @@ impl LlmEngine for LocalLlmEngine {
         Ok(response)
     }
 
-    #[instrument(name = "generate_embeddings_local_llm", skip(self, data), err(level = Level::INFO))]
+    #[instrument(name = "spin_llm_local.generate_embeddings", skip(self, data), err(level = Level::INFO))]
     async fn generate_embeddings(
         &mut self,
         model: wasi_llm::EmbeddingModel,

crates/llm-remote-http/Cargo.toml

@@ -16,3 +16,6 @@ spin-telemetry = { path = "../telemetry" }
 spin-world = { path = "../world" }
 reqwest = { version = "0.11", features = ["gzip", "json"] }
 tracing = { workspace = true }
+
+[lints]
+workspace = true

crates/llm-remote-http/src/lib.rs

@@ -55,7 +55,7 @@ struct EmbeddingResponseBody {
 
 #[async_trait]
 impl LlmEngine for RemoteHttpLlmEngine {
-    #[instrument(name = "generate_inference_remote_llm", skip(self, prompt), err(level = Level::INFO), fields(otel.kind = "client"))]
+    #[instrument(name = "spin_llm_remote_http.infer", skip(self, prompt), err(level = Level::INFO), fields(otel.kind = "client"))]
     async fn infer(
         &mut self,
         model: wasi_llm::InferencingModel,
@@ -118,7 +118,7 @@ impl LlmEngine for RemoteHttpLlmEngine {
         }
     }
 
-    #[instrument(name = "generate_embeddings_remote_llm", skip(self, data), err(level = Level::INFO), fields(otel.kind = "client"))]
+    #[instrument(name = "spin_llm_remote_http.generate_embeddings", skip(self, data), err(level = Level::INFO), fields(otel.kind = "client"))]
     async fn generate_embeddings(
         &mut self,
         model: wasi_llm::EmbeddingModel,