diff --git a/example.py b/example.py index 49bfaca9..47ee3f12 100644 --- a/example.py +++ b/example.py @@ -26,7 +26,7 @@ print(task) -########## V2 +# ######### V2 # from swarms.agents.models.openai import OpenAI # chat = OpenAI() @@ -35,7 +35,7 @@ print(task) -# ############# v3 +# # ############# v3 # import os # from swarms.workers.vortex_worker import VortexWorkerAgent diff --git a/swarms/agents/base.py b/swarms/agents/base.py index 51f3cb31..cb98bd50 100644 --- a/swarms/agents/base.py +++ b/swarms/agents/base.py @@ -3,6 +3,7 @@ from __future__ import annotations from typing import List, Optional from langchain.chains.llm import LLMChain + from pydantic import ValidationError from swarms.agents.memory.base import VectorStoreRetriever @@ -26,7 +27,6 @@ from swarms.agents.utils.human_input import HumanInputRun class Agent: """Base Agent class""" - def __init__( self, ai_name: str, diff --git a/swarms/swarms/swarms.py b/swarms/swarms/swarms.py index 82663896..3f10b3ec 100644 --- a/swarms/swarms/swarms.py +++ b/swarms/swarms/swarms.py @@ -9,12 +9,9 @@ from swarms.workers.worker_node import WorkerNode logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s') - # ---------- Constants ---------- ROOT_DIR = "./data/" - - class HierarchicalSwarm: def __init__( self, @@ -50,7 +47,7 @@ class HierarchicalSwarm: self.boss_node = BossNode( api_key=self.openai_api_key, - worker_node=self.worker_node, + # worker_node=self.worker_node, llm_class=OpenAI, max_iterations=self.max_iterations, verbose=self.verbose