mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-08-01 13:33:48 +08:00
Refactor: reformat all code for lefthook using ruff and gofmt (#16585)
This commit is contained in:
@@ -42,9 +42,7 @@ class TestAggregateByField:
|
||||
assert set(out) == {("user", 2), ("assistant", 1)}
|
||||
|
||||
def test_single_doc_result(self):
|
||||
messages = [
|
||||
{"id": "m1", "message_type_kwd": "user", "content_ltks": "x", "message_id": "msg1", "memory_id": "mem1", "status_int": 1}
|
||||
]
|
||||
messages = [{"id": "m1", "message_type_kwd": "user", "content_ltks": "x", "message_id": "msg1", "memory_id": "mem1", "status_int": 1}]
|
||||
out = aggregate_by_field(messages, "message_type_kwd")
|
||||
assert out == [("user", 1)]
|
||||
|
||||
|
||||
@@ -57,9 +57,7 @@ class TestGetHighlightFromMessages:
|
||||
assert get_highlight_from_messages(None, ["k"], "content_ltks") == {}
|
||||
|
||||
def test_empty_keywords_returns_empty_dict(self):
|
||||
assert get_highlight_from_messages(
|
||||
[{"id": "m1", "content_ltks": "hello"}], [], "content_ltks"
|
||||
) == {}
|
||||
assert get_highlight_from_messages([{"id": "m1", "content_ltks": "hello"}], [], "content_ltks") == {}
|
||||
|
||||
def test_returns_id_to_highlighted_text(self):
|
||||
messages = [
|
||||
|
||||
Reference in New Issue
Block a user