From d3a2367932c5de9cf57716ec34cb5024aac867ce Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Mon, 26 Jun 2023 13:27:34 -0300
Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=80=20feat(custom.py):=20add=20AgentTy?=
 =?UTF-8?q?pe=20enum=20to=20improve=20readability=20and=20type=20safety=20?=
 =?UTF-8?q?The=20`AgentType`=20enum=20is=20added=20to=20the=20`langchain.a?=
 =?UTF-8?q?gents.custom`=20module=20to=20improve=20readability=20and=20typ?=
 =?UTF-8?q?e=20safety.=20The=20`InitializeAgent`=20class=20now=20uses=20th?=
 =?UTF-8?q?e=20`AgentType`=20enum=20to=20ensure=20that=20the=20`agent`=20p?=
 =?UTF-8?q?arameter=20is=20a=20valid=20value=20from=20the=20enum.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/interface/agents/custom.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/backend/langflow/interface/agents/custom.py b/src/backend/langflow/interface/agents/custom.py
index a4191a8b7..aa0cfb5db 100644
--- a/src/backend/langflow/interface/agents/custom.py
+++ b/src/backend/langflow/interface/agents/custom.py
@@ -6,6 +6,7 @@ from langchain.agents import (
     Tool,
     ZeroShotAgent,
     initialize_agent,
+    AgentType,
 )
 from langchain.agents.agent_toolkits import (
     SQLDatabaseToolkit,
@@ -297,6 +298,9 @@ class InitializeAgent(CustomAgentExecutor):
         agent: str,
         memory: Optional[BaseChatMemory] = None,
     ):
+        # Find which value in the AgentType enum corresponds to the string
+        # passed in as agent
+        agent = AgentType(agent)
         return initialize_agent(
             tools=tools,
             llm=llm,