Class OpenAIModel

src/webwright/models/openai_model.py:111–157

class OpenAIModel(BaseModel):
    _API_KEY_FIELD = "openai_api_key"
    _ENV_VAR = "OPENAI_API_KEY"
    _LOG_SOURCE = "openai"
    _MAX_RATE_LIMIT_RETRIES = 5
    _MAX_TRANSIENT_RETRIES = 5
    _DEFAULT_CONFIG_CLASS = OpenAIModelConfig

    def _request_headers(self) -> dict[str, str]:
        return {
            "Content-Type": "application/json",
            "Authorization": f"Bearer {self.config.openai_api_key}",
        }

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

    def _build_payload(self, messages: list[dict[str, Any]]) -> dict[str, Any]:
        return {
            "model": self.config.model_name,
            "input": _serialize_response_input(messages),
            "max_output_tokens": self.config.max_output_tokens,
            "text": {
                "format": {
                    "type": "json_schema",
                    "name": "playwright_step",
                    "schema": self._response_schema(),
                    "strict": True,
                }
            },
        }

    def _build_text_payload(self, messages: list[dict[str, Any]]) -> dict[str, Any]:
        return {
            "model": self.config.model_name,
            "input": _serialize_response_input(messages),
            "max_output_tokens": self.config.max_output_tokens,
        }

    def _request_metrics_input(self, payload: dict[str, Any]) -> list[dict[str, Any]]:
        return payload.get("input") or []

    def _extract_text(self, payload: dict[str, Any]) -> str:
        return _extract_response_text(payload)

    def _usage_metrics_from_payload(self, payload: dict[str, Any]) -> dict[str, int]:
        return _usage_metrics_from_response_payload(payload)