Citations

thread = ks.threads.create(title="Q4 deep-dive")

for ev in ks.threads.stream(thread_id=thread.id, message="What drove revenue?"):
    if ev.type == "message_delta":
        print(ev.text, end="", flush=True)
    elif ev.type == "citation":
        # ev.chunk_id, ev.start_char, ev.length, ev.quote
        print(f"\n  ↳ [{ev.chunk_id}] {ev.quote!r}")

const thread = await ks.threads.create({ title: "Q4 deep-dive" });

for await (const ev of ks.threads.stream({
  threadId: thread.id,
  message: "What drove revenue?",
})) {
  if (ev.type === "message_delta") process.stdout.write(ev.text);
  if (ev.type === "citation") {
    console.log(`\n  ↳ [${ev.chunkId}] ${JSON.stringify(ev.quote)}`);
  }
}

curl -N -X POST $KS_BASE_URL/v1/threads/$THREAD/stream \
  -H "Authorization: Bearer $KS_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{"message": "What drove revenue?"}'
# event: message_delta
# data: {"text": "Revenue grew 18% in Q4..."}
# event: citation
# data: {"chunk_id": "01J...", "start_char": 0, "length": 21, "quote": "Revenue grew 18% in Q4"}

chunk = ks.chunks.get(chunk_id="01J7X...")
print(chunk.document_id, chunk.page_number, chunk.bbox)
print(chunk.content)

const chunk = await ks.chunks.get({ chunkId: "01J7X..." });
console.log(chunk.documentId, chunk.pageNumber, chunk.bbox);