def get_tools()

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

• 23 lines of code
• 12 McCabe index (conditional complexity)

def get_tools(
        industry_type: str,
        orchestration_framework: str
    ) -> list:
    """Grabs a list of tools based on the user's configselection"""

    tools_list = []
    # if industry_type == IndustryType.FINANCE_INDUSTRY.value:
    #     tools_list.append(yahoo_finance_tool)
    if industry_type == IndustryType.HEALTHCARE_INDUSTRY.value:
        tools_list.append(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([
            google_search_tool,
            google_scholar_tool,
            google_trends_tool,
            google_finance_tool
        ])
    # The Vertex AI Search Tool is only used if the user specifies a DATA_STORE_ID
    if DATA_STORE_ID != "unset":
        tools_list.append(retrieve_info)

    tools_list.extend([
        # fallback,
        should_continue
    ])

    # If using langchain or langgraph, then the tools must be defined as structured tools
    if (orchestration_framework == OrchestrationFramework.LANGCHAIN_PREBUILT_AGENT.value or
        orchestration_framework == OrchestrationFramework.LANGGRAPH_PREBUILT_AGENT.value):
        tools_list = [StructuredTool.from_function(tool) for tool in tools_list]

    return tools_list