Build grounded answers into your product

import os
import requests

BASE = f"{os.environ['MITHUNAI_URL']}/arukz/api/v1"
HEADERS = {"Authorization": f"Bearer {os.environ['MITHUNAI_API_KEY']}"}


def ask(assistant_id: str, question: str, conversation_id: str | None = None) -> dict:
    if conversation_id is None:
        conversation = requests.post(
            f"{BASE}/conversations", headers=HEADERS, json={"assistant_id": assistant_id}, timeout=30
        )
        conversation.raise_for_status()
        conversation_id = conversation.json()["id"]

    turn = requests.post(
        f"{BASE}/conversations/{conversation_id}/messages",
        headers=HEADERS,
        json={"text": question},
        timeout=120,
    )
    turn.raise_for_status()
    answer = turn.json()["answer"]
    return {
        "conversation_id": conversation_id,
        "text": answer["text"],
        "abstained": answer["abstained"],
        "citations": [{"title": c["title"], "url": c["url"]} for c in answer["citations"]],
    }

const BASE = `${process.env.MITHUNAI_URL}/arukz/api/v1`
const HEADERS = {
  Authorization: `Bearer ${process.env.MITHUNAI_API_KEY}`,
  'Content-Type': 'application/json',
}

export async function ask(assistantId, question, conversationId) {
  if (!conversationId) {
    const res = await fetch(`${BASE}/conversations`, {
      method: 'POST',
      headers: HEADERS,
      body: JSON.stringify({ assistant_id: assistantId }),
    })
    if (!res.ok) throw new Error(`MITHUNAI ${res.status}`)
    conversationId = (await res.json()).id
  }

  const res = await fetch(`${BASE}/conversations/${conversationId}/messages`, {
    method: 'POST',
    headers: HEADERS,
    body: JSON.stringify({ text: question }),
  })
  if (!res.ok) throw new Error(`MITHUNAI ${res.status}`)
  const { answer } = await res.json()
  return {
    conversationId,
    text: answer.text,
    abstained: answer.abstained,
    citations: answer.citations.map(({ title, url }) => ({ title, url })),
  }
}