Ask questions

import os
import requests

API_ROOT = "https://futuresmart.ai/document-intelligence/api"
COLLECTION_ID = os.environ["DOCUMENT_INTELLIGENCE_COLLECTION_ID"]
HEADERS = {"x-api-key": os.environ["DOCUMENT_INTELLIGENCE_API_KEY"]}


def ask(query, session_id=None):
    body = {"query": query, "stream": False}
    if session_id:
        body["session_id"] = session_id
    r = requests.post(f"{API_ROOT}/v1/{COLLECTION_ID}/chat", headers=HEADERS, json=body, timeout=120)
    r.raise_for_status()
    return r.json()


first = ask("What is the total due on the invoice?")
print(first["response"])

follow_up = ask("And when is it due?", session_id=first["session_id"])
print(follow_up["response"])
print("Sources:", [s["title"] for s in follow_up["data"]["knowledge"]["relevant_sources"]["data"]])

const API_ROOT = "https://futuresmart.ai/document-intelligence/api";
const COLLECTION_ID = process.env.DOCUMENT_INTELLIGENCE_COLLECTION_ID;

async function ask(query, session_id) {
  const res = await fetch(`${API_ROOT}/v1/${COLLECTION_ID}/chat`, {
    method: "POST",
    headers: {
      "x-api-key": process.env.DOCUMENT_INTELLIGENCE_API_KEY,
      "Content-Type": "application/json",
    },
    body: JSON.stringify({ query, session_id, stream: false }),
  });
  if (!res.ok) throw new Error(`${res.status} ${await res.text()}`);
  return res.json();
}

const first = await ask("What is the total due on the invoice?");
console.log(first.response);

const followUp = await ask("And when is it due?", first.session_id);
console.log(followUp.response);
console.log("Sources:", followUp.data.knowledge.relevant_sources.data.map((s) => s.title));

curl -sS -X POST "https://futuresmart.ai/document-intelligence/api/v1/$DOCUMENT_INTELLIGENCE_COLLECTION_ID/chat" \
  -H "x-api-key: $DOCUMENT_INTELLIGENCE_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{"query": "What is the total due on the invoice?", "stream": false}'