File size: 1,096 Bytes
4ad97e9 09fd315 0c5219d 09fd315 0c5219d 09fd315 |
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 |
# from enum import StrEnum
from pydantic import BaseModel, ConfigDict
class GoogleModelID():
GEMINI_2_0_FLASH = "gemini-2.0-flash"
GEMINI_2_5_FLASH_PREVIEW = "gemini-2.5-flash-preview"
class OpenRouterModelID():
QWEN_3_14B_FREE = "openrouter/qwen/qwen3-14b:free"
GPT_4_1_MINI = "openrouter/openai/gpt-4.1-mini"
GPT_O4_MINI = "openrouter/openai/o4-mini"
GROK_3_MINI_BETA = "openrouter/x-ai/grok-3-mini-beta"
GROK_3_BETA = "openrouter/x-ai/grok-3-beta"
class Question(BaseModel):
model_config = ConfigDict(validate_by_name=True, validate_by_alias=True)
task_id: str
question: str
file_name: str
class Answer(BaseModel):
task_id: str
answer: str
class QuestionAnswerPair(BaseModel):
task_id: str
question: str
answer: str
def get_answer(self) -> dict[str, str]:
return {"task_id": self.task_id, "submitted_answer": self.answer}
class Results(BaseModel):
model_config = ConfigDict(from_attributes=True)
username: str
score: int
correct_count: int
total_attempted: int
message: str
timestamp: str
|