Build a statute lookup chatbot

import requests

BASE = "https://api.vaquill.ai/api/v1"
HEADERS = {"Authorization": "Bearer vq_key_..."}


def find_section(message, state=None, corpus_type="STATE"):
    body = {"query": message, "limit": 3}
    if corpus_type:
        body["corpusType"] = corpus_type
    if state:
        body["state"] = state
    resp = requests.post(f"{BASE}/statutes/search", headers=HEADERS, json=body)
    resp.raise_for_status()
    results = resp.json()["results"]
    return results[0] if results else None


def section_text(act_id):
    resp = requests.get(f"{BASE}/statutes/section/{act_id}/body", headers=HEADERS)
    resp.raise_for_status()
    data = resp.json()
    return data.get("plain") if data.get("available") else None


def answer(message, state=None):
    hit = find_section(message, state=state)
    if not hit:
        return "I could not find a statute on that. Try rephrasing your question."

    text = section_text(hit["actId"])
    if not text:
        return "I found a matching section but its text is not available right now."

    citation = hit.get("citation") or hit["actId"]
    source_url = (
        hit.get("stateHtmlUrl")
        or hit.get("htmlUrl")
        or hit.get("govInfoHtmlUrl")
        or hit.get("externalUrl")
    )

    prompt = (
        "Answer the user's question using ONLY the statute text below. "
        "If it does not answer the question, say you could not find it. "
        "Cite the statute by its citation in your answer.\n\n"
        f"Citation: {citation}\n"
        f"Statute text:\n{text}\n\n"
        f"Question: {message}\n"
    )
    reply = call_llm(prompt)  # your LLM call; see note below
    return f"{reply}\n\nSource: {citation}\n{source_url}"


# call_llm(prompt) is a placeholder for the LLM you run in your stack.
# Swap in whatever you use; it takes a prompt string and returns text.

# Plain terminal loop. Replace input()/print() with your Slack, Discord,
# or web handler and the rest is unchanged.
if __name__ == "__main__":
    while True:
        message = input("You: ")
        if message.strip().lower() in {"quit", "exit"}:
            break
        print("Bot:", answer(message, state="ca"))