From 7fc9c28a94d177851c205d4958be88014d16eba6 Mon Sep 17 00:00:00 2001
From: luohy15 <luohycs@gmail.com>
Date: Fri, 26 Sep 2025 23:58:51 +0800
Subject: [PATCH] Add environment variable configuration support
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- Add .env.example file with API key placeholders
- Update README.md with .env file setup instructions
- Add dotenv loading in main.py for environment variables

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>
---
 .env.example | 2 ++
 README.md    | 6 ++++++
 main.py      | 5 +++++
 3 files changed, 13 insertions(+)
 create mode 100644 .env.example

diff --git a/.env.example b/.env.example
new file mode 100644
index 00000000..1e257c3c
--- /dev/null
+++ b/.env.example
@@ -0,0 +1,2 @@
+ALPHA_VANTAGE_API_KEY=alpha_vantage_api_key_placeholder
+OPENAI_API_KEY=openai_api_key_placeholder
\ No newline at end of file
diff --git a/README.md b/README.md
index ad293904..4557fc29 100644
--- a/README.md
+++ b/README.md
@@ -124,6 +124,12 @@ You will need the OpenAI API for all the agents.
 export OPENAI_API_KEY=$YOUR_OPENAI_API_KEY
 ```
 
+Alternatively, you can create a `.env` file in the project root with your API keys (see `.env.example` for reference):
+```bash
+cp .env.example .env
+# Edit .env with your actual API keys
+```
+
 ### CLI Usage
 
 You can also try out the CLI directly by running:
diff --git a/main.py b/main.py
index 8bf3d3c7..fe4e4cb3 100644
--- a/main.py
+++ b/main.py
@@ -1,6 +1,11 @@
 from tradingagents.graph.trading_graph import TradingAgentsGraph
 from tradingagents.default_config import DEFAULT_CONFIG
 
+from dotenv import load_dotenv
+
+# Load environment variables from .env file
+load_dotenv()
+
 # Create a custom config
 config = DEFAULT_CONFIG.copy()
 config["llm_provider"] = "google"  # Use a different model