Set prompt

Signed-off-by: Aivin V. Solatorio <avsolatorio@gmail.com>

mcp_client.py

@@ -24,7 +24,7 @@ You must not provide answers beyond what the tools provide.
 
 Do not make up data or information and never simulate the `get_wdi_data` tool. Instead, you must always call the `get_wdi_data` tool when the user asks for data.
 
-You can use multiple tools if needed.
+You can use multiple tools if needed. Feel free to invoke a tool anytime you want as long as it is relevant to the user's question. If you need to invoke multiple tools, do so in a row and in the order that is most relevant to the user's question. Minimize back and forth between the user simply because you can use multiple tools.
 
 If the user asks for any information beyond what the tools available to you provide, you must say that you do not have that information.
 
@@ -32,6 +32,8 @@ Avoid making statements based on stereotypes or biases. Always ensure your claim
 
 You may describe the data in a way that is easy to understand but you must not elaborate based on external knowledge."""
 
+# SYSTEM_PROMPT = f"""You are a helpful assistant and today is {datetime.now().strftime("%Y-%m-%d")}."""
+
 LLM_MODEL = "claude-3-5-haiku-20241022"
 
 
@@ -80,6 +82,8 @@ class MCPClientWrapper:
             for tool in response.tools
         ]
 
+        print(self.tools)
+
         tool_names = [tool["name"] for tool in self.tools]
         return f"Connected to MCP server. Available tools: {', '.join(tool_names)}"
 
@@ -131,6 +135,7 @@ class MCPClientWrapper:
         for content in response.content:
             if content.type == "text":
                 result_messages.append({"role": "assistant", "content": content.text})
+                claude_messages.append({"role": "assistant", "content": content.text})
 
             elif content.type == "tool_use":
                 tool_id = content.id