theroyallab/tabbyAPI
1
0
Fork 0
mirror of https://github.com/theroyallab/tabbyAPI.git synced 2026-04-29 10:42:03 +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

2
common/downloader.py
View File

@@ -121,7 +121,7 @@ async def hf_repo_download(
repo_type = "lora" repo_type = "lora"
if include or exclude: if include or exclude:
include_patterns = unwrap(include, []) include_patterns = unwrap(include, ["*"])
exclude_patterns = unwrap(exclude, []) exclude_patterns = unwrap(exclude, [])
file_list = [ file_list = [

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

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