fix missing change

run-llama · Sep 26, 2024 · a22ecf9 · a22ecf9
1 parent bfef745
commit a22ecf9
Showing 1 changed file with 3 additions and 5 deletions.
diff --git 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: