[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 # Create an instance of the ToolAgent class
agent = ToolAgent( 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 # Run the agent to generate the person's information

@ -61,11 +61,11 @@ class ToolAgent(AbstractLLM):
def __init__( def __init__(
self, self,
name: str, name: str = None,
description: str, description: str = None,
model: Any, model: Any = None,
tokenizer: Any, tokenizer: Any = None,
json_schema: Any, json_schema: Any = None,
max_number_tokens: int = 500, max_number_tokens: int = 500,
*args, *args,
**kwargs, **kwargs,

Loading…
Cancel
Save