feat: 支持多语言模型

DavideYang125 · May 4, 2023 · 37d98b4 · 37d98b4
1 parent 18c4366
commit 37d98b4
Show file tree

Hide file tree

Showing 10 changed files with 178 additions and 66 deletions.
diff --git a/README.md b/README.md
@@ -51,6 +51,7 @@ release dir 可以直接下载 release 版本，也可以:
 - [x] 全局状态管理
 - [x] 多轮次对话
 - [x] prompt 支持
+- [ ] 支持 ChatGlm 等其他模型
 - [ ] 一键到出会话支持
 - [ ] 支持高级搜索
 - [ ] 等你 issue 来支持

diff --git a/lib/components/conversation.dart b/lib/components/conversation.dart
@@ -333,6 +333,39 @@ class _ConversationWindowState extends State<ConversationWindow> {
                     const SizedBox(
                       height: 28,
                     ),
+                    DropdownButtonFormField(
+                      value:
+                          BlocProvider.of<UserSettingCubit>(context).state.llm,
+                      decoration: InputDecoration(
+                        labelText: AppLocalizations.of(context)!.llmHint,
+                        hintText: AppLocalizations.of(context)!.llmHint,
+                        floatingLabelBehavior: FloatingLabelBehavior.auto,
+                        contentPadding: const EdgeInsets.symmetric(
+                            horizontal: 16, vertical: 8),
+                        border: OutlineInputBorder(
+                          borderRadius: BorderRadius.circular(5),
+                          borderSide: BorderSide.none,
+                        ),
+                        filled: true,
+                      ),
+                      items: <String>['OpenAI', 'ChatGlm', 'IF']
+                          .map<DropdownMenuItem<String>>((String value) {
+                        return DropdownMenuItem<String>(
+                          value: value,
+                          child: Text(
+                            value,
+                          ),
+                        );
+                      }).toList(),
+                      onChanged: (String? newValue) {
+                        if (newValue == null) return;
+                        BlocProvider.of<UserSettingCubit>(context)
+                            .setLlm(newValue);
+                      },
+                    ),
+                    const SizedBox(
+                      height: 28,
+                    ),
                     TextFormField(
                       controller: controllerApiKey,
                       decoration: InputDecoration(

diff --git a/lib/cubit/setting_cubit.dart b/lib/cubit/setting_cubit.dart
@@ -7,7 +7,7 @@ part 'setting_state.dart';
 class UserSettingCubit extends Cubit<UserSettingState> with HydratedMixin {
   UserSettingCubit()
       : super(UserSettingState(lightTheme, const Locale('en'), "",
-            "https://api.openai-proxy.com", false, "gpt-3.5-turbo")) {
+            "https://api.openai-proxy.com", false, "OpenAI", "gpt-3.5-turbo")) {
     hydrate();
   }
 
@@ -22,17 +22,18 @@ class UserSettingCubit extends Cubit<UserSettingState> with HydratedMixin {
         state.key,
         state.baseUrl,
         state.useStream,
+        state.llm,
         state.gptModel));
   }
 
   void setKey(String key) {
     emit(UserSettingState(state.themeData, state.locale, key, state.baseUrl,
-        state.useStream, state.gptModel));
+        state.useStream, state.llm, state.gptModel));
   }
 
   void setProxyUrl(String baseUrl) {
     emit(UserSettingState(state.themeData, state.locale, state.key, baseUrl,
-        state.useStream, state.gptModel));
+        state.useStream, state.llm, state.gptModel));
   }
 
   void switchLocale() {
@@ -42,17 +43,23 @@ class UserSettingCubit extends Cubit<UserSettingState> with HydratedMixin {
         state.key,
         state.baseUrl,
         state.useStream,
+        state.llm,
         state.gptModel));
   }
 
   void setUseStream(bool useStream) {
     emit(UserSettingState(state.themeData, state.locale, state.key,
-        state.baseUrl, useStream, state.gptModel));
+        state.baseUrl, useStream, state.llm, state.gptModel));
   }
 
   void setGptModel(String value) {
     emit(UserSettingState(state.themeData, state.locale, state.key,
-        state.baseUrl, state.useStream, value));
+        state.baseUrl, state.useStream, state.llm, value));
+  }
+
+  void setLlm(String newValue) {
+    emit(UserSettingState(state.themeData, state.locale, state.key,
+        state.baseUrl, state.useStream, newValue, state.gptModel));
   }
 
   @override
