MCPcopy Create free account
hub / github.com/OpenBMB/ToolBench / run

Method run

toolbench/inference/Downstream_tasks/rapidapi.py:531–551  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

529 return result
530
531 def run(self):
532 task_list = self.task_list
533 random.seed(42)
534 random.shuffle(task_list)
535 print(f"total tasks: {len(task_list)}")
536 new_task_list = []
537 for task in task_list:
538 out_dir_path = task[-2]
539 query_id = task[2]
540 output_file_path = os.path.join(out_dir_path,f"{query_id}_{self.args.method}.json")
541 if not os.path.exists(output_file_path):
542 new_task_list.append(task)
543 task_list = new_task_list
544 print(f"undo tasks: {len(task_list)}")
545 if self.add_retrieval:
546 retriever = self.get_retriever()
547 else:
548 retriever = None
549 for k, task in enumerate(task_list):
550 print(f"process[{self.process_id}] doing task {k}/{len(task_list)}: real_task_id_{task[2]}")
551 result = self.run_single_task(*task, retriever=retriever, process_id=self.process_id)
552

Callers 3

qa_pipeline.pyFile · 0.45

Calls 2

get_retrieverMethod · 0.95
run_single_taskMethod · 0.95

Tested by

no test coverage detected