diff --git a/facefusion/apis/endpoints/root.py b/facefusion/apis/endpoints/root.py new file mode 100644 index 00000000..f4a43c70 --- /dev/null +++ b/facefusion/apis/endpoints/root.py @@ -0,0 +1,14 @@ +from starlette.requests import Request +from starlette.responses import JSONResponse +from starlette.status import HTTP_200_OK + +from facefusion import metadata + + +async def get_root(request : Request) -> JSONResponse: + root =\ + { + 'name': metadata.get('name'), + 'version': metadata.get('version') + } + return JSONResponse(root, status_code = HTTP_200_OK) diff --git a/tests/test_api_root.py b/tests/test_api_root.py index 843cb490..79b57b05 100644 --- a/tests/test_api_root.py +++ b/tests/test_api_root.py @@ -19,10 +19,9 @@ def test_client() -> Iterator[TestClient]: def test_get_root(test_client : TestClient) -> None: - root_response = test_client.get('/') - root_body = root_response.json() + get_root_response = test_client.get('/') + get_root_body = get_root_response.json() - assert root_body.get('name') == metadata.get('name') - assert root_body.get('description') == metadata.get('description') - assert root_body.get('version') == metadata.get('version') - assert root_response.status_code == 200 + assert get_root_body.get('name') == metadata.get('name') + assert get_root_body.get('version') == metadata.get('version') + assert get_root_response.status_code == 200