修复：类型不正确导致的编译失败

DavideYang125 · Apr 28, 2023 · b42739b · b42739b
1 parent 3744872
commit b42739b
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 7 deletions.
diff --git a/lib/repository/conversation.dart b/lib/repository/conversation.dart
@@ -1,3 +1,4 @@
+import 'package:dart_openai/openai.dart';
 import 'package:path/path.dart';
 import 'package:sqflite/sqflite.dart';
 
@@ -48,6 +49,19 @@ enum Role {
   assistant,
 }
 
+extension Convert on Role {
+  OpenAIChatMessageRole get asOpenAIChatMessageRole {
+    switch (this) {
+      case Role.assistant:
+        return OpenAIChatMessageRole.assistant;
+      case Role.system:
+        return OpenAIChatMessageRole.system;
+      case Role.user:
+        return OpenAIChatMessageRole.user;
+    }
+  }
+}
+
 class ConversationRepository {
   static const String _tableConversationName = 'conversations';
   static const String _tableMessageName = 'messages';

diff --git a/lib/repository/message.dart b/lib/repository/message.dart
@@ -1,5 +1,3 @@
-import 'dart:ui';
-
 import 'package:dart_openai/openai.dart';
 import 'package:flutter/foundation.dart';
 import 'package:flutter_chatgpt/cubit/setting_cubit.dart';
@@ -45,11 +43,12 @@ class MessageRepository {
         if (content.length < 1800) {
           // 插入到 openAIMessages 第一个位置
           openAIMessages.insert(
-              0,
-              OpenAIChatCompletionChoiceMessageModel(
-                content: message.text,
-                role: message.role.toString().split('.').last,
-              ));
+            0,
+            OpenAIChatCompletionChoiceMessageModel(
+              content: message.text,
+              role: message.role.asOpenAIChatMessageRole,
+            ),
+          );
         }
       }
       break;