refactor(agent): use dedent for cleaner system message formatting and enhance logging

ShrewdAgent.py

@@ -1,4 +1,5 @@
 import os
+from textwrap import dedent
 from typing import TypedDict, Annotated, Optional, Any, Callable, Sequence, Union
 
 from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage
@@ -29,18 +30,19 @@ class AgentState(TypedDict):
 
 
 class ShrewdAgent:
-    message_system = """You are a general AI assistant equipped with a suite of external tools. Your task is to
-                        answer the following question as accurately and helpfully as possible by using the tools
-                        provided. Do not write or execute code yourself. For any operation requiring computation,
-                        data retrieval, or external access, explicitly invoke the appropriate tool.
+    message_system = dedent("""
+        You are a general AI assistant equipped with a suite of external tools. Your task is to
+        answer the following question as accurately and helpfully as possible by using the tools
+        provided. Do not write or execute code yourself. For any operation requiring computation,
+        data retrieval, or external access, explicitly invoke the appropriate tool.
         
-                        Follow these guidelines:"
-                        - Clearly explain your reasoning step by step.
-                        - Justify your choice of tool(s) at each step.
-                        - If multiple interpretations are possible, outline them and explain your reasoning for selecting one.
-                        - If the answer requires external data or inference, retrieve or deduce it via the available tools.
+        Follow these guidelines:
+        - Clearly explain your reasoning step by step.
+        - Justify your choice of tool(s) at each step.
+        - If multiple interpretations are possible, outline them and explain your reasoning for selecting one.
+        - If the answer requires external data or inference, retrieve or deduce it via the available tools.
         
-                        Important: Your final output must be only a number or a short phrase, with no additional text or explanation."""
+        Important: Your final output must be only a number or a short phrase, with no additional text or explanation.""")
 
     def __init__(self):
         self.tools = [
@@ -67,7 +69,8 @@ class ShrewdAgent:
             }
 
         self.agent = _build_state_graph(AgentState, assistant_node, self.tools)
-        logger.info("ShrewdAgent initialized.")
+        logger.info(f"Agent initialized with tools: {[tool.name for tool in self.tools]}")
+        logger.debug(f"system message:\n{self.message_system}")
 
     def __call__(self, question: str) -> str:
         logger.info(f"Agent received question:\n{question}")