From 177ffd2f3f66d57408db7988c1af7fca838e462f Mon Sep 17 00:00:00 2001
From: evelynmitchell <efmphone@gmail.com>
Date: Fri, 16 Feb 2024 17:56:50 -0700
Subject: [PATCH] load_env in  utils

---
 playground/agents/simple_agent_example.py |  8 +++-----
 swarms/utils/__init__.py                  |  3 +++
 swarms/utils/load_environment.py          | 11 +++++++++++
 3 files changed, 17 insertions(+), 5 deletions(-)
 create mode 100644 swarms/utils/load_environment.py

diff --git a/playground/agents/simple_agent_example.py b/playground/agents/simple_agent_example.py
index 5d9d57ed..febacfcd 100644
--- a/playground/agents/simple_agent_example.py
+++ b/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()
-
-# Get the API key from the environment
-api_key = os.environ.get("OPENAI_API_KEY")
+api_key = load_environment()
 
 # Initialize the language model
 llm = OpenAIChat(openai_api_key=api_key)
diff --git a/swarms/utils/__init__.py b/swarms/utils/__init__.py
index 220bc3e6..cd660000 100644
--- a/swarms/utils/__init__.py
+++ b/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",
 ]
diff --git a/swarms/utils/load_environment.py b/swarms/utils/load_environment.py
new file mode 100644
index 00000000..7b318193
--- /dev/null
+++ b/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
+