From db03be31b75c70b36506a15dbcae785870b5a5f2 Mon Sep 17 00:00:00 2001 From: Kye Date: Sun, 16 Jul 2023 17:09:42 -0400 Subject: [PATCH] clean eval output parser Former-commit-id: 242dd00d21215b3d92706fd9a8ed6c11f78caf78 --- swarms/agents/utils/agent_creator.py | 3 ++- swarms/agents/utils/agent_setup.py | 2 +- swarms/agents/utils/{callback.py => output_parser.py} | 0 3 files changed, 3 insertions(+), 2 deletions(-) rename swarms/agents/utils/{callback.py => output_parser.py} (100%) diff --git a/swarms/agents/utils/agent_creator.py b/swarms/agents/utils/agent_creator.py index 9e6f82d4..bc850822 100644 --- a/swarms/agents/utils/agent_creator.py +++ b/swarms/agents/utils/agent_creator.py @@ -10,7 +10,8 @@ from swarms.tools.main import BaseToolSet, ToolsFactory from swarms.prompts.prompts import EVAL_PREFIX, EVAL_SUFFIX from swarms.agents.utils.agent_setup import AgentSetup -from .callback import EVALCallbackHandler, ExecutionTracingCallbackHandler +# from .callback import EVALCallbackHandler, ExecutionTracingCallbackHandler +from swarms.agents.utils.Calback import EVALCallbackHandler, ExecutionTracingCallbackHandler callback_manager_instance = CallbackManager(EVALCallbackHandler()) diff --git a/swarms/agents/utils/agent_setup.py b/swarms/agents/utils/agent_setup.py index 57811087..15669169 100644 --- a/swarms/agents/utils/agent_setup.py +++ b/swarms/agents/utils/agent_setup.py @@ -12,7 +12,7 @@ from langchain.callbacks.base import BaseCallbackManager from .ConversationalChatAgent import ConversationalChatAgent # from .ChatOpenAI import ChatOpenAI from langchain.chat_models import ChatOpenAI -from .callback import EvalOutputParser +from .output_parser import EvalOutputParser diff --git a/swarms/agents/utils/callback.py b/swarms/agents/utils/output_parser.py similarity index 100% rename from swarms/agents/utils/callback.py rename to swarms/agents/utils/output_parser.py