Fix: Avoid fix_memory_inputs when langchain_object is instance of AgentExecutor (#1142)

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-16 08:27:56 -03:00 • committed by GitHub
commit 384fb84950
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1,5 +1,6 @@
import json import json
from pathlib import Path from pathlib import Path
from langchain.agents import AgentExecutor
from langchain.schema import AgentAction from langchain.schema import AgentAction
from langflow.interface.run import ( from langflow.interface.run import (
build_sorted_vertices, build_sorted_vertices,
@ -70,7 +71,11 @@ def get_result_and_thought(langchain_object: Any, inputs: dict):
if hasattr(langchain_object, "return_intermediate_steps"): if hasattr(langchain_object, "return_intermediate_steps"):
langchain_object.return_intermediate_steps = False langchain_object.return_intermediate_steps = False
fix_memory_inputs(langchain_object) try:
if not isinstance(langchain_object, AgentExecutor):
fix_memory_inputs(langchain_object)
except Exception as exc:
logger.error(f"Error fixing memory inputs: {exc}")
try: try:
output = langchain_object(inputs, return_only_outputs=True) output = langchain_object(inputs, return_only_outputs=True)