[BUFG][ToolAgent]

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

@ -26,7 +26,7 @@ task = (
# Create an instance of the ToolAgent class
agent = ToolAgent(
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,11 +61,11 @@ class ToolAgent(AbstractLLM):
def __init__(
self,
name: str,
description: str,
model: Any,
tokenizer: Any,
json_schema: Any,
name: str = None,
description: str = None,
model: Any = None,
tokenizer: Any = None,
json_schema: Any = None,
max_number_tokens: int = 500,
*args,
**kwargs,

Loading…
Cancel
Save