mlflow--mlflow
wehub-resource-sync
24 行
500 B
Python
24 行
500 B
Python
from mlflow.genai.agent_server.server import (
|
|
AgentServer,
|
|
get_invoke_function,
|
|
get_stream_function,
|
|
invoke,
|
|
stream,
|
|
)
|
|
from mlflow.genai.agent_server.utils import (
|
|
get_request_headers,
|
|
set_request_headers,
|
|
setup_mlflow_git_based_version_tracking,
|
|
)
|
|
|
|
__all__ = [
|
|
"set_request_headers",
|
|
"get_request_headers",
|
|
"AgentServer",
|
|
"invoke",
|
|
"stream",
|
|
"get_invoke_function",
|
|
"get_stream_function",
|
|
"setup_mlflow_git_based_version_tracking",
|
|
]
|