diff --git a/olah/proxy/meta.py b/olah/proxy/meta.py index 60e73f3..f171740 100644 --- a/olah/proxy/meta.py +++ b/olah/proxy/meta.py @@ -97,6 +97,6 @@ async def meta_generator( yield item else: async for item in _meta_proxy_generator( - app, headers, meta_url, allow_cache, method, save_path + app, headers, meta_url, method, allow_cache, save_path ): yield item diff --git a/olah/proxy/pathsinfo.py b/olah/proxy/pathsinfo.py index bbcdd39..a1b6b15 100644 --- a/olah/proxy/pathsinfo.py +++ b/olah/proxy/pathsinfo.py @@ -29,9 +29,9 @@ async def _pathsinfo_proxy( app: FastAPI, headers: Dict[str, str], pathsinfo_url: str, - allow_cache: bool, method: str, path: str, + allow_cache: bool, save_path: str, ): headers = {k: v for k, v in headers.items()} @@ -94,7 +94,7 @@ async def pathsinfo_generator( status, headers, content = await _pathsinfo_cache(save_path) else: status, headers, content = await _pathsinfo_proxy( - app, headers, pathsinfo_url, allow_cache, method, path, save_path + app, headers, pathsinfo_url, method, path, allow_cache, save_path ) try: diff --git a/olah/proxy/tree.py b/olah/proxy/tree.py index f5a21b6..45a008d 100644 --- a/olah/proxy/tree.py +++ b/olah/proxy/tree.py @@ -28,9 +28,9 @@ async def _tree_proxy_generator( app: FastAPI, headers: Dict[str, str], tree_url: str, - allow_cache: bool, method: str, recursive: bool, + allow_cache: bool, save_path: str, ): async with httpx.AsyncClient(follow_redirects=True) as client: @@ -102,6 +102,6 @@ async def tree_generator( yield item else: async for item in _tree_proxy_generator( - app, headers, tree_url, allow_cache, method, recursive, save_path + app, headers, tree_url, method, recursive, allow_cache, save_path ): yield item diff --git a/pyproject.toml b/pyproject.toml index a17661d..3468655 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -15,7 +15,7 @@ classifiers = [ dependencies = [ "fastapi", "fastapi-utils", "httpx", "numpy", "pydantic<=2.8.2", "pydantic-settings<=2.4.0", "requests", "toml", "rich>=10.0.0", "shortuuid", "uvicorn", "tenacity>=8.2.2", "pytz", "cachetools", "GitPython", - "PyYAML", "typing_inspect>=0.9.0", "huggingface_hub", "jinja2" + "PyYAML", "typing_inspect>=0.9.0", "huggingface_hub", "jinja2", "python-multipart" ] [project.optional-dependencies]