Method to_jsonl_line

scrapegraphai/utils/batch_api.py:50–67

Convert to a JSONL line for the Batch API input file.

(self)

    """Optional response format (e.g., {"type": "json_object"})."""

    def to_jsonl_line(self) -> str:
        """Convert to a JSONL line for the Batch API input file."""
        body = {
            "model": self.model,
            "messages": self.messages,
            "temperature": self.temperature,
        }
        if self.max_tokens is not None:
            body["max_tokens"] = self.max_tokens
        if self.response_format is not None:
            body["response_format"] = self.response_format

        return json.dumps({
            "custom_id": self.custom_id,
            "method": "POST",
            "url": "/v1/chat/completions",
            "body": body,
        })


@dataclass