diff --git a/swarms/agents/base.py b/swarms/agents/base.py
index 4e216e58..de63f608 100644
--- a/swarms/agents/base.py
+++ b/swarms/agents/base.py
@@ -11,13 +11,11 @@ from swarms.agents.models.base import AbstractModel
 from swarms.agents.prompts.agent_output_parser import AgentOutputParser
 from swarms.agents.prompts.agent_prompt_auto import PromptConstructor, MessageFormatter
 from swarms.agents.prompts.agent_prompt import AIMessage, HumanMessage, SystemMessage
-
+from swarms.agents.tools.base import BaseTool
 
 from langchain.chains.llm import LLMChain
 from langchain.memory import ChatMessageHistory
 from langchain.schema import (BaseChatMessageHistory, Document)
-
-from langchain.tools.base import BaseTool
 from langchain.vectorstores.base import VectorStoreRetriever
 
 class Agent:
diff --git a/swarms/agents/tools/code_intepretor.py b/swarms/agents/tools/code_intepretor.py
index 11018652..a832dd3e 100644
--- a/swarms/agents/tools/code_intepretor.py
+++ b/swarms/agents/tools/code_intepretor.py
@@ -1,3 +1,4 @@
+#props to shroominic
 from swarms.agents.tools.base import Tool, ToolException
 from typing import Callable, Any, List
 from codeinterpreterapi import CodeInterpreterSession, File, ToolException