feat(keyboard): ship AI mode surface with streaming search answers

Add the AI keyboard tab, Agent settings, and user-owned LLM key path for 1.7.0, including streaming answers and web-search transports without the built-in DeepSeek fallback.
2026-08-11 01:06:27 +08:00
parent f6212dcd2c
commit 3de665d254
81 changed files with 4467 additions and 398 deletions
@@ -0,0 +1,157 @@
+import XCTest
+@testable import OSGKeyboardShared
+
+final class AISessionStateTests: XCTestCase {
+    func testSuccessfulAnswerReplacesPreviousOnlyAtTerminalResult() throws {
+        var state = AISessionState()
+        let conversationID = UUID()
+        let firstUtteranceID = UUID()
+        state.enter(conversationID: conversationID)
+        state.beginPreparing(utteranceID: firstUtteranceID)
+        state.beginListening(utteranceID: firstUtteranceID)
+        state.beginRecognizing(utteranceID: firstUtteranceID)
+        state.beginGenerating(question: "问题一", utteranceID: firstUtteranceID)
+        state.receiveAnswer("答案一", utteranceID: firstUtteranceID)
+
+        let secondUtteranceID = UUID()
+        state.beginPreparing(utteranceID: secondUtteranceID)
+        XCTAssertEqual(state.answer?.text, "答案一")
+
+        state.beginListening(utteranceID: secondUtteranceID)
+        state.beginGenerating(question: "问题二", utteranceID: secondUtteranceID)
+        XCTAssertEqual(state.answer?.text, "答案一")
+
+        state.receiveAnswer("答案二", utteranceID: secondUtteranceID)
+        XCTAssertEqual(state.answer?.text, "答案二")
+        XCTAssertTrue(state.canInsert)
+    }
+
+    func testCancellationRestoresPreviousAnswerState() {
+        var state = AISessionState()
+        let firstUtteranceID = UUID()
+        state.enter()
+        state.beginPreparing(utteranceID: firstUtteranceID)
+        state.beginListening(utteranceID: firstUtteranceID)
+        state.receiveAnswer("可用答案", utteranceID: firstUtteranceID)
+
+        let secondUtteranceID = UUID()
+        state.beginPreparing(utteranceID: secondUtteranceID)
+        state.updateTranscript("未完成问题", utteranceID: secondUtteranceID)
+        state.cancelCurrentWork()
+
+        XCTAssertEqual(state.phase, .ready)
+        XCTAssertEqual(state.answer?.text, "可用答案")
+        XCTAssertEqual(state.transcript, "")
+    }
+
+    func testAnswerRequiresInsertBeforeSend() {
+        var state = AISessionState()
+        let utteranceID = UUID()
+        state.enter()
+        state.beginPreparing(utteranceID: utteranceID)
+        state.receiveAnswer("答案", utteranceID: utteranceID)
+
+        XCTAssertTrue(state.canInsert)
+        XCTAssertFalse(state.canSend)
+
+        state.markAnswerInserted(offersSend: true)
+
+        XCTAssertEqual(state.phase, .awaitingSend)
+        XCTAssertFalse(state.canInsert)
+        XCTAssertTrue(state.canSend)
+        XCTAssertEqual(state.answer?.isInserted, true)
+
+        state.markAnswerSent()
+
+        XCTAssertEqual(state.phase, .sent)
+        XCTAssertFalse(state.canSend)
+        XCTAssertEqual(state.answer?.isSent, true)
+    }
+
+    func testNonSendFieldFinishesAfterInsertion() {
+        var state = AISessionState()
+        let utteranceID = UUID()
+        state.enter()
+        state.beginPreparing(utteranceID: utteranceID)
+        state.receiveAnswer("答案", utteranceID: utteranceID)
+
+        state.markAnswerInserted(offersSend: false)
+
+        XCTAssertEqual(state.phase, .inserted)
+        XCTAssertFalse(state.canPerformAnswerAction)
+        XCTAssertEqual(state.answer?.isInserted, true)
+        XCTAssertEqual(state.answer?.isSent, false)
+    }
+
+    func testCancellationRestoresPendingSendState() {
+        var state = AISessionState()
+        let firstUtteranceID = UUID()
+        state.enter()
+        state.beginPreparing(utteranceID: firstUtteranceID)
+        state.receiveAnswer("已插入答案", utteranceID: firstUtteranceID)
+        state.markAnswerInserted(offersSend: true)
+
+        let secondUtteranceID = UUID()
+        state.beginPreparing(utteranceID: secondUtteranceID)
+        state.cancelCurrentWork()
+
+        XCTAssertEqual(state.phase, .awaitingSend)
+        XCTAssertTrue(state.canSend)
+    }
+
+    func testStaleResultCannotReplaceCurrentAnswer() {
+        var state = AISessionState()
+        let currentUtteranceID = UUID()
+        state.enter()
+        state.beginPreparing(utteranceID: currentUtteranceID)
+
+        state.receiveAnswer("迟到答案", utteranceID: UUID())
+
+        XCTAssertNil(state.answer)
+        XCTAssertEqual(state.phase, .preparing)
+    }
+
+    func testPartialAnswerKeepsPreviousCommittedAnswerUntilFinal() {
+        var state = AISessionState()
+        let firstUtteranceID = UUID()
+        state.enter()
+        state.beginPreparing(utteranceID: firstUtteranceID)
+        state.receiveAnswer("答案一", utteranceID: firstUtteranceID)
+
+        let secondUtteranceID = UUID()
+        state.beginPreparing(utteranceID: secondUtteranceID)
+        state.beginGenerating(question: "问题二", utteranceID: secondUtteranceID)
+        state.receivePartialAnswer("草", utteranceID: secondUtteranceID)
+
+        XCTAssertEqual(state.phase, .generating)
+        XCTAssertEqual(state.draftAnswerText, "草")
+        XCTAssertEqual(state.answer?.text, "答案一")
+        XCTAssertFalse(state.canInsert)
+
+        state.receivePartialAnswer("草稿答案", utteranceID: secondUtteranceID)
+        XCTAssertEqual(state.draftAnswerText, "草稿答案")
+
+        state.receiveAnswer("答案二", utteranceID: secondUtteranceID)
+        XCTAssertNil(state.draftAnswerText)
+        XCTAssertEqual(state.answer?.text, "答案二")
+        XCTAssertTrue(state.canInsert)
+    }
+
+    func testCancelClearsDraftAndRestoresPreviousAnswer() {
+        var state = AISessionState()
+        let firstUtteranceID = UUID()
+        state.enter()
+        state.beginPreparing(utteranceID: firstUtteranceID)
+        state.receiveAnswer("可用答案", utteranceID: firstUtteranceID)
+
+        let secondUtteranceID = UUID()
+        state.beginPreparing(utteranceID: secondUtteranceID)
+        state.beginGenerating(question: "新问题", utteranceID: secondUtteranceID)
+        state.receivePartialAnswer("半截", utteranceID: secondUtteranceID)
+        state.cancelCurrentWork()
+
+        XCTAssertNil(state.draftAnswerText)
+        XCTAssertEqual(state.phase, .ready)
+        XCTAssertEqual(state.answer?.text, "可用答案")
+    }
+}