WebSockets

import { WebSocket } from 'ws'; // Node.js ws library

const orgId = 'org_123';
const apiKey = 'your-api-key-token'; // Settings → API Keys
const apiBase = 'https://api.useinvent.com';
const wsUrl = `${apiBase.replace(/^http/, 'ws')}/orgs/${orgId}/inbox/ws`;

const ws = new WebSocket(wsUrl, {
  headers: {
    Authorization: `Bearer ${apiKey}`,
  },
});

ws.on('open', () => {
  console.log('Connected to org inbox WS');
});

ws.on('message', (data) => {
  const event = JSON.parse(data.toString());
  console.log('Chat event:', event);
});

ws.on('error', (error) => {
  console.error('WebSocket error:', error);
});

ws.on('close', () => {
  console.log('Disconnected from org inbox WS');
});

type ChatEvent =
  | { type: 'create'; data: ChatSchema; identifier: string | null }
  | { type: 'update'; data: ChatSchema }
  | { type: 'typing'; chat_id: string; member_id: string }
  | { type: 'message'; chat_id: string; data: ChatMessageSchema }
  | { type: 'message-part'; chat_id: string; message_id: string; data: ModelMessagePartSchema }
  | { type: 'delete-message'; chat_id: string; message_id: string }
  | { type: 'message-reaction'; chat_id: string; message_id: string; member_id?: string; emoji: string; action: 'react' | 'unreact' }
  | { type: 'delete'; chat_id: string }
  | { type: 'tier-limits'; tier: ModelTierSchema; data: ChatLimitSchema }
  | { type: 'tool-limits'; tool: string; data: ChatLimitSchema }
  | { type: 'conversation-memory-update'; chat_id: string; data: ConversationMemorySchema }
  | { type: 'conversation-memory-delete'; chat_id: string; memory_id: string };

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

  if (event.type === 'message-part') {
    // Accumulate streaming parts
    // data is { id, role, part }: role is 'assistant' for model parts, 'tool' for results
    const { chat_id, message_id, data } = event;
    const part = data.part;

    // Part types include:
    // - text_delta: Streaming text chunks
    // - tool_call: Tool invocation
    // - tool_result: Tool execution result
    // - text: Complete text (after streaming)
    // - file, image: File attachments
    // - reasoning: Model reasoning steps

    if (part.type === 'text_delta') {
      // Append streaming text
      console.log('Streaming text:', part.text);
    }
  }
});

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

  if (event.type === 'message') {
    // Complete message with all parts
    const { chat_id, data: message } = event;

    // message contains:
    // - id: Message ID
    // - role: 'user' | 'assistant' | 'system' | 'tool' | 'event' | 'note'
    // - model: Model used
    // - messages: Array of complete message parts
    // - status: 'RUNNING' | 'COMPLETED' | 'FAILED' | 'CANCELED'
    // - usage: Token usage information
    // - error: Error message if failed

    console.log('Complete message:', message);
  }
});

const streamingMessages = new Map<string, string>();

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

  switch (event.type) {
    case 'message-part': {
      const { message_id, data: part } = event;

      if (part.type === 'text_delta') {
        // Accumulate streaming text
        const current = streamingMessages.get(message_id) || '';
        streamingMessages.set(message_id, current + part.text);
        console.log('Current text:', current + part.text);
      }
      break;
    }

    case 'message': {
      // Complete message received - use this as source of truth
      const { data: message } = event;
      streamingMessages.delete(message.id);

      // Full message with all parts is now available
      console.log('Final message:', message.messages);
      break;
    }
  }
});