Ask Question

import base64
import json
import mimetypes
import sys
from pathlib import Path
from typing import List

import requests

# ---------- Config ----------
BASE_URL = "https://api.gurubase.io/api/v1"
API_KEY = "your-api-key-here"
GURU_SLUG = "your-guru-slug"
QUESTION = "Summarize the attached files."
STREAM = False

# ---------- Attachments (absolute paths) ----------
IMAGE_PATHS: List[str] = [
    # "/path/to/screenshot.png",
]
LOG_PATHS: List[str] = [
    # "/path/to/app.log",
]
PDF_PATHS: List[str] = [
    # "/path/to/doc.pdf",
]


def encode_binary(path: Path, fallback_mime: str) -> dict:
    mime = mimetypes.guess_type(path.name)[0] or fallback_mime
    b64 = base64.b64encode(path.read_bytes()).decode("ascii")
    return {
        "data": f"data:{mime};base64,{b64}",
        "name": path.name,
        "type": mime,
    }


def read_log(path: Path) -> dict:
    return {
        "data": path.read_text(encoding="utf-8", errors="replace"),
        "name": path.name,
    }


def build_payload() -> dict:
    payload: dict = {"question": QUESTION, "stream": STREAM}
    if IMAGE_PATHS:
        payload["images"] = [encode_binary(Path(p), "image/png") for p in IMAGE_PATHS]
    if LOG_PATHS:
        payload["log_files"] = [read_log(Path(p)) for p in LOG_PATHS]
    if PDF_PATHS:
        payload["pdf_files"] = [encode_binary(Path(p), "application/pdf") for p in PDF_PATHS]
    return payload


def main() -> int:
    url = f"{BASE_URL.rstrip('/')}/{GURU_SLUG}/answer/"
    headers = {"x-api-key": API_KEY, "Content-Type": "application/json"}
    payload = build_payload()

    resp = requests.post(url, headers=headers, json=payload, timeout=180, stream=STREAM)

    if STREAM:
        for chunk in resp.iter_content(chunk_size=None, decode_unicode=True):
            if chunk:
                sys.stdout.write(chunk)
                sys.stdout.flush()
        sys.stdout.write("\n")
        return 0 if resp.ok else 1

    try:
        print(json.dumps(resp.json(), indent=2, ensure_ascii=False))
    except ValueError:
        print(resp.text)
    return 0 if resp.ok else 1


if __name__ == "__main__":
    sys.exit(main())