Update base_agent.py

base_agent.py

@@ -9,12 +9,17 @@ class BaseAgent:
     
     def __init__(self, model):
         """Initialize the base agent with the specified model."""
-        print("Initializing Base Agent with " + model)
+        print(f"Initializing Base Agent with {model}")
         if model is None:
             model = os.getenv("DEFAULT_LLM_MODEL")
-            
+            if model is None:
+                raise ValueError("No model specified and DEFAULT_LLM_MODEL not set")
+        
         openai_key = os.getenv("OPENAI_API_KEY")
-        print("OpenAI Key = " + openai_key)
+        if openai_key is None:
+            raise ValueError("OPENAI_API_KEY environment variable not set")
+        print(f"OpenAI Key = {openai_key}")
+        
         self.model = model
         print("Init Client")
         self.client = OpenAI(api_key=openai_key)