const ws = new WebSocket("wss://api.trellis.sh/v1/chats/ws");

let streamingResponse = false;
let streamingBuffer = "";
let authenticated = false;

ws.onopen = () => {
  // Browser clients cannot set headers on the WS upgrade —
  // authenticate via the first frame instead.
  ws.send(JSON.stringify({ action: "authenticate", token: "YOUR_TOKEN" }));
};

ws.onmessage = (msg) => {
  const event = JSON.parse(msg.data);

  switch (event.event) {
    case "authenticated":
      authenticated = true;
      // Now that we're authenticated, send a message.
      ws.send(JSON.stringify({
        action: "send_message",
        message: "Show me orders by region as a bar chart",
        chat_id: "chat_abc123",
        integration_id: "550e8400-e29b-41d4-a716-446655440000",
      }));
      break;
    case "chat_metadata":
      console.log("Chat:", event.data.id, "| User message:", event.data.user_message_id);
      streamingResponse = true;
      streamingBuffer = "";
      break;
    case "visualization":
      if (event.data.type === "chart") {
        renderChart(event.data.chart_type, event.data.title, event.data.data);
      } else if (event.data.type === "table") {
        renderTable(event.data.headers, event.data.rows);
      }
      break;
    case "text_delta":
      // Append the incremental chunk and re-render
      streamingBuffer += event.data.content;
      renderStreamingMessage(streamingBuffer);
      break;
    case "message":
      // Replace buffer with authoritative full content + id
      displayMessage(event.data.content, event.data.id);
      streamingBuffer = "";
      streamingResponse = false;
      break;
    case "interrupted":
      // Any text already in streamingBuffer has been persisted server-side
      console.log("Response was interrupted");
      streamingResponse = false;
      break;
    case "deleted":
      console.log("Message deleted:", event.data.message_id);
      break;
    case "error":
      showError(event.data.error);
      streamingResponse = false;
      break;
    case "heartbeat":
      // Server keepalive — no action required
      break;
  }
};

// Interrupt a streaming response
function interruptResponse() {
  if (streamingResponse) {
    ws.send(JSON.stringify({ action: "interrupt" }));
  }
}

// Delete a message
function deleteMessage(messageId) {
  ws.send(JSON.stringify({ action: "delete_message", message_id: messageId }));
}

// Edit a message
function editMessage(messageId, newContent, integrationId) {
  ws.send(JSON.stringify({
    action: "edit_message",
    message_id: messageId,
    content: newContent,
    integration_id: integrationId,
  }));
}

import asyncio
import json
import websockets

async def main():
    uri = "wss://api.trellis.sh/v1/chats/ws"

    async with websockets.connect(
        uri,
        additional_headers=[("Authorization", "Bearer YOUR_TOKEN")],
    ) as ws:
        # Wait for auth confirmation frame
        auth = await ws.recv()
        print(json.loads(auth))

        # Send message — chat_id is required
        await ws.send(json.dumps({
            "action": "send_message",
            "message": "Show me orders by region as a bar chart",
            "chat_id": "chat_abc123",
            "integration_id": "550e8400-e29b-41d4-a716-446655440000"
        }))
        
        # Receive events
        streaming_buffer = ""
        while True:
            response = await ws.recv()
            event = json.loads(response)

            if event["event"] == "chat_metadata":
                print("Chat:", event["data"]["id"], "| User message:", event["data"]["user_message_id"])
            elif event["event"] == "visualization":
                data = event["data"]
                if data["type"] == "chart":
                    print(f"Chart ({data['chart_type']}): {data.get('title')}")
                    print(f"  Values: {data['data']['values']}")
                elif data["type"] == "table":
                    print(f"Table headers: {data['headers']}")
                    print(f"  Rows: {len(data['rows'])}")
            elif event["event"] == "text_delta":
                # Append the incremental chunk
                streaming_buffer += event["data"]["content"]
            elif event["event"] == "message":
                # Authoritative full content
                print("Response:", event["data"]["content"])
                break
            elif event["event"] == "interrupted":
                print("Response interrupted; partial so far:", streaming_buffer)
                break
            elif event["event"] == "error":
                print("Error:", event["data"]["error"])
                break
            elif event["event"] == "heartbeat":
                # Server keepalive — no action required
                continue

asyncio.run(main())

{"event": "authenticated", "data": {"status": "ok"}}

{"event": "chat_metadata", "data": {"id": "chat_abc123", "user_message_id": "msg_001"}}
{"event": "processing", "data": {"status": "thinking"}}
{"event": "processing", "data": {"status": "analyzing"}}
{"event": "visualization", "data": {"id": "ab12xy3456", "type": "chart", "chart_type": "bar", "title": "Orders by Region", "data": {"values": [{"label": "North", "value": 412}, {"label": "South", "value": 289}, {"label": "East", "value": 531}], "x_axis_label": "Region", "y_axis_label": "Orders"}}}
{"event": "text_delta", "data": {"content": "There were 1,232 orders "}}
{"event": "text_delta", "data": {"content": "placed last month, with the East "}}
{"event": "text_delta", "data": {"content": "region leading at 531."}}
{"event": "message", "data": {"content": "There were 1,232 orders placed last month, with the East region leading at 531.", "id": "msg_xyz789"}}

{"event": "chat_metadata", "data": {"id": "chat_abc123", "user_message_id": "msg_001"}}
{"event": "processing", "data": {"status": "thinking"}}
{"event": "text_delta", "data": {"content": "Looking at the regional breakdown,"}}
{"event": "text_delta", "data": {"content": " the East region shows"}}
{"event": "interrupted", "data": {"status": "interrupted"}}

{"event": "deleted", "data": {"message_id": "msg_xyz789"}}

{"event": "error", "data": {"error": "Datasource not found", "code": "INTEGRATION_NOT_FOUND", "retryable": false, "debug_id": "4a8c0ae6", "error_type": "ValueError"}}