set tools in constructor

Author: ks6088ts

template_langgraph/agents/chat_with_tools_agent/agent.py

@@ -36,9 +36,9 @@ def __call__(self, inputs: dict):
 
 
 class ChatWithToolsAgent:
-    def __init__(self):
+    def __init__(self, tools=DEFAULT_TOOLS):
         self.llm = AzureOpenAiWrapper().chat_model
-        self.tools = DEFAULT_TOOLS
+        self.tools = tools
 
     def create_graph(self):
         """Create the main graph for the agent."""

template_langgraph/agents/kabuto_helpdesk_agent/agent.py

@@ -8,9 +8,7 @@
 
 
 class KabutoHelpdeskAgent:
-    def __init__(self, tools=None):
-        if tools is None:
-            tools = DEFAULT_TOOLS
+    def __init__(self, tools=DEFAULT_TOOLS):
         self.agent = create_react_agent(
             model=AzureOpenAiWrapper().chat_model,
             tools=tools,