xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-02 12:05:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(pipeline): api for updating pipeline

Browse Source
This commit is contained in:
Junyan Qin
2025-03-27 00:47:54 +08:00
parent d8c730341a
commit 6c1ee922de
2 changed files with 16 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pkg/api/http/controller/groups/pipelines.py
View File

@@ -30,7 +30,7 @@ class PipelinesRouterGroup(group.RouterGroup):
'configs': await self.ap.pipeline_service.get_pipeline_metadata()
})
@self.route('/<pipeline_uuid>', methods=['GET', 'DELETE'])
@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)
@@ -41,6 +41,12 @@ class PipelinesRouterGroup(group.RouterGroup):
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)

9
pkg/api/http/service/pipeline.py
View File

@@ -58,6 +58,15 @@ class PipelineService:
)
# TODO: 更新到pipeline manager
async def update_pipeline(self, pipeline_uuid: str, pipeline_data: dict) -> None:
del pipeline_data['uuid']
del pipeline_data['for_version']
del pipeline_data['stages']
await self.ap.persistence_mgr.execute_async(
sqlalchemy.update(persistence_pipeline.LegacyPipeline).where(persistence_pipeline.LegacyPipeline.uuid == pipeline_uuid).values(**pipeline_data)
)
# TODO: 更新到pipeline manager
async def delete_pipeline(self, pipeline_uuid: str) -> None:
await self.ap.persistence_mgr.execute_async(
sqlalchemy.delete(persistence_pipeline.LegacyPipeline).where(persistence_pipeline.LegacyPipeline.uuid == pipeline_uuid)