Class OpenRouterModel

src/webwright/models/openrouter_model.py:141–206

class OpenRouterModel(BaseModel):
    _API_KEY_FIELD = "openrouter_api_key"
    _ENV_VAR = "OPENROUTER_API_KEY"
    _LOG_SOURCE = "openrouter"
    _MAX_RATE_LIMIT_RETRIES = 5
    _MAX_TRANSIENT_RETRIES = 5
    _DEFAULT_CONFIG_CLASS = OpenRouterModelConfig

    def _request_headers(self) -> dict[str, str]:
        headers = {
            "Content-Type": "application/json",
            "Authorization": f"Bearer {self.config.openrouter_api_key}",
        }
        if _is_openrouter_endpoint(self.config.openrouter_endpoint) and self.config.http_referer:
            headers["HTTP-Referer"] = self.config.http_referer
        if _is_openrouter_endpoint(self.config.openrouter_endpoint) and self.config.app_title:
            headers["X-Title"] = self.config.app_title
        return headers

    def _post_url(self) -> str:
        return self.config.openrouter_endpoint

    def _build_payload(self, messages: list[dict[str, Any]]) -> dict[str, Any]:
        payload: dict[str, Any] = {
            "model": self.config.model_name,
            "messages": _serialize_chat_messages(messages),
            "stream": False,
            "response_format": {
                "type": "json_schema",
                "json_schema": {
                    "name": "playwright_step",
                    "strict": True,
                    "schema": self._response_schema(),
                },
            },
        }
        if _is_openai_endpoint(self.config.openrouter_endpoint) and self.config.model_name.startswith("gpt-5"):
            payload["max_completion_tokens"] = self.config.max_output_tokens
        else:
            payload["max_tokens"] = self.config.max_output_tokens
        if self.config.provider_require_parameters and _is_openrouter_endpoint(self.config.openrouter_endpoint):
            payload["provider"] = {"require_parameters": True}
        return payload

    def _build_text_payload(self, messages: list[dict[str, Any]]) -> dict[str, Any]:
        payload: dict[str, Any] = {
            "model": self.config.model_name,
            "messages": _serialize_chat_messages(messages),
            "stream": False,
        }
        if _is_openai_endpoint(self.config.openrouter_endpoint) and self.config.model_name.startswith("gpt-5"):
            payload["max_completion_tokens"] = self.config.max_output_tokens
        else:
            payload["max_tokens"] = self.config.max_output_tokens
        if self.config.provider_require_parameters and _is_openrouter_endpoint(self.config.openrouter_endpoint):
            payload["provider"] = {"require_parameters": True}
        return payload