Spaces:
Configuration error
Configuration error
File size: 1,296 Bytes
aa15bce |
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 |
from __future__ import annotations
from typing import Any, Dict, List, Optional
from pydantic import BaseModel, ConfigDict, Field, model_validator
class ChatMessage(BaseModel):
model_config = ConfigDict(extra="ignore")
role: str = Field(..., min_length=1)
content: str = Field(...)
timestamp: Optional[str] = Field(default=None)
@model_validator(mode="before")
@classmethod
def _coerce_content(cls, data: Any) -> Any:
if isinstance(data, dict) and "content" in data:
data["content"] = "" if data["content"] is None else str(data["content"])
return data
def as_openrouter(self) -> Dict[str, str]:
return {"role": self.role.strip(), "content": self.content}
class ChatRequest(BaseModel):
model_config = ConfigDict(populate_by_name=True, extra="ignore")
messages: List[ChatMessage] = Field(default_factory=list)
model: Optional[str] = None
system: Optional[str] = None
stream: bool = True
def openrouter_messages(self) -> List[Dict[str, str]]:
return [msg.as_openrouter() for msg in self.messages if msg.content.strip()]
class ChatHistoryResponse(BaseModel):
messages: List[ChatMessage] = Field(default_factory=list)
class ChatHistoryClearResponse(BaseModel):
ok: bool = True
|