feat: llm history

guchengxi1994 · Apr 30, 2024 · 76fa59f · 76fa59f
1 parent a0649ae
commit 76fa59f
Show file tree

Hide file tree

Showing 10 changed files with 371 additions and 138 deletions.
diff --git a/lib/isar/database.dart b/lib/isar/database.dart
@@ -24,7 +24,7 @@ class IsarDatabase {
     ScheduleItemSchema,
     RecentlyUsedSchema,
     LLMHistorySchema,
-    LLMHistoryMessagesSchema
+    LLMHistoryMessageSchema
   ];
 
   Future initialDatabase() async {

diff --git a/lib/isar/llm_history.dart b/lib/isar/llm_history.dart
@@ -11,17 +11,18 @@ class LLMHistory {
   @enumerated
   late LLMType llmType = LLMType.openai;
 
-  final messages = IsarLinks<LLMHistoryMessages>();
+  final messages = IsarLinks<LLMHistoryMessage>();
 }
 
 enum MessageType { query, response }
 
 @collection
-class LLMHistoryMessages {
+class LLMHistoryMessage {
   @enumerated
   late MessageType messageType;
 
   Id id = Isar.autoIncrement;
+  late int roleType = 0;
 
   String? content;
 }