theroyallab/tabbyAPI
1
0
Fork 0
mirror of https://github.com/theroyallab/tabbyAPI.git synced 2026-04-22 23:38:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Downloader: Fix handling of include pattern

Browse Source 
If an include or exclude pattern is provided, include should include
all files by default.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri
2024-04-30 01:13:06 -04:00
parent 21a01741c9
commit 50e0b71690
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
endpoints/OAI/types/download.py
View File

@@ -2,6 +2,10 @@ from pydantic import BaseModel, Field
from typing import List, Optional
def _generate_include_list():
return ["*"]
class DownloadRequest(BaseModel):
"""Parameters for a HuggingFace repo download."""
@@ -10,7 +14,7 @@ class DownloadRequest(BaseModel):
folder_name: Optional[str] = None
revision: Optional[str] = None
token: Optional[str] = None
include: List[str] = Field(default_factory=list)
include: List[str] = Field(default_factory=_generate_include_list)
exclude: List[str] = Field(default_factory=list)
chunk_limit: Optional[int] = None