32 lines
548 B
JavaScript
32 lines
548 B
JavaScript
|
import {AI_AVATAR} from "./env";
|
||
|
|
|
||
|
|
class BaseModel {
|
||
|
|
constructor(props) {
|
||
|
|
for (const k in props) {
|
||
|
|
this[k] = props[k];
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
export class ChatGPT extends BaseModel {
|
||
|
|
constructor() {
|
||
|
|
super({
|
||
|
|
avatar: AI_AVATAR,
|
||
|
|
name: 'ChatGPT',
|
||
|
|
id: "gpt-3.5-turbo",
|
||
|
|
desc: "ChatGPT-3.5所基于的模型"
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
export class ChatGLM extends BaseModel {
|
||
|
|
constructor() {
|
||
|
|
super({
|
||
|
|
avatar: AI_AVATAR,
|
||
|
|
name: 'ChatGLM',
|
||
|
|
id: "chatglm-6b",
|
||
|
|
desc: "ChatGLM-6B所基于的模型"
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|