reggpt/tools/qa_agent.py

"""
 /*************************************************************************
 *
 * CONFIDENTIAL
 * __________________
 *
 *  Copyright (2024-2025) AI Labs, IronOne Technologies, LLC
 *  All Rights Reserved
 *
 *  Author  : Theekshana Samaradiwakara
 *  Description : This file contains the server application routing logic. 
 *  CreatedDate : 17/10/2024
 *  LastModifiedDate : 
 *************************************************************************/
"""

from langchain.agents import Tool
from reggpt.routers.qa import run_qa_chain_answer_only

qa_agent_tools = [
    Tool(
        name="qa_chain",
        func=run_qa_chain_answer_only,
        description="Useful for when you need to answer questions about Rules and regulations made by central bank. Input should be a fully formed question.",
    ),
]