mii/AzooKeyKanaKanjiConverter
1
0
Fork 0
mirror of https://github.com/mii443/AzooKeyKanaKanjiConverter.git synced 2025-08-22 15:05:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add session CLI for dynamic user dictionary

Browse Source
This commit is contained in:
Miwa / Ensan
2025-07-21 11:18:18 +09:00
parent a055679f7e
commit 80be12f3ef
3 changed files with 33 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Sources/CliTool/Subcommands/EvaluateCommand.swift
View File

@ -146,6 +146,7 @@ extension Subcommands {
/// ユーザ辞書
var user_dictionary: [InputUserDictionaryItem]? = nil
}
struct InputUserDictionaryItem: Codable {
/// 漢字
@ -155,7 +156,6 @@ extension Subcommands {
/// ヒント
var hint: String? = nil
}
}
struct EvaluateResult: Codable {
internal init(n_best: Int, timestamp: TimeInterval = Date().timeIntervalSince1970, execution_time: TimeInterval, items: [Subcommands.EvaluateItem]) {

17
Sources/CliTool/Subcommands/SessionCommand.swift
View File

@ -30,6 +30,8 @@ extension Subcommands {
var reportScore = false
@Flag(name: [.customLong("roman2kana")], help: "Use roman2kana input.")
var roman2kana = false
@Option(name: [.customLong("config_user_dictionary")], help: "User Dictionary JSON file path")
var configUserDictionary: String? = nil
@Option(name: [.customLong("config_zenzai_inference_limit")], help: "inference limit for zenzai.")
var configZenzaiInferenceLimit: Int = .max
@Flag(name: [.customLong("config_zenzai_rich_n_best")], help: "enable rich n_best generation for zenzai.")
@ -70,6 +72,15 @@ extension Subcommands {
}
}
private func parseUserDictionaryFile() throws -> [InputUserDictionaryItem] {
guard let configUserDictionary else {
return []
}
let url = URL(fileURLWithPath: configUserDictionary)
let data = try Data(contentsOf: url)
return try JSONDecoder().decode([InputUserDictionaryItem].self, from: data)
}
@MainActor mutating func run() async {
if self.zenzV1 || self.zenzV2 {
print("\(bold: "We strongly recommend to use zenz-v3 models")")
@ -80,6 +91,11 @@ extension Subcommands {
if !self.zenzWeightPath.isEmpty && (!self.zenzV1 && !self.zenzV2 && !self.zenzV3) {
print("zenz version is not specified. By default, zenz-v3 will be used.")
}
let userDictionary = try! self.parseUserDictionaryFile().map {
DicdataElement(word: $0.word, ruby: $0.reading.toKatakana(), cid: CIDData.固有名詞.cid, mid: MIDData.一般.mid, value: -10)
}
let learningType: LearningType = if self.readOnlyMemoryPath != nil {
// 読み取りのみ
.onlyOutput
@ -107,6 +123,7 @@ extension Subcommands {
converter.sendToDicdataStore(
.setRequestOptions(requestOptions(learningType: learningType, memoryDirectory: memoryDirectory, leftSideContext: nil))
)
converter.sendToDicdataStore(.importDynamicUserDict(userDictionary))
var composingText = ComposingText()
let inputStyle: InputStyle = self.roman2kana ? .roman2kana : .direct
var lastCandidates: [Candidate] = []

8
Tests/KanaKanjiConverterModuleTests/ConverterTests/TemplateConversionTests.swift Normal file
View File

@ -0,0 +1,8 @@
//
// File.swift
// AzooKeyKanakanjiConverter
//
// Created by miwa on 2025/07/21.
//
import Foundation