axnwork-cli 0.2.0__py3-none-any.whl

axon/providers.py

@@ -0,0 +1,44 @@
+"""Fetch available models from LLM provider APIs."""
+import httpx
+
+
+def fetch_models(provider: str, api_key: str, api_base: str = "") -> list[dict]:
+    """Fetch models from provider. Returns list of {label, value}."""
+    try:
+        if provider == "anthropic" and api_key:
+            resp = httpx.get("https://api.anthropic.com/v1/models", headers={
+                "x-api-key": api_key, "anthropic-version": "2023-06-01",
+            }, timeout=10)
+            if resp.status_code == 200:
+                models = [m["id"] for m in resp.json().get("data", [])]
+                models = [m for m in models if "claude-2" not in m and "instant" not in m]
+                models.sort(reverse=True)
+                return [{"label": m, "value": f"anthropic/{m}"} for m in models]
+
+        if provider == "openai" and api_key:
+            resp = httpx.get("https://api.openai.com/v1/models", headers={
+                "Authorization": f"Bearer {api_key}",
+            }, timeout=10)
+            if resp.status_code == 200:
+                prefixes = ("gpt-", "o1", "o3", "o4", "gpt4", "gpt5", "chatgpt")
+                models = [m["id"] for m in resp.json().get("data", []) if m["id"].startswith(prefixes)]
+                models.sort(reverse=True)
+                return [{"label": m, "value": f"openai/{m}"} for m in models]
+
+        if provider == "deepseek" and api_key:
+            resp = httpx.get("https://api.deepseek.com/v1/models", headers={
+                "Authorization": f"Bearer {api_key}",
+            }, timeout=10)
+            if resp.status_code == 200:
+                models = sorted(m["id"] for m in resp.json().get("data", []))
+                return [{"label": m, "value": f"deepseek/{m}"} for m in models]
+
+        if provider == "ollama":
+            base = api_base or "http://localhost:11434"
+            resp = httpx.get(f"{base}/api/tags", timeout=5)
+            if resp.status_code == 200:
+                models = sorted(m["name"] for m in resp.json().get("models", []))
+                return [{"label": m, "value": f"ollama/{m}"} for m in models]
+    except Exception:
+        pass
+    return []

axon/session.py

@@ -0,0 +1,26 @@
+"""Mining session persistence."""
+import json
+
+from axon.config import AXON_HOME
+
+SESSIONS_DIR = AXON_HOME / "sessions"
+
+
+def load_session(task_id: str) -> dict | None:
+    path = SESSIONS_DIR / f"{task_id}.json"
+    if not path.exists():
+        return None
+    try:
+        return json.loads(path.read_text())
+    except Exception:
+        return None
+
+
+def save_session(task_id: str, data: dict):
+    SESSIONS_DIR.mkdir(parents=True, exist_ok=True)
+    (SESSIONS_DIR / f"{task_id}.json").write_text(json.dumps(data, indent=2))
+
+
+def delete_session(task_id: str):
+    path = SESSIONS_DIR / f"{task_id}.json"
+    path.unlink(missing_ok=True)

axon/wallet.py

@@ -0,0 +1,45 @@
+"""ETH wallet — generate, load, sign."""
+import json
+
+from eth_account import Account
+from eth_account.messages import encode_defunct
+
+from axon.config import AXON_HOME
+
+WALLET_FILE = AXON_HOME / "wallet.json"
+
+
+def generate_wallet() -> dict:
+    """Generate a new ETH wallet. Returns {address, private_key}."""
+    acct = Account.create()
+    return {"address": acct.address, "private_key": acct.key.hex()}
+
+
+def save_wallet(wallet: dict):
+    """Save wallet to ~/.axon/wallet.json"""
+    WALLET_FILE.parent.mkdir(parents=True, exist_ok=True)
+    WALLET_FILE.write_text(json.dumps(wallet, indent=2) + "\n")
+    WALLET_FILE.chmod(0o600)  # owner-only read/write
+
+
+def load_wallet() -> dict | None:
+    """Load wallet from disk. Returns {address, private_key} or None."""
+    if not WALLET_FILE.exists():
+        return None
+    try:
+        return json.loads(WALLET_FILE.read_text())
+    except Exception:
+        return None
+
+
+def sign_message(message: str, private_key: str) -> str:
+    """Sign a message with private key. Returns hex signature."""
+    msg = encode_defunct(text=message)
+    signed = Account.sign_message(msg, private_key=private_key)
+    return signed.signature.hex()
+
+
+def get_address() -> str:
+    """Get wallet address or empty string."""
+    wallet = load_wallet()
+    return wallet["address"] if wallet else ""