agent.py

# Copyright (C) 2024 Intel Corporation
# SPDX-License-Identifier: Apache-2.0


def instantiate_agent(args, strategy="react_langchain"):
    if strategy == "react_langchain":
        from .strategy.react import ReActAgentwithLangchain

        return ReActAgentwithLangchain(args)
    elif strategy == "react_langgraph":
        from .strategy.react import ReActAgentwithLanggraph

        return ReActAgentwithLanggraph(args)
    elif strategy == "plan_execute":
        from .strategy.planexec import PlanExecuteAgentWithLangGraph

        return PlanExecuteAgentWithLangGraph(args)
    elif strategy == "agentic_rag":
        from .strategy.agentic_rag import RAGAgentwithLanggraph

        return RAGAgentwithLanggraph(args)
    elif strategy == "docgrader":
        from .strategy.docgrader import RAGAgentDocGraderV1

        return RAGAgentDocGraderV1(args)
    else:
        raise ValueError(f"Agent strategy: {strategy} not supported!")