refactored

assafelovic · May 5, 2024 · 5f9059f · 5f9059f
1 parent 342279c
commit 5f9059f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/multi_agents/main.py b/multi_agents/main.py
@@ -4,15 +4,16 @@
 import json
 import os
 
+# Run with LangSmith if API key is set
 if os.environ.get("LANGCHAIN_API_KEY"):
     os.environ["LANGCHAIN_TRACING_V2"] = "true"
 load_dotenv()
 
-with open('task.json', 'r') as f:
-    task = json.load(f)
-
 
 async def main():
+    with open('task.json', 'r') as f:
+        task = json.load(f)
+
     master_agent = MasterAgent(task)
     research_report = await master_agent.run()
     print(research_report)