fix: Error handling in loading components/features in Agent Component (#5320)

This commit is contained in:
Edwin Jose 2024-12-17 20:30:34 -05:00 • committed by GitHub
commit d31cb351f3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 56 additions and 34 deletions

View file

@ -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