robodock-cli 0.1.4__tar.gz → 0.1.5.1__tar.gz

{robodock_cli-0.1.4 → robodock_cli-0.1.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: robodock-cli
-Version: 0.1.4
+Version: 0.1.5.1
 Summary: Robodock 平台命令行 & SDK 工具
 Author-email: KirigiriSuzumiya <boyifan1@126.com>
 Project-URL: Homepage, https://github.com/FII-AILab/robodock-cli
@@ -20,10 +20,10 @@ Classifier: Programming Language :: Python :: 3.14
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
-Requires-Dist: typer>=0.9
-Requires-Dist: rich
+Requires-Dist: typer<0.28.0,>=0.20.0
+Requires-Dist: rich<16.0.0,>=13.0.0
 Requires-Dist: httpx<0.29.0,>=0.23.1
-Requires-Dist: attrs>=22.2.0
+Requires-Dist: attrs<27.0.0,>=22.2.0
 Provides-Extra: dev
 Requires-Dist: build; extra == "dev"
 Requires-Dist: twine; extra == "dev"

{robodock_cli-0.1.4 → robodock_cli-0.1.5.1}/app/task_management.py

@@ -30,6 +30,7 @@ def download_checkpoint(
     checkpoint_prefix:str,
     output_path:str=str(Path.cwd()),
     overwrite:bool=False,
+    download_onnx:bool=True,
 ):
     local_path = Path(output_path) / run_uuid / Path(checkpoint_prefix).stem
     checkpoint_model_path = checkpoint_prefix + MODEL_PATH_SUFIX
@@ -49,8 +50,11 @@ def download_checkpoint(
 
     model_url = get_model_url()
     typer.echo("已获取模型下载链接")
-    onnx_url = get_onnx_url()
-    typer.echo("已获取onnx模型下载链接")
+
+    if download_onnx:
+        onnx_url = get_onnx_url()
+        typer.echo("已获取onnx模型下载链接")
+    
     typer.echo(f"开始下载模型文件到{str(local_path)}")
 
     
@@ -68,18 +72,19 @@ def download_checkpoint(
     )
 
     with progress:
-        onnx_task = progress.add_task(f"正在下载onnx文件", total=None)
-        download_file(
-            presigned_url=onnx_url,
-            local_path=local_path / Path(checkpoint_onnx_path).name,
-            get_presigned_url=get_model_url,
-            on_progress=lambda downloaded, total: progress.update(
-                onnx_task,
-                completed=downloaded,
-                total=total,
-            ),
-            overwrite=overwrite
-        )
+        if download_onnx:
+            onnx_task = progress.add_task(f"正在下载onnx文件", total=None)
+            download_file(
+                presigned_url=onnx_url,
+                local_path=local_path / Path(checkpoint_onnx_path).name,
+                get_presigned_url=get_model_url,
+                on_progress=lambda downloaded, total: progress.update(
+                    onnx_task,
+                    completed=downloaded,
+                    total=total,
+                ),
+                overwrite=overwrite
+            )
 
         model_task = progress.add_task(f"正在下载模型文件", total=None)
         download_file(

{robodock_cli-0.1.4 → robodock_cli-0.1.5.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "robodock-cli"
-version = "0.1.4"
+version = "0.1.5.1"
 description = "Robodock 平台命令行 & SDK 工具"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -26,10 +26,10 @@ classifiers = [
     "Topic :: Software Development :: Libraries :: Python Modules",
 ]
 dependencies = [
-    "typer>=0.9",
-    "rich",
+    "typer>=0.20.0,<0.28.0",
+    "rich>=13.0.0,<16.0.0",
     "httpx>=0.23.1,<0.29.0",
-    "attrs>=22.2.0",
+    "attrs>=22.2.0,<27.0.0",
 ]
 
 [project.urls]

{robodock_cli-0.1.4 → robodock_cli-0.1.5.1}/robodock_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: robodock-cli
-Version: 0.1.4
+Version: 0.1.5.1
 Summary: Robodock 平台命令行 & SDK 工具
 Author-email: KirigiriSuzumiya <boyifan1@126.com>
 Project-URL: Homepage, https://github.com/FII-AILab/robodock-cli
@@ -20,10 +20,10 @@ Classifier: Programming Language :: Python :: 3.14
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
-Requires-Dist: typer>=0.9
-Requires-Dist: rich
+Requires-Dist: typer<0.28.0,>=0.20.0
+Requires-Dist: rich<16.0.0,>=13.0.0
 Requires-Dist: httpx<0.29.0,>=0.23.1
-Requires-Dist: attrs>=22.2.0
+Requires-Dist: attrs<27.0.0,>=22.2.0
 Provides-Extra: dev
 Requires-Dist: build; extra == "dev"
 Requires-Dist: twine; extra == "dev"

robodock_cli-0.1.5.1/robodock_cli.egg-info/requires.txt

@@ -0,0 +1,9 @@
+typer<0.28.0,>=0.20.0
+rich<16.0.0,>=13.0.0
+httpx<0.29.0,>=0.23.1
+attrs<27.0.0,>=22.2.0
+
+[dev]
+build
+twine
+openapi-python-client

robodock_cli-0.1.4/robodock_cli.egg-info/requires.txt

@@ -1,9 +0,0 @@
-typer>=0.9
-rich
-httpx<0.29.0,>=0.23.1
-attrs>=22.2.0
-
-[dev]
-build
-twine
-openapi-python-client