brief-rags-bench/app/models/analysis.py

44 lines
803 B
Python

"""Analysis session Pydantic models."""
from typing import Any
from pydantic import BaseModel
class SessionCreate(BaseModel):
"""Create new analysis session."""
environment: str
api_mode: str
request: list[Any]
response: dict
annotations: dict
class SessionResponse(BaseModel):
"""Analysis session response."""
session_id: str
user_id: str
environment: str
api_mode: str
request: list[Any]
response: dict
annotations: dict
created_at: str
updated_at: str
class SessionListItem(BaseModel):
"""Session list item for listing sessions."""
session_id: str
environment: str
created_at: str
class SessionList(BaseModel):
"""List of sessions with total count."""
sessions: list[SessionListItem]
total: int