21 lines
475 B
Python
21 lines
475 B
Python
from pydantic import BaseModel
|
|
from typing import Optional
|
|
|
|
class ProfileBase(BaseModel):
|
|
default_ticker: str = "SPY"
|
|
preferred_research_depth: int = 3
|
|
preferred_shallow_thinker: str = "gpt-4o-mini"
|
|
preferred_deep_thinker: str = "gpt-4o"
|
|
|
|
class ProfileCreate(ProfileBase):
|
|
pass
|
|
|
|
class ProfileUpdate(ProfileBase):
|
|
openai_api_key: Optional[str] = None
|
|
|
|
class Profile(ProfileBase):
|
|
has_openai_api_key: bool
|
|
|
|
class Config:
|
|
orm_mode = True
|