validator.py

import os
from dotenv import load_dotenv
from langchain_openai import ChatOpenAI

load_dotenv()
openai_api_key = os.getenv("OPENAI_API_KEY")

def validate_query(query: str) -> bool:
    prompt = (
        f"Can the following task be performed using a web browser, such as searching, reading, or interacting with websites? "
        f"Respond with only YES or NO.\n\nTask: \"{query}\""
    )
    try:
        llm = ChatOpenAI(model_name="gpt-3.5-turbo", openai_api_key=openai_api_key)
        response = llm.invoke(prompt)
        answer = response.content.strip().lower()
        print(f"Validation response: {answer}")
        return answer.startswith("yes")
    except Exception as e:
        print("Validation error:", e)
        return False