@@ -62,6 +69,7 @@ class UserSettingCubit extends Cubit<UserSettingState> with HydratedMixin {
     String key = json['user_key_value'] as String;
     String baseUrl = json['user_proxy_url_value'] as String;
     bool useStream = json['user_use_stream_value'] as bool;
+    String llm = json['user_llm_value'] as String;
     String gptModel = json['user_gpt_model_value'] as String;
 
     return UserSettingState(
@@ -70,6 +78,7 @@ class UserSettingCubit extends Cubit<UserSettingState> with HydratedMixin {
         key,
         baseUrl,
         useStream,
+        llm.isEmpty ? "OpenAI" : llm,
         gptModel.isEmpty ? "gpt-3.5-turbo" : gptModel);
   }
 
@@ -96,6 +105,7 @@ class UserSettingCubit extends Cubit<UserSettingState> with HydratedMixin {
       'user_key_value': state.key,
       'user_proxy_url_value': state.baseUrl,
       'user_use_stream_value': state.useStream,
+      'user_llm_value': state.llm,
       'user_gpt_model_value': state.gptModel
     };
   }

diff --git a/lib/cubit/setting_state.dart b/lib/cubit/setting_state.dart
@@ -12,6 +12,7 @@ class UserSettingState {
   final String baseUrl;
   final bool useStream;
   final String gptModel;
+  final String llm; //大语言模型，OpenAi，ChatGlm，IF
   const UserSettingState(this.themeData, this.locale, this.key, this.baseUrl,
-      this.useStream, this.gptModel);
+      this.useStream, this.llm, this.gptModel);
 }
diff --git a/lib/data/glm.dart b/lib/data/glm.dart
@@ -0,0 +1,15 @@
+import 'package:flutter/foundation.dart';
+import 'package:flutter_chatgpt/data/llm.dart';
+import 'package:flutter_chatgpt/repository/conversation.dart';
+
+class ChatGlM extends LLM {
+  @override
+  getResponse(
+      List<Message> messages,
+      ValueChanged<Message> onResponse,
+      ValueChanged<Message> errorCallback,
+      ValueChanged<Message> onSuccess) async {
+    // TODO: implement getResponse
+    throw UnimplementedError();
+  }
+}
diff --git a/lib/data/if.dart b/lib/data/if.dart
@@ -0,0 +1,15 @@
+import 'package:flutter/foundation.dart';
+import 'package:flutter_chatgpt/data/llm.dart';
+import 'package:flutter_chatgpt/repository/conversation.dart';
+
+class ChatIF extends LLM {
+  @override
+  getResponse(
+      List<Message> messages,
+      ValueChanged<Message> onResponse,
+      ValueChanged<Message> errorCallback,
+      ValueChanged<Message> onSuccess) async {
+    // TODO: implement getResponse
+    throw UnimplementedError();
+  }
+}
diff --git a/lib/data/llm.dart b/lib/data/llm.dart
@@ -0,0 +1,76 @@
+import 'package:dart_openai/openai.dart';
+import 'package:flutter/foundation.dart';
+import 'package:flutter_chatgpt/cubit/setting_cubit.dart';
+import 'package:flutter_chatgpt/repository/conversation.dart';
+import 'package:get_it/get_it.dart';
+
+abstract class LLM {
+  getResponse(List<Message> messages, ValueChanged<Message> onResponse,
+      ValueChanged<Message> errorCallback, ValueChanged<Message> onSuccess);
+}
+
+class ChatGpt extends LLM {
+  @override
+  getResponse(
+      List<Message> messages,
+      ValueChanged<Message> onResponse,
+      ValueChanged<Message> errorCallback,
+      ValueChanged<Message> onSuccess) async {
+    List<OpenAIChatCompletionChoiceMessageModel> openAIMessages = [];
+    //将messages反转
+    messages = messages.reversed.toList();
+    // 将messages里面的每条消息的内容取出来拼接在一起
+    String content = "";
+    for (Message message in messages) {
+      content = content + message.text;
+      if (content.length < 1800) {
+        // 插入到 openAIMessages 第一个位置
+        openAIMessages.insert(
+          0,
+          OpenAIChatCompletionChoiceMessageModel(
+            content: message.text,
+            role: message.role.asOpenAIChatMessageRole,
+          ),
+        );
+      }
+    }
+    var message = Message(
+        conversationId: messages.first.conversationId,
+        text: "",
+        role: Role.assistant); //仅仅第一个返回了角色
+    if (GetIt.instance.get<UserSettingCubit>().state.useStream) {
+      Stream<OpenAIStreamChatCompletionModel> chatStream = OpenAI.instance.chat
+          .createStream(
+              model: GetIt.instance.get<UserSettingCubit>().state.gptModel,
+              messages: openAIMessages);
+      chatStream.listen(
+        (chatStreamEvent) {
+          if (chatStreamEvent.choices.first.delta.content != null) {
+            message.text =
+                message.text + chatStreamEvent.choices.first.delta.content!;
+            onResponse(message);
+          }
+        },
+        onError: (error) {
+          message.text = error.message;
+          errorCallback(message);
+        },
+        onDone: () {
+          onSuccess(message);
+        },
+      );
+    } else {
+      try {
+        var response = await OpenAI.instance.chat.create(
+          model: GetIt.instance.get<UserSettingCubit>().state.gptModel,
+          messages: openAIMessages,
+        );
+        message.text = response.choices.first.message.content;
+        onSuccess(message);
+      } catch (e) {
+        message.text = e.toString();
+        errorCallback(message);
+      }
+    }
+  }
+}
diff --git a/lib/l10n/app_en.arb b/lib/l10n/app_en.arb
@@ -20,5 +20,6 @@
   "ok": "OK",
   "cancel": "Cancel",
   "useStreamApi": "Use Stream API",
