[README][CLEANUP][SwarmNetwork]

README.md

@@ -571,9 +571,7 @@ import os
 from dotenv import load_dotenv
 
 # Import the OpenAIChat model and the Agent struct
-from swarms.models import OpenAIChat
-from swarms.structs import Agent
-from swarms.structs.swarm_net import SwarmNetwork
+from swarms import OpenAIChat, Agent, SwarmNetwork
 
 # Load the environment variables
 load_dotenv()
@@ -615,7 +613,6 @@ out = swarmnet.run_many_agents(
 )
 print(out)
 
-
 ```
 
 ---

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarms"
-version = "3.1.0"
+version = "3.1.1"
 description = "Swarms - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <kye@apac.ai>"]

swarm_network.py

@@ -3,9 +3,7 @@ import os
 from dotenv import load_dotenv
 
 # Import the OpenAIChat model and the Agent struct
-from swarms.models import OpenAIChat
-from swarms.structs import Agent
-from swarms.structs.swarm_net import SwarmNetwork
+from swarms import OpenAIChat, Agent, SwarmNetwork
 
 # Load the environment variables
 load_dotenv()

swarms/swarms/__init__.py

@@ -2,6 +2,7 @@ from swarms.structs.autoscaler import AutoScaler
 from swarms.swarms.model_parallizer import ModelParallelizer
 from swarms.swarms.multi_agent_collab import MultiAgentCollaboration
 from swarms.swarms.base import AbstractSwarm
+# from swarms.swarms.team import Team
 
 __all__ = [
     "AutoScaler",