From 8cafdaeecf9a77919ae68376c1a127630a9489f6 Mon Sep 17 00:00:00 2001 From: Kye Date: Sun, 16 Jul 2023 16:21:45 -0400 Subject: [PATCH] ckean up agent craetpr Former-commit-id: 24276636180efb751ba1d5931cb68ddcc57a19b4 --- swarms/agents/workers/WorkerUltraNode.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/swarms/agents/workers/WorkerUltraNode.py b/swarms/agents/workers/WorkerUltraNode.py index fd90e924..4c2846a9 100644 --- a/swarms/agents/workers/WorkerUltraNode.py +++ b/swarms/agents/workers/WorkerUltraNode.py @@ -4,7 +4,7 @@ import logging from pathlib import Path from typing import Dict, List -from swarms.agents.utils.AgentManager import AgentManager +from swarms.agents.utils.AgentManager import AgentCreator from swarms.utils.main import BaseHandler, FileHandler, FileType from swarms.tools.main import ExitConversation, RequestsGet, CodeEditor, Terminal from swarms.utils.main import CsvToDataframe @@ -60,7 +60,7 @@ class WorkerUltraNode: handlers[FileType.IMAGE] = ImageCaptioning("cuda") try: - self.agent_manager = AgentManager.create(toolsets=toolsets) + self.agent_manager = AgentCreator.create(toolsets=toolsets) self.file_handler = FileHandler(handlers=handlers, path=BASE_DIR) self.uploader = StaticUploader.from_settings( path=BASE_DIR / "static", endpoint="static"