aisapi
Version:
A JavaScript/TypeScript API library for multiple AI providers
63 lines • 1.77 kB
TypeScript
import { BaseProvider } from './base';
import { TextGenerationParams, TextGenerationResponse, ProviderOptions, ChatCompletionParams } from '../types';
/**
* 讯飞星火API提供商特定选项
*/
export interface SparkOptions extends ProviderOptions {
model?: string;
appId?: string;
apiSecret?: string;
}
/**
* 星火API可用模型
*/
export declare enum SparkModel {
LITE = "spark-lite",// 轻量版,4k上下文
PRO = "spark-pro",// 专业版,8k上下文
PRO_128K = "spark-pro-128k",// 专业长文本版,128k上下文
MAX = "spark-max",// 高级版,8k上下文
MAX_32K = "spark-max-32k",// 高级长文本版,32k上下文
ULTRA = "spark-ultra"
}
/**
* 讯飞星火API提供商实现
*/
export declare class SparkProvider extends BaseProvider {
readonly name = "Spark";
private model;
private appId?;
private apiSecret?;
/**
* 创建讯飞星火API提供商实例
*/
constructor(options?: SparkOptions);
/**
* @inheritdoc
*/
protected getDefaultBaseUrl(): string;
/**
* 获取请求头,包括鉴权信息
*/
protected getAuthHeaders(): Promise<Record<string, string>>;
/**
* @inheritdoc
*/
generateText(params: TextGenerationParams): Promise<TextGenerationResponse>;
/**
* 聊天完成API
*/
chatCompletion(params: ChatCompletionParams): Promise<TextGenerationResponse>;
/**
* 转换消息格式为星火API格式
*/
private convertMessagesToSparkFormat;
/**
* 获取星火API的领域/模型代码
*/
private getSparkDomain;
/**
* 生成JSON格式输出
*/
generateJSON(params: TextGenerationParams): Promise<any>;
}
//# sourceMappingURL=spark.d.ts.map