basyx commited on
Commit
76dc4d0
·
verified ·
1 Parent(s): 6278319

Create core/builders/api_builder.py

Browse files
Files changed (1) hide show
  1. core/builders/api_builder.py +22 -0
core/builders/api_builder.py ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import APIRouter
2
+ from core.registry.loader import get_tasks
3
+ from core.execution.executor import execute_task
4
+
5
+ router = APIRouter()
6
+
7
+
8
+ def build_api_router():
9
+
10
+ for task in get_tasks():
11
+
12
+ async def endpoint(payload: dict, task=task):
13
+ return await execute_task(task.name, payload)
14
+
15
+ router.add_api_route(
16
+ f"/execute/{task.name}",
17
+ endpoint,
18
+ methods=["POST"],
19
+ name=f"execute_{task.name}"
20
+ )
21
+
22
+ return router