54 lines
1.9 KiB
Swift
54 lines
1.9 KiB
Swift
import Foundation
|
|
import Testing
|
|
@testable import Clawdbot
|
|
|
|
@Suite
|
|
struct ModelCatalogLoaderTests {
|
|
@Test
|
|
func loadParsesModelsFromTypeScriptAndSorts() async throws {
|
|
let src = """
|
|
export const MODELS = {
|
|
openai: {
|
|
"gpt-4o-mini": { name: "GPT-4o mini", contextWindow: 128000 } satisfies any,
|
|
"gpt-4o": { name: "GPT-4o", contextWindow: 128000 } as any,
|
|
"gpt-3.5": { contextWindow: 16000 },
|
|
},
|
|
anthropic: {
|
|
"claude-3": { name: "Claude 3", contextWindow: 200000 },
|
|
},
|
|
};
|
|
"""
|
|
|
|
let tmp = FileManager().temporaryDirectory
|
|
.appendingPathComponent("models-\(UUID().uuidString).ts")
|
|
defer { try? FileManager().removeItem(at: tmp) }
|
|
try src.write(to: tmp, atomically: true, encoding: .utf8)
|
|
|
|
let choices = try await ModelCatalogLoader.load(from: tmp.path)
|
|
#expect(choices.count == 4)
|
|
#expect(choices.first?.provider == "anthropic")
|
|
#expect(choices.first?.id == "claude-3")
|
|
|
|
let ids = Set(choices.map(\.id))
|
|
#expect(ids == Set(["claude-3", "gpt-4o", "gpt-4o-mini", "gpt-3.5"]))
|
|
|
|
let openai = choices.filter { $0.provider == "openai" }
|
|
let openaiNames = openai.map(\.name)
|
|
#expect(openaiNames == openaiNames.sorted { a, b in
|
|
a.localizedCaseInsensitiveCompare(b) == .orderedAscending
|
|
})
|
|
}
|
|
|
|
@Test
|
|
func loadWithNoExportReturnsEmptyChoices() async throws {
|
|
let src = "const NOPE = 1;"
|
|
let tmp = FileManager().temporaryDirectory
|
|
.appendingPathComponent("models-\(UUID().uuidString).ts")
|
|
defer { try? FileManager().removeItem(at: tmp) }
|
|
try src.write(to: tmp, atomically: true, encoding: .utf8)
|
|
|
|
let choices = try await ModelCatalogLoader.load(from: tmp.path)
|
|
#expect(choices.isEmpty)
|
|
}
|
|
}
|