fix: Error handling in loading components/features in Agent Component (#5320)
This commit is contained in:
parent
c3f36b69a2
commit
d31cb351f3
8 changed files with 56 additions and 34 deletions
|
|
@ -13,6 +13,7 @@ from langflow.components.langchain_utilities.tool_calling import (
|
||||||
)
|
)
|
||||||
from langflow.custom.utils import update_component_build_config
|
from langflow.custom.utils import update_component_build_config
|
||||||
from langflow.io import BoolInput, DropdownInput, MultilineInput, Output
|
from langflow.io import BoolInput, DropdownInput, MultilineInput, Output
|
||||||
|
from langflow.logging import logger
|
||||||
from langflow.schema.dotdict import dotdict
|
from langflow.schema.dotdict import dotdict
|
||||||
from langflow.schema.message import Message
|
from langflow.schema.message import Message
|
||||||
|
|
||||||
|
|
@ -62,14 +63,25 @@ class AgentComponent(ToolCallingAgentComponent):
|
||||||
outputs = [Output(name="response", display_name="Response", method="message_response")]
|
outputs = [Output(name="response", display_name="Response", method="message_response")]
|
||||||
|
|
||||||
async def message_response(self) -> Message:
|
async def message_response(self) -> Message:
|
||||||
|
try:
|
||||||
llm_model, display_name = self.get_llm()
|
llm_model, display_name = self.get_llm()
|
||||||
self.model_name = get_model_name(llm_model, display_name=display_name)
|
|
||||||
if llm_model is None:
|
if llm_model is None:
|
||||||
msg = "No language model selected"
|
msg = "No language model selected"
|
||||||
raise ValueError(msg)
|
raise ValueError(msg)
|
||||||
|
self.model_name = get_model_name(llm_model, display_name=display_name)
|
||||||
|
except Exception as e:
|
||||||
|
# Log the error for debugging purposes
|
||||||
|
logger.error(f"Error retrieving language model: {e}")
|
||||||
|
raise
|
||||||
|
|
||||||
|
try:
|
||||||
self.chat_history = await self.get_memory_data()
|
self.chat_history = await self.get_memory_data()
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Error retrieving chat history: {e}")
|
||||||
|
raise
|
||||||
|
|
||||||
if self.add_current_date_tool:
|
if self.add_current_date_tool:
|
||||||
|
try:
|
||||||
if not isinstance(self.tools, list): # type: ignore[has-type]
|
if not isinstance(self.tools, list): # type: ignore[has-type]
|
||||||
self.tools = []
|
self.tools = []
|
||||||
# Convert CurrentDateComponent to a StructuredTool
|
# Convert CurrentDateComponent to a StructuredTool
|
||||||
|
|
@ -78,11 +90,17 @@ class AgentComponent(ToolCallingAgentComponent):
|
||||||
self.tools.append(current_date_tool)
|
self.tools.append(current_date_tool)
|
||||||
else:
|
else:
|
||||||
msg = "CurrentDateComponent must be converted to a StructuredTool"
|
msg = "CurrentDateComponent must be converted to a StructuredTool"
|
||||||
raise ValueError(msg)
|
raise TypeError(msg)
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Error adding current date tool: {e}")
|
||||||
|
raise
|
||||||
|
|
||||||
if not self.tools:
|
if not self.tools:
|
||||||
msg = "Tools are required to run the agent."
|
msg = "Tools are required to run the agent."
|
||||||
|
logger.error(msg)
|
||||||
raise ValueError(msg)
|
raise ValueError(msg)
|
||||||
|
|
||||||
|
try:
|
||||||
self.set(
|
self.set(
|
||||||
llm=llm_model,
|
llm=llm_model,
|
||||||
tools=self.tools,
|
tools=self.tools,
|
||||||
|
|
@ -91,6 +109,10 @@ class AgentComponent(ToolCallingAgentComponent):
|
||||||
system_prompt=self.system_prompt,
|
system_prompt=self.system_prompt,
|
||||||
)
|
)
|
||||||
agent = self.create_agent_runnable()
|
agent = self.create_agent_runnable()
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Error setting up the agent: {e}")
|
||||||
|
raise
|
||||||
|
|
||||||
return await self.run_agent(agent)
|
return await self.run_agent(agent)
|
||||||
|
|
||||||
async def get_memory_data(self):
|
async def get_memory_data(self):
|
||||||
|
|
|
||||||
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
Loading…
Add table
Add a link
Reference in a new issue