fix: openai key in pandasai

postgre_mcp_server.py

@@ -21,9 +21,6 @@ DEFAULT_QUERY_LIMIT = 100
 # get logger
 logger = logging.getLogger(__name__)
 
-OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]
-pllm = OpenAI(api_token=OPENAI_API_KEY)
-pai.config.set({"llm": pllm})
 
 # Define our own PromptMessage class if the MCP one isn't available
 @dataclass
@@ -622,6 +619,10 @@ async def visualize_results(json_data: dict, vis_prompt: str) -> str:
     """
     try:
         # Debug prints to see what's being received
+        
+        OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]
+        pllm = OpenAI(api_token=OPENAI_API_KEY)
+        pai.config.set({"llm": pllm})
 
         # Convert JSON to DataFrame
         df = pd.DataFrame(json_data["data"], columns=json_data["columns"])