Replace SQLAgent with AgentExecutor in SQLAgentComponent constructor

This commit is contained in:
anovazzi1 2024-01-10 18:34:14 -03:00
commit 369ca587fa

View file

@ -22,4 +22,4 @@ class SQLAgentComponent(CustomComponent):
# Assuming there is a constructor for SQLAgent that takes these parameters
# Since the actual implementation is not provided, this is a placeholder
# Replace SQLAgent with the actual class name if different
return SQLAgent(llm=llm, database_uri=database_uri)
return AgentExecutor(llm=llm, database_uri=database_uri)