doushi-cli 1.0.2__tar.gz → 1.0.4__tar.gz

{doushi_cli-1.0.2/doushi_cli.egg-info → doushi_cli-1.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: doushi-cli
-Version: 1.0.2
+Version: 1.0.4
 Summary: Official CLI for Doushi.ai - Instant Autonomous ML Modeling from your Terminal
 Author-email: Doushi AI Team <support@doushi.ai>
 License: MIT

{doushi_cli-1.0.2 → doushi_cli-1.0.4}/doushi/__init__.py

@@ -1,4 +1,4 @@
 """Doushi CLI - Instant Autonomous ML Modeling from your Terminal."""
 
-__version__ = "1.0.2"
+__version__ = "1.0.4"
 __app_name__ = "doushi"

{doushi_cli-1.0.2 → doushi_cli-1.0.4}/doushi/client.py

@@ -253,3 +253,32 @@ class DoushiClient:
                 headers=self._get_headers()
             )
             return self._handle_response(resp)
+
+    def download_model_artifact(self, project_id: str, dest_file: Path) -> bool:
+        """Download trained model.pkl artifact to destination file."""
+        self.check_auth_or_exit()
+        try:
+            with httpx.Client(timeout=self.timeout) as client:
+                resp = client.get(f"{self.base_url}/api/projects/{project_id}/download-model", headers=self._get_headers())
+                data = self._handle_response(resp)
+                presigned_url = data.get("url")
+        except Exception:
+            return False
+
+        if not presigned_url:
+            return False
+
+        try:
+            with httpx.Client(timeout=180.0, follow_redirects=True) as client:
+                with client.stream("GET", presigned_url) as stream_resp:
+                    if stream_resp.status_code == 404:
+                        return False
+                    if not stream_resp.is_success:
+                        return False
+                    with open(dest_file, "wb") as f:
+                        for chunk in stream_resp.iter_bytes(chunk_size=65536):
+                            f.write(chunk)
+            return dest_file.exists() and dest_file.stat().st_size > 0
+        except Exception:
+            return False
+

{doushi_cli-1.0.2 → doushi_cli-1.0.4}/doushi/commands/export.py

@@ -66,9 +66,17 @@ def export_model(
     with open(out_dir / "README.md", "w", encoding="utf-8") as f:
         f.write(generate_export_readme(project_id=project_id, project_name=project_name))
 
+    # 4. Download trained model.pkl artifact
+    model_file = out_dir / "model.pkl"
+    has_model_file = False
+    with console.status(f"[bold cyan]Downloading model artifact (model.pkl)...[/bold cyan]"):
+        has_model_file = client.download_model_artifact(project_id=project_id, dest_file=model_file)
+
     console.print("\n")
     print_success(f"Project exported successfully to [bold cyan]{out_dir}[/bold cyan]!")
     console.print("\n[dim]Files created:[/dim]")
+    if has_model_file:
+        console.print(f"  ├── [bold green]{out_dir}/model.pkl[/bold green] (trained model weights)")
     console.print(f"  ├── {out_dir}/pipeline.py")
     console.print(f"  ├── {out_dir}/serve.py")
     console.print(f"  ├── {out_dir}/Dockerfile")

{doushi_cli-1.0.2 → doushi_cli-1.0.4}/doushi/templates.py

@@ -77,6 +77,8 @@ pandas>=2.0.0
 numpy>=1.24.0
 scikit-learn>=1.4.0
 xgboost>=2.0.0
+lightgbm>=4.0.0
+catboost>=1.2.0
 joblib>=1.3.0
 '''
 

{doushi_cli-1.0.2 → doushi_cli-1.0.4/doushi_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: doushi-cli
-Version: 1.0.2
+Version: 1.0.4
 Summary: Official CLI for Doushi.ai - Instant Autonomous ML Modeling from your Terminal
 Author-email: Doushi AI Team <support@doushi.ai>
 License: MIT

{doushi_cli-1.0.2 → doushi_cli-1.0.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "doushi-cli"
-version = "1.0.2"
+version = "1.0.4"
 description = "Official CLI for Doushi.ai - Instant Autonomous ML Modeling from your Terminal"
 readme = "README.md"
 requires-python = ">=3.10"

{doushi_cli-1.0.2 → doushi_cli-1.0.4}/tests/test_cli.py

@@ -39,9 +39,10 @@ class TestDoushiCLI(unittest.TestCase):
 
     def test_version_output(self):
         """Verify --version flag outputs version."""
+        from doushi import __version__
         result = self.runner.invoke(app, ["--version"])
         self.assertEqual(result.exit_code, 0)
-        self.assertIn("1.0.2", result.output)
+        self.assertIn(__version__, result.output)
 
     def test_subcommand_helps(self):
         """Verify subcommands provide descriptive help."""