|
@@ -11,4 +11,8 @@ def get():
|
|
env = os.environ.get('AI_AGENT_ENV', 'dev')
|
|
env = os.environ.get('AI_AGENT_ENV', 'dev')
|
|
if env not in ('dev', 'pre', 'prod'):
|
|
if env not in ('dev', 'pre', 'prod'):
|
|
raise ValueError(f"Invalid environment: {env}. Expected one of ('dev', 'pre', 'prod').")
|
|
raise ValueError(f"Invalid environment: {env}. Expected one of ('dev', 'pre', 'prod').")
|
|
- return yaml.safe_load(open(f'{dirname}/{env}.yaml').read())
|
|
|
|
|
|
+ return yaml.safe_load(open(f'{dirname}/{env}.yaml').read())
|
|
|
|
+
|
|
|
|
+def get_env():
|
|
|
|
+ env = os.environ.get('AI_AGENT_ENV', 'dev')
|
|
|
|
+ return env
|