From c24b0d9ac4b8205040f50fbbf4bb6fa9dad00de7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=8Dtalo=20Johnny?= Date: Sat, 19 Oct 2024 10:47:31 -0300 Subject: [PATCH] fix: handle unsupported LLM exception message (#4199) * fix: handle unsupported LLM exception message * Update src/backend/base/langflow/components/agents/ToolCallingAgent.py Co-authored-by: Gabriel Luiz Freitas Almeida * [autofix.ci] apply automated fixes --------- Co-authored-by: Gabriel Luiz Freitas Almeida Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> --- .../base/langflow/components/agents/ToolCallingAgent.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/components/agents/ToolCallingAgent.py b/src/backend/base/langflow/components/agents/ToolCallingAgent.py index 3701d31cc..ec4603192 100644 --- a/src/backend/base/langflow/components/agents/ToolCallingAgent.py +++ b/src/backend/base/langflow/components/agents/ToolCallingAgent.py @@ -43,4 +43,8 @@ class ToolCallingAgentComponent(LCToolsAgentComponent): ("placeholder", "{agent_scratchpad}"), ] prompt = ChatPromptTemplate.from_messages(messages) - return create_tool_calling_agent(self.llm, self.tools, prompt) + try: + return create_tool_calling_agent(self.llm, self.tools, prompt) + except NotImplementedError as e: + message = f"{self.display_name} does not support tool calling." "Please try using a compatible model." + raise NotImplementedError(message) from e