handle errors when getting sql database tools

Author: ks6088ts

.env.template

@@ -16,7 +16,7 @@ LANGSMITH_API_KEY="lsv2_xxx"
 AZURE_OPENAI_ENDPOINT="https://xxx.openai.azure.com/"
 AZURE_OPENAI_API_KEY="xxx"
 AZURE_OPENAI_API_VERSION="2025-04-01-preview"
-AZURE_OPENAI_MODEL_CHAT="gpt-4o"
+AZURE_OPENAI_MODEL_CHAT="gpt-5-chat"
 AZURE_OPENAI_MODEL_EMBEDDING="text-embedding-3-small"
 AZURE_OPENAI_MODEL_REASONING="o4-mini"
 

template_langgraph/agents/chat_with_tools_agent/agent.py

@@ -6,7 +6,7 @@
 from template_langgraph.agents.chat_with_tools_agent.models import AgentState
 from template_langgraph.llms.azure_openais import AzureOpenAiWrapper
 from template_langgraph.loggers import get_logger
-from template_langgraph.tools.common import DEFAULT_TOOLS
+from template_langgraph.tools.common import get_default_tools
 
 logger = get_logger(__name__)
 
@@ -24,19 +24,29 @@ def __call__(self, inputs: dict):
             raise ValueError("No message found in input")
         outputs = []
         for tool_call in message.tool_calls:
-            tool_result = self.tools_by_name[tool_call["name"]].invoke(tool_call["args"])
-            outputs.append(
-                ToolMessage(
-                    content=json.dumps(tool_result.__str__(), ensure_ascii=False),
-                    name=tool_call["name"],
-                    tool_call_id=tool_call["id"],
+            try:
+                tool_result = self.tools_by_name[tool_call["name"]].invoke(tool_call["args"])
+                outputs.append(
+                    ToolMessage(
+                        content=json.dumps(tool_result.__str__(), ensure_ascii=False),
+                        name=tool_call["name"],
+                        tool_call_id=tool_call["id"],
+                    )
+                )
+            except Exception as e:
+                logger.error(f"Error occurred while invoking tools: {e}")
+                outputs.append(
+                    ToolMessage(
+                        content=json.dumps({"error": str(e)}, ensure_ascii=False),
+                        name=tool_call["name"],
+                        tool_call_id=tool_call["id"],
+                    )
                 )
-            )
         return {"messages": outputs}
 
 
 class ChatWithToolsAgent:
-    def __init__(self, tools=DEFAULT_TOOLS):
+    def __init__(self, tools=get_default_tools()):
         self.llm = AzureOpenAiWrapper().chat_model
         self.tools = tools
 

template_langgraph/agents/kabuto_helpdesk_agent/agent.py

@@ -2,13 +2,13 @@
 
 from template_langgraph.llms.azure_openais import AzureOpenAiWrapper
 from template_langgraph.loggers import get_logger
-from template_langgraph.tools.common import DEFAULT_TOOLS
+from template_langgraph.tools.common import get_default_tools
 
 logger = get_logger(__name__)
 
 
 class KabutoHelpdeskAgent:
-    def __init__(self, tools=DEFAULT_TOOLS):
+    def __init__(self, tools=get_default_tools()):
         self.agent = create_react_agent(
             model=AzureOpenAiWrapper().chat_model,
             tools=tools,

template_langgraph/tools/common.py

@@ -1,15 +1,25 @@
 from template_langgraph.llms.azure_openais import AzureOpenAiWrapper
+from template_langgraph.loggers import get_logger
 from template_langgraph.tools.cosmosdb_tool import search_cosmosdb
 from template_langgraph.tools.dify_tool import run_dify_workflow
 from template_langgraph.tools.elasticsearch_tool import search_elasticsearch
 from template_langgraph.tools.qdrant_tool import search_qdrant
 from template_langgraph.tools.sql_database_tool import SqlDatabaseClientWrapper
 
-DEFAULT_TOOLS = [
-    search_cosmosdb,
-    run_dify_workflow,
-    search_qdrant,
-    search_elasticsearch,
-] + SqlDatabaseClientWrapper().get_tools(
-    llm=AzureOpenAiWrapper().chat_model,
-)
+logger = get_logger(__name__)
+
+
+def get_default_tools():
+    try:
+        sql_database_tools = SqlDatabaseClientWrapper().get_tools(
+            llm=AzureOpenAiWrapper().chat_model,
+        )
+    except Exception as e:
+        logger.error(f"Error occurred while getting SQL database tools: {e}")
+        sql_database_tools = []
+    return [
+        search_cosmosdb,
+        run_dify_workflow,
+        search_qdrant,
+        search_elasticsearch,
+    ] + sql_database_tools