-  "gptModel": "Select GPT Model"
+  "gptModel": "Select GPT Model",
+  "llmHint": "Select LLM Model"
 }
diff --git a/lib/l10n/app_zh.arb b/lib/l10n/app_zh.arb
@@ -20,5 +20,6 @@
   "ok": "确定",
   "cancel": "取消",
   "useStreamApi": "使用流式(Stream) API",
-  "gptModel": "选择 GPT Model"
+  "gptModel": "选择 GPT Model",
+  "llmHint": "选择大语言模型"
 }
diff --git a/lib/repository/message.dart b/lib/repository/message.dart
@@ -1,6 +1,9 @@
 import 'package:dart_openai/openai.dart';
 import 'package:flutter/foundation.dart';
 import 'package:flutter_chatgpt/cubit/setting_cubit.dart';
+import 'package:flutter_chatgpt/data/glm.dart';
+import 'package:flutter_chatgpt/data/if.dart';
+import 'package:flutter_chatgpt/data/llm.dart';
 import 'package:flutter_chatgpt/repository/conversation.dart';
 import 'package:get_it/get_it.dart';
 
@@ -32,64 +35,20 @@ class MessageRepository {
       ValueChanged<Message> onResponse,
       ValueChanged<Message> errorCallback,
       ValueChanged<Message> onSuccess) async {
-    List<OpenAIChatCompletionChoiceMessageModel> openAIMessages = [];
-    //将messages反转
-    messages = messages.reversed.toList();
-    while (true) {
-      // 将messages里面的每条消息的内容取出来拼接在一起
-      String content = "";
-      for (Message message in messages) {
-        content = content + message.text;
-        if (content.length < 1800) {
-          // 插入到 openAIMessages 第一个位置
-          openAIMessages.insert(
-            0,
-            OpenAIChatCompletionChoiceMessageModel(
-              content: message.text,
-              role: message.role.asOpenAIChatMessageRole,
-            ),
-          );
-        }
-      }
-      break;
-    }
-    var message = Message(
-        conversationId: messages.first.conversationId,
-        text: "",
-        role: Role.assistant); //仅仅第一个返回了角色
-    if (GetIt.instance.get<UserSettingCubit>().state.useStream) {
-      Stream<OpenAIStreamChatCompletionModel> chatStream = OpenAI.instance.chat
-          .createStream(
-              model: GetIt.instance.get<UserSettingCubit>().state.gptModel,
-              messages: openAIMessages);
-      chatStream.listen(
-        (chatStreamEvent) {
-          if (chatStreamEvent.choices.first.delta.content != null) {
-            message.text =
-                message.text + chatStreamEvent.choices.first.delta.content!;
-            onResponse(message);
-          }
-        },
-        onError: (error) {
-          message.text = error.message;
-          errorCallback(message);
-        },
-        onDone: () {
-          onSuccess(message);
-        },
-      );
-    } else {
-      try {
-        var response = await OpenAI.instance.chat.create(
-          model: GetIt.instance.get<UserSettingCubit>().state.gptModel,
-          messages: openAIMessages,
-        );
-        message.text = response.choices.first.message.content;
-        onSuccess(message);
-      } catch (e) {
-        message.text = e.toString();
-        errorCallback(message);
-      }
+    String llm = GetIt.instance.get<UserSettingCubit>().state.llm;
+
+    switch (llm.toUpperCase()) {
+      case "OPENAI":
+        ChatGpt().getResponse(messages, onResponse, errorCallback, onSuccess);
+        break;
+      case "CHATGLM":
+        ChatGlM().getResponse(messages, onResponse, errorCallback, onSuccess);
+        break;
+      case "IF":
+        ChatIF().getResponse(messages, onResponse, errorCallback, onSuccess);
+        break;
+      default:
+        ChatGpt().getResponse(messages, onResponse, errorCallback, onSuccess);
     }
   }