pylint cleanup agents

pull/377/head
evelynmitchell 1 year ago
parent 543f8f0353
commit e70acbaea5

@ -3,7 +3,7 @@
from swarms.structs.agent import Agent from swarms.structs.agent import Agent
def agent_wrapper(ClassToWrap): def agent_wrapper(ClassToWrap): # noqa C0103
""" """
This function takes a class 'ClassToWrap' and returns a new class that This function takes a class 'ClassToWrap' and returns a new class that
inherits from both 'ClassToWrap' and 'Agent'. The new class overrides inherits from both 'ClassToWrap' and 'Agent'. The new class overrides

@ -74,11 +74,11 @@ class OmniModalAgent:
# self.task_executor = TaskExecutor # self.task_executor = TaskExecutor
self.history = [] self.history = []
def run(self, inputtoagent: str) -> str: def run(self, input_to_agent: str) -> str:
"""Run the OmniAgent""" """Run the OmniAgent"""
plan = self.chat_planner.plan( plan = self.chat_planner.plan(
inputs={ inputs={
"input": inputtoagent, "input": input_to_agent,
"hf_tools": self.tools, "hf_tools": self.tools,
} }
) )

Loading…
Cancel
Save