mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 03:55:55 +00:00
1ecb0735cb
* Initial plan * Add component-kind filtering to list_plugins and filter pipeline extensions to only show plugins with Command, EventListener, or Tool components Co-authored-by: RockChinQ <45992437+RockChinQ@users.noreply.github.com> * fix: testing path --------- 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>
92 lines
4.4 KiB
Python
92 lines
4.4 KiB
Python
from __future__ import annotations
|
|
|
|
import quart
|
|
|
|
from ... import group
|
|
|
|
|
|
@group.group_class('pipelines', '/api/v1/pipelines')
|
|
class PipelinesRouterGroup(group.RouterGroup):
|
|
async def initialize(self) -> None:
|
|
@self.route('', methods=['GET', 'POST'], auth_type=group.AuthType.USER_TOKEN_OR_API_KEY)
|
|
async def _() -> str:
|
|
if quart.request.method == 'GET':
|
|
sort_by = quart.request.args.get('sort_by', 'created_at')
|
|
sort_order = quart.request.args.get('sort_order', 'DESC')
|
|
return self.success(
|
|
data={'pipelines': await self.ap.pipeline_service.get_pipelines(sort_by, sort_order)}
|
|
)
|
|
elif quart.request.method == 'POST':
|
|
json_data = await quart.request.json
|
|
|
|
pipeline_uuid = await self.ap.pipeline_service.create_pipeline(json_data)
|
|
|
|
return self.success(data={'uuid': pipeline_uuid})
|
|
|
|
@self.route('/_/metadata', methods=['GET'], auth_type=group.AuthType.USER_TOKEN_OR_API_KEY)
|
|
async def _() -> str:
|
|
return self.success(data={'configs': await self.ap.pipeline_service.get_pipeline_metadata()})
|
|
|
|
@self.route(
|
|
'/<pipeline_uuid>', methods=['GET', 'PUT', 'DELETE'], auth_type=group.AuthType.USER_TOKEN_OR_API_KEY
|
|
)
|
|
async def _(pipeline_uuid: str) -> str:
|
|
if quart.request.method == 'GET':
|
|
pipeline = await self.ap.pipeline_service.get_pipeline(pipeline_uuid)
|
|
|
|
if pipeline is None:
|
|
return self.http_status(404, -1, 'pipeline not found')
|
|
|
|
return self.success(data={'pipeline': pipeline})
|
|
elif quart.request.method == 'PUT':
|
|
json_data = await quart.request.json
|
|
|
|
await self.ap.pipeline_service.update_pipeline(pipeline_uuid, json_data)
|
|
|
|
return self.success()
|
|
elif quart.request.method == 'DELETE':
|
|
await self.ap.pipeline_service.delete_pipeline(pipeline_uuid)
|
|
|
|
return self.success()
|
|
|
|
@self.route(
|
|
'/<pipeline_uuid>/extensions', methods=['GET', 'PUT'], auth_type=group.AuthType.USER_TOKEN_OR_API_KEY
|
|
)
|
|
async def _(pipeline_uuid: str) -> str:
|
|
if quart.request.method == 'GET':
|
|
# Get current extensions and available plugins
|
|
pipeline = await self.ap.pipeline_service.get_pipeline(pipeline_uuid)
|
|
if pipeline is None:
|
|
return self.http_status(404, -1, 'pipeline not found')
|
|
|
|
# Only include plugins with pipeline-related components (Command, EventListener, Tool)
|
|
# Plugins that only have KnowledgeRetriever components are not suitable for pipeline extensions
|
|
pipeline_component_kinds = ['Command', 'EventListener', 'Tool']
|
|
plugins = await self.ap.plugin_connector.list_plugins(component_kinds=pipeline_component_kinds)
|
|
mcp_servers = await self.ap.mcp_service.get_mcp_servers(contain_runtime_info=True)
|
|
|
|
extensions_prefs = pipeline.get('extensions_preferences', {})
|
|
return self.success(
|
|
data={
|
|
'enable_all_plugins': extensions_prefs.get('enable_all_plugins', True),
|
|
'enable_all_mcp_servers': extensions_prefs.get('enable_all_mcp_servers', True),
|
|
'bound_plugins': extensions_prefs.get('plugins', []),
|
|
'available_plugins': plugins,
|
|
'bound_mcp_servers': extensions_prefs.get('mcp_servers', []),
|
|
'available_mcp_servers': mcp_servers,
|
|
}
|
|
)
|
|
elif quart.request.method == 'PUT':
|
|
# Update bound plugins and MCP servers for this pipeline
|
|
json_data = await quart.request.json
|
|
enable_all_plugins = json_data.get('enable_all_plugins', True)
|
|
enable_all_mcp_servers = json_data.get('enable_all_mcp_servers', True)
|
|
bound_plugins = json_data.get('bound_plugins', [])
|
|
bound_mcp_servers = json_data.get('bound_mcp_servers', [])
|
|
|
|
await self.ap.pipeline_service.update_pipeline_extensions(
|
|
pipeline_uuid, bound_plugins, bound_mcp_servers, enable_all_plugins, enable_all_mcp_servers
|
|
)
|
|
|
|
return self.success()
|