wehub-resource-sync
97 lines
2.3 KiB
Python
97 lines
2.3 KiB
Python
from mlflow.genai import (
|
|
datasets,
|
|
judges,
|
|
scorers,
|
|
)
|
|
from mlflow.genai.agent_tester import test_agent
|
|
from mlflow.genai.datasets import (
|
|
create_dataset,
|
|
delete_dataset,
|
|
delete_dataset_tag,
|
|
get_dataset,
|
|
search_datasets,
|
|
set_dataset_tags,
|
|
)
|
|
from mlflow.genai.evaluation import evaluate, to_predict_fn
|
|
from mlflow.genai.git_versioning import disable_git_model_versioning, enable_git_model_versioning
|
|
from mlflow.genai.judges import make_judge
|
|
from mlflow.genai.labeling import (
|
|
Agent,
|
|
LabelingSession,
|
|
ReviewApp,
|
|
create_labeling_session,
|
|
delete_labeling_session,
|
|
get_labeling_session,
|
|
get_labeling_sessions,
|
|
get_review_app,
|
|
)
|
|
from mlflow.genai.optimize import optimize_prompt, optimize_prompts
|
|
from mlflow.genai.prompts import (
|
|
delete_prompt_alias,
|
|
delete_prompt_model_config,
|
|
delete_prompt_tag,
|
|
delete_prompt_version_tag,
|
|
get_prompt_tags,
|
|
load_prompt,
|
|
register_prompt,
|
|
search_prompts,
|
|
set_prompt_alias,
|
|
set_prompt_model_config,
|
|
set_prompt_tag,
|
|
set_prompt_version_tag,
|
|
)
|
|
from mlflow.genai.scheduled_scorers import (
|
|
ScorerScheduleConfig,
|
|
)
|
|
from mlflow.genai.scorers import Scorer, delete_scorer, get_scorer, list_scorers, scorer
|
|
from mlflow.genai.simulators import ConversationSimulator
|
|
|
|
__all__ = [
|
|
"datasets",
|
|
"test_agent",
|
|
"evaluate",
|
|
"to_predict_fn",
|
|
"Scorer",
|
|
"scorer",
|
|
"get_scorer",
|
|
"list_scorers",
|
|
"delete_scorer",
|
|
"judges",
|
|
"make_judge",
|
|
"scorers",
|
|
"create_dataset",
|
|
"delete_dataset",
|
|
"delete_dataset_tag",
|
|
"get_dataset",
|
|
"search_datasets",
|
|
"set_dataset_tags",
|
|
"load_prompt",
|
|
"register_prompt",
|
|
"search_prompts",
|
|
"delete_prompt_alias",
|
|
"set_prompt_alias",
|
|
"optimize_prompts",
|
|
"optimize_prompt",
|
|
"get_prompt_tags",
|
|
"set_prompt_tag",
|
|
"set_prompt_version_tag",
|
|
"delete_prompt_tag",
|
|
"delete_prompt_version_tag",
|
|
"set_prompt_model_config",
|
|
"delete_prompt_model_config",
|
|
"ScorerScheduleConfig",
|
|
"Agent",
|
|
"LabelingSession",
|
|
"ReviewApp",
|
|
"get_review_app",
|
|
"create_labeling_session",
|
|
"get_labeling_sessions",
|
|
"get_labeling_session",
|
|
"delete_labeling_session",
|
|
# git model versioning
|
|
"disable_git_model_versioning",
|
|
"enable_git_model_versioning",
|
|
# conversation simulation
|
|
"ConversationSimulator",
|
|
]
|