mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 03:55:55 +00:00
a8481e43f0
* Initial plan * Add backend support for external knowledge bases Co-authored-by: RockChinQ <45992437+RockChinQ@users.noreply.github.com> * Add frontend support for external knowledge bases with tabs UI Co-authored-by: RockChinQ <45992437+RockChinQ@users.noreply.github.com> * Add i18n translations for all languages (Traditional Chinese and Japanese) Co-authored-by: RockChinQ <45992437+RockChinQ@users.noreply.github.com> * Update knowledge base tab list styling to match plugins page Co-authored-by: RockChinQ <45992437+RockChinQ@users.noreply.github.com> * perf: margin-top for kb page * refactor: switch RetrievalResultEntry to langbot_plugin pkg ones * feat: knowledge retriever listing and creating * stash * refactor: unify sync mechanism for polymorphic components * feat: use unified retireval result struct in retrieval test page * chore: remove unused methods * feat: retriever icon displaying * feat: localagent retrieval with external kbs * chore: bump version of langbot-plugin to 0.2.0b1 * fix: i18n --------- Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com> Co-authored-by: RockChinQ <45992437+RockChinQ@users.noreply.github.com> Co-authored-by: Junyan Qin <rockchinq@gmail.com>
62 lines
2.6 KiB
Python
62 lines
2.6 KiB
Python
import quart
|
|
from ... import group
|
|
|
|
|
|
@group.group_class('external_knowledge_base', '/api/v1/knowledge/external-bases')
|
|
class ExternalKnowledgeBaseRouterGroup(group.RouterGroup):
|
|
async def initialize(self) -> None:
|
|
@self.route('/retrievers', methods=['GET'])
|
|
async def list_knowledge_retrievers() -> quart.Response:
|
|
"""List all available knowledge retrievers from plugins."""
|
|
retrievers = await self.ap.plugin_connector.list_knowledge_retrievers()
|
|
return self.success(data={'retrievers': retrievers})
|
|
|
|
@self.route('', methods=['POST', 'GET'])
|
|
async def handle_external_knowledge_bases() -> quart.Response:
|
|
if quart.request.method == 'GET':
|
|
external_kbs = await self.ap.external_kb_service.get_external_knowledge_bases()
|
|
return self.success(data={'bases': external_kbs})
|
|
|
|
elif quart.request.method == 'POST':
|
|
json_data = await quart.request.json
|
|
kb_uuid = await self.ap.external_kb_service.create_external_knowledge_base(json_data)
|
|
return self.success(data={'uuid': kb_uuid})
|
|
|
|
return self.http_status(405, -1, 'Method not allowed')
|
|
|
|
@self.route(
|
|
'/<kb_uuid>',
|
|
methods=['GET', 'DELETE', 'PUT'],
|
|
)
|
|
async def handle_specific_external_knowledge_base(kb_uuid: str) -> quart.Response:
|
|
if quart.request.method == 'GET':
|
|
external_kb = await self.ap.external_kb_service.get_external_knowledge_base(kb_uuid)
|
|
|
|
if external_kb is None:
|
|
return self.http_status(404, -1, 'external knowledge base not found')
|
|
|
|
return self.success(
|
|
data={
|
|
'base': external_kb,
|
|
}
|
|
)
|
|
|
|
elif quart.request.method == 'PUT':
|
|
json_data = await quart.request.json
|
|
await self.ap.external_kb_service.update_external_knowledge_base(kb_uuid, json_data)
|
|
return self.success({})
|
|
|
|
elif quart.request.method == 'DELETE':
|
|
await self.ap.external_kb_service.delete_external_knowledge_base(kb_uuid)
|
|
return self.success({})
|
|
|
|
@self.route(
|
|
'/<kb_uuid>/retrieve',
|
|
methods=['POST'],
|
|
)
|
|
async def retrieve_external_knowledge_base(kb_uuid: str) -> str:
|
|
json_data = await quart.request.json
|
|
query = json_data.get('query')
|
|
results = await self.ap.external_kb_service.retrieve_external_knowledge_base(kb_uuid, query)
|
|
return self.success(data={'results': results})
|