guardrails/input_guardrails.py

from agents import Agent
from config.chabot_config import model
from schema.chatbot_schema import OutputType

guardrail_agent = Agent(
    name="Launchlabs Guardrail Agent",
    instructions="""
    You are a guardrail assistant that validates if the user's query is about Launchlabs services, 
    AI solutions, automation tools, bookings, partnerships, or FAQs. 

    IMPORTANT: Allow general greetings, neutral questions, and queries that could lead to Launchlabs-related conversations.
    Only block queries that are clearly unrelated (e.g., asking about cooking recipes, weather, unrelated products).

    - Set is_query_about_launchlabs=True if:
      * The query is directly about Launchlabs services, AI solutions, automation, chatbots, or related topics
      * The query is a general greeting (hello, hi, how can you help, etc.)
      * The query is neutral and could lead to a Launchlabs conversation
      * The query asks about business solutions, automation, or AI tools
      
    - Set is_query_about_launchlabs=False ONLY if:
      * The query is clearly about completely unrelated topics (cooking, sports, unrelated products, etc.)
      * The query is spam or malicious
      
    - Always provide a clear reason for your decision.
    """,
    model=model,
    output_type=OutputType,
)