Document AI Overview

1
import os, time
2
from sarvamai import SarvamAI
3
4
client = SarvamAI(api_subscription_key=os.environ["SARVAM_API_KEY"])
5
6
# Create and submit a digitise job in one call
7
with open("/tmp/test-doc.pdf", "rb") as f:
8
    job = client.doc_ai.digitise(
9
        file=[("test-doc.pdf", f, "application/pdf")],
10
        language="en-IN",
11
        output_format="md",
12
    )
13
print("job created:", job.job_id, "| status:", job.status)
14
15
# Poll until the job reaches a terminal state
16
TERMINAL = {"completed", "partially_completed", "failed", "rejected"}
17
while True:
18
    st = client.doc_ai.get_status(job_id=job.job_id)
19
    print(f"status={st.status} pages={st.usage.pages_processed}/{st.usage.pages_total}")
20
    if st.status.lower() in TERMINAL:
21
        break
22
    time.sleep(5)
23
24
# Fetch the output
25
if st.status.lower() in ("completed", "partially_completed"):
26
    dl = client.doc_ai.get_download_url(job_id=job.job_id)
27
    print("download:", dl.method, dl.url)

1
import json, os, time
2
from sarvamai import SarvamAI
3
4
client = SarvamAI(api_subscription_key=os.environ["SARVAM_API_KEY"])
5
6
# Define the fields you want back
7
schema = {
8
    "type": "object",
9
    "properties": {
10
        "policy_number": {"type": "string", "description": "Insurance policy number"},
11
        "insured_name":  {"type": "string", "description": "Name of the insured person"},
12
        "sum_insured":   {"type": "number", "description": "Total sum insured, in INR"},
13
    },
14
}
15
16
# Create and submit an extract job in one call
17
with open("/tmp/policy.pdf", "rb") as f:
18
    job = client.doc_ai.extract(
19
        file=[("policy.pdf", f, "application/pdf")],
20
        schema=json.dumps(schema),   # or: config_id="cfg_abc123"
21
        language="en-IN",
22
        output_format="json",
23
    )
24
print("job created:", job.job_id, "| status:", job.status)
25
26
# Poll until the job reaches a terminal state
27
TERMINAL = {"completed", "partially_completed", "failed", "rejected"}
28
while True:
29
    st = client.doc_ai.get_status(job_id=job.job_id)
30
    print(f"status={st.status} pages={st.usage.pages_processed}/{st.usage.pages_total}")
31
    if st.status.lower() in TERMINAL:
32
        break
33
    time.sleep(5)
34
35
# Fetch the extracted fields
36
if st.status.lower() in ("completed", "partially_completed"):
37
    results = client.doc_ai.get_results(job_id=job.job_id)
38
    print(results.result)        # {"policy_number": ..., "insured_name": ..., "sum_insured": ...}