jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor build method to be asynchronous

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-24 16:35:38 -03:00
commit 1ef7914a2c
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/langflow/components/agents/XMLAgent.py
View file

@ -63,7 +63,7 @@ class XMLAgentComponent(CustomComponent):
}, },
} }
def build( async def build(
self, self,
inputs: str, inputs: str,
llm: BaseLLM, llm: BaseLLM,
@ -89,6 +89,6 @@ class XMLAgentComponent(CustomComponent):
for var in input_variables: for var in input_variables:
if var not in ["agent_scratchpad", "input"]: if var not in ["agent_scratchpad", "input"]:
input_dict[var] = "" input_dict[var] = ""
result = runnable.invoke(input_dict) result = await runnable.ainvoke(input_dict)
self.status = result self.status = result
return result return result.get("output")