Bring Your Own Model - Wraith Protocol

By default, Wraith agents use a hosted Gemini model. You can replace it with OpenAI, Claude, or your own Gemini key to reduce cost or use a specific model.

Configuration

Pass the ai option when creating the Wraith client:

import { Wraith, Chain } from "@wraith-protocol/sdk";

const wraith = new Wraith({
  apiKey: "wraith_live_abc123",
  ai: {
    provider: "openai",
    apiKey: "sk-...",
  },
});

All agents created from this client use your specified AI provider.

Supported Providers

Provider	`ai.provider`	API Key Format
Google Gemini	`"gemini"`	Gemini API key
OpenAI	`"openai"`	`sk-...`
Anthropic Claude	`"claude"`	Anthropic API key

OpenAI

const wraith = new Wraith({
  apiKey: "wraith_live_abc123",
  ai: {
    provider: "openai",
    apiKey: "sk-proj-...",
  },
});

const agent = await wraith.createAgent({
  name: "alice",
  chain: Chain.Horizen,
  wallet: "0x...",
  signature: "0x...",
});

// Chat uses your OpenAI key
const res = await agent.chat("send 0.1 ETH to bob.wraith");

Claude

const wraith = new Wraith({
  apiKey: "wraith_live_abc123",
  ai: {
    provider: "claude",
    apiKey: "sk-ant-...",
  },
});

Your Own Gemini Key

const wraith = new Wraith({
  apiKey: "wraith_live_abc123",
  ai: {
    provider: "gemini",
    apiKey: "AIza...",
  },
});

How It Works

The AI configuration is passed to the TEE server via HTTP headers on every request:

Header	Value
`Authorization`	`Bearer wraith_...` (your Wraith API key)
`X-AI-Provider`	`openai`, `claude`, or `gemini`
`X-AI-Key`	Your AI provider API key

The TEE server:

Receives your chat message
Builds the system prompt with agent identity and tools
Sends the conversation to your specified AI provider
Executes any tool calls via the chain connector
Returns the final response

Your AI key is only used server-side in the TEE. It’s never stored — used for the duration of the request only.

Default vs. BYOM

Aspect	Default (Gemini)	Bring Your Own
Cost	Included in Wraith API usage	You pay your AI provider directly
Model	Wraith’s chosen Gemini model	Your choice
Configuration	None	Pass `ai` option

Tool Compatibility

All tools work identically regardless of the AI provider. The tool declarations are adapted to each provider’s function-calling format by the TEE server. Your code doesn’t change.

// Works the same with Gemini, OpenAI, or Claude
const res = await agent.chat("send 0.1 ETH to bob.wraith");
const res = await agent.chat("run a privacy check");
const res = await agent.chat("create an invoice for 1 ETH");

​Configuration

​Supported Providers

​OpenAI

​Claude

​Your Own Gemini Key

​How It Works

​Default vs. BYOM

​Tool Compatibility