From 1f1b33fb376a73b32c59405f34d7a4645130edfa Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Wed, 19 Jun 2024 10:27:04 -0700 Subject: [PATCH] Fix --- backend/app/api/assistants.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/backend/app/api/assistants.py b/backend/app/api/assistants.py index b8ea7e53..13923582 100644 --- a/backend/app/api/assistants.py +++ b/backend/app/api/assistants.py @@ -18,6 +18,14 @@ class AssistantPayload(BaseModel): public: bool = Field(default=False, description="Whether the assistant is public.") +class AssistantPatch(BaseModel): + """Payload for creating an assistant.""" + + name: str | None = Field(None, description="The name of the assistant.") + config: dict | None = Field(None, description="The assistant config.") + public: bool | None = Field(None, description="Whether the assistant is public.") + + AssistantID = Annotated[str, Path(description="The ID of the assistant.")] @@ -67,7 +75,7 @@ async def create_assistant( async def patch_assistant( user: AuthedUser, aid: AssistantID, - payload: AssistantPayload, + payload: AssistantPatch, ) -> Assistant: """Create or update an assistant.""" if payload.config and not payload.config.get("configurable", {}).get("type"):