You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
swarms/swarms/structs/__init__.py

158 lines
3.9 KiB

from swarms.schemas.plan import Plan
from swarms.schemas.step import Step
from swarms.structs.agent import Agent
from swarms.structs.agent_job import AgentJob
from swarms.structs.agent_process import (
AgentProcess,
AgentProcessQueue,
)
from swarms.structs.auto_swarm import AutoSwarm, AutoSwarmRouter
from swarms.structs.base_structure import BaseStructure
from swarms.structs.base_swarm import BaseSwarm
from swarms.structs.base_workflow import BaseWorkflow
from swarms.structs.concurrent_workflow import ConcurrentWorkflow
from swarms.structs.conversation import Conversation
from swarms.structs.groupchat import GroupChat
from swarms.structs.hiearchical_swarm import HiearchicalSwarm
from swarms.structs.majority_voting import (
MajorityVoting,
majority_voting,
most_frequent,
parse_code_completion,
)
from swarms.structs.message import Message
from swarms.structs.message_pool import MessagePool
from swarms.structs.multi_agent_collab import MultiAgentCollaboration
from swarms.structs.multi_process_workflow import (
MultiProcessWorkflow,
)
from swarms.structs.multi_threaded_workflow import (
MultiThreadedWorkflow,
)
from swarms.structs.swarm_net import SwarmNetwork
from swarms.structs.rearrange import AgentRearrange, rearrange
from swarms.structs.recursive_workflow import RecursiveWorkflow
from swarms.structs.round_robin import RoundRobinSwarm
from swarms.structs.sequential_workflow import SequentialWorkflow
# New Swarms
# from swarms.structs.swarm_load_balancer import AgentLoadBalancer
# from swarms.structs.swarm_net import SwarmNetwork
from swarms.structs.swarming_architectures import (
broadcast,
circular_swarm,
exponential_swarm,
fibonacci_swarm,
geometric_swarm,
grid_swarm,
harmonic_swarm,
linear_swarm,
log_swarm,
mesh_swarm,
one_to_one,
one_to_three,
power_swarm,
prime_swarm,
pyramid_swarm,
sigmoid_swarm,
staircase_swarm,
star_swarm,
)
from swarms.structs.task import Task
from swarms.structs.task_queue_base import (
TaskQueueBase,
synchronized_queue,
)
from swarms.structs.utils import (
detect_markdown,
distribute_tasks,
extract_key_from_json,
extract_tokens_from_text,
find_agent_by_id,
find_token_in_text,
parse_tasks,
)
from swarms.structs.yaml_model import (
YamlModel,
create_yaml_schema_from_dict,
get_type_name,
pydantic_type_to_yaml_schema,
)
from swarms.structs.mixture_of_agents import MixtureOfAgents
from swarms.structs.graph_workflow import (
GraphWorkflow,
Node,
NodeType,
Edge,
)
__all__ = [
"Agent",
"AgentJob",
"AgentProcess",
"AgentProcessQueue",
"AutoSwarm",
"AutoSwarmRouter",
"BaseStructure",
"BaseSwarm",
"BaseWorkflow",
"ConcurrentWorkflow",
"Conversation",
"GroupChat",
"HiearchicalSwarm",
"MajorityVoting",
"majority_voting",
"most_frequent",
"parse_code_completion",
"Message",
"MessagePool",
"MultiAgentCollaboration",
"MultiProcessWorkflow",
"MultiThreadedWorkflow",
"SwarmNetwork",
"AgentRearrange",
"rearrange",
"RecursiveWorkflow",
"RoundRobinSwarm",
"SequentialWorkflow",
"Task",
"TaskQueueBase",
"synchronized_queue",
"detect_markdown",
"distribute_tasks",
"extract_key_from_json",
"extract_tokens_from_text",
"find_agent_by_id",
"find_token_in_text",
"parse_tasks",
"YamlModel",
"create_yaml_schema_from_dict",
"get_type_name",
"pydantic_type_to_yaml_schema",
"MixtureOfAgents",
"GraphWorkflow",
"Node",
"NodeType",
"Edge",
"Plan",
"Step",
"broadcast",
"circular_swarm",
"exponential_swarm",
"fibonacci_swarm",
"geometric_swarm",
"grid_swarm",
"harmonic_swarm",
"linear_swarm",
"log_swarm",
"mesh_swarm",
"one_to_one",
"one_to_three",
"power_swarm",
"prime_swarm",
"pyramid_swarm",
"sigmoid_swarm",
"staircase_swarm",
"star_swarm",
]