fully fix the bug

SylphAI-Inc · Dec 12, 2024 · 343fc63 · 343fc63
1 parent fbe3348
commit 343fc63
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/benchmarks/hotpot_qa/adal_exp/build_multi_hop_rag.py b/benchmarks/hotpot_qa/adal_exp/build_multi_hop_rag.py
@@ -326,7 +326,11 @@ def forward(self, *, input: str, id: str = None) -> adal.Parameter:
                 if x.full_response
                 and x.full_response.data
                 and x.full_response.data.query
-                else x.data.raw_response if x.data and x.data.raw_response else None
+                else (
+                    x.full_response.raw_response
+                    if x.full_response and x.full_response.raw_response
+                    else None
+                )
             )
             print(f"query {i}: {success_map_fn(gen_out)}")