added chat history
This commit is contained in:
@@ -162,6 +162,7 @@ class GenerateTableRequest(BaseModel):
|
||||
class LlmChatRequest(BaseModel):
|
||||
message: str = Field(min_length=1)
|
||||
table_type: str
|
||||
chat_id: Optional[int] = None
|
||||
|
||||
|
||||
class LlmEdit(BaseModel):
|
||||
@@ -171,10 +172,20 @@ class LlmEdit(BaseModel):
|
||||
fields: dict[str, Any] = {}
|
||||
|
||||
|
||||
class LlmContextUsage(BaseModel):
|
||||
used: int = 0
|
||||
limit: int = 0
|
||||
percent: float = 0
|
||||
prompt_tokens: Optional[int] = None
|
||||
completion_tokens: Optional[int] = None
|
||||
|
||||
|
||||
class LlmChatResponse(BaseModel):
|
||||
reply: str
|
||||
edits: list[LlmEdit] = []
|
||||
message_id: Optional[int] = None
|
||||
chat_id: Optional[int] = None
|
||||
context: Optional[LlmContextUsage] = None
|
||||
|
||||
|
||||
class LlmApplyRequest(BaseModel):
|
||||
@@ -182,6 +193,39 @@ class LlmApplyRequest(BaseModel):
|
||||
edits: list[LlmEdit]
|
||||
|
||||
|
||||
class LlmChatCreate(BaseModel):
|
||||
title: Optional[str] = None
|
||||
table_type: Optional[str] = None
|
||||
|
||||
|
||||
class LlmChatOut(BaseModel):
|
||||
id: int
|
||||
title: str
|
||||
table_type: Optional[str] = None
|
||||
created_at: datetime
|
||||
updated_at: datetime
|
||||
message_count: int = 0
|
||||
|
||||
|
||||
class LlmMessageOut(BaseModel):
|
||||
id: int
|
||||
role: str
|
||||
content: str
|
||||
table_type: Optional[str] = None
|
||||
proposed_edits: Optional[list[Any]] = None
|
||||
created_at: datetime
|
||||
|
||||
|
||||
class LlmChatDetail(BaseModel):
|
||||
id: int
|
||||
title: str
|
||||
table_type: Optional[str] = None
|
||||
created_at: datetime
|
||||
updated_at: datetime
|
||||
messages: list[LlmMessageOut] = []
|
||||
context: Optional[LlmContextUsage] = None
|
||||
|
||||
|
||||
class ExportRequest(BaseModel):
|
||||
table_type: str
|
||||
format: str = "pdf" # pdf | xlsx
|
||||
|
||||
Reference in New Issue
Block a user