[REFACTOR][ToolAgent][name and description]

pull/378/head
Kye 11 months ago
parent 5569a5daf8
commit 18b8a5d602

@ -26,7 +26,11 @@ task = (
# Create an instance of the ToolAgent class
agent = ToolAgent(
name="dolly-function-agent", description="Ana gent to create a child data", model=model, tokenizer=tokenizer, json_schema=json_schema
name="dolly-function-agent",
description="Ana gent to create a child data",
model=model,
tokenizer=tokenizer,
json_schema=json_schema,
)
# Run the agent to generate the person's information

@ -61,8 +61,8 @@ class ToolAgent(AbstractLLM):
def __init__(
self,
name: str = None,
description: str = None,
name: str = "Function Calling Agent",
description: str = "Generates a function based on the input json schema and the task",
model: Any = None,
tokenizer: Any = None,
json_schema: Any = None,

Loading…
Cancel
Save