From c7660e8d5850d165a302eee139eba683c9f37514 Mon Sep 17 00:00:00 2001 From: Kye Date: Thu, 3 Aug 2023 13:29:15 -0400 Subject: [PATCH] chipping sway at Agent implementation classs Former-commit-id: b6b166128c71cfbaa83c2d90d8f9b96c9224bae9 --- 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