fix type error

coaidev · Oct 24, 2023 · b9b0b94 · b9b0b94
1 parent 5e67bd4
commit b9b0b94
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/manager/conversation/storage.go b/manager/conversation/storage.go
@@ -54,11 +54,20 @@ func LoadConversation(db *sql.DB, userId int64, conversationId int64) *Conversat
 		Id:     conversationId,
 	}
 
-	var data string
+	var (
+		data  string
+		model interface{}
+	)
 	err := db.QueryRow(`
 		SELECT conversation_name, model, data FROM conversation
 		WHERE user_id = ? AND conversation_id = ?
-		`, userId, conversationId).Scan(&conversation.Name, &data, &conversation.Model)
+		`, userId, conversationId).Scan(&conversation.Name, &model, &data)
+	if value, ok := model.(string); ok {
+		conversation.Model = value
+	} else {
+		conversation.Model = globals.GPT3Turbo
+	}
+
 	if err != nil {
 		return nil
 	}