# 定义所有数据模型
from pydantic import BaseModel
from typing import Optional

class RequestProps(BaseModel):
    prompt: str
    options: Optional[dict] = {}
    systemMessage: Optional[str] = None
    temperature: Optional[float] = None
    top_p: Optional[float] = None

class ChatMessage(BaseModel):
    message: str
    lastContext: Optional[dict] = None
    
class RequestOptions(BaseModel):
    message: str
    lastContext: Optional[dict] = None
    process: Optional[str] = None
    systemMessage: Optional[str] = None
    temperature: Optional[float] = None
    top_p: Optional[float] = None

class SendMessageOptions(BaseModel):
    timeoutMs: int = 5000
    systemMessage: Optional[str] = None
    completionParams: Optional[dict] = None
    parentMessageId: Optional[str] = None

ErrorCodeMessage = {
  401: '[OpenAI] 提供错误的API密钥 | Incorrect API key provided',
  403: '[OpenAI] 服务器拒绝访问，请稍后再试 | Server refused to access, please try again later',
  502: '[OpenAI] 错误的网关 |  Bad Gateway',
  503: '[OpenAI] 服务器繁忙，请稍后再试 | Server is busy, please try again later',
  504: '[OpenAI] 网关超时 | Gateway Time-out',
  500: '[OpenAI] 服务器繁忙，请稍后再试 | Internal Server Error',
}