aisapi
Version:
A JavaScript/TypeScript API library for multiple AI providers
241 lines • 9.76 kB
JavaScript
;
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.GrokProvider = void 0;
const base_1 = require("./base");
/**
* Grok (xAI) API实现
* 支持文本生成、聊天完成、流式响应和JSON输出
* 与OpenAI API兼容，仅需更改baseUrl和apiKey
*/
class GrokProvider extends base_1.BaseProvider {
/**
* 创建Grok提供商实例
*/
constructor(options = {}) {
super(options);
this.name = 'Grok';
this.model = options.model || 'grok-3-beta';
this.maxRetries = options.maxRetries || 3;
}
/**
* @inheritdoc
*/
getDefaultBaseUrl() {
return 'https://api.x.ai/v1';
}
/**
* @inheritdoc
*/
generateText(params) {
return __awaiter(this, void 0, void 0, function* () {
// 对于Grok，使用聊天完成API更为适合，因为它主要基于聊天模式
const messages = [];
// 添加系统消息（如果有）
if (params.systemMessage) {
messages.push({
role: 'system',
content: params.systemMessage
});
}
// 添加用户消息
messages.push({
role: 'user',
content: params.prompt
});
// 使用聊天完成API
return this.chatCompletion({
model: params.model || this.model,
messages,
maxTokens: params.maxTokens,
temperature: params.temperature,
topP: params.topP,
stream: params.stream
});
});
}
/**
* 聊天完成API - 主要接口
*/
chatCompletion(params) {
return __awaiter(this, void 0, void 0, function* () {
var _a, _b, _c, _d;
const model = params.model || this.model;
// 构建请求体
const requestBody = {
model,
messages: params.messages,
max_tokens: params.maxTokens,
temperature: (_a = params.temperature) !== null && _a !== void 0 ? _a : 0.7,
stream: params.stream || false
};
// 添加topP（如果指定）
if (params.topP !== undefined) {
requestBody.top_p = params.topP;
}
// 添加responseFormat（如果指定）
if (params.responseFormat) {
requestBody.response_format = params.responseFormat;
}
// 其他可选参数
if (params.stop)
requestBody.stop = params.stop;
if (params.presencePenalty !== undefined)
requestBody.presence_penalty = params.presencePenalty;
if (params.frequencyPenalty !== undefined)
requestBody.frequency_penalty = params.frequencyPenalty;
if (params.logitBias)
requestBody.logit_bias = params.logitBias;
if (params.user)
requestBody.user = params.user;
// 发送请求
let retries = 0;
while (retries <= this.maxRetries) {
try {
const response = yield this.sendRequest(`${this.baseUrl}/chat/completions`, 'POST', requestBody);
return {
text: response.choices[0].message.content || '',
usage: {
promptTokens: ((_b = response.usage) === null || _b === void 0 ? void 0 : _b.prompt_tokens) || 0,
completionTokens: ((_c = response.usage) === null || _c === void 0 ? void 0 : _c.completion_tokens) || 0,
totalTokens: ((_d = response.usage) === null || _d === void 0 ? void 0 : _d.total_tokens) || 0
},
rawResponse: response
};
}
catch (error) {
retries++;
if (retries > this.maxRetries) {
if (error instanceof Error) {
throw new Error(`[${this.name}] ${error.message}`);
}
throw error;
}
// 指数退避重试，特别是处理速率限制错误 (429)
const delay = Math.min(1000 * Math.pow(2, retries), 60000);
yield new Promise(resolve => setTimeout(resolve, delay));
}
}
throw new Error(`[${this.name}] 达到最大重试次数`);
});
}
/**
* 创建流式聊天响应
*/
createStreamingChatCompletion(params) {
return __awaiter(this, void 0, void 0, function* () {
var _a;
const model = params.model || this.model;
// 构建请求体
const requestBody = {
model,
messages: params.messages,
max_tokens: params.maxTokens,
temperature: (_a = params.temperature) !== null && _a !== void 0 ? _a : 0.7,
stream: true
};
// 添加其他参数，与chatCompletion相同
if (params.topP !== undefined)
requestBody.top_p = params.topP;
if (params.responseFormat)
requestBody.response_format = params.responseFormat;
if (params.stop)
requestBody.stop = params.stop;
if (params.presencePenalty !== undefined)
requestBody.presence_penalty = params.presencePenalty;
if (params.frequencyPenalty !== undefined)
requestBody.frequency_penalty = params.frequencyPenalty;
if (params.logitBias)
requestBody.logit_bias = params.logitBias;
if (params.user)
requestBody.user = params.user;
// 发送流式请求
try {
const headers = {
'Content-Type': 'application/json',
'Authorization': `Bearer ${this.apiKey}`,
'Accept': 'text/event-stream'
};
const controller = new AbortController();
const timeoutId = setTimeout(() => controller.abort(), this.timeout);
const response = yield fetch(`${this.baseUrl}/chat/completions`, {
method: 'POST',
headers,
body: JSON.stringify(requestBody),
signal: controller.signal
});
clearTimeout(timeoutId);
if (!response.ok) {
const errorData = yield response.json().catch(() => ({}));
throw new Error(`API请求失败: ${response.status} ${response.statusText} - ${JSON.stringify(errorData)}`);
}
if (!response.body) {
throw new Error('响应没有可读流');
}
return response.body;
}
catch (error) {
if (error instanceof Error) {
throw new Error(`[${this.name}] ${error.message}`);
}
throw error;
}
});
}
/**
* 生成JSON格式响应
*/
generateJSON(params) {
return __awaiter(this, void 0, void 0, function* () {
const messages = [];
// 添加系统消息（如果有）
if (params.systemMessage) {
messages.push({
role: 'system',
content: params.systemMessage + '\n请以有效的JSON格式返回数据，不要包含额外文本。'
});
}
else {
messages.push({
role: 'system',
content: '请以有效的JSON格式返回数据，不要包含额外文本。'
});
}
// 添加用户消息
messages.push({
role: 'user',
content: params.prompt
});
// 使用强制JSON格式的聊天完成
const result = yield this.chatCompletion({
model: params.model || this.model,
messages,
maxTokens: params.maxTokens,
temperature: params.temperature,
topP: params.topP,
responseFormat: { type: 'json_object' }
});
try {
// 解析JSON
const jsonData = JSON.parse(result.text);
return {
data: jsonData,
usage: result.usage
};
}
catch (error) {
throw new Error(`[${this.name}] 解析JSON响应失败: ${error.message}`);
}
});
}
}
exports.GrokProvider = GrokProvider;
//# sourceMappingURL=grok.js.map