Refactor AgentInitializerComponent to support optional memory parameter

src/backend/langflow/components/agents/AgentInitializer.py

@@ -1,7 +1,6 @@
-from typing import Callable, List, Union
+from typing import Callable, List, Optional, Union
 
 from langchain.agents import AgentExecutor, AgentType, initialize_agent, types
-
 from langflow import CustomComponent
 from langflow.field_typing import BaseChatMemory, BaseLanguageModel, Tool
 
@@ -20,12 +19,19 @@ class AgentInitializerComponent(CustomComponent):
             "memory": {"display_name": "Memory"},
             "tools": {"display_name": "Tools"},
             "llm": {"display_name": "Language Model"},
+            "code": {"advanced": True},
         }
 
     def build(
-        self, agent: str, llm: BaseLanguageModel, memory: BaseChatMemory, tools: List[Tool], max_iterations: int
+        self,
+        agent: str,
+        llm: BaseLanguageModel,
+        tools: List[Tool],
+        max_iterations: int,
+        memory: Optional[BaseChatMemory] = None,
     ) -> Union[AgentExecutor, Callable]:
         agent = AgentType(agent)
+        if memory:
             return initialize_agent(
                 tools=tools,
                 llm=llm,
@@ -35,3 +41,12 @@ class AgentInitializerComponent(CustomComponent):
                 handle_parsing_errors=True,
                 max_iterations=max_iterations,
             )
+        else:
+            return initialize_agent(
+                tools=tools,
+                llm=llm,
+                agent=agent,
+                return_intermediate_steps=True,
+                handle_parsing_errors=True,
+                max_iterations=max_iterations,
+            )