text-generation-webui/extensions/api/script.py

import extensions.api.blocking_api as blocking_api
import extensions.api.streaming_api as streaming_api
from modules import shared

def setup():
    blocking_api.start_server(shared.args.api_blocking_port, share=shared.args.public_api)
    streaming_api.start_server(shared.args.api_streaming_port, share=shared.args.public_api)
New universal API with streaming/blocking endpoints (#990) Previous title: Add api_streaming extension and update api-example-stream to use it * Merge with latest main * Add parameter capturing encoder_repetition_penalty * Change some defaults, minor fixes * Add --api, --public-api flags * remove unneeded/broken comment from blocking API startup. The comment is already correctly emitted in try_start_cloudflared by calling the lambda we pass in. * Update on_start message for blocking_api, it should say 'non-streaming' and not 'streaming' * Update the API examples * Change a comment * Update README * Remove the gradio API * Remove unused import * Minor change * Remove unused import --------- Co-authored-by: oobabooga <112222186+oobabooga@users.noreply.github.com> 2023-04-23 20:52:43 +02:00			`import extensions.api.blocking_api as blocking_api`
			`import extensions.api.streaming_api as streaming_api`
KoboldAI api 2023-03-15 21:52:46 +01:00			`from modules import shared`

Add setup method to extensions 2023-03-19 14:22:24 +01:00			`def setup():`
add api port options (#1990) 2023-05-16 01:44:16 +02:00			`blocking_api.start_server(shared.args.api_blocking_port, share=shared.args.public_api)`
			`streaming_api.start_server(shared.args.api_streaming_port, share=shared.args.public_api)`