设置
示例使用 AI SDK 7。将其与 OpenAI 提供商、React hooks 和 Zod 一起安装:npm install ai@7 @ai-sdk/openai@4 @ai-sdk/react@4 zod
提供商配置
使用 OpenAI 兼容的适配器创建一个 Venice 提供商:// lib/venice.ts
import { createOpenAI } from '@ai-sdk/openai';
const openai = createOpenAI({
apiKey: process.env.VENICE_API_KEY!,
baseURL: 'https://api.venice.ai/api/v1',
});
// Use .chat() to ensure compatibility with Venice's chat completions endpoint
export const venice = (modelId: string) => openai.chat(modelId);
使用
.chat() 可确保请求发送到 Venice 的 /chat/completions 端点。在较新的 AI SDK 版本中,默认的 openai('model') 语法改用 Responses API,Venice 在 /responses 提供该 API,并存在一些限制。使用前请参阅 Responses API 指南。流式聊天(Next.js App Router)
API 路由
// app/api/chat/route.ts
import {
convertToModelMessages,
createUIMessageStreamResponse,
streamText,
toUIMessageStream,
type UIMessage,
} from 'ai';
import { venice } from '@/lib/venice';
export async function POST(req: Request) {
const { messages }: { messages: UIMessage[] } = await req.json();
const result = streamText({
model: venice('venice-uncensored-1-2'),
instructions: 'You are a helpful, privacy-respecting AI assistant.',
messages: await convertToModelMessages(messages),
});
return createUIMessageStreamResponse({
stream: toUIMessageStream({ stream: result.stream }),
});
}
React 组件
// app/page.tsx
'use client';
import { useChat } from '@ai-sdk/react';
import { useState } from 'react';
export default function Chat() {
const [input, setInput] = useState('');
const { messages, sendMessage, status } = useChat();
const isLoading = status === 'submitted' || status === 'streaming';
return (
<div className="max-w-2xl mx-auto p-4">
<div className="space-y-4 mb-4">
{messages.map((m) => (
<div key={m.id} className={m.role === 'user' ? 'text-right' : 'text-left'}>
<span className="font-bold">{m.role === 'user' ? 'You' : 'Venice'}:</span>
{m.parts.map((part, i) =>
part.type === 'text' ? (
<p key={i} className="whitespace-pre-wrap">{part.text}</p>
) : null,
)}
</div>
))}
</div>
<form
onSubmit={(e) => {
e.preventDefault();
if (!input.trim()) return;
sendMessage({ text: input });
setInput('');
}}
className="flex gap-2"
>
<input
value={input}
onChange={(e) => setInput(e.target.value)}
placeholder="Ask anything..."
className="flex-1 border rounded px-3 py-2"
disabled={isLoading}
/>
<button type="submit" disabled={isLoading} className="bg-red-600 text-white px-4 py-2 rounded">
Send
</button>
</form>
</div>
);
}
生成文本(非流式)
import { generateText } from 'ai';
import { venice } from '@/lib/venice';
const { text } = await generateText({
model: venice('z-ai-glm-5-3-flash'),
prompt: 'Explain zero-knowledge proofs in simple terms.',
});
console.log(text);
结构化输出
import { generateText, Output } from 'ai';
import { venice } from '@/lib/venice';
import { z } from 'zod';
const { output } = await generateText({
model: venice('venice-uncensored-1-2'),
output: Output.object({
schema: z.object({
recipe: z.object({
name: z.string(),
ingredients: z.array(z.string()),
steps: z.array(z.string()),
prepTimeMinutes: z.number(),
}),
}),
}),
prompt: 'Generate a recipe for chocolate chip cookies.',
});
console.log(output.recipe.name);
console.log(`Prep time: ${output.recipe.prepTimeMinutes} minutes`);
工具调用
import { isStepCount, streamText, tool } from 'ai';
import { venice } from '@/lib/venice';
import { z } from 'zod';
const result = streamText({
model: venice('z-ai-glm-5-3-flash'),
messages: [{ role: 'user', content: 'What is the weather in Tokyo?' }],
tools: {
getWeather: tool({
description: 'Get current weather for a location',
inputSchema: z.object({
location: z.string().describe('City name'),
}),
execute: async ({ location }) => {
// Your weather API call here
return { temperature: 22, condition: 'Sunny', location };
},
}),
},
stopWhen: isStepCount(2),
});
for await (const part of result.stream) {
if (part.type === 'text-delta') {
process.stdout.write(part.text);
} else if (part.type === 'tool-result') {
console.log('Tool result:', part.output);
}
}
stopWhen: isStepCount(2) 让模型在工具运行后的第二步中给出回答。如果不设置,流会在工具结果之后结束。
图像生成
Venice 的图像生成可以与 AI SDK 并行直接调用:// app/api/image/route.ts
export async function POST(req: Request) {
const { prompt } = await req.json();
const response = await fetch('https://api.venice.ai/api/v1/image/generate', {
method: 'POST',
headers: {
'Authorization': `Bearer ${process.env.VENICE_API_KEY}`,
'Content-Type': 'application/json',
},
body: JSON.stringify({
model: 'qwen-image',
prompt,
aspect_ratio: '1:1',
}),
});
const data = await response.json();
return Response.json({ image: data.images[0] });
}
多模型聊天(模型选择器)
让用户在 Venice 模型之间选择:// app/api/chat/route.ts
import {
convertToModelMessages,
createUIMessageStreamResponse,
streamText,
toUIMessageStream,
type UIMessage,
} from 'ai';
import { venice } from '@/lib/venice';
const ALLOWED_MODELS = [
'venice-uncensored-1-2',
'z-ai-glm-5-3-flash',
'qwen3-vl-235b-a22b',
'nvidia-nemotron-3-nano-30b-a3b',
];
export async function POST(req: Request) {
const { messages, model: modelId }: { messages: UIMessage[]; model: string } =
await req.json();
if (!ALLOWED_MODELS.includes(modelId)) {
return new Response('Invalid model', { status: 400 });
}
const result = streamText({
model: venice(modelId),
messages: await convertToModelMessages(messages),
});
return createUIMessageStreamResponse({
stream: toUIMessageStream({ stream: result.stream }),
});
}
// Client component with model selector
'use client';
import { useChat } from '@ai-sdk/react';
import { useState } from 'react';
const MODELS = [
{ id: 'venice-uncensored-1-2', name: 'Venice Uncensored', desc: 'Fast & uncensored' },
{ id: 'z-ai-glm-5-3-flash', name: 'GLM 5.3 Flash', desc: 'Fast private reasoning' },
{ id: 'qwen3-vl-235b-a22b', name: 'Qwen Vision', desc: 'Advanced vision + text' },
{ id: 'nvidia-nemotron-3-nano-30b-a3b', name: 'Nemotron 3 Nano 30B', desc: 'Fast & low cost' },
];
export default function Chat() {
const [model, setModel] = useState('venice-uncensored-1-2');
const [input, setInput] = useState('');
const { messages, sendMessage } = useChat();
return (
<div>
<select value={model} onChange={(e) => setModel(e.target.value)}>
{MODELS.map((m) => (
<option key={m.id} value={m.id}>{m.name} — {m.desc}</option>
))}
</select>
{messages.map((m) => (
<p key={m.id}>
{m.role}: {m.parts.map((part) => (part.type === 'text' ? part.text : '')).join('')}
</p>
))}
<form
onSubmit={(e) => {
e.preventDefault();
sendMessage({ text: input }, { body: { model } });
setInput('');
}}
>
<input value={input} onChange={(e) => setInput(e.target.value)} />
</form>
</div>
);
}
sendMessage 会在每条消息的请求体中发送所选的 model,因此切换模型会应用于下一条消息。
Web 搜索集成
在模型 ID 上使用模型功能后缀来开启 Venice web 搜索。该后缀通过模型 ID 传递venice_parameters,因此无需在请求中添加字段:
import { streamText } from 'ai';
import { venice } from '@/lib/venice';
const result = streamText({
model: venice('venice-uncensored-1-2:enable_web_search=auto'),
messages: [{ role: 'user', content: 'What happened in AI news today?' }],
});
Embeddings
对于 embeddings,请直接在提供商上使用embeddingModel():
import { embed, embedMany } from 'ai';
import { createOpenAI } from '@ai-sdk/openai';
const openai = createOpenAI({
apiKey: process.env.VENICE_API_KEY!,
baseURL: 'https://api.venice.ai/api/v1',
});
// Single embedding
const { embedding } = await embed({
model: openai.embeddingModel('text-embedding-bge-m3'),
value: 'Privacy-first AI infrastructure',
});
// Batch embeddings
const { embeddings } = await embedMany({
model: openai.embeddingModel('text-embedding-bge-m3'),
values: [
'Venice AI provides private inference.',
'Zero data retention guaranteed.',
'OpenAI SDK compatible.',
],
});
环境变量
# .env.local
VENICE_API_KEY=your-venice-api-key
推荐模型
| 用例 | 模型 | 原因 |
|---|---|---|
| 聊天应用 | venice-uncensored-1-2 | 快速、便宜、无过滤 |
| 复杂任务 | z-ai-glm-5-3-flash | 快速的私有推理 |
| 视觉应用 | qwen3-vl-235b-a22b | 高级图像理解 |
| 大流量 | nvidia-nemotron-3-nano-30b-a3b | 低成本,输入 $0.075/1M、输出 $0.30/1M |
| 工具调用 | z-ai-glm-5-3-flash | 可靠的函数调用 |
Vercel AI SDK 文档
官方 Vercel AI SDK 文档
Venice 模型
浏览所有 Venice 模型