mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-05 05:16:03 +00:00
4a84bf2355
* feat: add persistence field * feat: add basic extension page in pipeline config * Merge pull request #1751 from langbot-app/copilot/add-plugin-extension-tab Implement pipeline-scoped plugin binding system * fix: i18n keys --------- Co-authored-by: Copilot <198982749+Copilot@users.noreply.github.com>
74 lines
3.0 KiB
Python
74 lines
3.0 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'])
|
|
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'])
|
|
async def _() -> str:
|
|
return self.success(data={'configs': await self.ap.pipeline_service.get_pipeline_metadata()})
|
|
|
|
@self.route('/<pipeline_uuid>', methods=['GET', 'PUT', 'DELETE'])
|
|
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'])
|
|
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')
|
|
|
|
plugins = await self.ap.plugin_connector.list_plugins()
|
|
|
|
return self.success(
|
|
data={
|
|
'bound_plugins': pipeline.get('extensions_preferences', {}).get('plugins', []),
|
|
'available_plugins': plugins,
|
|
}
|
|
)
|
|
elif quart.request.method == 'PUT':
|
|
# Update bound plugins for this pipeline
|
|
json_data = await quart.request.json
|
|
bound_plugins = json_data.get('bound_plugins', [])
|
|
|
|
await self.ap.pipeline_service.update_pipeline_extensions(pipeline_uuid, bound_plugins)
|
|
|
|
return self.success()
|