z
File size: 3,194 Bytes
712a3c7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""API 层异常类。



用于 API 端点层的错误,如认证、授权、验证等。

"""

from typing import Any, Dict, Optional

from app.exceptions.base import AppError


class AuthenticationError(AppError):
    """认证失败异常。



    当用户未提供认证信息或认证信息无效时抛出。



    Example:

        >>> raise AuthenticationError("Missing Authorization header")

        >>> raise AuthenticationError("Invalid API key")

    """

    def __init__(

        self,

        message: str = "Authentication failed",

        details: Optional[Dict[str, Any]] = None,

    ) -> None:
        super().__init__(
            message=message,
            code="authentication_error",
            status_code=401,
            details=details,
        )


class AuthorizationError(AppError):
    """授权失败异常。



    当用户已认证但权限不足时抛出。



    Example:

        >>> raise AuthorizationError("Access denied to this resource")

    """

    def __init__(

        self,

        message: str = "Permission denied",

        details: Optional[Dict[str, Any]] = None,

    ) -> None:
        super().__init__(
            message=message,
            code="authorization_error",
            status_code=403,
            details=details,
        )


class ValidationError(AppError):
    """请求验证失败异常。



    当请求数据不符合预期格式或约束时抛出。



    Example:

        >>> raise ValidationError("Invalid email format", field="email")

        >>> raise ValidationError("messages cannot be empty")

    """

    def __init__(

        self,

        message: str,

        field: Optional[str] = None,

        details: Optional[Dict[str, Any]] = None,

    ) -> None:
        final_details = details or {}
        if field:
            final_details["field"] = field
        super().__init__(
            message=message,
            code="validation_error",
            status_code=400,
            details=final_details,
        )


class ModelNotFoundError(AppError):
    """模型不存在异常。



    当请求的模型不在支持列表中时抛出。



    Example:

        >>> raise ModelNotFoundError("gpt-5")

    """

    def __init__(self, model: str) -> None:
        super().__init__(
            message=f"Model '{model}' not found",
            code="model_not_found",
            status_code=404,
            details={"model": model},
        )


class RateLimitError(AppError):
    """请求频率限制异常。



    当客户端请求过于频繁时抛出。



    Example:

        >>> raise RateLimitError(retry_after=60)

    """

    def __init__(

        self,

        message: str = "Rate limit exceeded",

        retry_after: Optional[int] = None,

    ) -> None:
        details = {}
        if retry_after is not None:
            details["retry_after"] = retry_after
        super().__init__(
            message=message,
            code="rate_limit_exceeded",
            status_code=429,
            details=details,
        )