File size: 540 Bytes
cb505ff | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | from __future__ import annotations
from src.core.model_manager import model_manager
def set_model_provider(provider_name: str) -> None:
"""Set the provider used by newly created Manager and Worker runtimes."""
normalized = provider_name.strip().lower()
if normalized:
model_manager.select_model(normalized, None)
def set_model_selection(provider_name: str, model_name: str = None) -> None:
"""Set the provider and model used by newly created agents."""
model_manager.select_model(provider_name, model_name)
|