UniToken Node.js SDK

Installation

pnpm add @uni-token/sdk

Usage

import process from 'node:process'
import { requestUniTokenOpenAI } from '@uni-token/sdk'
import { OpenAI } from 'openai'

function loadApiKey(): string | null {
  // ...
  return null
}

function saveApiKey(apiKey: string | null): void {
  // ...
  console.log('API key: ', apiKey)
}

async function main() {
  const { baseURL, apiKey } = await requestUniTokenOpenAI({
    appName: 'MyApp',
    description: 'This is a test application',
    // If not provided, the user will always be prompted to grant permission to this app.
    savedApiKey: loadApiKey(),
  })
  saveApiKey(apiKey)
  if (!apiKey) {
    console.error('User did not grant permission for OpenAI token.')
    return
  }

  const openai = new OpenAI({
    baseURL,
    apiKey,
  })

  chatDemo(openai)
}

async function chatDemo(openai: OpenAI) {
  const stream = await openai.chat.completions.create({
    model: 'gpt-4o-mini',
    stream: true,
    messages: [
      { role: 'system', content: 'You are a concise assistant.' },
      { role: 'user', content: 'Please write a one-sentence bedtime story.' },
    ],
  })

  for await (const chunk of stream) {
    const delta = chunk.choices[0].delta
    if (delta?.content) {
      process.stdout.write(delta.content)
    }
  }
  process.stdout.write('\n')
}

main()