Load .env vars on project run instead of in project's main.py. Add python-dotenv to dependencies. Verify with tests.

Author: tcdent

agentstack/cli/cli.py

@@ -13,6 +13,7 @@
 import os
 import importlib.resources
 from cookiecutter.main import cookiecutter
+from dotenv import load_dotenv
 
 from .agentstack_data import (
     FrameworkData,
@@ -184,6 +185,7 @@ def run_project(framework: str, path: str = ''):
         print(e)
         sys.exit(1)
 
+    load_dotenv(_path / '.env')  # explicitly load the project's .env file
     entrypoint = _path / frameworks.get_entrypoint_path(framework)
     os.system(f'python {entrypoint}')
 

agentstack/templates/crewai/{{cookiecutter.project_metadata.project_slug}}/src/main.py

@@ -2,8 +2,6 @@
 import sys
 from crew import {{cookiecutter.project_metadata.project_name|replace('-', '')|replace('_', '')|capitalize}}Crew
 import agentops
-from dotenv import load_dotenv
-load_dotenv()
 
 agentops.init(default_tags=['crewai', 'agentstack'])
 

pyproject.toml

@@ -28,6 +28,7 @@ dependencies = [
     "packaging==23.2",
     "requests>=2.32",
     "appdirs>=1.4.4",
+    "python-dotenv>=1.0.1",
 ]
 
 [project.optional-dependencies]

tests/test_project_run.py

@@ -0,0 +1,46 @@
+import os
+from pathlib import Path
+import shutil
+import unittest
+from parameterized import parameterized_class
+
+from agentstack import frameworks
+from agentstack.cli import run_project
+from agentstack.generation.files import ConfigFile
+
+BASE_PATH = Path(__file__).parent
+
+
+@parameterized_class([{"framework": framework} for framework in frameworks.SUPPORTED_FRAMEWORKS])
+class ProjectRunTest(unittest.TestCase):
+    def setUp(self):
+        self.project_dir = BASE_PATH / 'tmp' / self.framework
+
+        os.makedirs(self.project_dir)
+        os.makedirs(self.project_dir / 'src')
+        (self.project_dir / 'src' / '__init__.py').touch()
+
+        # set the framework in agentstack.json
+        shutil.copy(BASE_PATH / 'fixtures' / 'agentstack.json', self.project_dir / 'agentstack.json')
+        with ConfigFile(self.project_dir) as config:
+            config.framework = self.framework
+
+        # populate the entrypoint
+        entrypoint_path = frameworks.get_entrypoint_path(self.framework, self.project_dir)
+        shutil.copy(BASE_PATH / f"fixtures/frameworks/{self.framework}/entrypoint_max.py", entrypoint_path)
+
+        # write a basic .env file
+        shutil.copy(BASE_PATH / 'fixtures' / '.env', self.project_dir / '.env')
+
+    def tearDown(self):
+        shutil.rmtree(self.project_dir)
+
+    def test_run_project(self):
+        run_project(self.framework, self.project_dir)
+
+    def test_env_is_set(self):
+        """
+        After running a project, the environment variables should be set from project_dir/.env.
+        """
+        run_project(self.framework, self.project_dir)
+        assert os.getenv('ENV_VAR1') == 'value1'