feat: vibed out some slop over here
This commit is contained in:
@@ -0,0 +1,55 @@
|
||||
"""Minimal end-to-end example: load config, send a chat completion, stream one.
|
||||
|
||||
Prerequisites:
|
||||
1. `raycast-api init` has been run (so `config.json` exists next to this file
|
||||
or at the path you point `--config` to). See the README.
|
||||
2. `RAYCAST_BEARER` is exported (your Raycast OAuth access token, `rca_...`).
|
||||
3. `RAYCAST_DEVICE_ID` is exported (any stable 64-hex string; the CLI mints
|
||||
one on first `ask` and persists it to `~/.config/raycast-api/device_id`,
|
||||
so a one-liner is: `export RAYCAST_DEVICE_ID=$(cat ~/.config/raycast-api/device_id)`).
|
||||
|
||||
Run:
|
||||
python examples/basic_usage.py
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import asyncio
|
||||
import os
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
from raycast_api import Client, Config, Message
|
||||
|
||||
CONFIG_PATH = Path(os.environ.get("RAYCAST_CONFIG", "config.json"))
|
||||
MODEL = os.environ.get("RAYCAST_MODEL", "Claude Sonnet 4.6")
|
||||
|
||||
|
||||
async def main() -> None:
|
||||
bearer = os.environ.get("RAYCAST_BEARER")
|
||||
device_id = os.environ.get("RAYCAST_DEVICE_ID")
|
||||
if not bearer or not device_id:
|
||||
sys.exit("set RAYCAST_BEARER and RAYCAST_DEVICE_ID before running")
|
||||
if not CONFIG_PATH.is_file():
|
||||
sys.exit(f"no config at {CONFIG_PATH} — run `raycast-api init` first")
|
||||
|
||||
config = Config.load(CONFIG_PATH)
|
||||
async with Client(
|
||||
config=config, bearer_token=bearer, device_id=device_id
|
||||
) as client:
|
||||
result = await client.chat.complete(
|
||||
model=MODEL, messages=[Message.user("Reply with the single word: OK")]
|
||||
)
|
||||
print(f"[complete] {result.text!r}")
|
||||
|
||||
print("[stream] ", end="", flush=True)
|
||||
async for chunk in client.chat.stream(
|
||||
model=MODEL, messages=[Message.user("Write a one-line haiku about HMAC.")]
|
||||
):
|
||||
if chunk.text:
|
||||
print(chunk.text, end="", flush=True)
|
||||
print()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(main())
|
||||
Reference in New Issue
Block a user