Regulatory filings and compliance search

import os, requests, anthropic

claude = anthropic.Anthropic()
HEADERS = {"Authorization": f"Bearer {os.environ['SIFTQ_API_KEY']}"}

def find_filings(query: str, k: int = 5) -> list[dict]:
    r = requests.post(
        "https://api.siftq.com/v1/search",
        headers=HEADERS,
        json={
            "q": query,
            "scope": "document",
            "size": str(k),
            "includeRawContent": True,   # +3 credits / page, pulls full text
        },
        timeout=30,
    )
    return r.json().get("documents", [])

def material_risks(company_ticker: str) -> str:
    docs = find_filings(f"{company_ticker} 10-K risk factors")
    if not docs:
        return "No filings found."
    primary = docs[0]
    msg = claude.messages.create(
        model="claude-sonnet-4-6", max_tokens=1024,
        messages=[{
            "role": "user",
            "content": (
                f"Extract the top 5 material risks from this 10-K filing.\n"
                f"Cite section numbers when present.\n\n"
                f"FILING: {primary.get('title')}\n"
                f"SOURCE: {primary.get('link')}\n\n"
                f"{primary.get('snippet', '')}"
            ),
        }],
    )
    return msg.content[0].text

print(material_risks("NVDA"))