Spaces:

tabiwellbeing
/

AI_Agent_Server

Sleeping

App Files Files Community

AI_Agent_Server / backend /models.py

Sidreds06

Initial Commit

b2f01bd about 1 year ago

Raw History Blame Contribute Delete

307 Bytes

from pydantic import BaseModel
from typing import List, Literal

class ChatTurn(BaseModel):
    role: Literal["user", "assistant"]
    content: str

class ChatRequest(BaseModel):
    message: str
    history: List[ChatTurn] = []
    uid: str = None

class SummaryRequest(BaseModel):
    messages: List[dict]

	from pydantic import BaseModel
	from typing import List, Literal

	class ChatTurn(BaseModel):
	role: Literal["user", "assistant"]
	content: str

	class ChatRequest(BaseModel):
	message: str
	history: List[ChatTurn] = []
	uid: str = None

	class SummaryRequest(BaseModel):
	messages: List[dict]