ray-llm

1
from typing import Optional
2

3
from rayllm.backend.server.models import AviaryModelResponse
4

5

6
class OpenAIHTTPException(Exception):
7
    def __init__(
8
        self,
9
        status_code: int,
10
        message: str,
11
        type: str = "Unknown",
12
        internal_message: Optional[str] = None,
13
    ) -> None:
14
        self.status_code = status_code
15
        self.message = message
16
        self.type = type
17
        self.internal_message = internal_message
18

19
    @classmethod
20
    def from_model_response(
21
        cls, response: AviaryModelResponse
22
    ) -> "OpenAIHTTPException":
23
        return cls(
24
            status_code=response.error.code,
25
            message=response.error.message,
26
            type=response.error.type,
27
            internal_message=response.error.internal_message,
28
        )
29