Update agent.py

agent.py

@@ -5,6 +5,16 @@ from typing import Optional
 
 from agents import Agent, Runner, SQLiteSession
 from get_kb import fetch_kb, KB_PATH  # our tool
+import os
+from dotenv import load_dotenv
+load_dotenv()
+
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+
+if not OPENAI_API_KEY:
+    raise ValueError("Missing OPENAI_API_KEY ENV variable. Please export it before running.")
+
+os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
 
 # System-level instructions (short, precise)
 SYSTEM_PROMPT = """