agentflow/engine/factory.py

from typing import Any

from agentflow.engine.auto_fast import AutoFastLLM


def create_llm_engine(model_string: str = "auto", use_cache: bool = False, is_multimodal: bool = False, **kwargs: Any) -> AutoFastLLM:
    """
    Drop-in replacement for AgentFlow's engine factory.

    The upstream factory chooses among OpenAI, DashScope, Gemini, vLLM, etc.
    This HF Space build intentionally routes every AgentFlow LLM call through
    AutoFastLLM, while preserving the same create_llm_engine(...) call site.
    """
    return AutoFastLLM(model_string=model_string or "auto", **kwargs)