From 8b9e630e4485d07221e686bc8dd3bedc2ce868d4 Mon Sep 17 00:00:00 2001 From: Kye Gomez Date: Mon, 5 May 2025 17:04:00 -0700 Subject: [PATCH] cleanup --- swarms/schemas/__init__.py | 3 - swarms/schemas/agent_input_schema.py | 149 --------------------------- 2 files changed, 152 deletions(-) delete mode 100644 swarms/schemas/agent_input_schema.py diff --git a/swarms/schemas/__init__.py b/swarms/schemas/__init__.py index f81ae400..e4b33b8c 100644 --- a/swarms/schemas/__init__.py +++ b/swarms/schemas/__init__.py @@ -1,10 +1,7 @@ from swarms.schemas.agent_step_schemas import Step, ManySteps -from swarms.schemas.agent_input_schema import AgentSchema - __all__ = [ "Step", "ManySteps", - "AgentSchema", ] diff --git a/swarms/schemas/agent_input_schema.py b/swarms/schemas/agent_input_schema.py deleted file mode 100644 index 43365389..00000000 --- a/swarms/schemas/agent_input_schema.py +++ /dev/null @@ -1,149 +0,0 @@ -from typing import Any, Callable, Dict, List, Optional -from pydantic import BaseModel, Field -from pydantic.v1 import validator - - -class AgentSchema(BaseModel): - llm: Any = Field(..., description="The language model to use") - max_tokens: int = Field( - ..., description="The maximum number of tokens", ge=1 - ) - context_window: int = Field( - ..., description="The context window size", ge=1 - ) - user_name: str = Field(..., description="The user name") - agent_name: str = Field(..., description="The name of the agent") - system_prompt: str = Field(..., description="The system prompt") - template: Optional[str] = Field(default=None) - max_loops: Optional[int] = Field(default=1, ge=1) - stopping_condition: Optional[Callable[[str], bool]] = Field( - default=None - ) - loop_interval: Optional[int] = Field(default=0, ge=0) - retry_attempts: Optional[int] = Field(default=3, ge=0) - retry_interval: Optional[int] = Field(default=1, ge=0) - return_history: Optional[bool] = Field(default=False) - stopping_token: Optional[str] = Field(default=None) - dynamic_loops: Optional[bool] = Field(default=False) - interactive: Optional[bool] = Field(default=False) - dashboard: Optional[bool] = Field(default=False) - agent_description: Optional[str] = Field(default=None) - tools: Optional[List[Callable]] = Field(default=None) - dynamic_temperature_enabled: Optional[bool] = Field(default=False) - sop: Optional[str] = Field(default=None) - sop_list: Optional[List[str]] = Field(default=None) - saved_state_path: Optional[str] = Field(default=None) - autosave: Optional[bool] = Field(default=False) - self_healing_enabled: Optional[bool] = Field(default=False) - code_interpreter: Optional[bool] = Field(default=False) - multi_modal: Optional[bool] = Field(default=False) - pdf_path: Optional[str] = Field(default=None) - list_of_pdf: Optional[str] = Field(default=None) - tokenizer: Optional[Any] = Field(default=None) - long_term_memory: Optional[Any] = Field(default=None) - preset_stopping_token: Optional[bool] = Field(default=False) - traceback: Optional[Any] = Field(default=None) - traceback_handlers: Optional[Any] = Field(default=None) - streaming_on: Optional[bool] = Field(default=False) - docs: Optional[List[str]] = Field(default=None) - docs_folder: Optional[str] = Field(default=None) - verbose: Optional[bool] = Field(default=False) - parser: Optional[Callable] = Field(default=None) - best_of_n: Optional[int] = Field(default=None) - callback: Optional[Callable] = Field(default=None) - metadata: Optional[Dict[str, Any]] = Field(default=None) - callbacks: Optional[List[Callable]] = Field(default=None) - logger_handler: Optional[Any] = Field(default=None) - search_algorithm: Optional[Callable] = Field(default=None) - logs_to_filename: Optional[str] = Field(default=None) - evaluator: Optional[Callable] = Field(default=None) - output_json: Optional[bool] = Field(default=False) - stopping_func: Optional[Callable] = Field(default=None) - custom_loop_condition: Optional[Callable] = Field(default=None) - sentiment_threshold: Optional[float] = Field(default=None) - custom_exit_command: Optional[str] = Field(default="exit") - sentiment_analyzer: Optional[Callable] = Field(default=None) - limit_tokens_from_string: Optional[Callable] = Field(default=None) - custom_tools_prompt: Optional[Callable] = Field(default=None) - tool_schema: Optional[Any] = Field(default=None) - output_type: Optional[Any] = Field(default=None) - function_calling_type: Optional[str] = Field(default="json") - output_cleaner: Optional[Callable] = Field(default=None) - function_calling_format_type: Optional[str] = Field( - default="OpenAI" - ) - list_base_models: Optional[List[Any]] = Field(default=None) - metadata_output_type: Optional[str] = Field(default="json") - state_save_file_type: Optional[str] = Field(default="json") - chain_of_thoughts: Optional[bool] = Field(default=False) - algorithm_of_thoughts: Optional[bool] = Field(default=False) - tree_of_thoughts: Optional[bool] = Field(default=False) - tool_choice: Optional[str] = Field(default="auto") - execute_tool: Optional[bool] = Field(default=False) - rules: Optional[str] = Field(default=None) - planning: Optional[bool] = Field(default=False) - planning_prompt: Optional[str] = Field(default=None) - device: Optional[str] = Field(default=None) - custom_planning_prompt: Optional[str] = Field(default=None) - memory_chunk_size: Optional[int] = Field(default=2000, ge=0) - agent_ops_on: Optional[bool] = Field(default=False) - log_directory: Optional[str] = Field(default=None) - project_path: Optional[str] = Field(default=None) - tool_system_prompt: Optional[str] = Field( - default="tool_sop_prompt()" - ) - top_p: Optional[float] = Field(default=0.9, ge=0, le=1) - top_k: Optional[int] = Field(default=None) - frequency_penalty: Optional[float] = Field( - default=0.0, ge=0, le=1 - ) - presence_penalty: Optional[float] = Field(default=0.0, ge=0, le=1) - temperature: Optional[float] = Field(default=0.1, ge=0, le=1) - - @validator( - "tools", - "docs", - "sop_list", - "callbacks", - "list_base_models", - each_item=True, - ) - def check_list_items_not_none(cls, v): - if v is None: - raise ValueError("List items must not be None") - return v - - @validator( - "tokenizer", - "memory", - "traceback", - "traceback_handlers", - "parser", - "callback", - "search_algorithm", - "evaluator", - "stopping_func", - "custom_loop_condition", - "sentiment_analyzer", - "limit_tokens_from_string", - "custom_tools_prompt", - "output_cleaner", - ) - def check_optional_callable_not_none(cls, v): - if v is not None and not callable(v): - raise ValueError(f"{v} must be a callable") - return v - - -# # Example of how to use the schema -# agent_data = { -# "llm": "OpenAIChat", -# "max_tokens": 4096, -# "context_window": 8192, -# "user_name": "Human", -# "agent_name": "test-agent", -# "system_prompt": "Custom system prompt", -# } - -# agent = AgentSchema(**agent_data) -# print(agent)