(api_name, output_path)
| 300 | pbar.update(1) |
| 301 | |
| 302 | def main(api_name, output_path): |
| 303 | ds = load_dataset("callanwu/WebWalkerQA", split="main") |
| 304 | api_functions = { |
| 305 | "o1_api": o1_api, |
| 306 | "gemini_api": gemini_api, |
| 307 | "gemini_search_api": gemini_api, |
| 308 | "doubao_api": doubao_api, |
| 309 | "kimi_api": kimi_api, |
| 310 | "wenxin_api": wenxin_api, |
| 311 | } |
| 312 | if api_name == "all": |
| 313 | for api in api_functions: |
| 314 | print(api) |
| 315 | print(output_path + "/" + api+"_result.jsonl") |
| 316 | os.makedirs(output_path, exist_ok=True) |
| 317 | if api != "gemini_search_api": |
| 318 | api_functions[api](ds, output_path + "/" + api+"_result.jsonl") |
| 319 | else: |
| 320 | api_functions[api](ds, output_path + "/" + api+"_result.jsonl", search=True) |
| 321 | else: |
| 322 | if api_name in api_functions: |
| 323 | if api_name == "gemini_search_api": |
| 324 | gemini_api(ds, output_path, search=True) |
| 325 | asyncio.run(api_functions[api_name](ds, output_path)) |
| 326 | else: |
| 327 | print(f"API {api_name} is not supported.") |
| 328 | |
| 329 | if __name__ == "__main__": |
| 330 | import argparse |
no test coverage detected