giers10/Heimgeist
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Rename ImageAttachment to ChatAttachment and add new fields

Browse Source
This commit is contained in:
Victor Giers
2026-04-17 12:51:21 +02:00
parent 1c80ca4a06
commit 67195af4c1
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
backend/schemas.py
View File

@@ -2,16 +2,21 @@ from pydantic import BaseModel, ConfigDict
from typing import List, Optional from typing import List, Optional
from datetime import datetime from datetime import datetime
class ImageAttachment(BaseModel): class ChatAttachment(BaseModel):
kind: Optional[str] = None
name: str name: str
mime_type: Optional[str] = None mime_type: Optional[str] = None
data_url: str data_url: Optional[str] = None
source_path: Optional[str] = None
size: Optional[int] = None
record_type: Optional[str] = None
text: Optional[str] = None
class Message(BaseModel): class Message(BaseModel):
role: str role: str
content: str content: str
sources: Optional[List[str]] = None sources: Optional[List[str]] = None
attachments: Optional[List[ImageAttachment]] = None attachments: Optional[List[ChatAttachment]] = None
class ChatRequest(BaseModel): class ChatRequest(BaseModel):
session_id: str session_id: str
@@ -20,7 +25,9 @@ class ChatRequest(BaseModel):
enriched_message: Optional[str] = None enriched_message: Optional[str] = None
stream: Optional[bool] = False stream: Optional[bool] = False
sources: Optional[List[str]] = None sources: Optional[List[str]] = None
attachments: Optional[List[ImageAttachment]] = None attachments: Optional[List[ChatAttachment]] = None
vision_model: Optional[str] = None
transcription_model: Optional[str] = None
class ChatResponse(BaseModel): class ChatResponse(BaseModel):
reply: str reply: str
@@ -59,6 +66,8 @@ class RegenerateRequest(BaseModel):
enriched_message: Optional[str] = None enriched_message: Optional[str] = None
stream: bool = True stream: bool = True
sources: Optional[List[str]] = None sources: Optional[List[str]] = None
vision_model: Optional[str] = None
transcription_model: Optional[str] = None
# Request payload for the web search enrichment endpoint. # Request payload for the web search enrichment endpoint.
class WebSearchRequest(BaseModel): class WebSearchRequest(BaseModel):