79 lines
1.6 KiB
Python
79 lines
1.6 KiB
Python
# SGLang public APIs
|
|
|
|
# Frontend Language APIs
|
|
from sglang.api import (
|
|
Engine,
|
|
Runtime,
|
|
assistant,
|
|
assistant_begin,
|
|
assistant_end,
|
|
flush_cache,
|
|
function,
|
|
gen,
|
|
gen_int,
|
|
gen_string,
|
|
get_server_info,
|
|
image,
|
|
select,
|
|
set_default_backend,
|
|
system,
|
|
system_begin,
|
|
system_end,
|
|
user,
|
|
user_begin,
|
|
user_end,
|
|
video,
|
|
)
|
|
from sglang.lang.backend.runtime_endpoint import RuntimeEndpoint
|
|
from sglang.lang.choices import (
|
|
greedy_token_selection,
|
|
token_length_normalized,
|
|
unconditional_likelihood_normalized,
|
|
)
|
|
from sglang.utils import LazyImport
|
|
|
|
ServerArgs = LazyImport("sglang.srt.server_args", "ServerArgs")
|
|
Anthropic = LazyImport("sglang.lang.backend.anthropic", "Anthropic")
|
|
LiteLLM = LazyImport("sglang.lang.backend.litellm", "LiteLLM")
|
|
OpenAI = LazyImport("sglang.lang.backend.openai", "OpenAI")
|
|
VertexAI = LazyImport("sglang.lang.backend.vertexai", "VertexAI")
|
|
|
|
# Other configs
|
|
from sglang.global_config import global_config
|
|
from sglang.version import __version__
|
|
|
|
__all__ = [
|
|
"Engine",
|
|
"Runtime",
|
|
"assistant",
|
|
"assistant_begin",
|
|
"assistant_end",
|
|
"flush_cache",
|
|
"function",
|
|
"gen",
|
|
"gen_int",
|
|
"gen_string",
|
|
"get_server_info",
|
|
"image",
|
|
"select",
|
|
"set_default_backend",
|
|
"system",
|
|
"system_begin",
|
|
"system_end",
|
|
"user",
|
|
"user_begin",
|
|
"user_end",
|
|
"video",
|
|
"RuntimeEndpoint",
|
|
"greedy_token_selection",
|
|
"token_length_normalized",
|
|
"unconditional_likelihood_normalized",
|
|
"ServerArgs",
|
|
"Anthropic",
|
|
"LiteLLM",
|
|
"OpenAI",
|
|
"VertexAI",
|
|
"global_config",
|
|
"__version__",
|
|
]
|