From 6ea59f583d5ab667ee904c98beffdf5adb43b5cc Mon Sep 17 00:00:00 2001 From: Kye Date: Tue, 2 Jan 2024 21:22:47 -0500 Subject: [PATCH] [README][ToolAgent] --- README.md | 2 +- swarms/agents/tool_agent.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 0271130f..7de6608a 100644 --- a/README.md +++ b/README.md @@ -99,7 +99,7 @@ json_schema = { task = "Generate a person's information based on the following schema:" agent = ToolAgent(model, tokenizer, json_schema) -generated_data = agent(task) +generated_data = agent.run(task) print(generated_data) diff --git a/swarms/agents/tool_agent.py b/swarms/agents/tool_agent.py index ac01c5cc..58b4cde4 100644 --- a/swarms/agents/tool_agent.py +++ b/swarms/agents/tool_agent.py @@ -57,12 +57,11 @@ class ToolAgent(AbstractLLM): task = "Generate a person's information based on the following schema:" agent = ToolAgent(model, tokenizer, json_schema) - generated_data = agent(task) + generated_data = agent.run(task) print(generated_data) """ - def __init__( self, name: str,