Spaces:
Running
Running
| import os | |
| import sys | |
| sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))) | |
| from agentreview.agent import Player | |
| from agentreview.paper_review_player import PaperExtractorPlayer, AreaChair, Reviewer | |
| from agentreview.role_descriptions import get_ac_config, get_reviewer_player_config, get_author_config, \ | |
| get_paper_extractor_config | |
| def initialize_players(experiment_setting: dict, args): | |
| paper_id = experiment_setting['paper_id'] | |
| paper_decision = experiment_setting['paper_decision'] | |
| if args.task == "paper_decision": | |
| experiment_setting["players"] = {k: v for k, v in experiment_setting["players"].items() if k.startswith("AC")} | |
| players = [] | |
| for role, players_list in experiment_setting["players"].items(): | |
| for i, player_config in enumerate(players_list): | |
| if role == "AC": | |
| # For AC, `env_type` is either "paper_decision" or "paper_review" | |
| player_config = get_ac_config(env_type=args.task, | |
| scoring_method=args.ac_scoring_method, | |
| num_papers_per_area_chair=args.num_papers_per_area_chair, | |
| global_settings=experiment_setting['global_settings'], | |
| acceptance_rate=args.acceptance_rate, | |
| **player_config) | |
| player_config['model'] = args.model_name | |
| player = AreaChair(data_dir=args.data_dir, | |
| conference=args.conference, | |
| args=args, | |
| **player_config) | |
| elif args.task == "paper_review": | |
| if role == "Paper Extractor": | |
| player_config = get_paper_extractor_config(global_settings=experiment_setting['global_settings']) | |
| player = PaperExtractorPlayer(data_dir=args.data_dir, paper_id=paper_id, | |
| paper_decision=paper_decision, | |
| args=args, | |
| conference=args.conference, **player_config) | |
| elif role == "Author": | |
| # Author requires no behavior customization. | |
| # So we directly use the Player class | |
| player_config = get_author_config() | |
| player = Player(data_dir=args.data_dir, | |
| conference=args.conference, | |
| args=args, | |
| **player_config) | |
| elif role == "Reviewer": | |
| player_config = get_reviewer_player_config(reviewer_index=i + 1, | |
| global_settings=experiment_setting['global_settings'], | |
| **player_config) | |
| player_config['model'] = args.model_name | |
| player = Reviewer(data_dir=args.data_dir, conference=args.conference, args=args, **player_config) | |
| else: | |
| raise NotImplementedError(f"Unknown role for paper review (stage 1-4): {role}") | |
| else: | |
| raise NotImplementedError(f"Unknown role for paper decision (stage 5): {role}") | |
| players.append(player) | |
| return players | |