fix(query): Remove verbose debug and skip self in instrumentation

bosun-ai · Sep 16, 2024 · 133cf1d · 133cf1d
1 parent 39b6ecb
commit 133cf1d
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 12 deletions.
diff --git a/swiftide-query/src/answers/simple.rs b/swiftide-query/src/answers/simple.rs
@@ -82,14 +82,6 @@ impl Answer for Simple {
             query.current()
         };
 
-        let prompt = self
-            .prompt_template
-            .to_prompt()
-            .with_context_value("question", query.original())
-            .with_context_value("context", context);
-
-        tracing::debug!(prompt = ?prompt, "Prompting from Simple for answer");
-
         let answer = self
             .client
             .prompt(

diff --git a/swiftide-query/src/query_transformers/embed.rs b/swiftide-query/src/query_transformers/embed.rs
@@ -21,7 +21,7 @@ impl Embed {
 
 #[async_trait]
 impl TransformQuery for Embed {
-    #[tracing::instrument]
+    #[tracing::instrument(skip_self)]
     async fn transform_query(
         &self,
         mut query: Query<states::Pending>,

diff --git a/swiftide-query/src/query_transformers/generate_subquestions.rs b/swiftide-query/src/query_transformers/generate_subquestions.rs
@@ -71,7 +71,7 @@ fn default_prompt() -> PromptTemplate {
 
 #[async_trait]
 impl TransformQuery for GenerateSubquestions {
-    #[tracing::instrument]
+    #[tracing::instrument(skip_self)]
     async fn transform_query(
         &self,
         mut query: Query<states::Pending>,

diff --git a/swiftide-query/src/query_transformers/sparse_embed.rs b/swiftide-query/src/query_transformers/sparse_embed.rs
@@ -22,7 +22,7 @@ impl SparseEmbed {
 
 #[async_trait]
 impl TransformQuery for SparseEmbed {
-    #[tracing::instrument]
+    #[tracing::instrument(skip_self)]
     async fn transform_query(
         &self,
         mut query: Query<states::Pending>,

diff --git a/swiftide-query/src/response_transformers/summary.rs b/swiftide-query/src/response_transformers/summary.rs
@@ -70,7 +70,7 @@ fn default_prompt() -> PromptTemplate {
 
 #[async_trait]
 impl TransformResponse for Summary {
-    #[tracing::instrument]
+    #[tracing::instrument(skip_self)]
     async fn transform_response(
         &self,
         mut query: Query<states::Retrieved>,