LZH-YS1998
82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
|
|
from opc.plugins.office_ui.chat_store import ChatStore
|
|
|
|
|
|
class ChatStoreProgressFoldingTests(unittest.TestCase):
|
|
def test_stream_fold_preserves_first_timestamp(self) -> None:
|
|
first = {
|
|
"timestamp": 1_700_000_000.0,
|
|
"type": "thinking",
|
|
"summary": "Thinking",
|
|
"detail": "Need ",
|
|
"turn_id": "runtime-1:1",
|
|
"item_id": "runtime-1:1:thinking",
|
|
"seq": 1,
|
|
}
|
|
deltas = [
|
|
{
|
|
"timestamp": 1_700_000_000.1,
|
|
"type": "thinking",
|
|
"summary": "Thinking",
|
|
"detail": "more ",
|
|
"turn_id": "runtime-1:1",
|
|
"item_id": "runtime-1:1:thinking",
|
|
"seq": 2,
|
|
},
|
|
{
|
|
"timestamp": 1_700_000_000.2,
|
|
"type": "thinking",
|
|
"summary": "Thinking",
|
|
"detail": "context",
|
|
"turn_id": "runtime-1:1",
|
|
"item_id": "runtime-1:1:thinking",
|
|
"seq": 3,
|
|
},
|
|
]
|
|
|
|
folded = ChatStore._fold_progress_entries([first], deltas)
|
|
|
|
self.assertEqual(len(folded), 1)
|
|
self.assertEqual(folded[0]["timestamp"], first["timestamp"])
|
|
self.assertEqual(folded[0]["detail"], "Need more context")
|
|
self.assertEqual(folded[0]["seq"], 3)
|
|
# Folding builds a replacement row and must not mutate the persisted
|
|
# value supplied by the caller.
|
|
self.assertEqual(first["detail"], "Need ")
|
|
|
|
def test_stream_fold_within_one_batch_keeps_creation_timestamp(self) -> None:
|
|
folded = ChatStore._fold_progress_entries(
|
|
[],
|
|
[
|
|
{
|
|
"timestamp": 10.0,
|
|
"type": "assistant",
|
|
"summary": "Part one",
|
|
"detail": "Part one ",
|
|
"turn_id": "runtime-2:1",
|
|
"stream_id": "runtime-2:1:assistant",
|
|
"seq": 1,
|
|
},
|
|
{
|
|
"timestamp": 11.0,
|
|
"type": "assistant",
|
|
"summary": "part two",
|
|
"detail": "part two",
|
|
"turn_id": "runtime-2:1",
|
|
"stream_id": "runtime-2:1:assistant",
|
|
"seq": 2,
|
|
},
|
|
],
|
|
)
|
|
|
|
self.assertEqual(len(folded), 1)
|
|
self.assertEqual(folded[0]["timestamp"], 10.0)
|
|
self.assertEqual(folded[0]["detail"], "Part one part two")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|