diff --git a/docs/mkdocs.yml b/docs/mkdocs.yml index e9191e96..939ababf 100644 --- a/docs/mkdocs.yml +++ b/docs/mkdocs.yml @@ -131,8 +131,6 @@ nav: - Swarm Ecosystem: "swarms/ecosystem.md" - Framework: - Overview: "swarms/framework/index.md" - - Contributing: - - Tests: "swarms/framework/test.md" - Models: - Overview: "swarms/models/index.md" - How to Create A Custom Language Model: "swarms/models/custom_model.md" @@ -148,6 +146,8 @@ nav: - Overview: "swarms/structs/multi_agent_orchestration.md" - Swarm Architectures: "swarms/concept/swarm_architectures.md" - Workflows: "swarms/structs/workflows.md" + - Contributing: + - Tests: "swarms/framework/test.md" - Reference: - Overview: "swarms/structs/index.md" - Models: diff --git a/playground/agents/multion_agent.py b/playground/agents/3rd_party_agents/multion_agent.py similarity index 100% rename from playground/agents/multion_agent.py rename to playground/agents/3rd_party_agents/multion_agent.py diff --git a/playground/agents/mm_agent_example.py b/playground/agents/mm_agent_example.py deleted file mode 100644 index d564fc02..00000000 --- a/playground/agents/mm_agent_example.py +++ /dev/null @@ -1,17 +0,0 @@ -from swarms.agents import MultiModalAgent - -load_dict = {"ImageCaptioning": "cuda"} - -node = MultiModalAgent(load_dict) - -text = node.run_text("What is your name? Generate a picture of yourself") - -img = node.run_img("/image1", "What is this image about?") - -chat = node.chat( - ( - "What is your name? Generate a picture of yourself. What is" - " this image about?" - ), - streaming=True, -) diff --git a/multi_modal_example.py b/playground/agents/multi_modal_example.py similarity index 100% rename from multi_modal_example.py rename to playground/agents/multi_modal_example.py diff --git a/playground/agents/full_stack_agent.py b/playground/agents/tools/full_stack_agent.py similarity index 100% rename from playground/agents/full_stack_agent.py rename to playground/agents/tools/full_stack_agent.py diff --git a/playground/agents/tool.py b/playground/agents/tools/tool.py similarity index 100% rename from playground/agents/tool.py rename to playground/agents/tools/tool.py diff --git a/playground/agents/worker_example.py b/playground/agents/worker_example.py deleted file mode 100644 index a2117e46..00000000 --- a/playground/agents/worker_example.py +++ /dev/null @@ -1,32 +0,0 @@ -# Importing necessary modules -import os - -from dotenv import load_dotenv - -from swarms import OpenAIChat -from swarms.agents.worker_agent import Worker - -# Loading environment variables from .env file -load_dotenv() - -# Retrieving the OpenAI API key from environment variables -api_key = os.getenv("OPENAI_API_KEY") - -# Creating a Worker instance -worker = Worker( - name="My Worker", - role="Worker", - human_in_the_loop=False, - tools=[], - temperature=0.5, - llm=OpenAIChat(openai_api_key=api_key), - verbose=True, -) - -# Running the worker with a prompt -out = worker.run( - "Hello, how are you? Create an image of how your are doing!" -) - -# Printing the output -print(out) diff --git a/tests/structs/test_multi_agent_collab.py b/tests/structs/test_multi_agent_collab.py index d17e549a..e69e87e1 100644 --- a/tests/structs/test_multi_agent_collab.py +++ b/tests/structs/test_multi_agent_collab.py @@ -57,6 +57,7 @@ worker2 = Agent( # Create a list of agents agents = [director, worker1, worker2] + @pytest.fixture def collaboration(): return MultiAgentCollaboration(agents)