Add files via upload

Author: WENGENG-boop

burn_tokens (3).py

@@ -0,0 +1,138 @@
+"""
+API Token Burner — 交互式配置 + 多线程版
+"""
+
+import anthropic
+import time
+import threading
+from concurrent.futures import ThreadPoolExecutor, as_completed
+
+lock = threading.Lock()
+stats = {"input": 0, "output": 0, "requests": 0, "errors": 0}
+stop_flag = threading.Event()
+
+BURN_PROMPT = (
+    "Explain in detail how the internet works, including TCP/IP, DNS, HTTP, "
+    "TLS handshake, CDN, load balancing, and BGP routing. "
+    "Give concrete examples for each concept. Be thorough and detailed."
+)
+
+def ask(prompt, default=None, cast=str):
+    suffix = f" [{default}]" if default is not None else ""
+    while True:
+        val = input(f"{prompt}{suffix}: ").strip()
+        if val == "" and default is not None:
+            return default
+        if val == "":
+            print("  ⚠  不能为空，请重新输入")
+            continue
+        try:
+            return cast(val)
+        except ValueError:
+            print(f"  ⚠  请输入有效的数字")
+
+def single_request(client, model, max_tokens, thread_id):
+    if stop_flag.is_set():
+        return
+
+    try:
+        response = client.messages.create(
+            model=model,
+            max_tokens=max_tokens,
+            messages=[{"role": "user", "content": BURN_PROMPT}],
+        )
+        in_tok  = response.usage.input_tokens
+        out_tok = response.usage.output_tokens
+        with lock:
+            stats["input"]    += in_tok
+            stats["output"]   += out_tok
+            stats["requests"] += 1
+            total = stats["input"] + stats["output"]
+            print(f"  [T{thread_id:02d}] ✓  in={in_tok:,}  out={out_tok:,}  | cumulative={total:,}")
+
+    except anthropic.RateLimitError:
+        with lock:
+            stats["errors"] += 1
+            print(f"  [T{thread_id:02d}] ⚠  Rate limit — waiting 5s...")
+        time.sleep(5)
+
+    except anthropic.APIStatusError as e:
+        with lock:
+            stats["errors"] += 1
+            print(f"  [T{thread_id:02d}] ✗  {e.status_code}: {e.message}")
+        if e.status_code == 402:
+            stop_flag.set()
+            raise
+
+    except Exception as e:
+        with lock:
+            stats["errors"] += 1
+            print(f"  [T{thread_id:02d}] ✗  {type(e).__name__}: {e}")
+
+def print_summary(batch):
+    print("\n" + "=" * 60)
+    print(f"  Batches    : {batch}")
+    print(f"  Requests   : {stats['requests']}")
+    print(f"  Errors     : {stats['errors']}")
+    print(f"  Input tok  : {stats['input']:,}")
+    print(f"  Output tok : {stats['output']:,}")
+    print(f"  Total tok  : {stats['input'] + stats['output']:,}")
+    print("=" * 60)
+
+def main():
+    print("=" * 60)
+    print("  API Token Burner — 交互式配置")
+    print("=" * 60)
+    print()
+
+    base_url   = ask("Base URL  ", default="https://api.anthropic.com")
+    api_key    = ask("API Key   ")
+    model      = ask("Model     ", default="claude-haiku-4-5-20251001")
+    threads    = ask("并发线程数  ", default=5, cast=int)
+    max_tokens = ask("每个请求 output tokens", default=2000, cast=int)
+
+    print()
+    print("=" * 60)
+    print(f"  Base URL : {base_url}")
+    print(f"  Model    : {model}")
+    print(f"  Threads  : {threads}")
+    print(f"  Max out  : {max_tokens} tokens/request")
+    print("=" * 60)
+    print("  开始执行，Ctrl+C 随时停止\n")
+
+    client = anthropic.Anthropic(
+        api_key=api_key,
+        base_url=base_url,
+        timeout=120.0,
+    )
+
+    batch = 0
+    try:
+        while not stop_flag.is_set():
+            batch += 1
+            print(f"--- Batch #{batch} ({threads} concurrent) ---")
+            with ThreadPoolExecutor(max_workers=threads) as executor:
+                futures = {
+                    executor.submit(single_request, client, model, max_tokens, i+1): i
+                    for i in range(threads)
+                }
+                for future in as_completed(futures):
+                    try:
+                        future.result()
+                    except anthropic.APIStatusError as e:
+                        if e.status_code == 402:
+                            print("\n💸  余额已耗尽！")
+                            stop_flag.set()
+
+            if stop_flag.is_set():
+                break
+
+            time.sleep(0.3)
+
+    except KeyboardInterrupt:
+        print("\n⏹  手动停止。")
+
+    print_summary(batch)
+
+if __name__ == "__main__":
+    main()