Fix: Avoid fix_memory_inputs when langchain_object is instance of AgentExecutor (#1142)
This commit is contained in:
commit
384fb84950
1 changed files with 6 additions and 1 deletions
|
|
@ -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)
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue