Conversation Continuity

payload = {
    "chat_id": current_chat_id,
    "from_checkpoint_id": "INITIAL",
    "message": "Let's start fresh: analyze NVIDIA's data center segment.",
    "persistence_mode": "enabled",
    "research_effort": "standard",
}

import json
import requests


def turn(api_key: str, message: str, chat_id: str | None, checkpoint_id: str | None):
    endpoint = "https://agents.bigdata.com/v1/research-agent"
    headers = {"X-API-KEY": api_key, "Content-Type": "application/json"}
    payload: dict = {
        "message": message,
        "research_effort": "standard",
        "persistence_mode": "enabled",
    }
    if chat_id:
        payload["chat_id"] = chat_id
    if checkpoint_id:
        payload["from_checkpoint_id"] = checkpoint_id

    answer = ""
    new_chat_id: str | None = None
    new_checkpoint_id: str | None = None

    with requests.post(endpoint, headers=headers, json=payload, stream=True, timeout=300) as r:
        r.raise_for_status()
        for raw_line in r.iter_lines(decode_unicode=True):
            if not raw_line or not raw_line.startswith("data: "):
                continue
            try:
                event = json.loads(raw_line[6:])
            except json.JSONDecodeError:
                continue

            new_chat_id = event.get("chat_id") or new_chat_id
            msg = event.get("message", {})
            if msg.get("type") == "ANSWER":
                answer += msg.get("content", "")
            elif msg.get("type") == "COMPLETE":
                new_checkpoint_id = msg.get("checkpoint_id")
            elif msg.get("type") == "ERROR":
                raise RuntimeError(msg.get("error"))

    return answer, new_chat_id, new_checkpoint_id


# First turn: no chat_id yet, no checkpoint.
api_key = "..."
answer1, chat_id, ckpt = turn(api_key, "Summarize NVIDIA's Q2 FY26 results.", None, None)

# Follow-up turn: reuse chat_id; pass the checkpoint from the previous turn.
answer2, chat_id, ckpt = turn(api_key, "How does that compare to AMD?", chat_id, ckpt)

# Another follow-up: same pattern.
answer3, chat_id, ckpt = turn(api_key, "Which one has better margins?", chat_id, ckpt)

# Suppose we have a stored checkpoint after the first turn (`ckpt_after_turn_1`)
# and we want to branch off in a different direction.
branch_answer, chat_id, branch_ckpt = turn(
    api_key,
    "Actually, focus on the gaming segment instead.",
    chat_id,
    ckpt_after_turn_1,
)

import json
import requests


def workflow_turn(api_key: str, template_id: str, message: str, execution_id: str | None):
    endpoint = "https://agents.bigdata.com/v1/workflow/execute"
    headers = {"X-API-KEY": api_key, "Content-Type": "application/json"}
    payload: dict = {"template": template_id, "input": {"message": message}}
    if execution_id:
        payload["execution_id"] = execution_id

    answer = ""
    new_execution_id: str | None = None

    with requests.post(endpoint, headers=headers, json=payload, stream=True, timeout=600) as r:
        r.raise_for_status()
        for raw_line in r.iter_lines(decode_unicode=True):
            if not raw_line or not raw_line.startswith("data: "):
                continue
            try:
                event = json.loads(raw_line[6:])
            except json.JSONDecodeError:
                continue

            new_execution_id = event.get("execution_id") or new_execution_id
            delta = event.get("delta", {})
            if delta.get("type") == "ANSWER":
                answer += delta.get("content", "")
            elif delta.get("type") == "ERROR":
                raise RuntimeError(delta.get("error"))

    return answer, new_execution_id