Update base_agent.py

base_agent.py

@@ -2,7 +2,7 @@ from typing import Dict, Any, Optional
 import os
 import time
 import json
-import openai
+from openai import OpenAI
 
 class BaseAgent:
     """Base class for all agents with common functionality."""
@@ -16,7 +16,8 @@ class BaseAgent:
         openai_key = os.getenv("OPENAI_API_KEY")
         print("OpenAI Key = " + openai_key)
         self.model = model
-        self.client = openai.OpenAI(api_key=openai_key)
+        print("Init Client")
+        self.client = OpenAI(api_key=openai_key)
         
     def __call__(self, question: str, task_id: Optional[str] = None, **kwargs) -> str:
         """Process the question and return an answer."""