Merge pull request #3 from prassanna-ravishankar/feat/use-dotenv-for-api-key

Author: prassanna-ravishankar

.env.example

@@ -0,0 +1 @@
+GEMINI_API_KEY="your_api_key_here"

examples/01_simple_example.py

@@ -10,8 +10,19 @@
 
 from mcp import StdioServerParameters
 
+# Load environment variables from .env file
+try:
+    from dotenv import load_dotenv
+
+    load_dotenv()
+except ImportError:
+    print("dotenv package not found, skipping.")
+
 # Set up Gemini API key
-GEMINI_API_KEY = "AIzaSyDAdEToKdFt8SHs25ABz65bx6cedU_zreo"
+GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
+if not GEMINI_API_KEY:
+    raise ValueError("GEMINI_API_KEY environment variable not set")
+
 os.environ["GOOGLE_API_KEY"] = GEMINI_API_KEY
 
 from smolagents import CodeAgent, MCPClient

examples/02_openai_compatible_example.py

@@ -16,8 +16,18 @@
 
 from mcp import StdioServerParameters
 
+# Load environment variables from .env file
+try:
+    from dotenv import load_dotenv
+
+    load_dotenv()
+except ImportError:
+    print("dotenv package not found, skipping.")
+
 # Set up Gemini API key
-GEMINI_API_KEY = "AIzaSyDAdEToKdFt8SHs25ABz65bx6cedU_zreo"
+GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
+if not GEMINI_API_KEY:
+    raise ValueError("GEMINI_API_KEY environment variable not set")
 
 try:
     from rich.console import Console

examples/03_find_real_experiments.py

@@ -10,8 +10,18 @@
 
 from mcp import StdioServerParameters
 
+# Load environment variables from .env file
+try:
+    from dotenv import load_dotenv
+
+    load_dotenv()
+except ImportError:
+    print("dotenv package not found, skipping.")
+
 # Set up Gemini API key
-GEMINI_API_KEY = "AIzaSyDAdEToKdFt8SHs25ABz65bx6cedU_zreo"
+GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
+if not GEMINI_API_KEY:
+    raise ValueError("GEMINI_API_KEY environment variable not set")
 
 try:
     from rich.console import Console

examples/04_quick_scalar_demo.py

@@ -10,8 +10,18 @@
 
 from mcp import StdioServerParameters
 
+# Load environment variables from .env file
+try:
+    from dotenv import load_dotenv
+
+    load_dotenv()
+except ImportError:
+    print("dotenv package not found, skipping.")
+
 # Set up Gemini API key
-GEMINI_API_KEY = "AIzaSyDAdEToKdFt8SHs25ABz65bx6cedU_zreo"
+GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
+if not GEMINI_API_KEY:
+    raise ValueError("GEMINI_API_KEY environment variable not set")
 
 try:
     from rich.console import Console

examples/consolidated_debugger.py

@@ -15,6 +15,20 @@
 console = Console()
 
 
+# Load environment variables from .env file
+try:
+    from dotenv import load_dotenv
+
+    load_dotenv()
+except ImportError:
+    print("dotenv package not found, skipping.")
+
+# Set up Gemini API key
+GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
+if not GEMINI_API_KEY:
+    raise ValueError("GEMINI_API_KEY environment variable not set")
+
+
 class ClearMLDemo:
     """Demo class for ClearML MCP integration."""
 
@@ -23,7 +37,7 @@ def __init__(self):
         self.model = OpenAIServerModel(
             model_id="gemini-2.0-flash",
             api_base="https://generativelanguage.googleapis.com/v1beta/openai/",
-            api_key="AIzaSyDAdEToKdFt8SHs25ABz65bx6cedU_zreo",
+            api_key=GEMINI_API_KEY,
             temperature=0.1,
         )
         self.clearml_server_params = StdioServerParameters(

pyproject.toml

@@ -42,6 +42,7 @@ dev = [
 examples = [
     "smolagents[openai,mcp]>=1.20.0",
     "rich>=10.0.0",
+    "python-dotenv>=0.21.0",
 ]
 
 [project.scripts]