load_env in utils

playground/agents/simple_agent_example.py

@@ -1,6 +1,5 @@
 import os
 
-from dotenv import load_dotenv
 
 from swarms import (
     OpenAIChat,
@@ -11,16 +10,15 @@ from swarms import (
 
 from swarms.tools.code_executor import CodeExecutor
 
+from swarms.utils import load_environment
+
 conv = Conversation(
     autosave=False,
     time_enabled=True,
 )
 
 # Load the environment variables
-load_dotenv()
+api_key = load_environment()
-
-# Get the API key from the environment
-api_key = os.environ.get("OPENAI_API_KEY")
 
 # Initialize the language model
 llm = OpenAIChat(openai_api_key=api_key)

swarms/utils/__init__.py

@@ -44,6 +44,8 @@ from swarms.utils.supervision_masking import (
 )
 from swarms.utils.supervision_visualizer import MarkVisualizer
 
+from swarms.utils import load_environment
+
 __all__ = [
     "SubprocessCodeInterpreter",
     "display_markdown_message",
@@ -77,4 +79,5 @@ __all__ = [
     "masks_to_marks",
     "refine_marks",
     "MarkVisualizer",
+    "load_environment",
 ]

swarms/utils/load_environment.py

@@ -0,0 +1,11 @@
+from dotenv import load_dotenv
+import os
+
+# Load the environment variables
+def load_environment():
+    load_dotenv()
+    # Get the API key from the environment
+    api_key = os.environ.get("OPENAI_API_KEY")
+
+    return api_key, os.environ
+