from loggers.logger import logger
from helper.web_search_agent import WEB_SEARCH_TOOL

import sys
sys.dont_write_bytecode = True

tool = WEB_SEARCH_TOOL

def web_search(state):
    """
    Route question to web search
    Args:
        state (dict): The current graph state
    Returns:
        state (dict): Next node to call
    """
    
    logger.info("INSIDE THE WEB SEARCH FILE")

    query = state['search_query']

    logger.info(f"SEARCH QUERY ENTERED: {query}")

    result = WEB_SEARCH_TOOL.invoke(query)

    logger.info("WEB SEARCH TOOL INVOKED")

    return {
        "context": result
    }