From bb1cb0804048f9c2ccbe314327e757f88adbd1d4 Mon Sep 17 00:00:00 2001 From: Kye Date: Mon, 25 Sep 2023 20:50:14 -0400 Subject: [PATCH] clean up --- swarms/agents/__init__.py | 4 ++-- swarms/agents/profitpilot.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) 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()