Test an AI assistant before you launch

import os
import requests

BASE = f"{os.environ['MITHUNAI_URL']}/arukz/api/v1"
HEADERS = {"Authorization": f"Bearer {os.environ['MITHUNAI_API_KEY']}"}
ASSISTANT_ID = "<assistant id>"  # a UUID from GET /assistants

questions = [line.strip() for line in open("questions.txt") if line.strip()]

for question in questions:
    # A fresh conversation per question keeps the answers independent.
    conversation = requests.post(
        f"{BASE}/conversations", headers=HEADERS, json={"assistant_id": ASSISTANT_ID}, timeout=30
    ).json()
    turn = requests.post(
        f"{BASE}/conversations/{conversation['id']}/messages",
        headers=HEADERS,
        json={"text": question},
        timeout=120,
    ).json()
    answer = turn["answer"]
    sources = ", ".join(c["title"] for c in answer["citations"])
    print(f"{'ABSTAINED' if answer['abstained'] else 'ANSWERED '} | {question} | {sources}")