Feature/fix hyde pipeline (#344)

* filter logs * cleanup * fix hyde pipeline
SciPhi-AI · May 1, 2024 · 2b73935 · 2b73935
1 parent c79a0b4
commit 2b73935
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/r2r/pipelines/hyde/rag.py b/r2r/pipelines/hyde/rag.py
@@ -164,7 +164,7 @@ def run(
                 {"answers": answers},
             )
 
-        context = self._construct_joined_context(search_results_tuple, message)
+        context = self._construct_joined_context(search_results_tuple)
         prompt = self.construct_prompt({"query": message, "context": context})
 
         if not generation_config.stream:
@@ -222,7 +222,7 @@ def run_stream(
         ]
 
         return self._return_stream(
-            search_results, context, prompt, generation_config
+            search_results, context, prompt, generation_config, metadata={"answers": answers},
         )
 
     def _construct_joined_context(