File size: 992 Bytes
755dd12 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
import platform from './provider';
export type SearchFunc = (...args: any[]) => Promise<any[]>;
export enum ESearchEngine {
GOOGLE = 'GOOGLE',
BING = 'BING',
SOGOU = 'SOGOU',
SEARXNG = 'SEARXNG',
CHATGLM = 'CHATGLM'
}
export type ChatRoleType = 'user' | 'assistant' | 'system';
export interface IChatInputMessage {
content: string;
role: ChatRoleType;
}
export type Provider = 'ollama' | 'lmstudio';
export interface IChatResponse {
text: string;
usage?: {
outputTokens: number;
inputTokens: number;
};
}
export interface IModelInfo {
platform: keyof typeof platform;
type: string;
models: string[];
}
export interface IStreamHandler {
(message: string | null, done: boolean): void
}
// search engine result
export interface ISearchResponseResult {
name: string;
url: string;
snippet: string;
thumbnail?: string;
img?: string;
source?: string;
[key: string]: string | undefined;
}
export type TMode = 'simple' | 'deep' | 'research'
|