From 07899cdaad1d501a6c49e6261c1f7fcd187a0ad7 Mon Sep 17 00:00:00 2001 From: Kye <kye@apacmediasolutions.com> Date: Thu, 3 Aug 2023 13:29:15 -0400 Subject: [PATCH] chipping sway at Agent implementation classs Former-commit-id: 583c012f4c87ceca8efe38e6fbd283c9201ae73b --- swarms/agents/base.py | 4 +--- swarms/agents/tools/code_intepretor.py | 1 + 2 files changed, 2 insertions(+), 3 deletions(-) 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