mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2026-04-06 07:03:37 +00:00
API: Move OpenAI-compatible API from extensions/openai to modules/api
This commit is contained in:
parent
2e4232e02b
commit
bf6fbc019d
23 changed files with 51 additions and 65 deletions
15
server.py
15
server.py
|
|
@ -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()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue