init

models/debate.py

@@ -0,0 +1,92 @@
+from pydantic import BaseModel
+from typing import List, Optional
+from enum import Enum
+from datetime import datetime
+
+
+class ModelProvider(str, Enum):
+    OPENAI = "openai"
+    DEEPSEEK = "deepseek"
+    QWEN = "qwen"
+    CLAUDE = "claude"
+
+
+class DebateStance(str, Enum):
+    PRO = "pro"
+    CON = "con"
+
+
+class SearchResult(BaseModel):
+    title: str
+    url: str
+    snippet: str
+    score: Optional[float] = None
+
+
+class SearchEvidence(BaseModel):
+    query: str
+    results: List[SearchResult]
+    mode: str  # "auto", "tool", "both"
+
+
+class DebateRound(BaseModel):
+    round_number: int
+    speaker: str  # Model identifier
+    stance: DebateStance
+    content: str
+    timestamp: datetime
+    token_count: Optional[int] = None
+    search_evidence: Optional[SearchEvidence] = None
+
+
+class DebateParticipant(BaseModel):
+    model_config = {"protected_namespaces": ()}
+
+    model_identifier: str
+    provider: ModelProvider
+    stance: DebateStance
+    api_key: Optional[str] = None
+
+
+class DebateConstraints(BaseModel):
+    max_rounds: int = 5
+    max_tokens_per_turn: int = 500
+    max_total_tokens: Optional[int] = None
+    forbid_repetition: bool = True
+    must_respond_to_opponent: bool = True
+    web_search_enabled: bool = False
+    web_search_mode: str = "auto"  # "auto", "tool", "both"
+
+
+class DebateRequest(BaseModel):
+    topic: str
+    participants: List[DebateParticipant]
+    constraints: DebateConstraints
+    custom_system_prompt: Optional[str] = None
+
+
+class EvidenceReference(BaseModel):
+    round_number: int
+    speaker: str
+    stance: DebateStance
+
+
+class EvidenceEntry(BaseModel):
+    title: str
+    url: str
+    snippet: str
+    score: Optional[float] = None
+    references: List[EvidenceReference]
+
+
+class DebateSession(BaseModel):
+    session_id: str
+    topic: str
+    participants: List[DebateParticipant]
+    constraints: DebateConstraints
+    rounds: List[DebateRound]
+    status: str  # "active", "completed", "terminated"
+    created_at: datetime
+    completed_at: Optional[datetime] = None
+    summary: Optional[str] = None
+    evidence_library: List[EvidenceEntry] = []