Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

add new route for app config

Browse Source
This commit is contained in:
maryhipp 2023-07-12 08:54:42 -07:00 committed by psychedelicious
parent 8fc0ce7e38
commit fba25792f9
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
invokeai/app/api/routers/app_info.py
View File

@ -1,5 +1,6 @@
from fastapi.routing import APIRouter from fastapi.routing import APIRouter
from pydantic import BaseModel from pydantic import BaseModel
from invokeai.backend.image_util.patchmatch import PatchMatch
from invokeai.version import __version__ from invokeai.version import __version__
@ -10,9 +11,19 @@ class AppVersion(BaseModel):
"""App Version Response""" """App Version Response"""
version: str version: str
class AppConfig(BaseModel):
"""App Config Response"""
patchmatch_enabled: bool
@app_router.get('/version', operation_id="app_version", @app_router.get('/version', operation_id="app_version",
status_code=200, status_code=200,
response_model=AppVersion) response_model=AppVersion)
async def get_version() -> AppVersion: async def get_version() -> AppVersion:
return AppVersion(version=__version__) return AppVersion(version=__version__)
@app_router.get('/config', operation_id="get_config",
status_code=200,
response_model=AppConfig)
async def get_config() -> AppConfig:
return AppConfig(patchmatch_enabled=PatchMatch.patchmatch_available())