mirror of
https://github.com/Tencent/WeKnora.git
synced 2026-09-01 14:53:07 +08:00
01luyicheng
54c4309c65
Add an MCP tool that creates a knowledge entry from raw Markdown text by
calling the existing backend endpoint
POST /knowledge-bases/{id}/knowledge/manual (handler.CreateManualKnowledge).
This completes the "text" half of #323, whose "file" half was already
covered by create_knowledge_from_file.
The tool defaults to status="publish" so the created entry is chunked,
embedded and made searchable immediately — matching the API/MCP use case
where there is no UI to publish a draft. Callers may pass status="draft"
to save without indexing.
- New client method WeKnoraClient.create_knowledge_from_text.
- New @mcp.tool() create_knowledge_from_text, resolving kb_id by name or
UUID (consistent with create_session / hybrid_search).
- Add focused unittests verifying endpoint, request body shape, status
default/override, and kb_id resolution wiring.
- Bump the stdio tools-list count assertion in test_mcp_transports.py
from 29 to 30 for the newly registered tool.
60 lines
2.6 KiB
Python
60 lines
2.6 KiB
Python
#!/usr/bin/env python3
|
|
"""Tests for the create_knowledge_from_text MCP tool and client method.
|
|
|
|
unittest discover 会收集本文件中的 TestCase；也可直接运行:
|
|
python test_create_knowledge_from_text.py
|
|
"""
|
|
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
import weknora_mcp_server as srv
|
|
|
|
|
|
class CreateKnowledgeFromTextTest(unittest.TestCase):
|
|
def test_client_method_exists_and_callable(self):
|
|
client = srv.WeKnoraClient("http://localhost:8080/api/v1", "k")
|
|
self.assertTrue(callable(getattr(client, "create_knowledge_from_text", None)))
|
|
|
|
def test_posts_to_manual_endpoint_with_expected_body(self):
|
|
client = srv.WeKnoraClient("http://localhost:8080/api/v1", "k")
|
|
with mock.patch.object(client, "_request", return_value={"ok": True}) as req:
|
|
client.create_knowledge_from_text("kb-1", "T", "C", tag_ids=["t1"])
|
|
args, kwargs = req.call_args
|
|
self.assertEqual(args[0], "POST")
|
|
self.assertEqual(args[1], "/knowledge-bases/kb-1/knowledge/manual")
|
|
body = kwargs["json"]
|
|
self.assertEqual(body["title"], "T")
|
|
self.assertEqual(body["content"], "C")
|
|
self.assertEqual(body["tag_ids"], ["t1"])
|
|
# status defaults to "publish" so the entry is indexed/searchable.
|
|
self.assertEqual(body["status"], "publish")
|
|
|
|
def test_status_can_be_overridden_to_draft(self):
|
|
client = srv.WeKnoraClient("http://localhost:8080/api/v1", "k")
|
|
with mock.patch.object(client, "_request", return_value={"ok": True}) as req:
|
|
client.create_knowledge_from_text("kb-1", "T", "C", status="draft")
|
|
self.assertEqual(req.call_args.kwargs["json"]["status"], "draft")
|
|
|
|
def test_omits_tag_ids_when_none(self):
|
|
client = srv.WeKnoraClient("http://localhost:8080/api/v1", "k")
|
|
with mock.patch.object(client, "_request", return_value={"ok": True}) as req:
|
|
client.create_knowledge_from_text("kb-1", "T", "C")
|
|
body = req.call_args.kwargs["json"]
|
|
self.assertNotIn("tag_ids", body)
|
|
self.assertEqual(body["status"], "publish")
|
|
|
|
def test_tool_function_resolves_kb_id_and_forwards_status(self):
|
|
with mock.patch.object(
|
|
srv.client, "resolve_kb_id", return_value="uuid-1"
|
|
) as rid, mock.patch.object(
|
|
srv.client, "create_knowledge_from_text", return_value={"ok": True}
|
|
) as method:
|
|
srv.create_knowledge_from_text("my-kb", "T", "C", status="draft")
|
|
rid.assert_called_once_with("my-kb")
|
|
method.assert_called_once_with("uuid-1", "T", "C", tag_ids=None, status="draft")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|