programasweights 0.2.2.dev1__py3-none-any.whl → 0.2.4__py3-none-any.whl

programasweights/__init__.py

@@ -21,8 +21,9 @@ API reference:
     paw.api_url                Server URL (default: https://programasweights.com)
     paw.api_key                API key (set via login() or PAW_API_KEY env var)
 """
+from __future__ import annotations
 
-__version__ = "0.2.2.dev1"
+__version__ = "0.2.4"
 
 from .config import get_api_url, get_api_key, set_api_key
 

programasweights/client.py

@@ -108,25 +108,38 @@ class PAWClient:
         """Download a .paw bundle to the local cache.
 
         Returns the path to the extracted program directory.
-        Retries on 404 since freshly compiled programs may still be uploading.
+        Handles 202 Accepted (assets still generating) and 302 redirects (HF CDN).
         """
         program_dir = config.get_programs_dir() / program_id
         if (program_dir / "prompt_template.txt").exists():
             return program_dir
 
-        max_retries = 10
-        for attempt in range(max_retries):
+        max_wait = 30
+        elapsed = 0
+        resp = None
+        while elapsed < max_wait:
             resp = httpx.get(
                 f"{self._api_url}/api/v1/programs/{program_id}/download",
                 headers=self._headers(),
                 timeout=60.0,
                 follow_redirects=True,
             )
-            if resp.status_code == 404 and attempt < max_retries - 1:
+            if resp.status_code == 202:
+                retry_after = int(resp.headers.get("Retry-After", "3"))
+                time.sleep(retry_after)
+                elapsed += retry_after
+                continue
+            if resp.status_code == 404 and elapsed < max_wait - 3:
                 time.sleep(3)
+                elapsed += 3
                 continue
             resp.raise_for_status()
             break
+        else:
+            raise RuntimeError(
+                f"Program {program_id} assets not ready after {max_wait}s. "
+                "The program may still be generating. Try again shortly."
+            )
 
         program_dir.mkdir(parents=True, exist_ok=True)
         paw_path = program_dir / f"{program_id}.paw"

programasweights/runtime_llamacpp.py

@@ -67,24 +67,24 @@ class PawFunction:
                 n_gpu_layers=n_gpu_layers,
                 verbose=verbose,
             )
+
+            adapter_path = program_dir / "adapter.gguf"
+            if adapter_path.exists():
+                self._adapter = llama_cpp.llama_adapter_lora_init(
+                    self._llm.model, str(adapter_path).encode("utf-8"),
+                )
+                if self._adapter:
+                    llama_cpp.llama_set_adapter_lora(self._llm.ctx, self._adapter, 1.0)
+                else:
+                    raise RuntimeError(f"Failed to load LoRA adapter: {adapter_path}")
+            else:
+                self._adapter = None
         finally:
             if not verbose:
                 _os.dup2(_old_stderr, _stderr_fd)
                 _os.close(_devnull)
                 _os.close(_old_stderr)
 
-        adapter_path = program_dir / "adapter.gguf"
-        if adapter_path.exists():
-            self._adapter = llama_cpp.llama_adapter_lora_init(
-                self._llm.model, str(adapter_path).encode("utf-8"),
-            )
-            if self._adapter:
-                llama_cpp.llama_set_adapter_lora(self._llm.ctx, self._adapter, 1.0)
-            else:
-                raise RuntimeError(f"Failed to load LoRA adapter: {adapter_path}")
-        else:
-            self._adapter = None
-
         placeholder = "{INPUT_PLACEHOLDER}"
         self._use_special = interpreter not in ("gpt2",)
 

{programasweights-0.2.2.dev1.dist-info → programasweights-0.2.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: programasweights
-Version: 0.2.2.dev1
+Version: 0.2.4
 Summary: Compile natural language specifications into neural programs that run locally via llama.cpp.
 Project-URL: Homepage, https://programasweights.com
 Project-URL: Repository, https://github.com/programasweights/programasweights-python
@@ -58,6 +58,10 @@ program = paw.compile(
 )
 fn = paw.function(program.slug)    # or paw.function(program.id)
 fn("{name: 'Alice',}")  # '{"name": "Alice"}'
+
+# Or compile and load in one step
+fn = paw.compile_and_load("Classify sentiment as positive or negative")
+fn("I love this!")  # "positive"
 ```
 
 ## Two Compilers

{programasweights-0.2.2.dev1.dist-info → programasweights-0.2.4.dist-info}/RECORD

@@ -1,19 +1,19 @@
-programasweights/__init__.py,sha256=8SpAl6BpIBGHy1WsWZQ4EHGnHevzqsjVnMzKoaAjMpo,7852
+programasweights/__init__.py,sha256=PJKL0I4KZPFhMfhUbPmgXOZhMV9dfuKgsGrV6UfTVBE,7882
 programasweights/artifacts.py,sha256=bSRZgYadAYyuH9aIW6P3VocExfGDGAHeDuj8vod5-Bo,1968
 programasweights/cache.py,sha256=7L4D5juQLtARVT4aEqzy9N2Lz6xUv9uk7LBZMuJJFrY,3866
 programasweights/cli.py,sha256=X0M3OddGvsBJXjcaKw0qebaU9koe8r30BHgj9Wbf5WM,6840
-programasweights/client.py,sha256=DlgEEgluTZqQJnfYeVh5bW74ZcyxHmrE_3fFgufHsgQ,5153
+programasweights/client.py,sha256=RlMDMueDtMnZ-OkIe_Zqt_ayTyFDmr7X67dlzG_I5rU,5638
 programasweights/config.py,sha256=zQFnVNfYR4bFx2DsZheMnPy2yd9T4-fe0u2sG1GluE8,1553
 programasweights/convert_peft_to_paw.py,sha256=yvavaAzLUqc-lFHEC_9AOhsuI-fCobu6k-GDMVQdOHk,6521
 programasweights/paw_format.py,sha256=wRXolwtnPgKoopiJC8-yMVZZtiwB-PoAqaWpDFKMoj8,10679
-programasweights/runtime_llamacpp.py,sha256=7uuJx6sD1TVu1pj_qikT8CK4_I7aoYvy3UHOc-5gXOM,6929
+programasweights/runtime_llamacpp.py,sha256=zZ5KFZiH_PBjEXX8aJXl4moDQ9uzikKvajdjCvXZg7M,6973
 programasweights/compiler/__init__.py,sha256=_q0L02k6Cl9zyvxKQlMuW2o6w7cX6OZ_bZ9wHuZqs18,9508
 programasweights/compiler/dummy.py,sha256=PcLwijRNM4q2E9PNUcfCPf_y08QjsteVENR0TVsOIfY,1011
 programasweights/runtime/__init__.py,sha256=S4jp7-eWAcMa0X417U7P6HkTL1j3v8ffQA-nneh9wY0,549
 programasweights/runtime/interpreter.py,sha256=brYCtatSsu23lq87cuki4MViqZX6U32LBw7Z3USbQmA,19632
 programasweights/runtime/interpreter_onnx.py,sha256=kzYKmwg_h0tVhoQlixJGKoHPVhcCK22AcHG59oeYX2Q,21559
-programasweights-0.2.2.dev1.dist-info/METADATA,sha256=9EPrwbhEKMuLSLxN1NV_nZpcjiZhDrrLVecdqJFx5zg,5778
-programasweights-0.2.2.dev1.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
-programasweights-0.2.2.dev1.dist-info/entry_points.txt,sha256=l4ZnfCPU0oMzhGB9T2Fv9AnDbSCMM70KUUJVtSvMJBc,50
-programasweights-0.2.2.dev1.dist-info/licenses/LICENSE,sha256=6GRcDlVhbPMVgkKqBt0cgwyvtvFxpdp4s2vcYKST4r0,1073
-programasweights-0.2.2.dev1.dist-info/RECORD,,
+programasweights-0.2.4.dist-info/METADATA,sha256=AoTSUtBPVXNEyLjMYIoz4pz2AQDVwgpQoyLGZLjIGtY,5913
+programasweights-0.2.4.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+programasweights-0.2.4.dist-info/entry_points.txt,sha256=l4ZnfCPU0oMzhGB9T2Fv9AnDbSCMM70KUUJVtSvMJBc,50
+programasweights-0.2.4.dist-info/licenses/LICENSE,sha256=6GRcDlVhbPMVgkKqBt0cgwyvtvFxpdp4s2vcYKST4r0,1073
+programasweights-0.2.4.dist-info/RECORD,,