| 46 | END_SEARCH_RESULT = "<|end_search_result|>" |
| 47 | |
| 48 | def parse_args(): |
| 49 | parser = argparse.ArgumentParser(description="Run Search-o1 for various datasets and models.") |
| 50 | |
| 51 | # Dataset and split configuration |
| 52 | parser.add_argument( |
| 53 | '--dataset_name', |
| 54 | type=str, |
| 55 | required=True, |
| 56 | help="Name of the dataset to use." |
| 57 | ) |
| 58 | |
| 59 | parser.add_argument( |
| 60 | '--split', |
| 61 | type=str, |
| 62 | required=True, |
| 63 | help="Dataset split to use." |
| 64 | ) |
| 65 | |
| 66 | parser.add_argument( |
| 67 | '--subset_num', |
| 68 | type=int, |
| 69 | default=-1, |
| 70 | help="Number of examples to process. Defaults to all if not specified." |
| 71 | ) |
| 72 | |
| 73 | # Search and document retrieval configuration |
| 74 | parser.add_argument( |
| 75 | '--max_search_limit', |
| 76 | type=int, |
| 77 | default=10, |
| 78 | help="Maximum number of searches per question." |
| 79 | ) |
| 80 | |
| 81 | parser.add_argument( |
| 82 | '--max_turn', |
| 83 | type=int, |
| 84 | default=15, |
| 85 | help="Maximum number of turns." |
| 86 | ) |
| 87 | |
| 88 | parser.add_argument( |
| 89 | '--top_k', |
| 90 | type=int, |
| 91 | default=10, |
| 92 | help="Maximum number of search documents to return." |
| 93 | ) |
| 94 | |
| 95 | parser.add_argument( |
| 96 | '--max_doc_len', |
| 97 | type=int, |
| 98 | default=3000, |
| 99 | help="Maximum length of each searched document." |
| 100 | ) |
| 101 | |
| 102 | parser.add_argument( |
| 103 | '--use_jina', |
| 104 | type=bool, |
| 105 | default=False, |