feat: 配置模型参数

src/api/controllers/chat.ts

@@ -86,22 +86,18 @@ async function createCompletion(
 ) {
   let session: http2.ClientHttp2Session;
   return (async () => {
-    console.log(333333333);
     logger.info(messages);
 
     // 提取引用文件URL并上传qwen获得引用的文件ID列表
     const refFileUrls = extractRefFileUrls(messages);
-    console.log(44444444);
     const refs = refFileUrls.length
       ? await Promise.all(
           refFileUrls.map((fileUrl) => uploadFile(fileUrl, ticket))
         )
       : [];
-      console.log(55555555);
     // 如果引用对话ID不正确则重置引用
     if (!/[0-9a-z]{32}/.test(refConvId))
       refConvId = '';
-    console.log(666666666);
     // 请求流
     const session: http2.ClientHttp2Session = await new Promise(
       (resolve, reject) => {
@@ -110,7 +106,6 @@ async function createCompletion(
         session.on("error", reject);
       }
     );
-    console.log(7777777777);
     const [sessionId, parentMsgId = ''] = refConvId.split('-');
     const req = session.request({
       ":method": "POST",
@@ -120,12 +115,11 @@ async function createCompletion(
       ...FAKE_HEADERS,
       Accept: "text/event-stream",
     });
-    console.log(888888888888);
     req.setTimeout(120000);
     req.write(
       JSON.stringify({
         mode: "chat",
-        model: "",
+        model,
         action: "next",
         userAction: "chat",
         requestId: util.uuid(false),
@@ -141,8 +135,7 @@ async function createCompletion(
     req.setEncoding("utf8");
     const streamStartTime = util.timestamp();
     // 接收流为输出文本
-    console.log(99999999999);
-    const answer = await receiveStream(req);
+    const answer = await receiveStream(req, model);
     session.close();
     logger.success(
       `Stream has completed transfer ${util.timestamp() - streamStartTime}ms`
@@ -217,7 +210,7 @@ async function createCompletionStream(
     req.write(
       JSON.stringify({
         mode: "chat",
-        model: "",
+        model,
         action: "next",
         userAction: "chat",
         requestId: util.uuid(false),
@@ -233,7 +226,7 @@ async function createCompletionStream(
     req.setEncoding("utf8");
     const streamStartTime = util.timestamp();
     // 创建转换流将消息格式转换为gpt兼容格式
-    return createTransStream(req, (convId: string) => {
+    return createTransStream(req, model, (convId: string) => {
       // 关闭请求会话
       session.close();
       logger.success(
@@ -287,7 +280,7 @@ async function generateImages(
     req.write(
       JSON.stringify({
         mode: "chat",
-        model: "",
+        model,
         action: "next",
         userAction: "chat",
         requestId: util.uuid(false),
@@ -435,12 +428,12 @@ function checkResult(result: AxiosResponse) {
  *
  * @param stream 消息流
  */
-async function receiveStream(stream: any): Promise<any> {
+async function receiveStream(stream: any, model): Promise<any> {
   return new Promise((resolve, reject) => {
     // 消息初始化
     const data = {
       id: "",
-      model: MODEL_NAME,
+      model,
       object: "chat.completion",
       choices: [
         {
@@ -518,7 +511,7 @@ async function receiveStream(stream: any): Promise<any> {
  * @param stream 消息流
  * @param endCallback 传输结束回调
  */
-function createTransStream(stream: any, endCallback?: Function) {
+function createTransStream(stream: any, model, endCallback?: Function) {
   // 消息创建时间
   const created = util.unixTimestamp();
   // 创建转换流
@@ -528,7 +521,7 @@ function createTransStream(stream: any, endCallback?: Function) {
     transStream.write(
       `data: ${JSON.stringify({
         id: "",
-        model: MODEL_NAME,
+        model,
         object: "chat.completion.chunk",
         choices: [
           {
@@ -576,7 +569,7 @@ function createTransStream(stream: any, endCallback?: Function) {
           content += chunk;
           const data = `data: ${JSON.stringify({
             id: `${result.sessionId}-${result.msgId}`,
-            model: MODEL_NAME,
+            model,
             object: "chat.completion.chunk",
             choices: [
               { index: 0, delta: { content: chunk }, finish_reason: null },
@@ -593,7 +586,7 @@ function createTransStream(stream: any, endCallback?: Function) {
           delta.content += `服务暂时不可用，第三方响应错误：${result.errorCode}`;
         const data = `data: ${JSON.stringify({
           id: `${result.sessionId}-${result.msgId}`,
-          model: MODEL_NAME,
+          model,
           object: "chat.completion.chunk",
           choices: [
             {

src/api/routes/chat.ts

@@ -9,7 +9,6 @@ export default {
 
   post: {
     "/completions": async (request: Request) => {
-      console.log('111111');
       request
         .validate('body.conversation_id', v => _.isUndefined(v) || _.isString(v))
         .validate("body.messages", _.isArray)
@@ -20,7 +19,6 @@ export default {
       const token = _.sample(tokens);
       const { model, conversation_id: convId, messages, stream } = request.body;
 
-      console.log('222222');
       if (stream) {
         const stream = await chat.createCompletionStream(
           model,
@@ -32,7 +30,6 @@ export default {
           type: "text/event-stream",
         });
       } else 
-      console.log('3333333');
 
         return await chat.createCompletion(
           model,