Skip to main content
@nuwa-ai/ui-kit gives you a React hook (useNuwa) and a context provider (NuwaProvider) to talk to the Nuwa Client from your Cap UI. With it you can send prompts, persist UI state per chat thread, stream AI output, add selections, and manage iframe height/theme.

Connect To The Client

Wrap your UI with NuwaProvider, then call useNuwa() inside to access the nuwa client instance and connection state. For all props and return types, see: NuwaProvider and useNuwa
import { NuwaProvider, useNuwa } from '@nuwa-ai/ui-kit';

function MyCapUI() {
  const { nuwa, connected, theme } = useNuwa();

  const run = async () => {
    // Persist local UI state to the current chat thread
    await nuwa.saveState({ step: 'started' });
    // Send a prompt to the active conversation
    await nuwa.sendPrompt('Please analyze the selected notes.');
  };

  return (
    <div className={theme === 'dark' ? 'dark' : ''}>
      <button onClick={run} disabled={!connected}>Run</button>
    </div>
  );
}

export default function App() {
  return (
    <NuwaProvider onConnected={() => console.log('Nuwa connected')} onError={console.error}>
      <MyCapUI />
    </NuwaProvider>
  );
}

NuwaProvider

NuwaProvider sets up the client connection and exposes it via context.
  • automatically connects to the Nuwa Client
  • provides a nuwa client instance for calling Nuwa Client functions
  • optional autoHeight resizes the parent iframe to fit your content
  • syncs light/dark theme from the Nuwa Client to your UI wrapper
Example with options:
<NuwaProvider
  autoHeight
  debug
  methodTimeout={3000}
  streamTimeout={45000}
  onConnected={() => console.log('connected')}
  onError={(e) => console.error('nuwa error', e)}
>
  <App />
</NuwaProvider>
See full provider API and option types

Send Prompt

Send a message to the current chat thread with nuwa.sendPrompt(message) See full method signature
function SendPromptButton() {
  const { nuwa, connected } = useNuwa();
  return (
    <button
      disabled={!connected}
      onClick={() => nuwa.sendPrompt('Summarize the selected notes.')}
    >
      Ask AI
    </button>
  );
}

UI State

Persist and retrieve UI state per chat thread with nuwa.saveState() and nuwa.getState(). You can save UI state at any time, as any object types. When the user returns to the same chat thread, your latest saved state is delivered back to the UI. See full method signature
You can save any object types as UI state but you need to make sure your UI can handle the state object. For example, if you save a state object with a property title, you need to make sure your UI can handle the title property.
import { useEffect, useState } from 'react';

function ThreadStateExample() {
  const { nuwa } = useNuwa();
  const [title, setTitle] = useState('');

  // Load previously saved state for this chat thread
  useEffect(() => {
    let mounted = true;
    (async () => {
      const prev = await nuwa.getState<{ title?: string } | null>();
      if (mounted && prev?.title) setTitle(prev.title);
    })();
    return () => { mounted = false; };
  }, [nuwa]);

  // Save on change
  const onChange = async (e: React.ChangeEvent<HTMLInputElement>) => {
    const v = e.target.value;
    setTitle(v);
    await nuwa.saveState({ title: v });
  };

  return <input value={title} onChange={onChange} placeholder="Title" />;
}

AI Stream

Stream AI output and update the UI incrementally.
  • nuwa.createAIStream({ prompt, capId? }) -> StreamHandle
  • await handle.execute({ onChunk?, onError? }) -> { result, error }
  • handle.abort() to cancel; inspect handle.status | error | result
See full streaming API
This function only supports streaming text-based LLM output at the moment.
import { useState } from 'react';

function StreamingExample() {
  const { nuwa } = useNuwa();
  const [out, setOut] = useState('');
  const [busy, setBusy] = useState(false);

  const run = async () => {
    setOut('');
    setBusy(true);
    const stream = nuwa.createAIStream({ prompt: 'Write a haiku about Nuwa.' });
    const { result, error } = await stream.execute({
      onChunk: (chunk) => {
        if (chunk.type === 'content') setOut((s) => s + chunk.content);
      },
      onError: (e) => console.error('stream error', e),
    });
    setBusy(false);
    if (error) console.warn('ended with error', error);
    console.log('final result', result);
  };

  return (
    <div>
      <button onClick={run} disabled={busy}>Generate</button>
      <pre>{out}</pre>
    </div>
  );
}

Add Selection

Add a selection (label + message) to parent UI. The selection message is provided to the AI as part of the system prompt. You can set it with the prompt variable {{artifact_selections}} in the Cap’s prompt field.
  • nuwa.addSelection(label: string, message: string | object): Promise<void>
More details and types
function AddSelectionButtons() {
  const { nuwa } = useNuwa();

  const addText = () => nuwa.addSelection('Notes', 'User selected 3 notes.');
  const addObject = () =>
    nuwa.addSelection('Page', {
      url: window.location.href,
      title: document.title,
      excerpt: 'First 200 chars...'
    });

  return (
    <div>
      <button onClick={addText}>Add Text Selection</button>
      <button onClick={addObject}>Add Structured Selection</button>
    </div>
  );
}

Height And Theme

NuwaProvider with autoHeight keeps your iframe fully visible. You can also set height manually when needed.
  • Auto height: <NuwaProvider autoHeight> (enabled by default)
  • Manual: nuwa.setHeight(pxOrCss): Promise<void>
function ManualHeightSection() {
  const { nuwa } = useNuwa();
  // e.g., after expanding/collapsing panels
  const update = () => nuwa.setHeight(document.body.scrollHeight);
  return <button onClick={update}>Recalculate Height</button>;
}

See Also