apply black formatting to rearrange example

examples/swarms/rearrange/example.py

@@ -1,83 +1,91 @@
-import os
 import asyncio
+import os
+
 from dotenv import load_dotenv
+
 load_dotenv()
 
-from swarms.structs import Agent
 from swarms.models import Anthropic
+from swarms.structs import Agent
 from swarms.structs.rearrange import AgentRearrange
 
-llm = Anthropic(anthropic_api_key=os.getenv("ANTHROPIC_API_KEY"), streaming=True)
+llm = Anthropic(
+    anthropic_api_key=os.getenv("ANTHROPIC_API_KEY"), streaming=True
+)
+
 
 async def sequential():
 
-  agent1 = Agent(
+    agent1 = Agent(
-      agent_name="Blog generator",
+        agent_name="Blog generator",
-      system_prompt="Generate a blog post like Stephen King",
+        system_prompt="Generate a blog post like Stephen King",
-      llm=llm,
+        llm=llm,
-      dashboard=False,
+        dashboard=False,
-      streaming_on=True
+        streaming_on=True,
-  )
+    )
-
+
-  agent2 = Agent(
+    agent2 = Agent(
-      agent_name="Summarizer",
+        agent_name="Summarizer",
-      system_prompt="Summarize the blog post",
+        system_prompt="Summarize the blog post",
-      llm=llm,
+        llm=llm,
-      dashboard=False,
+        dashboard=False,
-      streaming_on=True
+        streaming_on=True,
-  )
+    )
-
+
-  flow = f"{agent1.name} -> {agent2.name}"
+    flow = f"{agent1.name} -> {agent2.name}"
-
+
-  agent_rearrange = AgentRearrange(
+    agent_rearrange = AgentRearrange(
-    [agent1, agent2], flow, verbose=False, logging=False
+        [agent1, agent2], flow, verbose=False, logging=False
-  )
+    )
-
+
-  result = await agent_rearrange.astream(
+    result = await agent_rearrange.astream(
-     "Generate a short blog post about Muhammad Ali."
+        "Generate a short blog post about Muhammad Ali."
-  )
+    )
-
+
-  # LEAVING THIS CALL BELOW FOR COMPARISON with "completion-style" .run() approach ;)
+    # LEAVING THIS CALL BELOW FOR COMPARISON with "completion-style" .run() approach ;)
-  # await agent_rearrange.run(
+    # await agent_rearrange.run(
-  #   "Generate a short blog post about Muhammad Ali."
+    #   "Generate a short blog post about Muhammad Ali."
-  # )
+    # )
+
 
 async def parallel():
 
-  writer1 = Agent(
+    writer1 = Agent(
-      agent_name="Writer 1",
+        agent_name="Writer 1",
-      system_prompt="Generate a blog post in the style of J.K. Rowling",
+        system_prompt="Generate a blog post in the style of J.K. Rowling",
-      llm=llm,
+        llm=llm,
-      dashboard=False,
+        dashboard=False,
-  )
+    )
+
+    writer2 = Agent(
+        agent_name="Writer 2",
+        system_prompt="Generate a blog post in the style of Stephen King",
+        llm=llm,
+        dashboard=False,
+    )
 
-  writer2 = Agent(
+    reviewer = Agent(
-      agent_name="Writer 2",
+        agent_name="Reviewer",
-      system_prompt="Generate a blog post in the style of Stephen King",
+        system_prompt="Select the writer that wrote the best story. There can only be one best story.",
-      llm=llm,
+        llm=llm,
-      dashboard=False
+        dashboard=False,
-  )
+    )
 
-  reviewer = Agent(
+    flow = f"{writer1.name}, {writer2.name} -> {reviewer.name}"
-      agent_name="Reviewer",
-      system_prompt="Select the writer that wrote the best story. There can only be one best story.",
-      llm=llm,
-      dashboard=False
-  )
 
-  flow = f"{writer1.name}, {writer2.name} -> {reviewer.name}"
+    agent_rearrange = AgentRearrange(
+        [writer1, writer2, reviewer], flow, verbose=False, logging=False
+    )
 
-  agent_rearrange = AgentRearrange(
+    result = await agent_rearrange.astream(
-    [writer1, writer2, reviewer], flow, verbose=False, logging=False
+        "Generate a 1 sentence story about Michael Jordan."
-  )
+    )
 
-  result = await agent_rearrange.astream("Generate a 1 sentence story about Michael Jordan.")
+    # LEAVING THIS CALL BELOW FOR COMPARISON with "completion-style" .run() approach ;)
+    # result = agent_rearrange.run(
+    #   "Generate a short blog post about Michael Jordan."
+    # )
 
-  # LEAVING THIS CALL BELOW FOR COMPARISON with "completion-style" .run() approach ;)
-  # result = agent_rearrange.run(
-  #   "Generate a short blog post about Michael Jordan."
-  # )
 
 # asyncio.run(sequential())
 asyncio.run(parallel())
-