diff --git a/swarms/agents/utils/llm.py b/swarms/agents/utils/llm.py
index a9fdb265..46984130 100644
--- a/swarms/agents/utils/llm.py
+++ b/swarms/agents/utils/llm.py
@@ -1,6 +1,7 @@
 """OpenAI chat wrapper."""
 from __future__ import annotations
 
+import os
 import logging
 import sys
 from typing import Any, Callable, Dict, List, Mapping, Optional, Tuple