File size: 384 Bytes
702fae5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
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
|