mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-15 01:21:48 +03:00
e53bc6807d
* 🗃️ add new tables for multiple brains * 🗑️ remove date input from fetch_user_id_from_credentials * ✨ new /brain endpoints * ♻️ refactor backend utils by splitting it into files * 💡 comments for next actions to update /upload
22 lines
605 B
Python
22 lines
605 B
Python
from typing import List, Optional, Tuple
|
|
from uuid import UUID
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class Brain(BaseModel):
|
|
brain_id: Optional[UUID] = None
|
|
brain_name: str = "New Brain"
|
|
status: str = "public"
|
|
model: str = "gpt-3.5-turbo-0613"
|
|
temperature: float = 0.0
|
|
max_tokens: int = 256
|
|
|
|
class BrainToUpdate(BaseModel):
|
|
brain_id: UUID
|
|
brain_name: Optional[str] = "New Brain"
|
|
status: Optional[str] = "public"
|
|
model: Optional[str] = "gpt-3.5-turbo-0613"
|
|
temperature: Optional[float] = 0.0
|
|
max_tokens: Optional[int] = 256
|
|
file_sha1: Optional[str] = '' |