设置
npm install ai @ai-sdk/openai
提供商配置
使用 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 端点。默认的 openai('model') 语法可能使用 Venice 尚不支持的较新 OpenAI 端点。流式聊天(Next.js App Router)
API 路由
// app/api/chat/route.ts
import { streamText } from 'ai';
import { venice } from '@/lib/venice';
export async function POST(req: Request) {
const { messages } = await req.json();
const result = streamText({
model: venice('venice-uncensored'),
system: 'You are a helpful, privacy-respecting AI assistant.',
messages,
});
return result.toDataStreamResponse();
}
React 组件
// app/page.tsx
'use client';
import { useChat } from '@ai-sdk/react';
export default function Chat() {
const { messages, input, handleInputChange, handleSubmit, isLoading } = useChat();
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>
<p className="whitespace-pre-wrap">{m.content}</p>
</div>
))}
</div>
<form onSubmit={handleSubmit} className="flex gap-2">
<input
value={input}
onChange={handleInputChange}
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('zai-org-glm-5-1'),
prompt: 'Explain zero-knowledge proofs in simple terms.',
});
console.log(text);
结构化输出
import { generateObject } from 'ai';
import { venice } from '@/lib/venice';
import { z } from 'zod';
const { object } = await generateObject({
model: venice('venice-uncensored'),
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(object.recipe.name);
console.log(`Prep time: ${object.recipe.prepTimeMinutes} minutes`);
工具调用
import { streamText, tool } from 'ai';
import { venice } from '@/lib/venice';
import { z } from 'zod';
const result = streamText({
model: venice('zai-org-glm-5-1'),
messages: [{ role: 'user', content: 'What is the weather in Tokyo?' }],
tools: {
getWeather: tool({
description: 'Get current weather for a location',
parameters: z.object({
location: z.string().describe('City name'),
}),
execute: async ({ location }) => {
// Your weather API call here
return { temperature: 22, condition: 'Sunny', location };
},
}),
},
});
for await (const part of result.fullStream) {
if (part.type === 'text-delta') {
process.stdout.write(part.textDelta);
} else if (part.type === 'tool-result') {
console.log('Tool result:', part.result);
}
}
图像生成
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,
width: 1024,
height: 1024,
}),
});
const data = await response.json();
return Response.json({ image: data.images[0] });
}
多模型聊天(模型选择器)
让用户在 Venice 模型之间选择:// app/api/chat/route.ts
import { streamText } from 'ai';
import { venice } from '@/lib/venice';
const ALLOWED_MODELS = [
'venice-uncensored',
'zai-org-glm-5-1',
'qwen3-vl-235b-a22b',
'qwen3-5-9b',
];
export async function POST(req: Request) {
const { messages, model: modelId } = await req.json();
if (!ALLOWED_MODELS.includes(modelId)) {
return new Response('Invalid model', { status: 400 });
}
const result = streamText({
model: venice(modelId),
messages,
});
return result.toDataStreamResponse();
}
// Client component with model selector
'use client';
import { useChat } from '@ai-sdk/react';
import { useState } from 'react';
const MODELS = [
{ id: 'venice-uncensored', name: 'Venice Uncensored', desc: 'Fast & uncensored' },
{ id: 'zai-org-glm-5-1', name: 'GLM 5.1', desc: 'Most intelligent (private)' },
{ id: 'qwen3-vl-235b-a22b', name: 'Qwen Vision', desc: 'Advanced vision + text' },
{ id: 'qwen3-5-9b', name: 'Qwen 3.5 9B', desc: 'Fastest & cheapest' },
];
export default function Chat() {
const [model, setModel] = useState('venice-uncensored');
const { messages, input, handleInputChange, handleSubmit } = useChat({
body: { model },
});
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>
{/* ... chat UI ... */}
</div>
);
}
Web 搜索集成
为 web 搜索传递 Venice 参数:import { streamText } from 'ai';
import { venice } from '@/lib/venice';
const result = streamText({
model: venice('venice-uncensored'),
messages: [{ role: 'user', content: 'What happened in AI news today?' }],
// Venice-specific parameters
experimental_providerMetadata: {
venice_parameters: {
enable_web_search: 'auto',
},
},
});
如果
experimental_providerMetadata 没有传递过去,您可以使用自定义 fetch 包装器或直接调用 Venice API 来使用 web 搜索功能。Embeddings
对于 embeddings,请直接在提供商上使用textEmbeddingModel():
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.textEmbeddingModel('text-embedding-bge-m3'),
value: 'Privacy-first AI infrastructure',
});
// Batch embeddings
const { embeddings } = await embedMany({
model: openai.textEmbeddingModel('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 | 快速、便宜、无过滤 |
| 复杂任务 | zai-org-glm-5-1 | 私有旗舰推理 |
| 视觉应用 | qwen3-vl-235b-a22b | 高级图像理解 |
| 大流量 | qwen3-5-9b | 最便宜,输入 0.10/1M、输出0.15/1M |
| 工具调用 | zai-org-glm-5-1 | 可靠的函数调用 |
Vercel AI SDK 文档
官方 Vercel AI SDK 文档
Venice 模型
浏览所有 Venice 模型