chore: import upstream snapshot with attribution

backend/test/unit/knowledge/test_sample_question_utils.py

@@ -0,0 +1,108 @@
+from types import SimpleNamespace
+
+import pytest
+from fastapi import HTTPException
+
+from yuxi.knowledge.utils import sample_question_utils as sq
+
+
+def test_parse_sample_questions_content_strips_json_fence():
+    questions = sq.parse_sample_questions_content('```json\n{"questions": ["什么是测试？"]}\n```')
+
+    assert questions == ["什么是测试？"]
+
+
+def test_parse_sample_questions_content_rejects_invalid_payload():
+    with pytest.raises(ValueError, match="问题格式"):
+        sq.parse_sample_questions_content('{"items": []}')
+
+
+@pytest.mark.asyncio
+async def test_generate_database_sample_questions_rejects_empty_files(monkeypatch):
+    class FakeKnowledgeBase:
+        async def get_database_info(self, kb_id: str, include_files: bool = False) -> dict:
+            return {"name": "空知识库", "kb_type": "milvus", "files": {}}
+
+    monkeypatch.setattr(sq, "knowledge_base", FakeKnowledgeBase())
+    monkeypatch.setattr(
+        sq.KnowledgeBaseFactory,
+        "get_kb_class",
+        lambda _kb_type: SimpleNamespace(supports_documents=True),
+    )
+
+    with pytest.raises(HTTPException) as exc_info:
+        await sq.generate_database_sample_questions("kb_1")
+
+    assert exc_info.value.status_code == 400
+    assert "没有文件" in exc_info.value.detail
+
+
+@pytest.mark.asyncio
+async def test_generate_database_sample_questions_saves_and_returns_questions(monkeypatch):
+    saved: dict = {}
+
+    class FakeKnowledgeBase:
+        async def get_database_info(self, kb_id: str, include_files: bool = False) -> dict:
+            return {
+                "name": "测试知识库",
+                "kb_type": "milvus",
+                "files": {"file_1": {"filename": "demo.md", "file_type": "md"}},
+            }
+
+    class FakeModel:
+        async def call(self, messages, stream: bool = False):
+            assert messages[0]["role"] == "system"
+            assert "demo.md" in messages[1]["content"]
+            return SimpleNamespace(content='{"questions": ["如何使用 demo？"]}')
+
+    class FakeRepository:
+        async def update(self, kb_id: str, data: dict) -> None:
+            saved[kb_id] = data["sample_questions"]
+
+        async def get_by_kb_id(self, kb_id: str):
+            return SimpleNamespace(name="测试知识库", sample_questions=saved.get(kb_id))
+
+    monkeypatch.setattr(sq, "knowledge_base", FakeKnowledgeBase())
+    monkeypatch.setattr(
+        sq.KnowledgeBaseFactory,
+        "get_kb_class",
+        lambda _kb_type: SimpleNamespace(supports_documents=True),
+    )
+    monkeypatch.setattr(sq, "select_model", lambda model_spec: FakeModel())
+    monkeypatch.setattr(sq, "KnowledgeBaseRepository", lambda: FakeRepository())
+
+    generated = await sq.generate_database_sample_questions("kb_1", count=1)
+    stored = await sq.get_database_sample_questions("kb_1")
+
+    assert generated["questions"] == ["如何使用 demo？"]
+    assert generated["count"] == 1
+    assert stored["questions"] == ["如何使用 demo？"]
+
+
+@pytest.mark.asyncio
+async def test_generate_database_sample_questions_maps_invalid_json(monkeypatch):
+    class FakeKnowledgeBase:
+        async def get_database_info(self, kb_id: str, include_files: bool = False) -> dict:
+            return {
+                "name": "测试知识库",
+                "kb_type": "milvus",
+                "files": {"file_1": {"filename": "demo.md", "file_type": "md"}},
+            }
+
+    class FakeModel:
+        async def call(self, messages, stream: bool = False):
+            return SimpleNamespace(content="not json")
+
+    monkeypatch.setattr(sq, "knowledge_base", FakeKnowledgeBase())
+    monkeypatch.setattr(
+        sq.KnowledgeBaseFactory,
+        "get_kb_class",
+        lambda _kb_type: SimpleNamespace(supports_documents=True),
+    )
+    monkeypatch.setattr(sq, "select_model", lambda model_spec: FakeModel())
+
+    with pytest.raises(HTTPException) as exc_info:
+        await sq.generate_database_sample_questions("kb_1")
+
+    assert exc_info.value.status_code == 500
+    assert "AI返回格式错误" in exc_info.value.detail