diff --git a/templates/types/multiagent/fastapi/app/agents/single.py b/templates/types/multiagent/fastapi/app/agents/single.py index b47662f8..754cc5ac 100644 --- a/templates/types/multiagent/fastapi/app/agents/single.py +++ b/templates/types/multiagent/fastapi/app/agents/single.py @@ -5,10 +5,8 @@ from llama_index.core.llms.function_calling import FunctionCallingLLM from llama_index.core.memory import ChatMemoryBuffer from llama_index.core.settings import Settings -from llama_index.core.tools import ToolOutput, ToolSelection +from llama_index.core.tools import FunctionTool, ToolOutput, ToolSelection from llama_index.core.tools.types import BaseTool -from llama_index.core.tools import FunctionTool - from llama_index.core.workflow import ( Context, Event, @@ -64,13 +62,13 @@ def __init__( timeout: float = 360.0, name: str, write_events: bool = True, - role: Optional[str] = None, + description: Optional[str] = None, **kwargs: Any, ) -> None: super().__init__(*args, verbose=verbose, timeout=timeout, **kwargs) self.tools = tools or [] self.name = name - self.role = role + self.description = description self.write_events = write_events if llm is None: