| 898 | return await self.step() |
| 899 | |
| 900 | async def step(self) -> Optional[Task]: |
| 901 | if self._index >= len(self._data_records): |
| 902 | return None |
| 903 | |
| 904 | record = self._data_records[self._index] |
| 905 | self._index += 1 |
| 906 | |
| 907 | task_id = str(record.get("id") or record.get("task_id", "")) |
| 908 | task_name = record.get("name") or record.get("problem_name") or "Unknown" |
| 909 | |
| 910 | # templates are in record |
| 911 | templates = record.get("code_template", {}) |
| 912 | # Map our language to potential keys in the template dict |
| 913 | lang_key = self.language.lower() |
| 914 | lang_config = self.LANGUAGE_CONFIG.get(lang_key, {}) |
| 915 | lang_tag = lang_config.get("lang_tag", lang_key) |
| 916 | template = templates.get(lang_tag) |
| 917 | |
| 918 | input_text = f""" |
| 919 | TASK ID: {task_id} |
| 920 | Problem Name: {task_name} |
| 921 | Problem: {record.get("question") or record.get("prompt") or "Unknown"} |
| 922 | Template: |
| 923 | ```{lang_key} |
| 924 | # |
| 925 | # @lc app=leetcode id={task_id} lang={lang_tag} |
| 926 | # |
| 927 | # [{task_id}] {task_name} |
| 928 | # |
| 929 | # @lc code=start |
| 930 | {template} |
| 931 | # @lc code=end |
| 932 | ``` |
| 933 | """ |
| 934 | file_ext = self.LANGUAGE_CONFIG[self.language]["ext"] |
| 935 | file_name = f"{task_id}.{inflection.parameterize(task_name)}" |
| 936 | |
| 937 | return Task( |
| 938 | task_id=task_id, |
| 939 | input=input_text, |
| 940 | system_prompt=self.system_prompt, |
| 941 | ground_truth=record.get("true_answer") or record.get("answer"), |
| 942 | extra={ |
| 943 | "file_name": file_name, |
| 944 | "file_ext": file_ext, |
| 945 | "task_start_time": time.time() # <--- 新增：记录任务开始时间 |
| 946 | } |
| 947 | ) |
| 948 | |
| 949 | def get_queue_size(self) -> int: |
| 950 | """获取当前队列中的任务数量""" |