Katya Beresneva commited on
Commit
94cd5f4
·
1 Parent(s): 523e34e
Files changed (1) hide show
  1. config.py +28 -0
config.py ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import dotenv
3
+ from typing import Optional
4
+
5
+ # Load environment variables from .env file if it exists
6
+ dotenv.load_dotenv()
7
+
8
+ def get_required_env(key: str, error_message: Optional[str] = None) -> str:
9
+ """Get a required environment variable or raise a helpful error."""
10
+ value = os.getenv(key)
11
+ if not value:
12
+ default_message = (
13
+ f"{key} environment variable is not set.\n"
14
+ f"Please create a .env file in the project root with:\n"
15
+ f"{key}=your_{key.lower()}_here\n\n"
16
+ f"If you don't have an API key:\n"
17
+ f"1. Go to https://makersuite.google.com/app/apikey\n"
18
+ f"2. Create a new API key\n"
19
+ f"3. Add it to your .env file"
20
+ )
21
+ raise ValueError(error_message or default_message)
22
+ return value
23
+
24
+ # Required environment variables
25
+ GOOGLE_API_KEY = get_required_env("GOOGLE_API_KEY")
26
+
27
+ # Optional environment variables with defaults
28
+ AGENT_MODEL_NAME = os.getenv("AGENT_MODEL_NAME", "gemini-2.0-flash")