airsmodel / utils /chat_request.py
tanbushi's picture
update
702fae5
raw
history blame contribute delete
384 Bytes
from pydantic import BaseModel
from typing import List, Optional, Dict, Any
class ChatRequest(BaseModel):
model: Optional[str] = "gpt-3.5-turbo"
messages: List[Dict[str, Any]]
temperature: Optional[float] = 1.0
max_tokens: Optional[int] = None
top_p: Optional[float] = 1.0
frequency_penalty: Optional[float] = 0.0
presence_penalty: Optional[float] = 0.0