4c271170b5
- Introduced a new `ChatHistoryMessageModel` to persist chat messages in the database. - Implemented a migration script to create the `chat_history_messages` table. - Enhanced chat services to support storing and retrieving chat history. - Added functionality for generating multiple media assets in a single API call. - Updated the chat UI to display assistant activities and tool usage more effectively. - Refactored API calls to use absolute URLs for better reliability.
81 lines
2.3 KiB
Python
81 lines
2.3 KiB
Python
import json
|
|
from typing import Any, Literal
|
|
|
|
import dirtyjson # type: ignore[import-untyped]
|
|
from pydantic import BaseModel, ConfigDict, Field, field_validator
|
|
|
|
|
|
class StrictSchemaModel(BaseModel):
|
|
model_config = ConfigDict(extra="forbid", strict=True)
|
|
|
|
|
|
class NoArgsInput(StrictSchemaModel):
|
|
pass
|
|
|
|
|
|
class GetSlideAtIndexInput(StrictSchemaModel):
|
|
index: int = Field(ge=0, le=1000)
|
|
include_full_content: bool = Field(alias="includeFullContent")
|
|
|
|
model_config = ConfigDict(extra="forbid", strict=True, populate_by_name=True)
|
|
|
|
|
|
class SearchSlidesInput(StrictSchemaModel):
|
|
query: str = Field(min_length=1, max_length=1000)
|
|
limit: int = Field(ge=1, le=10)
|
|
|
|
|
|
class GetContentSchemaFromLayoutIdInput(StrictSchemaModel):
|
|
layout_id: str = Field(alias="layoutId", min_length=1, max_length=200)
|
|
|
|
model_config = ConfigDict(extra="forbid", strict=True, populate_by_name=True)
|
|
|
|
|
|
class GenerateImageInput(StrictSchemaModel):
|
|
prompt: str = Field(min_length=1, max_length=4000)
|
|
|
|
|
|
class GenerateIconInput(StrictSchemaModel):
|
|
query: str = Field(min_length=1, max_length=1000)
|
|
|
|
|
|
class GenerateAssetItemInput(StrictSchemaModel):
|
|
kind: Literal["image", "icon"]
|
|
prompt: str = Field(
|
|
min_length=1,
|
|
max_length=4000,
|
|
description="Image prompt or icon search query.",
|
|
)
|
|
|
|
|
|
class GenerateAssetsInput(StrictSchemaModel):
|
|
assets: list[GenerateAssetItemInput] = Field(min_length=1, max_length=12)
|
|
|
|
|
|
class SaveSlideInput(StrictSchemaModel):
|
|
content: str = Field(
|
|
min_length=2,
|
|
max_length=200000,
|
|
description=(
|
|
"A JSON-serialized object for slide content. "
|
|
"Example: '{\"title\": \"Q4 Revenue\", \"bullets\": [\"North America +22%\"]}'"
|
|
),
|
|
)
|
|
layout_id: str = Field(alias="layoutId", min_length=1, max_length=200)
|
|
index: int = Field(ge=0, le=1000)
|
|
replace_old_slide_at_index: bool = Field(alias="replaceOldSlideAtIndex")
|
|
|
|
model_config = ConfigDict(extra="forbid", strict=True, populate_by_name=True)
|
|
|
|
@field_validator("content")
|
|
@classmethod
|
|
def validate_content(cls, value: str) -> str:
|
|
try:
|
|
parsed: Any = dirtyjson.loads(value)
|
|
except Exception:
|
|
parsed = json.loads(value)
|
|
|
|
if not isinstance(parsed, dict):
|
|
raise ValueError("'content' must be a JSON object.")
|
|
|
|
return value
|