Added load_dotenv to genesis cli command

VRSEN · Mar 27, 2024 · 352b629 · 352b629
1 parent 5f46c4e
commit 352b629
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/agency_swarm/cli.py b/agency_swarm/cli.py
@@ -1,6 +1,8 @@
 import argparse
 import os
 
+from dotenv import load_dotenv
+
 
 def main():
     parser = argparse.ArgumentParser(description='Create agent template.')
@@ -28,6 +30,7 @@ def main():
         from agency_swarm.util import create_agent_template
         create_agent_template(args.name, args.description, args.path, args.use_txt)
     elif args.command == "genesis":
+        load_dotenv()
         if not os.getenv('OPENAI_API_KEY') and not args.openai_key:
             print("OpenAI API key not set. "
                   "Please set it with --openai_key argument or by setting OPENAI_API_KEY environment variable.")