Playwright : add new test for configuration tab in datasets (#13365)

### What problem does this PR solve?

this pr adds new tests, for the full configuration tab in datasests

### Type of change

- [x] Other (please describe): new tests

test/unit_test/api/db/services/test_dialog_service_use_sql_source_columns.py

@@ -0,0 +1,221 @@
+#
+#  Copyright 2026 The InfiniFlow Authors. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+#
+import asyncio
+import sys
+import types
+import warnings
+
+import pytest
+
+# xgboost imports pkg_resources and emits a deprecation warning that is promoted
+# to error in our pytest configuration; ignore it for this unit test module.
+warnings.filterwarnings(
+    "ignore",
+    message="pkg_resources is deprecated as an API.*",
+    category=UserWarning,
+)
+
+
+def _install_cv2_stub_if_unavailable():
+    try:
+        import cv2  # noqa: F401
+        return
+    except Exception:
+        pass
+
+    stub = types.ModuleType("cv2")
+
+    # Constants referenced by deepdoc import-time defaults.
+    stub.INTER_LINEAR = 1
+    stub.INTER_CUBIC = 2
+    stub.BORDER_CONSTANT = 0
+    stub.BORDER_REPLICATE = 1
+    stub.COLOR_BGR2RGB = 0
+    stub.COLOR_BGR2GRAY = 1
+    stub.COLOR_GRAY2BGR = 2
+    stub.IMREAD_IGNORE_ORIENTATION = 128
+    stub.IMREAD_COLOR = 1
+    stub.RETR_LIST = 1
+    stub.CHAIN_APPROX_SIMPLE = 2
+
+    def _missing(*_args, **_kwargs):
+        raise RuntimeError("cv2 runtime call is unavailable in this test environment")
+
+    def _module_getattr(name):
+        if name.isupper():
+            return 0
+        return _missing
+
+    stub.__getattr__ = _module_getattr
+    sys.modules["cv2"] = stub
+
+
+_install_cv2_stub_if_unavailable()
+
+from api.db.services import dialog_service
+
+
+class _StubChatModel:
+    def __init__(self, outputs):
+        self._outputs = outputs
+        self.calls = []
+
+    async def async_chat(self, system_prompt, messages, llm_setting):
+        idx = len(self.calls)
+        if idx >= len(self._outputs):
+            raise AssertionError("async_chat called more times than expected")
+        self.calls.append(
+            {
+                "system_prompt": system_prompt,
+                "message": messages[0]["content"],
+                "llm_setting": llm_setting,
+            }
+        )
+        return self._outputs[idx]
+
+
+class _StubRetriever:
+    def __init__(self, results):
+        self._results = results
+        self.sql_calls = []
+
+    def sql_retrieval(self, sql, format="json"):
+        assert format == "json"
+        idx = len(self.sql_calls)
+        if idx >= len(self._results):
+            raise AssertionError("sql_retrieval called more times than expected")
+        self.sql_calls.append(sql)
+        return self._results[idx]
+
+
+@pytest.fixture
+def force_es_engine(monkeypatch):
+    monkeypatch.setattr(dialog_service.settings, "DOC_ENGINE_INFINITY", False)
+    monkeypatch.setattr(dialog_service.settings, "DOC_ENGINE_OCEANBASE", False)
+
+
+@pytest.mark.p2
+def test_use_sql_repairs_missing_source_columns_for_non_aggregate(monkeypatch, force_es_engine):
+    retriever = _StubRetriever(
+        [
+            {
+                "columns": [{"name": "product"}],
+                "rows": [["desk"], ["monitor"]],
+            },
+            {
+                "columns": [{"name": "doc_id"}, {"name": "docnm_kwd"}, {"name": "product"}],
+                "rows": [["doc-1", "products.xlsx", "desk"], ["doc-2", "products.xlsx", "monitor"]],
+            },
+        ]
+    )
+    chat_model = _StubChatModel(
+        [
+            "SELECT product FROM ragflow_tenant",
+            "SELECT doc_id, docnm_kwd, product FROM ragflow_tenant",
+        ]
+    )
+    monkeypatch.setattr(dialog_service.settings, "retriever", retriever, raising=False)
+
+    result = asyncio.run(
+        dialog_service.use_sql(
+            question="show me column of product",
+            field_map={"product": "product"},
+            tenant_id="tenant-id",
+            chat_mdl=chat_model,
+            quota=True,
+            kb_ids=None,
+        )
+    )
+
+    assert result is not None
+    assert "|product|Source|" in result["answer"]
+    assert len(chat_model.calls) == 2
+    assert len(retriever.sql_calls) == 2
+
+
+@pytest.mark.p2
+def test_use_sql_keeps_aggregate_flow_without_source_repair(monkeypatch, force_es_engine):
+    retriever = _StubRetriever(
+        [
+            {
+                "columns": [{"name": "count(star)"}],
+                "rows": [[6]],
+            },
+        ]
+    )
+    chat_model = _StubChatModel(
+        [
+            "SELECT COUNT(*) FROM ragflow_tenant",
+        ]
+    )
+    monkeypatch.setattr(dialog_service.settings, "retriever", retriever, raising=False)
+
+    result = asyncio.run(
+        dialog_service.use_sql(
+            question="how many rows are there",
+            field_map={"product": "product"},
+            tenant_id="tenant-id",
+            chat_mdl=chat_model,
+            quota=True,
+            kb_ids=None,
+        )
+    )
+
+    assert result is not None
+    assert "|COUNT(*)|" in result["answer"]
+    assert "Source" not in result["answer"]
+    assert len(chat_model.calls) == 1
+    assert len(retriever.sql_calls) == 1
+
+
+@pytest.mark.p2
+def test_use_sql_source_repair_is_bounded_to_single_retry(monkeypatch, force_es_engine):
+    retriever = _StubRetriever(
+        [
+            {
+                "columns": [{"name": "product"}],
+                "rows": [["desk"]],
+            },
+            {
+                "columns": [{"name": "product"}],
+                "rows": [["desk"]],
+            },
+        ]
+    )
+    chat_model = _StubChatModel(
+        [
+            "SELECT product FROM ragflow_tenant",
+            "SELECT product FROM ragflow_tenant WHERE product IS NOT NULL",
+        ]
+    )
+    monkeypatch.setattr(dialog_service.settings, "retriever", retriever, raising=False)
+
+    result = asyncio.run(
+        dialog_service.use_sql(
+            question="show me column of product",
+            field_map={"product": "product"},
+            tenant_id="tenant-id",
+            chat_mdl=chat_model,
+            quota=True,
+            kb_ids=None,
+        )
+    )
+
+    assert result is not None
+    assert "|product|" in result["answer"]
+    assert "Source" not in result["answer"]
+    assert len(chat_model.calls) == 2
+    assert len(retriever.sql_calls) == 2