Function message

src/services/api/openAICompatWsV2Native.test.ts:455–513

(ws, raw)

    },
    websocket: {
      message(ws, raw) {
        const message = JSON.parse(String(raw)) as {
          id?: string
          method?: string
          params?: unknown
        }
        if (message.method === 'initialize') {
          ws.send(JSON.stringify({ id: message.id, result: { ok: true } }))
          if (options?.closeAfterInitialize) {
            ws.close()
          }
          return
        }
        if (message.method === 'chat.completions.start') {
          requestIds.push(String(message.id))
          ws.send(JSON.stringify({ id: message.id, result: { accepted: true } }))
          if (options?.oversizedPayloadBytes) {
            ws.send(
              JSON.stringify({
                method: 'chat.completions.delta',
                params: {
                  id: message.id,
                  data: 'x'.repeat(options.oversizedPayloadBytes),
                },
              }),
            )
          }
          if (options?.delayedDeltaMs !== undefined) {
            setTimeout(() => {
              ws.send(
                JSON.stringify({
                  method: 'chat.completions.delta',
                  params: {
                    id: message.id,
                    data: JSON.stringify({ choices: [{ delta: { content: 'delayed' } }] }),
                  },
                }),
              )
            }, options.delayedDeltaMs)
          }
          if (!options?.keepRequestsOpen) {
            ws.send(
              JSON.stringify({
                method: 'chat.completions.completed',
                params: { id: message.id },
              }),
            )
          }
          return
        }
        if (message.method === 'chat.completions.cancel') {
          ws.send(
            JSON.stringify({
              method: 'chat.completions.completed',
              params: { id: (message.params as { id?: string })?.id },
            }),
          )
        }