wehub-resource-sync
129 lines
3.1 KiB
Swift
129 lines
3.1 KiB
Swift
//
|
|
// BatchTest.swift
|
|
// MNNLLMiOS
|
|
//
|
|
// Created by 游薪渝(揽清) on 2025/10/10.
|
|
//
|
|
|
|
import Foundation
|
|
|
|
/// Errors that can occur during batch file testing
|
|
enum BatchFileTestError: LocalizedError {
|
|
case unsupportedFileFormat
|
|
case invalidFileContent
|
|
case fileNotFound
|
|
case permissionDenied
|
|
|
|
var errorDescription: String? {
|
|
switch self {
|
|
case .unsupportedFileFormat:
|
|
return "Unsupported file format. Please use .txt, .json, or .jsonl files."
|
|
case .invalidFileContent:
|
|
return "Invalid file content. Please check the file format and try again."
|
|
case .fileNotFound:
|
|
return "The selected file could not be found."
|
|
case .permissionDenied:
|
|
return "Permission denied. Unable to access the selected file."
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Represents the type of batch test being performed
|
|
enum BatchTestType: String, CaseIterable {
|
|
case text = "text"
|
|
case image = "image"
|
|
case audio = "audio"
|
|
|
|
var displayName: String {
|
|
switch self {
|
|
case .text:
|
|
return "Text Test"
|
|
case .image:
|
|
return "Image Test"
|
|
case .audio:
|
|
return "Audio Test"
|
|
}
|
|
}
|
|
|
|
var fileExtension: String {
|
|
switch self {
|
|
case .text:
|
|
return "jsonl"
|
|
case .image:
|
|
return "jsonl"
|
|
case .audio:
|
|
return "jsonl"
|
|
}
|
|
}
|
|
|
|
var localBatchTestFolder: String {
|
|
switch self {
|
|
case .text:
|
|
return "Texts"
|
|
case .image:
|
|
return "Images"
|
|
case .audio:
|
|
return "Audios"
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Represents a single test item in a batch
|
|
struct BatchTestItem {
|
|
let prompt: String
|
|
let testType: BatchTestType
|
|
|
|
/// Creates a BatchTestItem with explicit test type
|
|
/// - Parameters:
|
|
/// - prompt: The test prompt text
|
|
/// - testType: The type of batch test
|
|
init(prompt: String, testType: BatchTestType) {
|
|
self.prompt = prompt
|
|
self.testType = testType
|
|
}
|
|
}
|
|
|
|
/// Represents the result of a single test execution
|
|
struct BatchTestResult {
|
|
let item: BatchTestItem
|
|
let response: String
|
|
let timestamp: Date
|
|
|
|
init(item: BatchTestItem, response: String, timestamp: Date) {
|
|
self.item = item
|
|
self.response = response
|
|
self.timestamp = timestamp
|
|
}
|
|
}
|
|
|
|
/// Available file formats for batch test output
|
|
enum BatchTestFileFormat: String, CaseIterable {
|
|
case txt
|
|
case json
|
|
case jsonl
|
|
|
|
var displayName: String {
|
|
switch self {
|
|
case .txt:
|
|
return "Text (.txt)"
|
|
case .json:
|
|
return "JSON (.json)"
|
|
case .jsonl:
|
|
return "JSON Lines (.jsonl)"
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Configuration for batch test operations
|
|
struct BatchTestConfiguration {
|
|
let outputFormat: BatchTestFileFormat
|
|
let includeTimestamp: Bool
|
|
let includePrompt: Bool
|
|
|
|
init(outputFormat: BatchTestFileFormat, includeTimestamp: Bool, includePrompt: Bool) {
|
|
self.outputFormat = outputFormat
|
|
self.includeTimestamp = includeTimestamp
|
|
self.includePrompt = includePrompt
|
|
}
|
|
}
|