Spaces:
Build error
Build error
File size: 1,815 Bytes
ad8ba8a |
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 |
from typing import List, Optional, Dict, Any
from pydantic import BaseModel, Field
from enum import Enum
import time
# --- Enums ---
class SessionStatus(str, Enum):
ACTIVE = "active"
STOPPED = "stopped"
ARCHIVED = "archived"
class MessageRole(str, Enum):
USER = "user"
ASSISTANT = "assistant"
SYSTEM = "system"
TOOL = "tool"
# --- Entities ---
class Message(BaseModel):
role: MessageRole
content: str
timestamp: int = Field(default_factory=lambda: int(time.time()))
tool_calls: Optional[List[Dict[str, Any]]] = None
tool_call_id: Optional[str] = None
class Session(BaseModel):
session_id: str
title: str
created_at: int = Field(default_factory=lambda: int(time.time()))
updated_at: int = Field(default_factory=lambda: int(time.time()))
status: SessionStatus = SessionStatus.ACTIVE
messages: List[Message] = []
metadata: Dict[str, Any] = {}
# --- API Request/Response Models ---
class StandardResponse(BaseModel):
code: int
msg: str
data: Optional[Any] = None
class SessionResponse(BaseModel):
session_id: str
title: str
latest_message: Optional[str] = None
latest_message_at: Optional[int] = None
status: SessionStatus
unread_message_count: int = 0
class SessionDetailResponse(BaseModel):
session_id: str
title: str
events: List[Dict[str, Any]] = [] # Simplified for now, can be specific Event models
class ChatRequest(BaseModel):
message: str
timestamp: Optional[int] = None
event_id: Optional[str] = None
class ShellRequest(BaseModel):
session_id: str
class FileRequest(BaseModel):
file: str
class ShellResponse(BaseModel):
output: str
session_id: str
console: List[Dict[str, str]]
class FileResponse(BaseModel):
content: str
file: str
|