wehub-resource-sync
58 lines
1.8 KiB
Swift
58 lines
1.8 KiB
Swift
//
|
|
// ChatHistoryManager.swift
|
|
// MNNLLMiOS
|
|
//
|
|
// Created by 游薪渝(揽清) on 2025/1/15.
|
|
//
|
|
|
|
import ExyteChat
|
|
|
|
class ChatHistoryManager {
|
|
static let shared = ChatHistoryManager()
|
|
|
|
private init() {}
|
|
|
|
func saveChat(historyId: String, modelInfo: ModelInfo, messages: [Message]) {
|
|
ChatHistoryDatabase.shared?.saveChat(
|
|
historyId: historyId,
|
|
modelInfo: modelInfo,
|
|
messages: messages
|
|
)
|
|
}
|
|
|
|
// For backward compatibility
|
|
func saveChat(historyId: String, modelId: String, modelName _: String, messages: [Message]) {
|
|
let modelInfo = createFallbackModelInfo(for: modelId)
|
|
saveChat(historyId: historyId, modelInfo: modelInfo, messages: messages)
|
|
}
|
|
|
|
/// Create fallback ModelInfo that correctly identifies local models
|
|
private func createFallbackModelInfo(for modelId: String) -> ModelInfo {
|
|
// Check if this modelId corresponds to a local model
|
|
let localModels = ModelInfo.getAvailableLocalModels()
|
|
|
|
// Try to find matching local model by name or id
|
|
if let localModel = localModels.first(where: { $0.modelName == modelId || $0.id.contains(modelId) }) {
|
|
return localModel
|
|
}
|
|
|
|
// Fallback to downloaded model
|
|
return ModelInfo(modelId: modelId, isDownloaded: true)
|
|
}
|
|
|
|
func getAllHistory() -> [ChatHistory] {
|
|
return ChatHistoryDatabase.shared?.getAllHistory() ?? []
|
|
}
|
|
|
|
func getHistory(for modelId: String) -> [ChatHistory] {
|
|
let allHistories = getAllHistory()
|
|
let filtered = allHistories.filter { $0.modelId == modelId }
|
|
print("Found \(filtered.count) histories for modelId: \(modelId)")
|
|
return filtered
|
|
}
|
|
|
|
func deleteHistory(_ history: ChatHistory) {
|
|
ChatHistoryDatabase.shared?.deleteHistory(history)
|
|
}
|
|
}
|