Clean up instructions

Author: anishathalye

src/airline_agent/backend/services/airline_chat.py

@@ -47,7 +47,7 @@
     get_tools_in_openai_format,
     run_cleanlab_validation_logging_tools,
 )
-from airline_agent.constants import AGENT_BASE_INSTRUCTIONS, AGENT_MODEL, DEMO_DATETIME
+from airline_agent.constants import AGENT_INSTRUCTIONS, AGENT_MODEL, DEMO_DATETIME
 from airline_agent.tools.booking import BookingTools
 from airline_agent.tools.knowledge_base import KnowledgeBase
 
@@ -57,25 +57,13 @@
 logger.setLevel(logging.INFO)
 
 
-def instructions() -> str:
-    current_date_str = DEMO_DATETIME.date().isoformat()
-    current_datetime_str = DEMO_DATETIME.strftime("%Y-%m-%d %H:%M:%S %Z")
-    return dedent(f"""
-        {AGENT_BASE_INSTRUCTIONS}
-
-        ## Context:
-        - Today's date: {current_date_str}
-        - Current time: {current_datetime_str}
-    """).strip()
-
-
 def create_agent(kb: KnowledgeBase, booking: BookingTools) -> Agent:
     """Create the airline support agent."""
     model = OpenAIChatModel(model_name=AGENT_MODEL, settings=ModelSettings(temperature=0.0))
 
     return Agent(
         model=model,
-        instructions=instructions,
+        instructions=AGENT_INSTRUCTIONS,
         tools=[
             kb.get_article,
             kb.search,

src/airline_agent/constants.py

@@ -17,7 +17,7 @@
 RAG_CHUNK_OVERLAP = 200
 CONTEXT_RETRIEVAL_TOOLS = ["search", "get_article", "list_directory"]
 AGENT_MODEL = "gpt-4o"
-AGENT_BASE_INSTRUCTIONS = """
+AGENT_INSTRUCTIONS = f"""
 You are an AI customer support agent for Frontier Airlines. You can use tools to access to a knowledge base of articles and documents about the airline's services, policies, and procedures.
 
 ## You have access to the following tools:
@@ -48,6 +48,10 @@
 - When a booking is successfully created, provide the booking ID and confirmation details clearly.
 - If you book flights, provide the booking ID and summarize the flights booked and total price.
 - If the user asks about anything unrelated to the airline, politely inform them that you can only assist with airline-related inquiries.
+
+## Context:
+- Today's date: {DEMO_DATETIME.date().isoformat()}
+- Current time: {DEMO_DATETIME.strftime("%H:%M:%S %Z")}
 """.strip()
 
 FALLBACK_RESPONSE = "I'm sorry, but I don't have the information you're looking for. Please rephrase the question or contact Frontier Airlines customer support for further assistance."