|
15 | 15 | router = APIRouter(tags=["system/assistant"], prefix="/system/assistant") |
16 | 16 |
|
17 | 17 | @router.get("/info/{id}") |
18 | | -async def info(request: Request, response: Response, session: SessionDep, id: int) -> dict: |
| 18 | +async def info(request: Request, response: Response, session: SessionDep, id: int) -> AssistantModel: |
| 19 | + if not id: |
| 20 | + raise Exception('miss assistant id') |
19 | 21 | db_model = await get_assistant_info(session=session, assistant_id=id) |
20 | 22 | if not db_model: |
21 | 23 | raise RuntimeError(f"assistant application not exist") |
22 | 24 | db_model = AssistantModel.model_validate(db_model) |
23 | 25 | response.headers["Access-Control-Allow-Origin"] = db_model.domain |
24 | 26 | origin = request.headers.get("origin") or request.headers.get("referer") |
25 | 27 | origin = origin.rstrip('/') |
26 | | - """ if origin != db_model.domain: |
27 | | - raise RuntimeError("invalid domain [{origin}]") """ |
28 | | - return db_model.model_dump() |
| 28 | + if origin != db_model.domain: |
| 29 | + raise RuntimeError("invalid domain [{origin}]") |
| 30 | + return db_model |
29 | 31 |
|
30 | 32 | @router.get("/validator", response_model=AssistantValidator) |
31 | | -async def info(session: SessionDep, id: str, virtual: Optional[int] = Query(None), online: Optional[bool] = Query(default=False)): |
| 33 | +async def validator(session: SessionDep, id: int, virtual: Optional[int] = Query(None)): |
| 34 | + if not id: |
| 35 | + raise Exception('miss assistant id') |
| 36 | + |
32 | 37 | db_model = await get_assistant_info(session=session, assistant_id=id) |
33 | 38 | if not db_model: |
34 | 39 | return AssistantValidator() |
|
0 commit comments