swarms 7.8.4__py3-none-any.whl → 7.8.8__py3-none-any.whl

swarms/prompts/prompt.py

@@ -99,9 +99,6 @@ class Prompt(BaseModel):
         if self.autosave:
             self._autosave()
 
-        if self.auto_generate_prompt and self.llm:
-            self.auto_generate_prompt()
-
     def edit_prompt(self, new_content: str) -> None:
         """
         Edits the prompt content and updates the version control.

swarms/schemas/agent_completion_response.py

@@ -0,0 +1,71 @@
+from datetime import datetime
+from typing import Any, List, Optional
+
+from pydantic import BaseModel, Field
+
+
+class Usage(BaseModel):
+    prompt_tokens: Optional[int] = Field(
+        default=None,
+        description="Number of tokens used in the prompt",
+    )
+    completion_tokens: Optional[int] = Field(
+        default=None,
+        description="Number of tokens used in the completion",
+    )
+    total_tokens: Optional[int] = Field(
+        default=None, description="Total number of tokens used"
+    )
+
+
+class ModelConfig(BaseModel):
+    model_name: Optional[str] = Field(
+        default=None,
+        description="Name of the model used for generation",
+    )
+    temperature: Optional[float] = Field(
+        default=None,
+        description="Temperature setting used for generation",
+    )
+    top_p: Optional[float] = Field(
+        default=None, description="Top-p setting used for generation"
+    )
+    max_tokens: Optional[int] = Field(
+        default=None,
+        description="Maximum number of tokens to generate",
+    )
+    frequency_penalty: Optional[float] = Field(
+        default=None,
+        description="Frequency penalty used for generation",
+    )
+    presence_penalty: Optional[float] = Field(
+        default=None,
+        description="Presence penalty used for generation",
+    )
+
+
+class AgentCompletionResponse(BaseModel):
+    id: Optional[str] = Field(
+        default=None, description="Unique identifier for the response"
+    )
+    agent_name: Optional[str] = Field(
+        default=None,
+        description="Name of the agent that generated the response",
+    )
+    agent_description: Optional[str] = Field(
+        default=None, description="Description of the agent"
+    )
+    outputs: Optional[List[Any]] = Field(
+        default=None,
+        description="List of outputs generated by the agent",
+    )
+    usage: Optional[Usage] = Field(
+        default=None, description="Token usage statistics"
+    )
+    model_config: Optional[ModelConfig] = Field(
+        default=None, description="Model configuration"
+    )
+    timestamp: Optional[str] = Field(
+        default_factory=lambda: datetime.now().isoformat(),
+        description="Timestamp of when the response was generated",
+    )

swarms/schemas/agent_rag_schema.py

@@ -0,0 +1,7 @@
+from pydantic import BaseModel
+
+
+class AgentRAGConfig(BaseModel):
+    """
+    Configuration for the AgentRAG class.
+    """

swarms/schemas/conversation_schema.py

@@ -0,0 +1,9 @@
+from typing import Optional
+from pydantic import BaseModel, Field
+
+
+class ConversationSchema(BaseModel):
+    time_enabled: Optional[bool] = Field(default=False)
+    message_id_on: Optional[bool] = Field(default=True)
+    autosave: Optional[bool] = Field(default=False)
+    count_tokens: Optional[bool] = Field(default=False)

swarms/schemas/llm_agent_schema.py

@@ -1,91 +1,109 @@
 from pydantic import BaseModel, Field
-from typing import List, Optional, Union, Any, Literal
-from litellm.types import (
-    ChatCompletionPredictionContentParam,
-)
+from typing import Optional
 
+# from litellm.types import (
+#     ChatCompletionPredictionContentParam,
+# )
 
-class LLMCompletionRequest(BaseModel):
-    """Schema for LLM completion request parameters."""
 
-    model: Optional[str] = Field(
-        default=None,
-        description="The name of the language model to use for text completion",
-    )
-    temperature: Optional[float] = Field(
-        default=0.5,
-        description="Controls randomness of the output (0.0 to 1.0)",
-    )
-    top_p: Optional[float] = Field(
-        default=None,
-        description="Controls diversity via nucleus sampling",
-    )
-    n: Optional[int] = Field(
-        default=None, description="Number of completions to generate"
-    )
-    stream: Optional[bool] = Field(
-        default=None, description="Whether to stream the response"
-    )
-    stream_options: Optional[dict] = Field(
-        default=None, description="Options for streaming response"
-    )
-    stop: Optional[Any] = Field(
-        default=None,
-        description="Up to 4 sequences where the API will stop generating",
-    )
-    max_completion_tokens: Optional[int] = Field(
-        default=None,
-        description="Maximum tokens for completion including reasoning",
-    )
-    max_tokens: Optional[int] = Field(
-        default=None,
-        description="Maximum tokens in generated completion",
-    )
-    prediction: Optional[ChatCompletionPredictionContentParam] = (
-        Field(
-            default=None,
-            description="Configuration for predicted output",
-        )
-    )
-    presence_penalty: Optional[float] = Field(
-        default=None,
-        description="Penalizes new tokens based on existence in text",
-    )
-    frequency_penalty: Optional[float] = Field(
-        default=None,
-        description="Penalizes new tokens based on frequency in text",
-    )
-    logit_bias: Optional[dict] = Field(
-        default=None,
-        description="Modifies probability of specific tokens",
-    )
-    reasoning_effort: Optional[Literal["low", "medium", "high"]] = (
-        Field(
-            default=None,
-            description="Level of reasoning effort for the model",
-        )
-    )
-    seed: Optional[int] = Field(
-        default=None, description="Random seed for reproducibility"
-    )
-    tools: Optional[List] = Field(
-        default=None,
-        description="List of tools available to the model",
-    )
-    tool_choice: Optional[Union[str, dict]] = Field(
-        default=None, description="Choice of tool to use"
-    )
-    logprobs: Optional[bool] = Field(
-        default=None,
-        description="Whether to return log probabilities",
-    )
-    top_logprobs: Optional[int] = Field(
+# class LLMCompletionRequest(BaseModel):
+#     """Schema for LLM completion request parameters."""
+
+#     model: Optional[str] = Field(
+#         default=None,
+#         description="The name of the language model to use for text completion",
+#     )
+#     temperature: Optional[float] = Field(
+#         default=0.5,
+#         description="Controls randomness of the output (0.0 to 1.0)",
+#     )
+#     top_p: Optional[float] = Field(
+#         default=None,
+#         description="Controls diversity via nucleus sampling",
+#     )
+#     n: Optional[int] = Field(
+#         default=None, description="Number of completions to generate"
+#     )
+#     stream: Optional[bool] = Field(
+#         default=None, description="Whether to stream the response"
+#     )
+#     stream_options: Optional[dict] = Field(
+#         default=None, description="Options for streaming response"
+#     )
+#     stop: Optional[Any] = Field(
+#         default=None,
+#         description="Up to 4 sequences where the API will stop generating",
+#     )
+#     max_completion_tokens: Optional[int] = Field(
+#         default=None,
+#         description="Maximum tokens for completion including reasoning",
+#     )
+#     max_tokens: Optional[int] = Field(
+#         default=None,
+#         description="Maximum tokens in generated completion",
+#     )
+#     prediction: Optional[ChatCompletionPredictionContentParam] = (
+#         Field(
+#             default=None,
+#             description="Configuration for predicted output",
+#         )
+#     )
+#     presence_penalty: Optional[float] = Field(
+#         default=None,
+#         description="Penalizes new tokens based on existence in text",
+#     )
+#     frequency_penalty: Optional[float] = Field(
+#         default=None,
+#         description="Penalizes new tokens based on frequency in text",
+#     )
+#     logit_bias: Optional[dict] = Field(
+#         default=None,
+#         description="Modifies probability of specific tokens",
+#     )
+#     reasoning_effort: Optional[Literal["low", "medium", "high"]] = (
+#         Field(
+#             default=None,
+#             description="Level of reasoning effort for the model",
+#         )
+#     )
+#     seed: Optional[int] = Field(
+#         default=None, description="Random seed for reproducibility"
+#     )
+#     tools: Optional[List] = Field(
+#         default=None,
+#         description="List of tools available to the model",
+#     )
+#     tool_choice: Optional[Union[str, dict]] = Field(
+#         default=None, description="Choice of tool to use"
+#     )
+#     logprobs: Optional[bool] = Field(
+#         default=None,
+#         description="Whether to return log probabilities",
+#     )
+#     top_logprobs: Optional[int] = Field(
+#         default=None,
+#         description="Number of most likely tokens to return",
+#     )
+#     parallel_tool_calls: Optional[bool] = Field(
+#         default=None,
+#         description="Whether to allow parallel tool calls",
+#     )
+
+#     class Config:
+#         allow_arbitrary_types = True
+
+
+class ModelConfigOrigin(BaseModel):
+    """Schema for model configuration origin."""
+
+    model_url: Optional[str] = Field(
         default=None,
-        description="Number of most likely tokens to return",
+        description="The URL of the model to use for text completion",
     )
-    parallel_tool_calls: Optional[bool] = Field(
+
+    api_key: Optional[str] = Field(
         default=None,
-        description="Whether to allow parallel tool calls",
+        description="The API key to use for the model",
     )
 
     class Config:

swarms/schemas/swarms_api_schemas.py

@@ -0,0 +1,164 @@
+from pydantic import BaseModel, Field
+from typing import Optional, List, Dict, Any, Union, Literal
+
+SwarmType = Literal[
+    "AgentRearrange",
+    "MixtureOfAgents",
+    "SpreadSheetSwarm",
+    "SequentialWorkflow",
+    "ConcurrentWorkflow",
+    "GroupChat",
+    "MultiAgentRouter",
+    "AutoSwarmBuilder",
+    "HiearchicalSwarm",
+    "auto",
+    "MajorityVoting",
+    "MALT",
+    "DeepResearchSwarm",
+    "CouncilAsAJudge",
+    "InteractiveGroupChat",
+]
+
+
+class AgentSpec(BaseModel):
+    agent_name: Optional[str] = Field(
+        # default=None,
+        description="The unique name assigned to the agent, which identifies its role and functionality within the swarm.",
+    )
+    description: Optional[str] = Field(
+        default=None,
+        description="A detailed explanation of the agent's purpose, capabilities, and any specific tasks it is designed to perform.",
+    )
+    system_prompt: Optional[str] = Field(
+        default=None,
+        description="The initial instruction or context provided to the agent, guiding its behavior and responses during execution.",
+    )
+    model_name: Optional[str] = Field(
+        default="gpt-4o-mini",
+        description="The name of the AI model that the agent will utilize for processing tasks and generating outputs. For example: gpt-4o, gpt-4o-mini, openai/o3-mini",
+    )
+    auto_generate_prompt: Optional[bool] = Field(
+        default=False,
+        description="A flag indicating whether the agent should automatically create prompts based on the task requirements.",
+    )
+    max_tokens: Optional[int] = Field(
+        default=8192,
+        description="The maximum number of tokens that the agent is allowed to generate in its responses, limiting output length.",
+    )
+    temperature: Optional[float] = Field(
+        default=0.5,
+        description="A parameter that controls the randomness of the agent's output; lower values result in more deterministic responses.",
+    )
+    role: Optional[str] = Field(
+        default="worker",
+        description="The designated role of the agent within the swarm, which influences its behavior and interaction with other agents.",
+    )
+    max_loops: Optional[int] = Field(
+        default=1,
+        description="The maximum number of times the agent is allowed to repeat its task, enabling iterative processing if necessary.",
+    )
+    tools_list_dictionary: Optional[List[Dict[Any, Any]]] = Field(
+        default=None,
+        description="A dictionary of tools that the agent can use to complete its task.",
+    )
+    mcp_url: Optional[str] = Field(
+        default=None,
+        description="The URL of the MCP server that the agent can use to complete its task.",
+    )
+
+    class Config:
+        arbitrary_types_allowed = True
+
+
+class AgentCompletion(BaseModel):
+    agent_config: Optional[AgentSpec] = Field(
+        None,
+        description="The configuration of the agent to be completed.",
+    )
+    task: Optional[str] = Field(
+        None, description="The task to be completed by the agent."
+    )
+    history: Optional[Union[Dict[Any, Any], List[Dict[str, str]]]] = (
+        Field(
+            default=None,
+            description="The history of the agent's previous tasks and responses. Can be either a dictionary or a list of message objects.",
+        )
+    )
+
+    model_config = {
+        "arbitrary_types_allowed": True,
+        "populate_by_name": True,
+    }
+
+
+class Agents(BaseModel):
+    """Configuration for a collection of agents that work together as a swarm to accomplish tasks."""
+
+    agents: List[AgentSpec] = Field(
+        description="A list containing the specifications of each agent that will participate in the swarm, detailing their roles and functionalities."
+    )
+
+
+class SwarmSpec(BaseModel):
+    name: Optional[str] = Field(
+        None,
+        description="The name of the swarm, which serves as an identifier for the group of agents and their collective task.",
+        max_length=100,
+    )
+    description: Optional[str] = Field(
+        None,
+        description="A comprehensive description of the swarm's objectives, capabilities, and intended outcomes.",
+    )
+    agents: Optional[List[AgentSpec]] = Field(
+        None,
+        description="A list of agents or specifications that define the agents participating in the swarm.",
+    )
+    max_loops: Optional[int] = Field(
+        default=1,
+        description="The maximum number of execution loops allowed for the swarm, enabling repeated processing if needed.",
+    )
+    swarm_type: Optional[SwarmType] = Field(
+        None,
+        description="The classification of the swarm, indicating its operational style and methodology.",
+    )
+    rearrange_flow: Optional[str] = Field(
+        None,
+        description="Instructions on how to rearrange the flow of tasks among agents, if applicable.",
+    )
+    task: Optional[str] = Field(
+        None,
+        description="The specific task or objective that the swarm is designed to accomplish.",
+    )
+    img: Optional[str] = Field(
+        None,
+        description="An optional image URL that may be associated with the swarm's task or representation.",
+    )
+    return_history: Optional[bool] = Field(
+        True,
+        description="A flag indicating whether the swarm should return its execution history along with the final output.",
+    )
+    rules: Optional[str] = Field(
+        None,
+        description="Guidelines or constraints that govern the behavior and interactions of the agents within the swarm.",
+    )
+    tasks: Optional[List[str]] = Field(
+        None,
+        description="A list of tasks that the swarm should complete.",
+    )
+    messages: Optional[
+        Union[List[Dict[Any, Any]], Dict[Any, Any]]
+    ] = Field(
+        None,
+        description="A list of messages that the swarm should complete.",
+    )
+    stream: Optional[bool] = Field(
+        False,
+        description="A flag indicating whether the swarm should stream its output.",
+    )
+    service_tier: Optional[str] = Field(
+        "standard",
+        description="The service tier to use for processing. Options: 'standard' (default) or 'flex' for lower cost but slower processing.",
+    )
+
+    class Config:
+        arbitrary_types_allowed = True

swarms/structs/__init__.py

@@ -1,10 +1,13 @@
 from swarms.structs.agent import Agent
 from swarms.structs.agent_builder import AgentsBuilder
+from swarms.structs.auto_swarm_builder import AutoSwarmBuilder
 from swarms.structs.base_structure import BaseStructure
 from swarms.structs.base_swarm import BaseSwarm
 from swarms.structs.base_workflow import BaseWorkflow
+from swarms.structs.batch_agent_execution import batch_agent_execution
 from swarms.structs.concurrent_workflow import ConcurrentWorkflow
 from swarms.structs.conversation import Conversation
+from swarms.structs.council_judge import CouncilAsAJudge
 from swarms.structs.de_hallucination_swarm import DeHallucinationSwarm
 from swarms.structs.deep_research_swarm import DeepResearchSwarm
 from swarms.structs.graph_workflow import (
@@ -20,6 +23,11 @@ from swarms.structs.groupchat import (
 from swarms.structs.hybrid_hiearchical_peer_swarm import (
     HybridHierarchicalClusterSwarm,
 )
+from swarms.structs.ma_blocks import (
+    aggregate,
+    find_agent_by_name,
+    run_agent,
+)
 from swarms.structs.majority_voting import (
     MajorityVoting,
     majority_voting,
@@ -32,8 +40,9 @@ from swarms.structs.meme_agent_persona_generator import (
 )
 from swarms.structs.mixture_of_agents import MixtureOfAgents
 from swarms.structs.model_router import ModelRouter
-from swarms.structs.multi_agent_collab import MultiAgentCollaboration
 from swarms.structs.multi_agent_exec import (
+    get_agents_info,
+    get_swarms_info,
     run_agent_with_timeout,
     run_agents_concurrently,
     run_agents_concurrently_async,
@@ -43,8 +52,6 @@ from swarms.structs.multi_agent_exec import (
     run_agents_with_resource_monitoring,
     run_agents_with_tasks_concurrently,
     run_single_agent,
-    get_agents_info,
-    get_swarms_info,
 )
 from swarms.structs.multi_agent_router import MultiAgentRouter
 from swarms.structs.rearrange import AgentRearrange, rearrange
@@ -55,7 +62,6 @@ from swarms.structs.swarm_arange import SwarmRearrange
 from swarms.structs.swarm_router import (
     SwarmRouter,
     SwarmType,
-    swarm_router,
 )
 from swarms.structs.swarming_architectures import (
     broadcast,
@@ -77,11 +83,7 @@ from swarms.structs.swarming_architectures import (
     staircase_swarm,
     star_swarm,
 )
-from swarms.structs.auto_swarm_builder import AutoSwarmBuilder
-from swarms.structs.council_judge import CouncilAsAJudge
-from swarms.structs.batch_agent_execution import batch_agent_execution
-from swarms.structs.ma_blocks import aggregate
-
+from swarms.structs.interactive_groupchat import InteractiveGroupChat
 
 __all__ = [
     "Agent",
@@ -95,7 +97,6 @@ __all__ = [
     "majority_voting",
     "most_frequent",
     "parse_code_completion",
-    "MultiAgentCollaboration",
     "AgentRearrange",
     "rearrange",
     "RoundRobinSwarm",
@@ -135,7 +136,6 @@ __all__ = [
     "run_agents_with_different_tasks",
     "run_agent_with_timeout",
     "run_agents_with_resource_monitoring",
-    "swarm_router",
     "run_agents_with_tasks_concurrently",
     "GroupChat",
     "expertise_based",
@@ -153,4 +153,7 @@ __all__ = [
     "CouncilAsAJudge",
     "batch_agent_execution",
     "aggregate",
+    "find_agent_by_name",
+    "run_agent",
+    "InteractiveGroupChat",
 ]