[CLEANUP][ToolAgent]

1 year ago · 95967e311e
parent 6dd8d98c97
commit 95967e311e
2 changed files with 2 additions and 3 deletions
--- a/README.md
+++ b/README.md
@ -98,7 +98,7 @@ json_schema = {
 }

 task = "Generate a person's information based on the following schema:"
-agent = ToolAgent(model, tokenizer, json_schema)
+agent = ToolAgent(model=model, tokenizer=tokenizer, json_schema=json_schema)
 generated_data = agent.run(task)

 print(generated_data)
--- a/swarms/agents/tool_agent.py
+++ b/swarms/agents/tool_agent.py
@ -56,11 +56,10 @@ class ToolAgent(AbstractLLM):
        }

        task = "Generate a person's information based on the following schema:"
-        agent = ToolAgent(model, tokenizer, json_schema)
+        agent = ToolAgent(model=model, tokenizer=tokenizer, json_schema=json_schema)
        generated_data = agent.run(task)

        print(generated_data)
-
    """
    def __init__(
        self,