diff --git a/swarms/agents/__init__.py b/swarms/agents/__init__.py index d589fa9b..946c60cf 100644 --- a/swarms/agents/__init__.py +++ b/swarms/agents/__init__.py @@ -17,7 +17,7 @@ #agents -from swarms.swarms.profitpilot import ProfitPilot -from swarms.aot import AoTAgent +from swarms.agents.profitpilot import ProfitPilot +from swarms.agents.aot import AoTAgent from swarms.agents.multi_modal_agent import MultiModalVisualAgent from swarms.agents.omni_modal_agent import OmniModalAgent \ No newline at end of file diff --git a/swarms/agents/profitpilot.py b/swarms/agents/profitpilot.py index 9f30127e..95afc637 100644 --- a/swarms/agents/profitpilot.py +++ b/swarms/agents/profitpilot.py @@ -316,7 +316,7 @@ Previous conversation history: """ -class SalesGPT(Chain, BaseModel): +class ProfitPilot(Chain, BaseModel): """Controller model for the Sales Agent.""" conversation_history: List[str] = [] @@ -513,7 +513,7 @@ config = dict( product_catalog="sample_product_catalog.txt", ) -sales_agent = SalesGPT.from_llm(llm, verbose=False, **config) +sales_agent = ProfitPilot.from_llm(llm, verbose=False, **config) # init sales agent sales_agent.seed_agent()