Spaces:
Sleeping
Sleeping
File size: 2,066 Bytes
dccc925 a43aaf6 dccc925 a43aaf6 dccc925 a43aaf6 dccc925 a43aaf6 dccc925 |
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 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
"""Pydantic models for API request/response validation."""
from pydantic import BaseModel
from datetime import datetime
class TaskIDRequest(BaseModel):
"""Request model for task-based endpoints."""
task_id: int
class SuggestGroupsResponse(BaseModel):
"""Response model for group suggestion endpoint."""
suggested_groups: list[str]
class ChatMessage(BaseModel):
"""Model for individual chat messages."""
role: str # 'user', 'ai', 'system'
content: str
created_at: datetime | None = None
class ChatRequest(BaseModel):
"""Request model for chat endpoint."""
message: str
task_id: int
project_title: str
current_stage: str
output_mode: str = "standard"
prompt_group_config: dict | None = None
custom_system_prompt: str | None = None
conversation_id: str | None = None # Optional conversation ID for persistence
current_group_name: str | None = None
selected_standards: list[dict] | None = None
current_alignment_work: str | None = None
message_history: list[ChatMessage] | None = None # Sliding window of recent messages
class ChatResponse(BaseModel):
"""Response model for chat endpoint."""
response: str
conversation_id: str | None = None # Return conversation ID for tracking
class ConversationSummary(BaseModel):
"""Summary model for conversation list."""
id: str
title: str | None
project_id: str | None
created_at: datetime
updated_at: datetime
class ConversationListResponse(BaseModel):
"""Response model for listing conversations."""
conversations: list[ConversationSummary]
class ConversationMessagesResponse(BaseModel):
"""Response model for fetching conversation messages."""
conversation_id: str
messages: list[ChatMessage]
class GenerateRubricRequest(BaseModel):
"""Request model for rubric generation endpoint."""
task_id: int
prompt_groups: dict
class GenerateRubricResponse(BaseModel):
"""Response model for rubric generation endpoint."""
rubric: dict
message: str
|