from common.utility.autogen_model_factory import AutoGenModelFactory

def get_model_client(provider:str = "google"):
    if provider.lower() == "google":
        return AutoGenModelFactory.get_model(
            provider="google",
            model_name="gemini-3-pro-preview",
            temperature=0,
        model_info={
            "family": "gemini",
            "vision": True,
            "function_calling": True,
            "json_output": True,
            "structured_output": True,
        },
    )
    elif provider.lower() == "openai":
        return AutoGenModelFactory.get_model(
            provider="openai",
            model_name="gpt-4o-mini",
            temperature=0,
        model_info={
            "family": "gpt",
            "vision": True,
            "function_calling": True,
            "json_output": True,
            "structured_output": True,
        },
    )
    else: 
        raise  ValueError(f"Unsupported provider: {provider}")