File size: 1,232 Bytes
62a1756 | 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 | from typing import List, Optional
from pydantic import BaseModel, Field
from bson import ObjectId
from datetime import datetime
ALLOWED_MODELS = [
"allam-2-7b", # Fixed typo from "allam-2-7b"
"groq/compound",
"groq/compound-mini",
"llama-3.1-8b-instant",
"llama-3.3-70b-versatile",
"llama-3.1-70b-versatile", # Assuming typo from "llama-3.3-70b-versatile"
"meta-llama/llama-4-maverick-17b-128e-instruct",
"meta-llama/llama-4-scout-17b-16e-instruct",
"meta-llama/llama-guard-4-12b",
"meta-llama/llama-prompt-guard-2-22m",
"meta-llama/llama-prompt-guard-2-86m",
"moonshotai/kimi-k2-instruct",
"moonshotai/kimi-k2-instruct-0905",
"openai/gpt-oss-120b",
"openai/gpt-oss-20b",
"openai/gpt-oss-safeguard-20b",
"qwen/qwen3-32b",
]
class Message(BaseModel):
role: str
content: str
class Conversation(BaseModel):
id: Optional[str] = None # str(ObjectId)
user_id: str
messages: List[Message] = []
created_at: datetime = Field(default_factory=datetime.utcnow)
class ChatRequest(BaseModel):
model: str = Field(..., description="Groq model", enum=ALLOWED_MODELS)
enable_web_search: bool = False
message: str = Field(..., min_length=1) |