drift-ml 0.1.6__py3-none-any.whl → 0.1.8__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
drift/engine_launcher.py CHANGED
@@ -16,7 +16,7 @@ try:
16
16
  except ImportError:
17
17
  requests = None
18
18
 
19
- ENGINE_TAG = "v0.1.3"
19
+ ENGINE_TAG = "v0.1.4"
20
20
  GITHUB_REPO = "lakshitsachdeva/drift"
21
21
  GITHUB_API = f"https://api.github.com/repos/{GITHUB_REPO}/releases/tags/{ENGINE_TAG}"
22
22
  ENGINE_PORT = os.environ.get("DRIFT_ENGINE_PORT", "8000")
@@ -62,6 +62,13 @@ def _engine_running() -> bool:
62
62
 
63
63
  def _get_asset_download_url(asset_name: str) -> str:
64
64
  """Resolve GitHub release asset to download URL."""
65
+ # Direct URL works for public repos, no API rate limits
66
+ direct_url = f"https://github.com/{GITHUB_REPO}/releases/download/{ENGINE_TAG}/{asset_name}"
67
+ r = requests.head(direct_url, timeout=10, allow_redirects=True)
68
+ if r.status_code == 200:
69
+ return r.url # follow redirects to final URL
70
+
71
+ # Fallback: API (needed for private repos or if direct fails)
65
72
  token = os.environ.get("DRIFT_GITHUB_TOKEN") or os.environ.get("GITHUB_TOKEN")
66
73
  headers = {
67
74
  "User-Agent": "Drift-Engine-Launcher/1.0",
@@ -79,14 +86,9 @@ def _get_asset_download_url(asset_name: str) -> str:
79
86
  data = r.json()
80
87
  for a in data.get("assets", []):
81
88
  if a.get("name") == asset_name:
82
- # Prefer browser_download_url for public repos (no auth needed)
83
- url = a.get("browser_download_url")
89
+ url = a.get("browser_download_url") or a.get("url")
84
90
  if url:
85
91
  return url
86
- # API URL requires Accept: application/octet-stream
87
- api_url = a.get("url")
88
- if api_url:
89
- return api_url
90
92
  raise RuntimeError(f"Asset {asset_name} not found in release {ENGINE_TAG}")
91
93
 
92
94
 
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: drift-ml
3
- Version: 0.1.6
3
+ Version: 0.1.8
4
4
  Summary: Terminal-first AutoML CLI - chat-based ML engineer
5
5
  Requires-Python: >=3.10
6
6
  Requires-Dist: requests>=2.28.0
@@ -1,6 +1,6 @@
1
1
  drift/__init__.py,sha256=X0NUP5ZAZSz-rBFfjvmmS4IYWP2_CZtu187mqwpWwqk,127
2
2
  drift/__main__.py,sha256=RnBMIqpDEMHknOKn-s5Q2MTUWm_UsKSkYuOArzShQeo,693
3
- drift/engine_launcher.py,sha256=mF1yNDJMV2ccWGUJNIxTIV8rpIOYXQowm2m0_hw2_5Q,5462
3
+ drift/engine_launcher.py,sha256=YFSqXFuwvVB6JHbwAsbsbD2uHx8fdwPnIBhTD07oiQc,5623
4
4
  drift/cli/__init__.py,sha256=OQt7M06e98e_L_60Qz-HsmSqGXKWdH53SvgETJ_BZZ0,172
5
5
  drift/cli/client.py,sha256=hGxz-J8fC_gGV9BZnpCBwkQZg4Q0_s9-Agk2PuXQ5MQ,5527
6
6
  drift/cli/repl.py,sha256=hRnx4nfP_StnXDbdRZXnnjA8aNvrdQSV-uTNsuaaO60,8262
@@ -9,8 +9,8 @@ drift/llm_adapters/__init__.py,sha256=y1UhZWlC8Ik_OKfLcOp0JZP-FKR3MBBCemWwsL6Tnk
9
9
  drift/llm_adapters/base.py,sha256=KlZUPYpvCI8pafklBWel0GHHLNtVKVwSHA92MZt3VsI,1331
10
10
  drift/llm_adapters/gemini_cli.py,sha256=Z61wY3yFiZqPrQrpJAQrBtMbBkr2qLWBkni-A4p9lZo,2163
11
11
  drift/llm_adapters/local_llm.py,sha256=Z6j6z1CXk2LMeQ5ZnY4o38PiYkHYmcIkgJGkHdU50M8,2279
12
- drift_ml-0.1.6.dist-info/METADATA,sha256=CIj7-l0RadnHdDOeKRPnH2RL-_MWfFzvQH-7TCUmWZA,168
13
- drift_ml-0.1.6.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
14
- drift_ml-0.1.6.dist-info/entry_points.txt,sha256=aCY7U9M8nhYj_tIfTXJmYkVmXY3ZoxF0tebDZzYswv8,46
15
- drift_ml-0.1.6.dist-info/top_level.txt,sha256=3u2KGqsciGZQ2uCoBivm55t3e8er8S4xnqkgdQ_8oeM,6
16
- drift_ml-0.1.6.dist-info/RECORD,,
12
+ drift_ml-0.1.8.dist-info/METADATA,sha256=lN6INF5pRo1iuVmyG3QeLmVEFX4bDkNMSNKWG2g24EI,168
13
+ drift_ml-0.1.8.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
14
+ drift_ml-0.1.8.dist-info/entry_points.txt,sha256=aCY7U9M8nhYj_tIfTXJmYkVmXY3ZoxF0tebDZzYswv8,46
15
+ drift_ml-0.1.8.dist-info/top_level.txt,sha256=3u2KGqsciGZQ2uCoBivm55t3e8er8S4xnqkgdQ_8oeM,6
16
+ drift_ml-0.1.8.dist-info/RECORD,,