From 343fc63f234803e8c1dddb9374cfe24a6d20f56e Mon Sep 17 00:00:00 2001 From: Li Yin Date: Thu, 12 Dec 2024 00:14:32 -0800 Subject: [PATCH] fully fix the bug --- benchmarks/hotpot_qa/adal_exp/build_multi_hop_rag.py | 6 +++++- 1 file changed, 5 insertions(+), 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 index 3ac38743..fed4d592 100644 --- 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)}")