From 1ef7914a2cc08c16f3fd837c0dabfbe6edb030d6 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Sun, 24 Mar 2024 16:35:38 -0300
Subject: [PATCH] Refactor build method to be asynchronous

---
 src/backend/langflow/components/agents/XMLAgent.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/backend/langflow/components/agents/XMLAgent.py b/src/backend/langflow/components/agents/XMLAgent.py
index fd931ae9b..47a3cc09a 100644
--- a/src/backend/langflow/components/agents/XMLAgent.py
+++ b/src/backend/langflow/components/agents/XMLAgent.py
@@ -63,7 +63,7 @@ class XMLAgentComponent(CustomComponent):
             },
         }
 
-    def build(
+    async def build(
         self,
         inputs: str,
         llm: BaseLLM,
@@ -89,6 +89,6 @@ class XMLAgentComponent(CustomComponent):
         for var in input_variables:
             if var not in ["agent_scratchpad", "input"]:
                 input_dict[var] = ""
-        result = runnable.invoke(input_dict)
+        result = await runnable.ainvoke(input_dict)
         self.status = result
-        return result
+        return result.get("output")