def get_llamaindex_tools()

in Runtime_env/app/orchestration/tools.py [0:0]

• 14 lines of code
• 7 McCabe index (conditional complexity)

def get_llamaindex_tools(industry_type: str = None) -> list:
    """Grabs a list of tools based on the user's configselection"""

    tools_list = []
    # if industry_type == IndustryType.FINANCE_INDUSTRY.value:
    #     tool_spec = YahooFinanceToolSpec()
    #     tools_list.extend(tool_spec.to_tool_list())
    if industry_type == IndustryType.HEALTHCARE_INDUSTRY.value:
        tools_list.append(FunctionTool.from_defaults(fn=medical_publications_tool))
    # elif industry_type == IndustryType.RETAIL_INDUSTRY.value:
    #     tools_list.append(retail_discovery_tool)

    # These tools are only used if the user specifies a SERPER_API_KEY
    if SERP_API_KEY != "unset":
        tools_list.extend([
            FunctionTool.from_defaults(fn=google_search_tool),
            FunctionTool.from_defaults(fn=google_scholar_tool),
            FunctionTool.from_defaults(fn=google_trends_tool),
            FunctionTool.from_defaults(fn=google_finance_tool)
        ])

    if DATA_STORE_ID != "unset":
        tools_list.append(FunctionTool.from_defaults(fn=llamaindex_query_engine_tool))

    return tools_list