Voice Conversations

from datagrid import Datagrid
import asyncio
import websockets
import json
import base64

client = Datagrid()

# 1. Prepare the session via REST
session = client.voice.start_session(agent_id="agent_abc123")

# 2. Connect to the WebSocket URL
async def voice_session():
    async with websockets.connect(session.url) as ws:
        # 3. Send the pre-built start message
        await ws.send(json.dumps(session.start_message))

        # 4. Wait for ready
        while True:
            msg = json.loads(await ws.recv())
            print(f"← {msg['type']}")
            if msg["type"] == "started":
                print(f"  Session: {msg['payload']['session_id']}")
            if msg["type"] == "ready":
                break

        # 5. Stream audio
        with open("recording.pcm", "rb") as f:
            while chunk := f.read(4096):
                await ws.send(json.dumps({
                    "type": "audio",
                    "payload": {"data": base64.b64encode(chunk).decode()}
                }))

        # 6. End session and get transcript
        await ws.send(json.dumps({"type": "stop"}))
        while True:
            msg = json.loads(await ws.recv())
            if msg["type"] == "audio":
                audio_bytes = base64.b64decode(msg["payload"]["data"])
                # Play or save audio_bytes...
            elif msg["type"] == "ended":
                print("Transcript:", msg["payload"]["transcript"])
                break

asyncio.run(voice_session())

import Datagrid from "datagrid-ai";
import WebSocket from "ws";

const client = new Datagrid();

// 1. Prepare the session via REST
const session = await client.voice.startSession({
  agent_id: "agent_abc123",
});

// 2. Connect to the WebSocket URL
const ws = new WebSocket(session.url);

ws.on("open", () => {
  // 3. Send the pre-built start message
  ws.send(JSON.stringify(session.start_message));
});

ws.on("message", (data) => {
  const msg = JSON.parse(data.toString());

  switch (msg.type) {
    case "started":
      console.log("Session:", msg.payload.session_id);
      break;
    case "ready":
      console.log("Ready — start sending audio");
      // Send audio chunks here...
      break;
    case "audio":
      // Decode and play: Buffer.from(msg.payload.data, "base64")
      break;
    case "transcript":
      console.log(`[${msg.payload.role}] ${msg.payload.text}`);
      break;
    case "ended":
      console.log("Transcript:", msg.payload.transcript);
      ws.close();
      break;
    case "error":
      console.error("Error:", msg.payload.message);
      break;
  }
});

# 1. Prepare the session
curl -X POST https://api.datagrid.com/v1/voice \
  -H "Authorization: Bearer $DATAGRID_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{"agent_id": "agent_abc123"}'

# Response:
# {
#   "object": "voice.session",
#   "url": "wss://api.datagrid.com/ws/voice?token=dg_live_...",
#   "agent_id": "agent_abc123",
#   "start_message": {"type":"start","payload":{"agent_id":"agent_abc123"}}
# }

# 2. Connect and send the start_message
wscat -c "wss://api.datagrid.com/ws/voice?token=$DATAGRID_API_KEY"
> {"type":"start","payload":{"agent_id":"agent_abc123"}}

import asyncio
import websockets
import json
import base64
import os

API_KEY = os.environ["DATAGRID_API_KEY"]

async def voice_session():
    uri = f"wss://api.datagrid.com/ws/voice?token={API_KEY}"

    async with websockets.connect(uri) as ws:
        # 1. Start a session
        await ws.send(json.dumps({
            "type": "start",
            "payload": {
                "agent_id": "agent_abc123"
            }
        }))

        # 2. Wait for ready
        while True:
            msg = json.loads(await ws.recv())
            print(f"← {msg['type']}")
            if msg["type"] == "ready":
                break

        # 3. Stream audio and collect responses...
        # (same as Option A from step 5 onward)

asyncio.run(voice_session())

const API_KEY = process.env.DATAGRID_API_KEY;
const WebSocket = require("ws");

const ws = new WebSocket(
  `wss://api.datagrid.com/ws/voice?token=${API_KEY}`
);

ws.on("open", () => {
  ws.send(JSON.stringify({
    type: "start",
    payload: { agent_id: "agent_abc123" }
  }));
});

ws.on("message", (data) => {
  const msg = JSON.parse(data.toString());
  // Handle messages (same as Option A)
});