mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-07-23 04:12:13 +00:00
196 lines
3.9 KiB
TypeScript
196 lines
3.9 KiB
TypeScript
import { ParamNeed } from './app';
|
||
|
||
// Define the content types for the message object
|
||
export interface UserContentItem {
|
||
type: string;
|
||
[key: string]: any; // This allows for additional properties based on the type
|
||
}
|
||
|
||
// Define the message format for object input
|
||
export interface UserMessageObject {
|
||
role: 'user';
|
||
content: UserContentItem[];
|
||
}
|
||
|
||
// Union type for the content parameter
|
||
export type UserChatContent = string | UserMessageObject;
|
||
|
||
type ChartValue = {
|
||
name: string;
|
||
type: string;
|
||
value: number;
|
||
};
|
||
|
||
/**
|
||
* dashboard chart type
|
||
*/
|
||
export type ChartData = {
|
||
chart_desc: string;
|
||
chart_name: string;
|
||
chart_sql: string;
|
||
chart_type: string;
|
||
chart_uid: string;
|
||
column_name: Array<string>;
|
||
values: Array<ChartValue>;
|
||
type?: string;
|
||
};
|
||
|
||
export type SceneResponse = {
|
||
chat_scene: string;
|
||
param_title: string;
|
||
scene_describe: string;
|
||
scene_name: string;
|
||
show_disable: boolean;
|
||
};
|
||
|
||
export type NewDialogueParam = {
|
||
chat_mode: string;
|
||
model?: string;
|
||
};
|
||
|
||
export type ChatHistoryResponse = IChatDialogueMessageSchema[];
|
||
|
||
export type IChatDialogueSchema = {
|
||
conv_uid: string;
|
||
user_input: UserChatContent;
|
||
user_name: string;
|
||
chat_mode:
|
||
| 'chat_with_db_execute'
|
||
| 'chat_excel'
|
||
| 'chat_with_db_qa'
|
||
| 'chat_knowledge'
|
||
| 'chat_dashboard'
|
||
| 'chat_execution'
|
||
| 'chat_agent'
|
||
| 'chat_flow'
|
||
| (string & {});
|
||
select_param: string;
|
||
app_code: string;
|
||
param_need?: ParamNeed[];
|
||
};
|
||
|
||
export type UserParam = {
|
||
user_channel: string;
|
||
user_no: string;
|
||
nick_name: string;
|
||
};
|
||
|
||
export type UserParamResponse = {
|
||
user_channel: string;
|
||
user_no: string;
|
||
user_id: string;
|
||
};
|
||
|
||
export type DialogueListResponse = IChatDialogueSchema[];
|
||
|
||
export type IChatDialogueMessageSchema = {
|
||
role: 'human' | 'view' | 'system' | 'ai';
|
||
context: string;
|
||
order: number;
|
||
time_stamp: number | string | null;
|
||
model_name: string;
|
||
retry?: boolean;
|
||
thinking?: boolean;
|
||
outing?: boolean;
|
||
feedback?: Record<string, any>;
|
||
};
|
||
|
||
export type ModelType =
|
||
| 'proxyllm'
|
||
| 'flan-t5-base'
|
||
| 'vicuna-13b'
|
||
| 'vicuna-7b'
|
||
| 'vicuna-13b-v1.5'
|
||
| 'vicuna-7b-v1.5'
|
||
| 'codegen2-1b'
|
||
| 'codet5p-2b'
|
||
| 'chatglm-6b-int4'
|
||
| 'chatglm-6b'
|
||
| 'chatglm2-6b'
|
||
| 'chatglm2-6b-int4'
|
||
| 'guanaco-33b-merged'
|
||
| 'falcon-40b'
|
||
| 'gorilla-7b'
|
||
| 'gptj-6b'
|
||
| 'proxyllm'
|
||
| 'chatgpt_proxyllm'
|
||
| 'bard_proxyllm'
|
||
| 'claude_proxyllm'
|
||
| 'wenxin_proxyllm'
|
||
| 'tongyi_proxyllm'
|
||
| 'zhipu_proxyllm'
|
||
| 'llama-2-7b'
|
||
| 'llama-2-13b'
|
||
| 'llama-2-70b'
|
||
| 'baichuan-7b'
|
||
| 'baichuan-13b'
|
||
| 'baichuan2-7b'
|
||
| 'baichuan2-13b'
|
||
| 'wizardlm-13b'
|
||
| 'llama-cpp'
|
||
| (string & {});
|
||
|
||
export type LLMOption = { label: string; icon: string };
|
||
|
||
export type FeedBack = {
|
||
information?: string;
|
||
just_fun?: string;
|
||
others?: string;
|
||
work_study?: string;
|
||
};
|
||
|
||
export type Reference = {
|
||
name: string;
|
||
chunks: Array<number>;
|
||
};
|
||
|
||
export type IDB = {
|
||
param: string;
|
||
type: string;
|
||
space_id?: number;
|
||
};
|
||
|
||
export interface UploadResponse {
|
||
file_learning: boolean;
|
||
file_path: string;
|
||
is_oss: boolean;
|
||
}
|
||
|
||
export interface RecommendQuestionParams {
|
||
valid?: string; // 是否仅选择生效的应用,true/false
|
||
app_code?: string; // 所属应用
|
||
chat_mode?: string; // 类型(chat_knwoledge, chat_excel...)
|
||
is_hot_question?: string;
|
||
}
|
||
|
||
export interface RecommendQuestionResponse {
|
||
id: string;
|
||
app_code: string;
|
||
question: string;
|
||
chat_mode: string;
|
||
user_code: string;
|
||
}
|
||
|
||
export interface FeedbackReasonsResponse {
|
||
reason_type: string;
|
||
reason: string;
|
||
}
|
||
|
||
export interface FeedbackAddParams {
|
||
conv_uid: string;
|
||
message_id: string; // 消息id, 对应order
|
||
feedback_type: string; // 反馈类型,like, unlike
|
||
reason_types?: string[]; // 原因类型
|
||
remark?: string; // 备注信息
|
||
}
|
||
|
||
export interface CancelFeedbackAddParams {
|
||
conv_uid: string;
|
||
message_id: string;
|
||
}
|
||
|
||
export interface StopTopicParams {
|
||
conv_id: string;
|
||
round_index: number;
|
||
}
|