API: Move OpenAI-compatible API from extensions/openai to modules/api

This commit is contained in:
oobabooga 2026-03-20 14:46:00 -03:00
parent 2e4232e02b
commit bf6fbc019d
23 changed files with 51 additions and 65 deletions

View file

@ -106,6 +106,11 @@ def create_interface():
if shared.args.extensions is not None and len(shared.args.extensions) > 0:
extensions_module.load_extensions()
# Start the API server if enabled
if shared.args.api or shared.args.public_api:
from modules.api.script import setup as api_setup
api_setup()
# Force some events to be triggered on page load
shared.persistent_interface_state.update({
'mode': shared.settings['mode'],
@ -273,6 +278,12 @@ if __name__ == "__main__":
# Activate the extensions listed on settings.yaml
extensions_module.available_extensions = utils.get_available_extensions()
for extension in shared.settings['default_extensions']:
# The openai extension was moved to modules/api and is now
# activated with --api. Treat it as an alias for backwards compat.
if extension == 'openai':
shared.args.api = True
continue
shared.args.extensions = shared.args.extensions or []
if extension not in shared.args.extensions:
shared.args.extensions.append(extension)
@ -337,6 +348,10 @@ if __name__ == "__main__":
shared.args.extensions = [x for x in (shared.args.extensions or []) if x != 'gallery']
if shared.args.extensions:
extensions_module.load_extensions()
if shared.args.api or shared.args.public_api:
from modules.api.script import setup as api_setup
api_setup()
else:
# Launch the web UI
create_interface()