Update api.py
Browse files
api.py
CHANGED
|
@@ -34,6 +34,7 @@ from main_process.main_router import router as main_router
|
|
| 34 |
from main_process.salamandra_router import router as salamandra_router
|
| 35 |
from main_process.moe_router import router as moe_router
|
| 36 |
from main_process.refinement_router import router as refinement_router
|
|
|
|
| 37 |
|
| 38 |
app = FastAPI(title="Veureu Engine API", version="0.2.0")
|
| 39 |
app.add_middleware(
|
|
@@ -63,6 +64,7 @@ class JobStatus(str, Enum):
|
|
| 63 |
|
| 64 |
jobs: Dict[str, dict] = {}
|
| 65 |
|
|
|
|
| 66 |
app.include_router(main_router)
|
| 67 |
app.include_router(salamandra_router)
|
| 68 |
app.include_router(moe_router)
|
|
|
|
| 34 |
from main_process.salamandra_router import router as salamandra_router
|
| 35 |
from main_process.moe_router import router as moe_router
|
| 36 |
from main_process.refinement_router import router as refinement_router
|
| 37 |
+
from storage.data_routers import router as data_router
|
| 38 |
|
| 39 |
app = FastAPI(title="Veureu Engine API", version="0.2.0")
|
| 40 |
app.add_middleware(
|
|
|
|
| 64 |
|
| 65 |
jobs: Dict[str, dict] = {}
|
| 66 |
|
| 67 |
+
app.include_router(data_router)
|
| 68 |
app.include_router(main_router)
|
| 69 |
app.include_router(salamandra_router)
|
| 70 |
app.include_router(moe_router)
|