StanGirard/quivr
1
1
Fork 0
mirror of https://github.com/StanGirard/quivr.git synced 2024-08-16 16:40:40 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix: make max_brain_size overwritable (#751)

Browse Source
This commit is contained in:
Mamadou DICKO 2023-07-24 14:17:33 +02:00 committed by GitHub
parent cf376fb59f
commit 25f859a31e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 5 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/core/models/brains.py
View File

@ -19,15 +19,11 @@ class Brain(BaseModel):
temperature: Optional[float] = 0.0
max_tokens: Optional[int] = 256
files: List[Any] = []
max_brain_size = BrainRateLimiting().max_brain_size
class Config:
arbitrary_types_allowed = True
@property
def max_brain_size(self) -> int:
brain_rate_limiting = BrainRateLimiting()
return brain_rate_limiting.max_brain_size
@property
def commons(self) -> CommonsDep:
return common_dependencies()

4
backend/core/models/settings.py
View File

@ -8,8 +8,8 @@ from vectorstore.supabase import SupabaseVectorStore
class BrainRateLimiting(BaseSettings):
max_brain_size = 52428800
max_brain_per_user = 5
max_brain_size: int = 52428800
max_brain_per_user: int = 5
class BrainSettings(BaseSettings):

5
backend/core/routes/upload_routes.py
View File

@ -46,9 +46,8 @@ async def upload_file(
commons = common_dependencies()
if request.headers.get("Openai-Api-Key"):
brain.max_brain_size = os.getenv(
"MAX_BRAIN_SIZE_WITH_KEY", 209715200
) # pyright: ignore reportPrivateUsage=none
brain.max_brain_size = int(os.getenv("MAX_BRAIN_SIZE_WITH_KEY", 209715200))
remaining_free_space = brain.remaining_brain_size
file_size = get_file_size(uploadFile)