Method process_item

eval/src/vllm_request.py:100–165

(self, item, image_root)

        ]

    def process_item(self, item, image_root):
        attempt = 0
        result = None
        start_time = time.monotonic()

        while attempt < self.max_retries:
            try:
                attempt += 1
                session = self._get_session()
                raw_messages = self.build_messages(item, image_root)

                payload = {
                    "model": "QwenVL",
                    "messages": raw_messages,
                    "do_sample": False,
                    "max_tokens": 4096,
                }

                response = session.post(
                    f"{self.api_url}/v1/chat/completions",
                    json=payload,
                    timeout=self.timeout_base + attempt * 5,
                )
                if response.status_code in {429, 500, 502, 503, 504}:
                    raise requests.HTTPError(
                        f"Retryable HTTP {response.status_code}",
                        response=response,
                    )
                response.raise_for_status()

                output = response.json()["choices"][0]["message"]["content"]

                item["model_output"] = output
                item["success"] = True
                result = item

                break

            except Exception as e:
                if attempt == self.max_retries:
                    print(f"请求失败（已达最大重试次数）: {str(e)}")
                    result = {
                        "idx": item.get("idx"),
                        "question": item["problem"],
                        "image_path": item.get("images", []),
                        "error": str(e),
                        "attempt": attempt,
                        "success": False
                    }
                else:
                    sleep_time = min(
                        self.backoff_base ** attempt + random.uniform(0, 1),
                        self.backoff_cap,
                    )
                    time.sleep(sleep_time)
        if result is None:
            result = {
                "idx": item.get("idx"),