add default tool set

Author: ks6088ts

template_langgraph/agents/chat_with_tools_agent/agent.py

@@ -6,10 +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.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.common import DEFAULT_TOOLS
 
 logger = get_logger(__name__)
 
@@ -41,12 +38,7 @@ def __call__(self, inputs: dict):
 class ChatWithToolsAgent:
     def __init__(self):
         self.llm = AzureOpenAiWrapper().chat_model
-        self.tools = [
-            search_cosmosdb,
-            run_dify_workflow,
-            search_qdrant,
-            search_elasticsearch,
-        ]
+        self.tools = DEFAULT_TOOLS
 
     def create_graph(self):
         """Create the main graph for the agent."""

template_langgraph/agents/kabuto_helpdesk_agent/agent.py

@@ -2,23 +2,15 @@
 
 from template_langgraph.llms.azure_openais import AzureOpenAiWrapper
 from template_langgraph.loggers import get_logger
-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.common import DEFAULT_TOOLS
 
 logger = get_logger(__name__)
 
 
 class KabutoHelpdeskAgent:
     def __init__(self, tools=None):
         if tools is None:
-            # Default tool for searching Qdrant
-            tools = [
-                run_dify_workflow,
-                search_qdrant,
-                search_elasticsearch,
-                # Add other tools as needed
-            ]
+            tools = DEFAULT_TOOLS
         self.agent = create_react_agent(
             model=AzureOpenAiWrapper().chat_model,
             tools=tools,

template_langgraph/tools/common.py

@@ -0,0 +1,11 @@
+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
+
+DEFAULT_TOOLS = [
+    search_cosmosdb,
+    run_dify_workflow,
+    search_qdrant,
+    search_elasticsearch,
+]