Read a live sales chat

import asyncio, json, os
from urllib.request import Request, urlopen
from websockets.asyncio.client import connect

URL = "wss://sage.levanto.ai/v1/stream"
HEADERS = {"Authorization": f"Bearer {os.environ['SAGE_API_KEY']}"}

SIGNALS = [
    {
        "id": "buying",
        "kind": "yesno",
        "instructions": "Will the customer buy the car?",
    },
    {
        "id": "shopping_around",
        "kind": "yesno",
        "instructions": "Is the customer considering an offer from another dealer?",
    },
    {
        "id": "needs_approval",
        "kind": "yesno",
        "instructions": "Does the customer still need someone else's approval before deciding?",
    },
]

CONCERN = {
    "id": "concern",
    "kind": "choice",
    "instructions": "What is the customer's main concern right now?",
    "options": [
        {"option": "price"},
        {"option": "trade-in value"},
        {"option": "reliability"},
        {"option": "timing"},
        {"option": "none"},
    ],
}

QUESTIONS = SIGNALS + [CONCERN]

CHAT = [
    "Hannah: Hi Priya, thanks for coming in. Still looking at the Aurora C5?",
    "Priya: Yes, the two-year-old grey one. You have it at 31,500 euros, right?",
    "Hannah: That's right. It has 28,000 km and one owner.",
    "Priya: Northgate Auto offered me the same model for 29,900 yesterday.",
    "Hannah: I can look at that. Are you trading anything in?",
    "Priya: My old hatchback. Your colleague said 4,000, which feels low.",
    "Hannah: If I go to 30,200 and 4,800 for the trade-in, could we close today?",
    "Priya: I'd have to run it past my husband first. But that's much better.",
    "Priya: He says yes! Can I pay the deposit now?",
]

async def feed(ws):
    for line in CHAT:  # in production: each message as it arrives
        await ws.send(json.dumps({"type": "append", "text": line + "\n"}))
        await asyncio.sleep(1)
    await ws.send(json.dumps({"type": "close"}))

async def main():
    async with connect(URL, additional_headers=HEADERS) as ws:
        await ws.send(json.dumps({"type": "start", "questions": QUESTIONS}))
        feeder = asyncio.create_task(feed(ws))
        async for raw in ws:  # ends on a normal close; any other close raises
            frame = json.loads(raw)
            if frame["type"] == "decision":
                for qid in frame["changed"]:
                    r = frame["answers"][qid]["result"]
                    print(qid, r.get("answer", r.get("chosen")), r["probability"])
            elif frame["type"] == "error":
                print("error:", frame.get("code"), frame["detail"])
        await feeder

asyncio.run(main())

FIELDS = {
    "car": "the car the customer wants",
    "rival_dealer": "another dealer the customer mentions",
    "rival_offer": {"type": "money", "definition": "the price another dealer offered"},
    "final_price": {
        "type": "money",
        "definition": "the latest price the salesperson offered",
    },
    "trade_in": {"type": "money", "definition": "the latest trade-in value offered"},
}

def extract(content, fields):
    body = json.dumps({"content": content, "extract": fields}).encode()
    req = Request("https://sage.levanto.ai/v1/extract", data=body, method="POST",
                  headers={**HEADERS, "Content-Type": "application/json"})
    with urlopen(req, timeout=30) as r:
        return json.load(r)["values"]

print(extract("\n".join(CHAT), FIELDS))