Fix bug

llamaindex_utils.py

@@ -350,8 +350,7 @@ async def asynthesize_results(queries: List[SubQuestion], contexts: Dict[str, Se
     for idx, response in enumerate(responses):
         # Parse json string to dictionary
         json_dict = parse_json_markdown(response)
-        
-        if len(json_dict["choices"]) > 1:
+        if len(json_dict["choices"]) > 0:
             for choice in json_dict["choices"]:
                 new_contexts[json_dict["summarized_text"]] = new_contexts[json_dict["summarized_text"]].union(contexts[contexts_batches[idx][choice - 1]])
         else:
@@ -527,14 +526,13 @@ class CustomQueryEngine:
             
             for task, image in answers:
                 response2images_mapping[str(task.result())].add(image)
-            
+                
             # Synthesize results
             synthesized_text, source_images = await asynthesize_results(queries=sub_queries,
                                                         contexts=response2images_mapping,
                                                         llm=self._llm,
                                                         num_children=self._num_children)
             
-            
             final_answer = await self._llm.apredict(self._qa_prompt,
                                             context_str=synthesized_text,
                                             query_str=query_str)